#!/usr/bin/python

import sys,os,area,shutil

dirname=sys.argv[1]
if not os.path.isdir(dirname):
  print "reconstruct.py dctc_download_dir_name"
  sys.exit(1)
a=open(dirname+'/.cmd').readline().rstrip().split('|')
asum=area.area()
for file in os.listdir(dirname):
  if file[0]!='.':
    size=os.stat("%s/%s" % (dirname,file))[6]
    asum.add(long(file.split('-')[0],16),size,file)
total=asum.sum()
print "%-70s %6.2f %%" % (dirname[:70],100.0*total/int(a[3]))
asum.show()
pos=0
length=int(a[3])
f=open(os.path.basename(dirname.rstrip('/')),'wb')
while pos<length:
  bs={}
  for start,end,fn in asum.begins(pos):
    pattern=open(os.path.join(dirname,fn),'rb').read(1024)
    if bs.has_key(pattern):
      bs[pattern].append((start,end,fn))
    else:
      bs[pattern]=[(start,end,fn)]
  # sort it by patter count
  bss=sorted(bs,key=lambda x: len(bs[x]))
  start,end,fn=bs[bss[-1]][-1]
  print " %6.2f %% complete, bs=%3d MB, count=%d/%d" \
        % (100.0*pos/length,(end-(pos-start))/1024/1024,
           len(bs[bss[-1]]),sum([len(bs[x]) for x in bs]))
  sys.stdout.flush()
  block=open(os.path.join(dirname,fn),'rb')
  block.seek(pos-start)
  #f.write(block.read())
  shutil.copyfileobj(block,f,1024*1024)
  pos=f.tell()