Techt3o's picture
c9887c76edf8bc22d935c6d311b9bc08e1b7b0e1fc7c155f7fc2fc66524fb1e7
caf1218 verified
raw
history blame
1.69 kB
#!/usr/bin/env python3
import sys
import os
import argparse
import numpy as np
import time
import json
# If this example doesn't work, it's probably because this path is wrong...
sys.path.append('../build/src')
import pypangolin as pango
def main(flags):
vid_uri = flags.pango
vout_uri = flags.pangoOut
vid = pango.VideoInput(vid_uri)
vout = pango.VideoOutput(vout_uri) if vout_uri else None
device_properties = vid.DeviceProperties()
# print metadata
print("Opened video uri: '{}' with {} x {} dimensions".format( vid_uri,vid.Width(),vid.Height()))
# user specified initial frame
vid.Seek(flags.startFrame)
# show each frame
streamsBitDepth = vid.GetStreamsBitDepth()
for frame in vid:
if vout:
vout.WriteStreams(frame, streamsBitDepth, vid.FrameProperties(), device_properties);
# frame is a list of Images! One per stream
# process(frame)
# printing
sys.stdout.write('\rframe: {} / {}'.format(vid.GetCurrentFrameId(), vid.GetTotalFrames()))
print('\nDONE')
if __name__ == "__main__":
# input flags
parser = argparse.ArgumentParser('Read a .pango file frame by frame. Optionally stream to another video output.')
parser.add_argument(
'--pango', type=str,
help='path to the input pango file.')
parser.add_argument(
'--startFrame', type=int, default=0,
help='index of the start frame (inclusive)')
parser.add_argument(
'--pangoOut', type=str, default=None,
help='path to the output pango file.')
# main function
main(parser.parse_args())