Exemplo n.º 1
0
def get_input(name):
    if os.path.isdir(name):
        sequences = sequence.sequences(os.listdir(name))
        keys = list(sequences)
        if not len(keys):
            raise ValueError('no image sequence found in "%s"' % name)
        if len(keys) > 1:
            for i, seq in enumerate(keys):
                numbers = sequences[seq]
                print '%d) %s#%s' % (
                    (i + 1, ) + seq), '[%s-%s]' % (numbers[0], numbers[-1])
            user_input = raw_input('select sequence> ')
            try:
                seq = keys[int(user_input) - 1]
            except (ValueError, IndexError):
                raise ValueError('invalid selection "%s"' % user_input)
        else:
            seq = keys[0]
        names = list(
            sequence.iterate_sequence(name, seq[0], seq[1], sequences[seq]))
        missing_frames = int(sequences[seq][-1]) - int(
            sequences[seq][0]) + 1 - len(names)
        if missing_frames:
            print >> sys.stderr, 'warning: sequence has %d missing frames' % missing_frames
        return names
    else:
        return name
Exemplo n.º 2
0
def _main():
    parser = optparse.OptionParser(
        "usage: %prog [options] input-dir output-dir")
    parser.add_option("--fps", dest="fps", help="set frames per second")
    parser.add_option("--yuv",
                      dest="yuv",
                      help="output raw yuv instead of encoding h264",
                      action='store_true',
                      default=False)
    options, args = parser.parse_args()
    if len(args) != 2:
        parser.error('one input and one output is required (-h for help)')
    input_dir = args[0]
    output_dir = args[1]
    for d in (input_dir, output_dir):
        if not os.path.isdir(d):
            parser.error('%s is not a directory!' % d)

    os.umask(2)

    class DecodeOptions(coding.OptionsBase):
        def error(self, returncode, output):
            print ''.join(output)

        # def status(self, format, info):
        #     print format, info
    class EncodeOptions(coding.OptionsBase):
        def error(self, returncode, output):
            print ''.join(output)

    decode_options = DecodeOptions()
    decode_options.options = {}
    if options.fps:
        decode_options.options['r'] = options.fps
    else:
        decode_options.options['r'] = 24

    encode_options = EncodeOptions()
    encode_options.options = {}
    encode_options.options['crf'] = 22
    encode_options.options['keyint'] = 24
    encode_options.options['preset'] = 'slower'

    sequences = sequence.sequences(os.listdir(input_dir))
    for (head, tail), names in sequences.iteritems():
        names = list(sequence.iterate_sequence(input_dir, head, tail, names))
        output_name = os.path.join(output_dir,
                                   os.path.splitext(head[:-1] + tail)[0])
        if options.yuv:
            coding.encode_yuv(names,
                              output_name + '_yuv.mov',
                              decode_options=decode_options)
        else:
            coding.encode_h264(names,
                               output_name + '.mp4',
                               decode_options=decode_options,
                               encode_options=encode_options)
Exemplo n.º 3
0
def _main():
    parser = optparse.OptionParser("usage: %prog [options] input-dir output-dir")
    parser.add_option("--fps", dest="fps", help="set frames per second")
    parser.add_option("--yuv", dest="yuv", help="output raw yuv instead of encoding h264",
                      action='store_true', default=False)
    options, args = parser.parse_args()
    if len(args) != 2:
        parser.error('one input and one output is required (-h for help)')
    input_dir = args[0]
    output_dir = args[1]
    for d in (input_dir, output_dir):
        if not os.path.isdir(d):
            parser.error('%s is not a directory!' % d)

    os.umask(2)

    class DecodeOptions(coding.OptionsBase):
        def error(self, returncode, output):
            print ''.join(output)
        # def status(self, format, info):
        #     print format, info
    class EncodeOptions(coding.OptionsBase):
        def error(self, returncode, output):
            print ''.join(output)

    decode_options = DecodeOptions()
    decode_options.options = {}
    if options.fps:
        decode_options.options['r'] = options.fps
    else:
        decode_options.options['r'] = 24

    encode_options = EncodeOptions()
    encode_options.options = {}
    encode_options.options['crf'] = 22
    encode_options.options['keyint'] = 24
    encode_options.options['preset'] = 'slower'

    sequences = sequence.sequences(os.listdir(input_dir))
    for (head, tail), names in sequences.iteritems():
        names = list(sequence.iterate_sequence(input_dir, head, tail, names))
        output_name = os.path.join(output_dir, os.path.splitext(head[:-1] + tail)[0])
        if options.yuv:
            coding.encode_yuv(names, output_name + '_yuv.mov', decode_options=decode_options)
        else:
            coding.encode_h264(names, output_name + '.mp4', decode_options=decode_options,
                               encode_options=encode_options)
Exemplo n.º 4
0
def get_input(name):
    if os.path.isdir(name):
        sequences = sequence.sequences(os.listdir(name))
        keys = list(sequences)
        if not len(keys):
            raise ValueError('no image sequence found in "%s"' % name)
        if len(keys) > 1:
            for i, seq in enumerate(keys):
                numbers = sequences[seq]
                print '%d) %s#%s' % ((i+1,) + seq), '[%s-%s]' % (numbers[0], numbers[-1])
            user_input = raw_input('select sequence> ')
            try:
                seq = keys[int(user_input)-1]
            except (ValueError, IndexError):
                raise ValueError('invalid selection "%s"' % user_input)
        else:
            seq = keys[0]
        names = list(sequence.iterate_sequence(name, seq[0], seq[1], sequences[seq]))
        missing_frames = int(sequences[seq][-1])-int(sequences[seq][0])+1 - len(names)
        if missing_frames:
            print >>sys.stderr, 'warning: sequence has %d missing frames' % missing_frames
        return names
    else:
        return name