some fixes

[rec10/rec10-git.git] / rec10 / trunk / src / tv2avi.py

diff --git a/rec10/trunk/src/tv2avi.py b/rec10/trunk/src/tv2avi.py
index ac62b95..b4e682d 100644 (file)
--- a/rec10/trunk/src/tv2avi.py
+++ b/rec10/trunk/src/tv2avi.py
@@ -12,6 +12,16 @@ import ts2x264
 import ts2xvid
 import tv2ts
 import recdb
+
+global Bitrate_HD
+global Bitrate_FHD
+global Bitrate_SD
+global Bitrate_WQVGA
+Bitrate_HD="2000"
+Bitrate_FHD="2500"
+Bitrate_SD="1250"
+Bitrate_WQVGA="200"
+
 def timetv2avi(pout, chtxt, btime, etime,opt):
     bt = datetime.datetime.strptime(btime, "%Y-%m-%d %H:%M:%S")
     et = datetime.datetime.strptime(etime, "%Y-%m-%d %H:%M:%S")
@@ -31,6 +41,9 @@ def timetv2avi(pout, chtxt, btime, etime,opt):
     tv2ts.tv2ts(tsout + ".ts", chdb.chtxtsearch(chtxt)['ch'], chdb.chtxtsearch(chtxt)['csch'], rectime)
     aviin = tsout + ".ts"
     dualaudio = 0
+    pentaaudio = 0
+    if re.search("5",opt):
+        pentaaudio = 1
     if re.search("d",opt):
         dualaudio = 1
     if re.search("\[二\]", pout):
@@ -57,9 +70,13 @@ def timetv2avi(pout, chtxt, btime, etime,opt):
         makexvid = 0
     else:
         makexvid = 1
-    """if dualaudio == 1:
+    if dualaudio == 1:
+        dualaudiots2avi(aviin, tsout + "sa.avi")
+        aviin = tsout + "sa.avi"
+    if pentaaudio ==1:
         dualaudiots2avi(aviin, tsout + "sa.avi")
         aviin = tsout + "sa.avi"
+    """
     if makexvid==1:
         ts2xvid.ts2xvid(aviin, pout,opts)
     else:
@@ -70,7 +87,6 @@ def timetv2avi(pout, chtxt, btime, etime,opt):
     et=datetime.datetime.now()+datetime.timedelta(seconds=600)
     etime=et.strftime("%Y-%m-%d %H:%M:%S")
     title=os.path.split(tsout)[1]
-    recdb.rec_reckey("ts2avi",title,chtxt,btime,etime,opt)
     #try:
     #    os.remove(aviin)
     #except:
@@ -89,6 +105,9 @@ def tv2avi(pout, chtxt, btime, etime,opt):
     tv2ts.tv2ts(tsout + ".ts", chdb.chtxtsearch(chtxt)['ch'], chdb.chtxtsearch(chtxt)['csch'], time)
     aviin = tsout + ".ts"
     dualaudio = 0
+    pentaaudio = 0 
+    if re.search("5",opt):
+        pentaaudio = 1
     if re.search("d",opt):
         dualaudio = 1
     if re.search('\[二\]', pout):
@@ -111,6 +130,9 @@ def tv2avi(pout, chtxt, btime, etime,opt):
     if dualaudio == 1:
         dualaudiots2avi(aviin, tsout + "sa.avi")
         aviin = tsout + "sa.avi"
+    if pentaaudio ==1:
+        dualaudiots2avi(aviin, tsout + "sa.avi")
+        aviin = tsout + "sa.avi"
     ts2x264.ts2x264(aviin, pout,opts)
     #os.remove(aviin)
 def b252avi(pout, chtxt,opt):
@@ -132,6 +154,9 @@ def b252avi(pout, chtxt,opt):
     #tv2ts.tv2ts(tsout + ".ts", chdb.chtxtsearch(chtxt)['ch'], chdb.chtxtsearch(chtxt)['csch'], rectime)
     aviin = tsout + ".ts"
     dualaudio = 0
+    pentaaudio = 0
+    if re.search("5",opt):
+        pentaaudio = 1
     if re.search("d",opt):
         dualaudio = 1
     if re.search('\[二\]', pout):
@@ -161,6 +186,9 @@ def b252avi(pout, chtxt,opt):
     if dualaudio == 1:
         dualaudiots2avi(aviin, tsout + "sa.avi")
         aviin = tsout + "sa.avi"
+    if pentaaudio ==1:
+        dualaudiots2avi(aviin, tsout + "sa.avi")
+        aviin = tsout + "sa.avi"
     if makexvid==1:
         ts2xvid.ts2xvid(aviin, pout,opts)
     else:
@@ -226,5 +254,24 @@ def dualaudiots2avi(pin, pout):
     os.remove(ffpin1)
     os.remove(ffpin2)
 
-
+def pentaaudiots2avi(pin, pout):
+    bontsdemux = configreader.getpath('bontsdemux')
+    wine = configreader.getpath('wine')
+    cmdwine = configreader.getpath('cmdwine')
+    bonpin = "Z:\\" + pin[1:]
+    outf = os.path.splitext(pin)[0]
+    bonpout = "Z:\\" + outf[1:]
+    exe = 'wine ' + bontsdemux + " -i \"" + bonpin + "\" -delay 167 -nd -sound 3 -o \"" + bonpout + "\" -start -quit"
+    #print exe
+    exe = 'xvfb-run -a ' + exe
+    #exe='export Display=:2&&'+exe
+    print exe
+    os.system(exe)
+    ffpin1 = pin.replace("ts", "wav")
+    ffpin2 = pin.replace("ts", "m2v")
+    exe = "ffmpeg -y  -i \'" + ffpin1 + "\' -i \'" + ffpin2 + "\' -r 29.97 -vsync 200 -vcodec copy -acodec libmp3lame -async 200 -ab 128k -f avi  \'" + pout + "\'"
+    #print exe
+    os.system(exe)
+    os.remove(ffpin1)
+    os.remove(ffpin2)