#!/usr/bin/python
#
# Encoding example.
# This is approximately a port of libvorbis's encoder_example.c
# It expects 44kHz 16-bit stereo input.
#
# With numarray version 0.9, this is only 6.4-15% slower than oggenc.c -
# only a second's difference on most songs.
#
# Daniel Holth <dholth@fastmail.fm>, 2004

import oggpy
import vorbispy
import sys
import numarray
import random
from numarray import array

# samples to read at one time, halving or quartering doesn't slow things down
READ = 8192

def test(indata):
    outfile = file("encodepy.ogg", "wb")
    
    og = oggpy.page()
    op = oggpy.packet()
    vi = vorbispy.info()
    vc = vorbispy.comment()
    vd = vorbispy.dsp()
    vb = vorbispy.block()

    vi.encode_init_vbr(2, 44100, -0.1)
    # vc.add_tag("ENCODER", "encode.py")
    vd.analysis_init(vi)
    vd.block_init(vb)
    
    randomnumber = random.randrange(0,2**30)
    os = oggpy.stream(randomnumber)

    header      = oggpy.packet()
    header_comm = oggpy.packet()
    header_code = oggpy.packet()

    vd.analysis_headerout(vc, header, header_comm, header_code)

    os.packetin(header)
    os.packetin(header_comm)
    os.packetin(header_code)

    # ensure the actual audio data will start on a new page
    eos = False
    while not eos:
        result = os.flush(og)
        if result == 0: break
        outfile.write(og.header())
        outfile.write(og.body())

    while not eos:
        data  = indata.read(READ*4)
        if not data:
            vd.analysis_write(None)
        else:
            interleaved = numarray.fromstring(data, numarray.Int16)
            # better convert to float, from -1.0 to 1.0
            deinterleaved = numarray.array((interleaved[0::2], interleaved[1::2]), numarray.Float32)
            deinterleaved *= 1.0/32768.0
            vd.analysis_write(deinterleaved)
 
        while(vd.analysis_blockout(vb)==1):
            # will have to be able to take null/None to use bitrate management:
            vb.analysis(op)
            os.packetin(op)
            
            while not eos:
                result = os.pageout(og)
                if result == 0: break
                outfile.write(og.header())
                outfile.write(og.body())

                if og.eos(): eos = 1
    
    del og


if __name__ == "__main__":
    test(file("./decode.raw", 'rb'))