import math
import sys
import os
from yt.mods import *
import numpy as np
from yt.analysis_modules.halo_finding.api import *
import glob
import re
import commands

sys.path.append('/Users/haoxu/software/seedme-v0.8')
import seedme
obj=seedme.SeedMe()

dlist = glob.glob('RD00??')
dlist.sort()
#dlist.reverse() 
#output = int(re.sub(r'\D',"",fn))
for folder in dlist:
  n = int(re.sub(r'\D',"",folder))
  if n < 72 :continue
  if glob.glob(folder+'/PopIII_histogram*.png') != []: continue  
  a=commands.getoutput('grep CosmologyCurrentRedshift '+folder+'/RedshiftOutput00??')
  redshift=float(a.split()[2])
  os.system('cp plot_PopIII_histogram.plt '+folder)
  current=os.getcwd()
  os.chdir(current+"/"+folder)
  os.system('gnuplot plot_PopIII_histogram.plt')
  os.chdir(current)
  os.system('cp plot_PopII_histogram.plt '+folder)
  os.chdir(current+"/"+folder)
  os.system('gnuplot plot_PopII_histogram.plt')
  os.system('convert -density 300 PopII_histogram.eps PopII_histogram_%.1f.png'%redshift)
  os.system('convert -density 300 PopIII_histogram.eps PopIII_histogram_%.1f.png'%redshift)
  my_plots = [
               {'filepath': './PopII_histogram_%.1f.png'%redshift,
                'title': 'Pop II historam (z = %4.1f)'%redshift,
                'description': 'z=%4.1f'%redshift,
               },
               {'filepath': './PopIII_histogram_%.1f.png'%redshift,
                'title': 'Pop III historam (z = %4.1f)'%redshift,
                'description': 'z=%4.1f'%redshift,
               }
            ]

  print("\nAdd new histograms of Void region")
  obj.update_collection(6960, plots=my_plots)