#!/usr/bin/env python3

import sys
import os
import vidi

#FONT_DESC = "Andale Mono, 72"
FONT_DESC = "Mono, 72"

SAMPLE_LENGTH_SECONDS = 4

# 48000 samples per seconds, at 1024 samples per buffer
NUM_AUDIO_BUFFERS = round(SAMPLE_LENGTH_SECONDS * 48000 / 1024)

LIVE_PIPELINE_TEMPLATE = """
  videotestsrc num-buffers=1 pattern=black ! \
          textoverlay valignment=center halignment=center font-desc="%s" text="{0}" ! \
          autovideosink \
  audiotestsrc num-buffers=%d freq={1:f} ! audio/x-raw,rate=48000 ! \
          autoaudiosink
""" % (FONT_DESC, NUM_AUDIO_BUFFERS)

FILE_PIPELINE_TEMPLATE = """
  matroskamux name=mux ! filesink location="{2}/sample_{0}.mkv"
  videotestsrc num-buffers=1 pattern=black ! \
          textoverlay valignment=center halignment=center font-desc="%s" text="{0}" ! \
          queue ! schroenc rate-control=3 ! mux.
  audiotestsrc num-buffers=%d freq={1:f} ! audio/x-raw,rate=48000 ! \
          queue ! audioconvert ! vorbisenc quality=0.5 ! queue ! mux.
""" % (FONT_DESC, NUM_AUDIO_BUFFERS)


def create_test_videofont(pipeline_template, notes_range, destination_dir=None):
    for i, note_number in enumerate(notes_range):
        note = vidi.SpnNote(note_number)

        print("%2d %s" % (i, note))

        pipeline_string = pipeline_template.format(note.name, note.frequency, destination_dir)

        player = vidi.Player.from_pipeline_string(pipeline_string)
        error = player.play()
        if error:
            break


def usage():
    print("usage: %s [<videofont_destination_dir>]" %
          os.path.basename(sys.argv[0]))


def main():
    if len(sys.argv) > 1 and sys.argv[1] in ["-h", "--help"]:
        usage()
        return 0

    notes_range = vidi.PIANO_88_KEYS_RANGE

    if len(sys.argv) > 1:
        destination_dir = os.path.realpath(sys.argv[1])
        if os.path.exists(destination_dir):
            sys.stderr.write("The destination already exists '%s'!\n"
                             % destination_dir)
            return 1

        os.mkdir(destination_dir)
        create_test_videofont(FILE_PIPELINE_TEMPLATE, notes_range,
                              destination_dir)
    else:
        create_test_videofont(LIVE_PIPELINE_TEMPLATE, notes_range)


if __name__ == "__main__":
    sys.exit(main())