From e3abc44a53aad84883c2b23a74fe72c4b4e729c2 Mon Sep 17 00:00:00 2001
From: Hiroshi Yamashita <piro@users.sourceforge.jp>
Date: Sun, 24 Apr 2011 01:18:13 +0900
Subject: [PATCH] add source

---
 cmcheckwave.c | 433 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 433 insertions(+)
 create mode 100644 cmcheckwave.c

diff --git a/cmcheckwave.c b/cmcheckwave.c
new file mode 100644
index 0000000..4b93fcf
--- /dev/null
+++ b/cmcheckwave.c
@@ -0,0 +1,433 @@
+/*********************************************************************
+ original base program  http://oku.edu.mie-u.ac.jp/~okumura/dumpwave.c
+ 
+  cmcheckwave.c
+
+  Usage: cmcheckwave filename.wav
+*********************************************************************/
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <assert.h>
+
+static void usage(void){
+  exit(1);
+}
+
+unsigned char *get_bytes(FILE *f, int n)
+{
+    static unsigned char s[16];
+
+    assert (n <= sizeof s);
+    if (fread(s, n, 1, f) != 1) {
+        fprintf(stderr, "Read error\n");
+        exit(1);
+    }
+    return s;
+}
+
+unsigned long get_ulong(FILE *f)
+{
+    unsigned char *s = get_bytes(f, 4);
+    return s[0] + 256LU * (s[1] + 256LU * (s[2] + 256LU * s[3]));
+}
+
+unsigned get_ushort(FILE *f)
+{
+    unsigned char *s = get_bytes(f, 2);
+    return s[0] + 256U * s[1];
+}
+
+typedef struct {
+	int  stsec;
+	int  edsec;
+	int  diffs;
+	char cmflg;
+	char honpen;
+}muonst;
+
+static muonst m[1000];
+static muonst h[100];
+static int verbose=0;
+static char *wkfilename=NULL;
+static int defmuon=250;
+static int defmax=9;
+
+static char *MP4BOXCMD="/usr/local/bin/MP4Box -quiet -noprog";
+static char *SOXCMD="/usr/local/bin/sox";
+static char *FFMPEGCMD="/usr/local/bin/ffmpeg";
+static char *AACENCCMD="/usr/local/bin/aacplusenc";
+
+void dumpinfo(int mcnt)
+{
+    int honstart,hcnt,totalsec;
+    int i;
+
+    honstart=0;
+    hcnt=0;
+    totalsec=0;
+    printf("#!/bin/sh\n# cmcheckwave %s\n#\n",wkfilename?wkfilename:"");
+    //カットするため、一連のCM,本編時間を結合
+    for(i=0;i<mcnt;i++) {
+	    printf("# %.2f %.2f diff %.2f %s\n",m[i].stsec/1000.0,m[i].edsec/1000.0,m[i].diffs/1000.0,m[i].cmflg?"CM":"");
+	    //本編開始位置をマーク
+	    if ((m[i].cmflg==0)&&(honstart==0)) {
+		    honstart=1;
+		    if (i==0) h[hcnt].stsec = 0;
+		    else h[hcnt].stsec = m[i-1].edsec+(defmuon*0.5);
+	    }
+	    else {
+		    //終了位置をマーク
+	      if ((m[i].cmflg==1)&&(honstart==1)) {
+		    honstart=0;
+		    h[hcnt].edsec = m[i-1].stsec;
+		    totalsec += h[hcnt].edsec - h[hcnt].stsec;
+		    hcnt++;
+	      }
+	    }
+    }
+    if (honstart==1) {
+	h[hcnt].edsec = m[i-1].stsec;
+	totalsec += h[hcnt].edsec - h[hcnt].stsec;
+	hcnt++;
+    }
+    
+    printf("# total %.2f\n\n",totalsec/1000.0);
+
+    for(i=0;i<hcnt;i++) {
+	    if (wkfilename) {
+	      printf("%s -splitx %.2f:%.2f %s -out %s.%d\n",MP4BOXCMD,h[i].stsec/1000.0,h[i].edsec/1000.0,wkfilename,wkfilename,i);
+   	printf("%s -v 0 -i %s.%d -vn %s.%d.wav\n",FFMPEGCMD,wkfilename,i,wkfilename,i);
+	printf("%s -v 0 -i %s.%d -an -vcodec copy %s.%d.mp4\n",FFMPEGCMD,wkfilename,i,wkfilename,i);
+	    }
+	    else {
+	      printf("# %s -splitx %.2f:%.2f \n",MP4BOXCMD,h[i].stsec/1000.0,h[i].edsec/1000.0);
+	    }
+    }
+    if (wkfilename) {
+	    printf("%s ",SOXCMD);
+	for(i=0;i<hcnt;i++) {
+		printf(" %s.%d.wav ",wkfilename,i);
+	}
+	printf(" %s.wav\n",wkfilename);
+	printf("%s %s.wav %s.aac 60\n",AACENCCMD,wkfilename,wkfilename);
+	printf("%s ",MP4BOXCMD);
+	for(i=0;i<hcnt;i++) {
+		printf(" -cat %s.%d.mp4 ",wkfilename,i);
+	}
+	printf(" %s-new.mp4\n",wkfilename);
+	printf("%s -add %s.aac %s-new.mp4\n",MP4BOXCMD,wkfilename,wkfilename);
+
+	printf("rm -f %s.*\n\n",wkfilename);
+    }
+}
+
+int rechecktext(FILE *f)
+{
+	char rbuf[1024];
+	char fname[1024];
+	char cm[100];
+	char wk1[100];
+	char wk2[100];
+	char wk3[100];
+	char wk4[100];
+	int cnt,pgst,len;
+	float in1,in2,in3;
+
+
+	rewind(f);
+	cnt=0;
+
+	pgst=0;
+	fname[0]=0x00;
+
+	while(fgets(rbuf,1024,f)!=NULL){
+		if (strstr(rbuf,"# cmcheckwave ")) {
+				pgst=1;
+				if (strlen(rbuf+14)-1>0) {
+					wkfilename = malloc(strlen(rbuf+14));
+					strncpy(wkfilename,rbuf+14,strlen(rbuf+14)-1);
+				}
+				fgets(rbuf,1024,f);
+				continue;
+		}
+		if (strstr(rbuf,"# total "))
+			pgst=0;
+		if (pgst) {
+			wk4[0]=0x00;
+			sscanf(rbuf,"# %s %s diff %s %s",wk1,wk2,wk3,wk4);
+
+			m[cnt].stsec = (int)(atof(wk1)*1000.0);
+			m[cnt].edsec = (int)(atof(wk2)*1000.0);
+			m[cnt].diffs = (int)(atof(wk3)*1000.0);
+			if (strstr(wk4,"CM")) m[cnt].cmflg=1;
+			else m[cnt].cmflg=0;
+			m[cnt].honpen = 0;
+			cnt++;
+		}
+
+	}
+	dumpinfo(cnt);
+
+
+}
+
+int cmcheckwave(FILE *f)
+{
+    int i,j, x, channels, bits;
+    unsigned long len;
+    long count;
+    unsigned char s[5];
+    unsigned long bsec;
+    int readed,loop,max,totalsec,muonstartsec,kankaku;
+    double dul,diffs;
+    char cm[10];
+    int mcnt,hcnt,rcnt,readbufsz;
+    unsigned char *readbuf;
+    int honstart;
+    int cmwork;
+
+
+    if (memcmp(get_bytes(f, 4), "RIFF", 4) != 0) {
+        return -1;
+    }
+    if (memcmp(get_bytes(f, 8), "WAVEfmt ", 8) != 0) {
+        return -1;
+    }
+    len = get_ulong(f);
+    channels = get_ushort(f);
+    bsec = get_ulong(f);
+    bits = get_ushort(f);
+    while (fread(s, 4, 1, f) == 1) {
+        len = get_ulong(f);
+        s[4] = 0;
+        if (memcmp(s, "data", 4) == 0) break;
+    }
+
+    readed=max=totalsec=kankaku=mcnt=0;
+    muonstartsec=-1;
+    loop=1;
+    readbuf=malloc(4096*1000);
+    while(rcnt=fread(readbuf,1,4096*1000,f)) {
+      for(readbufsz=0;readbufsz<rcnt;) {
+        for (i = 0; i < channels; i++) {
+            if (bits <= 8) {
+                //if ((x = fgetc(f)) == EOF) {loop=0;break;}
+		x = readbuf[readbufsz];
+		readed++;
+		readbufsz++;
+                x -= 128;
+            } else {
+                //if (fread(s, 2, 1, f) != 1) {loop=0;break;}
+                //x = (short)(s[0] + 256 * s[1]);
+		x = (short)(readbuf[readbufsz+0] + 256 * readbuf[readbufsz+1]);
+		readed+=2;
+		readbufsz+=2;
+            }
+	    if (x > max) max = x;
+            //printf("%d", x);
+            //if (i != channels - 1) printf("\t");
+        }
+        //printf("\n");
+	if (readed % (bsec/100) == 0) {
+		//無音開始
+		if (muonstartsec==-1 && max < defmax) {
+			muonstartsec = totalsec;
+		}
+		else {
+		   // 無音終わり
+		   if (muonstartsec>=0 && max >= defmax) {
+			// 無音が300(defmuon)msより大きい
+			if (totalsec - muonstartsec > defmuon){
+				
+				m[mcnt].stsec = muonstartsec;
+				m[mcnt].edsec = totalsec;
+				m[mcnt].diffs = totalsec - kankaku;
+				m[mcnt].cmflg = 0;
+				m[mcnt].honpen = 0;
+
+				dul = (totalsec-muonstartsec)/1000.0;
+				diffs = (totalsec-kankaku)/1000.0;
+
+				if ((diffs >  14.5) && (diffs < 15.5)) m[mcnt].cmflg=1;
+				if ((diffs >  29.5) && (diffs < 30.5)) m[mcnt].cmflg=1;
+				if ((diffs >  59.5) && (diffs < 60.5)) m[mcnt].cmflg=1;
+
+				kankaku=totalsec;
+				mcnt++;
+			}
+			muonstartsec=-1;
+		   }
+		}
+		max=0;
+		totalsec += 10;
+	}
+      }
+    }
+    if (mcnt > 1) {
+            //本編前CMチェック
+	    if ((m[0].cmflg==0) && (m[0].diffs < 15000))
+		    m[0].cmflg=1;
+	    // 最終CMチェック
+	    if ((m[mcnt-1].cmflg==0) && (m[mcnt-1].diffs < 15000))
+		    m[mcnt-1].cmflg=1;
+	    //細切れCMのたしこみ
+    	for(i=1;i<mcnt-1;i++) {
+		// 本編で31秒以下が連続だったら、次の31秒以上の本編もしくはCMまでの時間をチェック
+		if (m[i].cmflg==0 && m[i].diffs < 31000  && m[i+1].cmflg==0 && m[i+1].diffs < 31000) {
+			cmwork=0;
+			for(j=i;j<mcnt-1;j++) {
+				if (m[j].cmflg==1) break;
+				if (m[j].diffs > 31000) break;
+				cmwork = cmwork + m[j].diffs;
+			}
+			//合計時間を15秒で割ってcm時間っぽいならばCMとする。
+			// TODO 30(15)秒以下の条件付けがいる？60秒どうする？
+			if (cmwork%15000>14500 || cmwork%15000<500) {
+				for(j=i;j<mcnt-1;j++) {
+					if (m[j].cmflg==1) break;
+					if (m[j].diffs > 31000) break;
+					m[j].cmflg=1;
+				}
+			}
+		}
+
+	}
+	//短い本編・提供などの処理
+    	for(i=1;i<mcnt-1;i++) {
+		//本編で46秒以下かつ、前後がCMの場合CM14-16,29-31,44-46秒でもCMとする。
+		if (m[i].cmflg==0 && m[i].diffs < 46000 && m[i-1].cmflg==1 && m[i+1].cmflg==1) {
+			if ((m[i].diffs > 14000) && (m[i].diffs < 16000)) m[i].cmflg=1;
+			if ((m[i].diffs > 29000) && (m[i].diffs < 31000)) m[i].cmflg=1;
+			if ((m[i].diffs > 44000) && (m[i].diffs < 46000)) m[i].cmflg=1;
+			// 0.9秒以下(おそらく前後ＣＭのあまり時間)
+			if (m[i].diffs < 900) m[i].cmflg=1;
+
+			//10,5秒のときは提供とみなし、その前を本編にする。
+			//TODO 15秒提供は判別不能・・・
+			if ((m[i].diffs > 9500) && (m[i].diffs < 10500)) m[i-1].cmflg=0;
+			if ((m[i].diffs > 4500) && (m[i].diffs < 5500)) m[i-1].cmflg=0;
+		}
+	}
+
+    }
+    honstart=0;
+    hcnt=0;
+    totalsec=0;
+    printf("#!/bin/sh\n# cmcheckwave %s\n#\n",wkfilename?wkfilename:"");
+    //カットするため、一連のCM,本編時間を結合
+    for(i=0;i<mcnt;i++) {
+	    printf("# %.2f %.2f diff %.2f %s\n",m[i].stsec/1000.0,m[i].edsec/1000.0,m[i].diffs/1000.0,m[i].cmflg?"CM":"");
+	    //本編開始位置をマーク
+	    if ((m[i].cmflg==0)&&(honstart==0)) {
+		    honstart=1;
+		    if (i==0) h[hcnt].stsec = 0;
+		    else h[hcnt].stsec = m[i-1].edsec+(defmuon*0.5);
+	    }
+	    else {
+		    //終了位置をマーク
+	      if ((m[i].cmflg==1)&&(honstart==1)) {
+		    honstart=0;
+		    h[hcnt].edsec = m[i-1].stsec;
+		    totalsec += h[hcnt].edsec - h[hcnt].stsec;
+		    hcnt++;
+	      }
+	    }
+    }
+    if (honstart==1) {
+	h[hcnt].edsec = m[i-1].stsec;
+	totalsec += h[hcnt].edsec - h[hcnt].stsec;
+	hcnt++;
+    }
+    
+    printf("# total %.2f\n\n",totalsec/1000.0);
+
+    for(i=0;i<hcnt;i++) {
+	    if (wkfilename) {
+	      printf("%s -splitx %.2f:%.2f %s -out %s.%d\n",MP4BOXCMD,h[i].stsec/1000.0,h[i].edsec/1000.0,wkfilename,wkfilename,i);
+   	printf("%s -v 0 -i %s.%d -vn %s.%d.wav\n",FFMPEGCMD,wkfilename,i,wkfilename,i);
+	printf("%s -v 0 -i %s.%d -an -vcodec copy %s.%d.mp4\n",FFMPEGCMD,wkfilename,i,wkfilename,i);
+	    }
+	    else {
+	      printf("# %s -splitx %.2f:%.2f \n",MP4BOXCMD,h[i].stsec/1000.0,h[i].edsec/1000.0);
+	    }
+    }
+    if (wkfilename) {
+	    printf("%s ",SOXCMD);
+	for(i=0;i<hcnt;i++) {
+		printf(" %s.%d.wav ",wkfilename,i);
+	}
+	printf(" %s.wav\n",wkfilename);
+	printf("%s %s.wav %s.aac 60\n",AACENCCMD,wkfilename,wkfilename);
+	printf("%s ",MP4BOXCMD);
+	for(i=0;i<hcnt;i++) {
+		printf(" -cat %s.%d.mp4 ",wkfilename,i);
+	}
+	printf(" %s-new.mp4\n",wkfilename);
+	printf("%s -add %s.aac %s-new.mp4\n",MP4BOXCMD,wkfilename,wkfilename);
+
+	printf("rm -f %s.*\n\n",wkfilename);
+    }
+
+    return hcnt;
+}
+
+int main(int argc, char *argv[])
+{
+extern char *optarg;
+extern int optind, opterr;
+    int i,ch;
+    FILE *f;
+    int ret;
+    char *tmpenv;
+    ret = -1;
+
+  while ((ch = getopt(argc, argv, "vb:m:n:")) != -1){
+    switch (ch){
+      case 'v':
+	verbose=1;
+	break;
+      case 'b':
+	wkfilename=optarg;
+	break;
+      case 'm':
+	defmuon=atoi(optarg);
+	break;
+      case 'n':
+	defmax=atoi(optarg);
+	break;
+      default:
+        usage();
+    }
+  }
+  argc -= optind;
+  argv += optind;
+
+  if (argc != 1) {
+        usage();
+	return 0;
+  }
+  if (tmpenv=getenv("FFMPEG")) FFMPEGCMD=tmpenv;
+  if (tmpenv=getenv("SOX")) SOXCMD=tmpenv;
+  if (tmpenv=getenv("MP4BOX")) MP4BOXCMD=tmpenv;
+  if (tmpenv=getenv("AACENC")) AACENCCMD=tmpenv;
+
+  ret=0;
+  if (strcmp(argv[0],"-")==0)
+	  f = stdin;
+  else
+  	f = fopen(argv[0],"rb");
+  if (f) {
+    ret = cmcheckwave(f);
+    // -1 のときはテキストとして再チェック
+    if (ret == -1) rechecktext(f);
+    fclose(f);
+  }
+  return ret;
+}
+
+
+
+
-- 
2.11.0