def ReEncode(outFileName):
"""\
Prefab.
Takes in audio and video frames and encodes them to a compressed video file
using ffmpeg to do the compression.
Inboxes:
- "inbox" -- NOT USED
- "control" -- Shutdown signalling
- "video" -- Video frames to be saved
- "audio" -- Auio frames to be saved
Outboxes:
- "outbox" -- NOT USED
- "signal" -- Shutdown signalling
"""
vidpipe = tmpFilePath+"vidPipe2.yuv"
try:
os.remove(vidpipe)
except:
pass
audpipe = tmpFilePath+"audPipe2.wav"
try:
os.remove(audpipe)
except:
pass
vidpipe=vidpipe.replace(" ","\ ")
audpipe=audpipe.replace(" ","\ ")
outFileName=outFileName.replace(" ","\ ")
encoder = "ffmpeg -f yuv4mpegpipe -i "+vidpipe+" -f wav -i "+audpipe+" -y "+outFileName
print encoder
return Graphline( \
VIDEO = FrameToYUV4MPEG(),
AUDIO = Carousel( lambda format : WAVWriter(**format),
make1stRequest=False),
ENCODE = UnixProcess2(encoder,buffersize=327680,inpipes={vidpipe:"video",audpipe:"audio"},boxsizes={"inbox":2,"video":2,"audio":2}),
DEBUG = ConsoleEchoer(),
linkages = {
("","audioformat") : ("AUDIO","next"),
("","video") : ("VIDEO","inbox"),
("VIDEO","outbox") : ("ENCODE","video"),
("","audio") : ("AUDIO", "inbox"),
("AUDIO","outbox") : ("ENCODE", "audio"),
("","control") : ("VIDEO","control"),
("VIDEO","signal") : ("AUDIO","control"),
("AUDIO","signal") : ("ENCODE", "control"),
("ENCODE", "signal") : ("DEBUG", "control"),
("DEBUG", "signal") : ("", "signal"),
("ENCODE","outbox") : ("DEBUG","inbox"),
("ENCODE","error") : ("DEBUG","inbox"),
},
boxsizes = {
("VIDEO", "inbox") : 2,
("AUDIO", "inbox") : 2,
}
)
("SPLIT", "signal" ) : ("FIRST", "control"),
("SPLIT", "signal2") : ("PLAY", "control"),
("PLAY", "signal" ) : ("", "signal"),
},
boxsizes = {
("SPLIT","inbox") : 1,
},
)
Graphline(
DECODE=UnixProcess2(
"ffmpeg -i " + infile +
" -f yuv4mpegpipe -y vidpipe.yuv -f wav -y audpipe.wav",
outpipes={
"vidpipe.yuv": "video",
"audpipe.wav": "audio"
},
buffersize=131072,
),
VIDEO=Pipeline(
1,
YUV4MPEGToFrame(),
FrameRateLimitedPlayback(VideoOverlay()),
),
AUDIO=Graphline(
PARSE=WAVParser(),
OUT=Carousel(lambda format: Output(format['sample_rate'],
format['channels'],
format['sample_format'],
maximumLag=0.5)),
def DecodeAndSeparateFrames(inFileName, tmpFilePath, edlfile,maxframe):
"""\
Prefab.
Decompresses audio and video from the specified file (using ffmpeg) and
saves them as individual files into the provided temp directory. Only
reads up to maxframes frames from the video file.
Arguments:
- inFileName -- The video file to be decompressed
- tmpFilePath -- temp directory into which frames of audio and video should be saved
- edlfile -- full filepathname of the EDL xml file
- maxframe -- the number of frames to decompress
Inboxes:
- "inbox" -- NOT USED
- "control" -- Shutdown signalling
Outboxes:
- "outbox" -- NOT USED
- "signal" -- Shutdown signalling
"""
vidpipe = tmpFilePath+"vidPipe.yuv"
try:
os.remove(vidpipe)
except:
pass
audpipe = tmpFilePath+"audPipe.wav"
try:
os.remove(audpipe)
except:
pass
mplayer = "ffmpeg -vframes %d -i %s -f yuv4mpegpipe -y %s -f wav -y %s" % ((maxframe+25),inFileName.replace(" ","\ "),vidpipe,audpipe)
print mplayer
return Graphline(
DECODER = UnixProcess2(mplayer, 2000000, {vidpipe:"video",audpipe:"audio"}),
FRAMES = YUV4MPEGToFrame(),
SPLIT = TwoWaySplitter(),
FIRST = FirstOnly(),
VIDEO = SaveVideoFrames(tmpFilePath,edlfile),
AUDIO = Carousel(lambda vformat: SaveAudioFrames(vformat['frame_rate'],tmpFilePath,edlfile)),
linkages = {
("DECODER","video") : ("FRAMES","inbox"),
("FRAMES","outbox") : ("SPLIT","inbox"),
("SPLIT","outbox") : ("VIDEO","inbox"),
("SPLIT","outbox2") : ("FIRST","inbox"),
("FIRST","outbox") : ("AUDIO","next"),
("DECODER","audio") : ("AUDIO","inbox"),
("DECODER","signal") : ("FRAMES","control"),
("FRAMES","signal") : ("SPLIT","control"),
("SPLIT","signal") : ("VIDEO","control"),
("SPLIT","signal2") : ("FIRST","control"),
("FIRST","signal") : ("AUDIO","control"),
("AUDIO","signal") : ("","signal"),
},
boxsizes = {
("FRAMES", "inbox") : 2,
("SPLIT", "inbox") : 1,
("FIRST", "inbox") : 2,
("VIDEO", "inbox") : 2,
("AUDIO", "inbox") : 10,
}
)
if len(files) != 1 or threshold is None or threshold <= 0.0:
sys.stderr.write(
"Usage:\n\n " + sys.argv[0] +
" [--show] [threshold] videofile\n\n* threshold is a floating point value greater than zero (default=0.9)\n\n"
)
sys.exit(1)
infile = files[0].replace(" ", "\ ")
if not show:
# simple cut detector
Pipeline(
UnixProcess2(
"ffmpeg -i " + infile + " -f yuv4mpegpipe -y /dev/stdout",
32768),
2,
YUV4MPEGToFrame(),
1,
TagWithSequenceNumber(),
1,
DetectShotChanges(threshold),
FormatOutput(),
ConsoleEchoer(),
StopSelector(waitForTrigger=True),
).run()
else:
# cut detector plus playback at the same time