#!/usr/bin/python

# coding=utf8

from optparse import OptionParser

from subprocess import *

from re import *

from shutil import *

import os		# os.open koliduje s built-in open

global predchozi_index

def strt(time):

    secs=int(time)

    fract=time-secs

    hrs=str(100+secs//3600)[-2:]

    min=str(100+(secs%3600)//60)[-2:]

    sec=str(100+secs%60)[-2:]

    fra=str(1000+1000*fract)[1:4]

    return hrs+":"+min+":"+sec+":"+fra

def dect(time):

    s=split(":", time)

    return int(s[0])*3600+int(s[1])*60+int(s[2])+float(s[3])/1000

def deb(line):

    if 'DEB' in os.environ: 

        try: 

            if int(os.environ['DEB'])>0:

                for x in line: print x, 

                print

        except: return

class jazyk:				# titulky jednoho jazyka

    def __init__(self, idx):

        self.idx=idx			# pořadí jazyka z indexu

        self.id=idxi.line.split(',')[0].split(': ')[1]

        self.head=idxi.line		# header jazyka

        self.index=[index(), index()]	# dílčí indexy pro jazyk v jednotlivých dílech

        self.image=["", ""]		# dílčí images pro jazyk v jednotlivých dílech 

    def nacti(self):

        deb(("jazyk", self.id))

        line=idxi.next()

        while line and not match(r"^timestamp:", line): line=idxi.next()		# dojeď na začátek záznamů s timestampy

        if not line: return

        self.index[dil].index_lines()

        print "konec dílu jazyka", self.id

        deb(("délka indexu", len(self.index[dil].time_pos)))

class index:				# dílčí index jazyka za jeden díl

    def __init__(self):

        self.time_pos=[]

        self.image=""

        self.img_beg=0

        self.img_end=0

        self.img_len=0

        self.img_del=0

    def index_lines(self):

        global predchozi_index

        zacatek=True

        line=idxi.line

        while line and match(r"^timestamp:", line):

            timestamp=search(r"^timestamp:\s(\d\d:\d\d:\d\d:\d\d\d)", line).group(1)

            position=eval("0x"+search(r"filepos:\s([a-z0-9]+)", line).group(1))

            if zacatek:

                if predchozi_index and predchozi_index.img_len==0:

                    deb(("načítá se image k předchozímu indexu",))

                    predchozi_index.getimage(position)

                predchozi_index=self

                self.img_beg=position

                zacatek=False

            if dil==0 and timestamp>off_set: break	# výchozí díly filmu se mohou překrývat, a proto se případně 1.díl zařezává a s ním i jeho titulky    

            self.time_pos.append((timestamp, position))

            line=idxi.next()

        if line and match(r"^timestamp:", line):

            deb(("konec zkráceného image",))

            self.getimage(position)

        if not line:

            deb(("konec image na konci file",))

            self.getimage(None)

        if line:

            deb(("konec dílu image",))

        while line and match(r"^timestamp:", line):	# projedeme ignorovanou část zařízutého 1.dílu

            line=idxi.next()

        #if timePosList.size()>0: img_beg=time_posList.getFirst().position

    def getimage(self, position):	# načti díl image k právě načtenému dílu indexu

        if position:	self.img_len=position-self.img_beg

        else:     	self.img_len=img_sizes[dil+1]-self.img_beg

        deb(("čte se ", self.img_len, "na offsetu", self.img_beg))

        self.image=imgi.read(self.img_len)

        deb(("precteno", self.img_len, "z", imgi_fn))

class index_file:

    def __init__(self, fn):

        self.line=None

        self.inf=open(fn, "r")

    def next(self):

        try: self.line=self.inf.next()

        except StopIteration: self.line=None

        return self.line

    def close(self): self.inf.close()        

usage = "usage: %prog <output_base> <1.part>.vob <2.part>.vob <1.part>.avi"

parser = OptionParser(usage=usage)

options, args = parser.parse_args()

if len(args)<4:

    print ">>> Povinné jsou 4 argumenty.\n"

    parser.print_help()

    exit(1)

try:	p=Popen(["mplayer", "-v", "-endpos", "0", "-vo", "null", "-ao", "null", args[3]], stdout=PIPE, stderr=STDOUT).communicate()

except: print "Nelze otevřít první část filmu."; exit(1)

try:	offset2dilu=float(search(r"ID_LENGTH=([\d.]+)", p[0]).group(1))

except: print "Nelze zjistit časovou délku první části filmu."; exit(1)

off_set=strt(offset2dilu)

print "Odstup druhé části:", off_set

N=(args[0], args[1], args[2])

try: p=os.stat(N[0]+".vob.ifo")

except:

    try: copy2(N[1]+".ifo", N[0]+".vob.ifo")

    except: print "Problém s vytvořením", N[0]+".vob.ifo"; exit(1)

img_sizes={}

for i in (1, 2):

    try:	img_sizes[i]=os.stat(N[i]+".sub").st_size

    except:     print "Chybí", N[i]+".sub"; exit(1)

    if img_sizes[i]%2048: print "Velikost", N[i]+".sub", "není násobkem 2048."; exit(1)

idxi=None

line=None

idxo=open(N[0]+".vob.idx", "w")

jazyky=dict()

for dil in (0, 1):			# zpracování jednotlivých dílů na vstupu

    print "dil", str(dil+1)

    predchozi_index=None

    if idxi: idxi.close()

    idxi=index_file(N[dil+1]+".idx")

    #idxi=open(N[dil+1]+".idx","r")

    line=idxi.next()

    imgi_fn=N[dil+1]+".sub"

    imgi=open(imgi_fn, "rb")

    while line:				# načtení dílčích indexů pro jednotlivé jazyky

        while line and not match(r"^id:\s", line):

            if dil==0: idxo.write(line)	# v prvním díle se zapíšou do výstupu úvodní řádky indexu

            line=idxi.next()

        if line:

            #jid=line.split(',')[0].split(': ')[1]

            jid=int(line.split('index: ')[1])

            deb(("nalezen jazyk", jid, line))

            if not jid in jazyky: jazyky[jid]=jazyk(jid)

            jazyky[jid].nacti()

            line=idxi.next()

deb(("počet načtených jazyků:", len(jazyky)))

imgo=open(N[0]+".vob.sub", "wb")

offset=0; jid=0

#for jid in jazyky.keys():		# sestavení výstupu

while jid in jazyky:

    print "zápis jazyka", jazyky[jid].id

    #idxo.write("id: "+jid+", index: "+str(poradi_jazyka)+"\n")

    idxo.write(jazyky[jid].head)

    for dil in (0, 1):

        index=jazyky[jid].index[dil]

        for tp in index.time_pos:

            time = strt(dect(tp[0])+dil*offset2dilu)

            position = ("00000000"+(hex(tp[1]-index.img_beg+offset))[2:])[-9:]

            idxo.write("timestamp: "+time+", filepos: "+position+"\n")

        imgo.write(index.image); offset+=len(index.image)

    jid+=1

idxo.close()

imgo.close()