import matplotlib
matplotlib.use('pdf')
matplotlib.rcParams.update({'dpi':300})
matplotlib.rcParams.update({'font.size': 12})
#from IPython.core.interactiveshell import InteractiveShell
#inst = InteractiveShell.instance()
#ip = inst.get_ipython()
#ip.enable_pylab("inline", import_all=False)
import sys
import gc
import yt
import matplotlib.pylab as pl
from matplotlib.ticker import FixedLocator
from mpl_toolkits.mplot3d import Axes3D
from yt.mods import *
from yt.data_objects.profiles import Profile2D


#def _xkpc(field,data):
#    return(data["x"]*26.)

#def _ykpc(field,data):
#    return(data["y"]*26.) 

#def _zkpc(field,data):
#    return(data["z"]*26.)
#def _cyl_Rkpc(field,data):
#    return(data["cyl_RCode"]*26.)

#def _CellMassSolar(field,data):
#    return(data["cel1_mass"]/1.99e33)

#def _zvelocity8(field,data):
#    return(data["z-velocity"]/1e8)
#def _logDensity(field,data):
#    return(np.log10(data["Density"]))
#def _totPressure(field,data):
#    return(data["Pressure"] + data["MagneticPressure"])

#add_field("xkpc",function=_xkpc,units="kpc")
#add_field("ykpc",function=_ykpc,units="kpc")
#add_field("zkpc",function=_zkpc,units="kpc")
#add_field("cyl_Rkpc",function=_cyl_Rkpc,units="kpc")
#add_field("CellMassSolar",function=_CellMassSolar,units=r'M$_{\odot}$')
#add_field("zvelocity8",function=_zvelocity8,units=r'10$^8$ cm/s)')
#add_field("logDensity",function=_logDensity,units=r'log(cm/s)')
#add_field("totPressure",function=_totPressure,units=r'erg cm$^{-3}$')

#outstring = ['0030','0040','0050','0060','0070','0080','0090','0100','0110','0120','0130']
#i = 0
#while i < len(outstring):
print sys.argv[1]
#pf = load("id0/rps."+sys.argv[1]+".vtk",parameters={"TimeUnits":3.086e14,"LengthUnits":8.0236e22,"DensityUnits":9.999e-30,"VelocityUnits":2.6e8})
pf = yt.load("DD"+sys.argv[1]+"/DD44sMBhn25wyu7hdh"+sys.argv[1])

pf.index
pf.field_info["gas","density"].take_log=True
#pf.field_info["Pressure"].take_log=True
#pf.field_info["totPressure"].take_log=True
#pf.field_info["cyl_Rkpc"].take_log=False
#pf.field_info["z-velocity"].take_log=False
#pf.field_info["zvelocity8"].take_log=False
#pf.field_info["zkpc"].take_log=False
#pf.field_info["Temperature"].take_log=True

#alld = pf.h.all_data()

from yt.visualization.volume_rendering.api import ColorTransferFunction

tf = ColorTransferFunction((-28,-25.5))
tf.add_layers(8,w=0.05,colormap="gist_ncar")#"spectral")#"gist_rainbow")#"Hue Sat Value 2")
tf.sample_colormap(-27.,w=0.02,colormap="gist_ncar")#"spectral")#"gist_rainbow")#"Hue Sat Value 2")
tf.sample_colormap(-27.5,w=0.01,colormap="gist_ncar")#"spectral")#"gist_rainbow")#"Hue Sat Value 2")
tf.sample_colormap(-27.9,w=0.01,alpha=0.3,colormap="gist_ncar")#
cam = pf.camera([0.5,0.5,0.65],[0.0,-1.0,-0.2],(60,'kpc'),1024,tf,fields=[('gas','density')])

im = cam.snapshot('DD44sMBhn25wyu7hdh_'+sys.argv[1]+'_volumeall_vertcbar.png')
yt.write_projection(im,"DD44sMBhn25wyu7hdh_"+sys.argv[1]+"_volumeall_vertcbar2.png",colorbar_label="Density")
tf.vert_cbar()
cam.show_tf()
nim=cam.draw_domain(im)
f=pl.figure()
timef = (58-6)*5
pl.text(0.2,0.2,np.str(timef),transform=f.transFigure, color="white")
pl.text(0.2,0.82,"Tonnesen",transform=f.transFigure, color='y')
cam._render_figure=f
cam.save_annotated('DD44sMBhn25wyu7hdh_'+sys.argv[1]+'_volumeall_vertcbar_ann.png',nim,dpi=300)