让FFMPEG支持实时流“伴随”转码

FFMPEG命令行工具目前无法针对视频实时流进行转码，而我们可以用它的新版本提供的支持TCP SERVER的特性来实现这一功能。实现边录边用FFMPEG提供的命令行工具进行转码。（实践证明，FFMPEG 的SDK针对实时流开发比较繁琐，命令行工具相对来说稳定）

我们将实时流录制成正在增长的文件，这样问题就变成了FFMPEG如何针对正在增长的视频文件转码。（首先我们保证视频格式是可以支持正在增长文件解码，这不在本文的讨论范畴）

而我们如果直接使用它对正在增长的文件转码，我们会发现FFMPEG转码到文件末尾（转码速度比录制速度快）就会退出。我们要做的就是在FFMPEG转码到末尾时如何将其挂起。若直接修改FFMPEG的源代码则比较繁琐，我们搭建一个TCP的FILE SERVER，来控制给FFMPEG发送数据的速度，而FFMPEG使用TCP的方式来获取视频流，即可实现这个功能。

以下是代码，我们用python来实现：

FFMPEG version:N-32611-gd55b06b

#encoding=utf8
'''
Created on 2011-9-25

@author: chenggong

ffmpeg扩展工具
'''
import SocketServer
import os
import threading
import time

BIN_DIR = "bin\\"
BLOCK_SIZE = 188*1024
FRONTOFFSET = 1024 * 1024 * 1

gparam={'output':'','source':'','port':9333,
        'start':0,'length':0,'ffmpeg_argvs':'',
        'ffmpegsema':None,'tcpserversema':None}

class MyFfmpegThread(threading.Thread):   
    def run(self):
        if os.path.exists(gparam['output']):
            os.remove(gparam['output'])
        argvs = gparam['ffmpeg_argvs'].replace("[filelocate]","tcp://127.0.0.1:%d"%gparam['port']) + " " + gparam['output']
        cmd = "%s\\avconv %s"%(BIN_DIR,argvs)
        os.system(cmd)
        gparam['ffmpegsema'].release()

class MyTcpServerThread(threading.Thread):    
    def run(self):
        while True:
            try:
                ADDR = ("127.0.0.1", gparam['port'])
                self.tcpServ = SocketServer.ThreadingTCPServer(ADDR, MyRequestHandler)
                break
            except Exception,e:
                print str(e)
                print "port:%d 被占用..更换"%gparam['port']
                gparam['port']+=1
        self.tcpServ.serve_forever()
        
    def close(self):
        self.tcpServ.shutdown()
        self.tcpServ.server_close()

class MyRequestHandler(SocketServer.BaseRequestHandler):   
    def handle(self):
        print 'connected from:', self.client_address      
        file = open(gparam['source'],"rb")
        
        startoffset = gparam['start']
        if startoffset<0:startoffset=0
            
        print "start offset="+str(startoffset)
        file.seek(startoffset)
        
        left = gparam['length']
        while(True):
            try:
                if(left<BLOCK_SIZE and left!=-1):
                    buffer_size = left
                else:
                    buffer_size = BLOCK_SIZE
                data=file.read(buffer_size)
                
                #print "read data,size="+str(len(data))
                #if(left!=-1): print "left size=%d"%left
                if not data:
                    print "data empty! read to file_end,wait.."
                    finish_filename=gparam['source'].replace(os.path.splitext(gparam['source'])[1],".finish")
                    if(os.path.exists(finish_filename)):
                        print ".finish file founded"
                        break
                    time.sleep(1)
                    continue

                self.request.sendall(data)
                
                if(left!=-1): 
                    left -= buffer_size
                    if(left<=0):
                        print "transfer finished"
                        break
                if(len(data)<buffer_size): 
                    finish_filename=gparam['source'].replace(os.path.splitext(gparam['source'])[1],".finish")
                    if(os.path.exists(finish_filename)):
                        print ".finish file founded"
                        break
                    else:
                        print "read to file_end,wait.."
                        time.sleep(1)
            except:
                print "exception occured,client stopped"
                break
        print "work finished"
        file.close()
        gparam['tcpserversema'].release()
        
def ffmpegex(source,output,start,length,ffmpeg_argvs):
    gparam['source']=source
    gparam['output']=output
    gparam['start']=start
    gparam['length']=length
    gparam['ffmpeg_argvs']=ffmpeg_argvs
    
    try:
        while True:
            gparam['ffmpegsema']=threading.Semaphore(0)
            gparam['tcpserversema']=threading.Semaphore(0)
            tcpServerThread = MyTcpServerThread()
            tcpServerThread.start()
            MyFfmpegThread().start()
            
            gparam['ffmpegsema'].acquire()
            gparam['tcpserversema'].acquire()
            tcpServerThread.close()
            
            filesize = os.path.getsize(gparam['output'])
            if(filesize<length/10): #生成失败,将start提前，重做任务
                if gparam['start']==0: return False
                if gparam['start']-FRONTOFFSET>=0:
                    gparam['start']-=FRONTOFFSET
                else:
                    gparam['start']=0
            else:
                return True
    except:
        return False
        
if __name__ == "__main__":
    
    for i in range(0,100):
        import random
        start = random.randrange(0,1024*1024*1024)
        print ffmpegex('C:\\16942.ts','C:\\test2\\%d-%d.wmv'%(i,start),start,1024*1024*5,\
                       "-i [filelocate] -acodec wmav2 -vcodec wmv2 -qscale 1 -ab 256k -r 25")