def ProcessEditDecision(tmpFilePath, edit, width, height):
"""\
Prefab.
Applies an edit decision - reading in the relevant video frames and applying
the reframing. Outputs the reframed video frames out of the "outbox" outbox.
Arguments:
- tmpFilePath -- temp directory into which video frames have been saved
- edit -- the edit instruction (dictionary containing: "start","end","left","top","right","bottom")
- width -- width (in pixels) for output video frames
- height -- height (in pixels) for output video frames
Inboxes:
- "inbox" -- NOT USED
- "control" -- Shutdown signalling
Outboxes:
- "outbox" -- NOT USED
- "signal" -- Shutdown signalling
"""
print " Video segment: ", edit
filenames = [
tmpFilePath + "%08d.yuv" % i
for i in range(edit["start"], edit["end"] + 1)
]
newsize = (width, height)
cropbounds = (edit["left"], edit["top"], edit["right"], edit["bottom"])
return Graphline( \
FILENAMES = ForwardIteratingChooser(filenames),
FRAME_LOADER = Carousel( lambda filename :
Pipeline(
2, MaxSpeedFileReader(filename,chunksize=1024*1024),
2, YUV4MPEGToFrame(),
),
make1stRequest=False ),
REFRAMING = Pipeline( 2, ToRGB_interleaved(),
2, CropAndScale(newsize, cropbounds),
2, ToYUV420_planar(),
),
linkages = {
("FRAME_LOADER", "requestNext") : ("FILENAMES", "inbox"),
("FILENAMES", "outbox") : ("FRAME_LOADER", "next"),
("FRAME_LOADER", "outbox") : ("REFRAMING", "inbox"),
("REFRAMING", "outbox") : ("", "outbox"),
("FILENAMES", "signal") : ("FRAME_LOADER", "control"),
("FRAME_LOADER", "signal") : ("REFRAMING", "control"),
("REFRAMING", "signal") : ("", "signal"),
},
boxsizes = {
},
)
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,
}
)
)
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)),
linkages={
("", "inbox"): ("PARSE", "inbox"),
("PARSE", "outbox"): ("OUT", "inbox"),
("PARSE", "all_meta"): ("OUT", "next"),
("", "control"): ("PARSE", "control"),
("PARSE", "signal"): ("OUT", "control"),
("OUT", "signal"): ("", "signal"),