# /opt/local/stow/python-2.6.6/bin/python2.6 make_JHKs_fig_zoomest.py

import aplpy
import numpy
import os
import matplotlib
import matplotlib.pyplot as mpl

def ra2deg(s):
    token = s.split(':')
    deg = 15*(float(token[0])+float(token[1])/60.+float(token[2])/3600.)
    return(deg)

def dec2deg(s):
    token = s.split(':')
    deg = (abs(float(token[0]))+float(token[1])/60.+float(token[2])/3600.)
    if (token[0].find('-') >= 0):
        deg = -deg
    return(deg)

# target_ra1=84.694   # Bigger FoV with R136
# target_dec1=-69.083   # Bigger FoV with R136
# image_radius1=0.0175   # Bigger FoV with R136

target_ra1=ra2deg('05:38:49.5')
target_dec1=dec2deg('-69:04:44')
image_radius1=0.0075  

os.system('rm -rf 30Dor_JHKs_mol_zoomest.png')
#os.system('rm -rf 30Dor_JHKs_mol.png')

fig = mpl.figure(figsize=(8, 8))

# Plot
f1 = aplpy.FITSFigure('rgb_JHKs_2d.fits', figure=fig, subplot=[0.0,0.5,0.5,0.5])
f2 = aplpy.FITSFigure('rgb_JHKs_2d.fits', figure=fig, subplot=[0.5,0.5,0.5,0.5])
f3 = aplpy.FITSFigure('rgb_JHKs_2d.fits', figure=fig, subplot=[0.0,0.0,0.5,0.5])
f4 = aplpy.FITSFigure('rgb_JHKs_2d.fits', figure=fig, subplot=[0.5,0.0,0.5,0.5])

f1.show_rgb('30Dor_JHKs_3color_log.png')
f2.show_rgb('30Dor_JHKs_3color_log.png')
f3.show_rgb('30Dor_JHKs_3color_log.png')
f4.show_rgb('30Dor_JHKs_3color_log.png')

f1.recenter(target_ra1,target_dec1,radius=image_radius1)
f2.recenter(target_ra1,target_dec1,radius=image_radius1)
f3.recenter(target_ra1,target_dec1,radius=image_radius1)
f4.recenter(target_ra1,target_dec1,radius=image_radius1)

########################################
# Labels
########################################

labsize='medium'

# Panels

f1.add_label(x=0.9,y=0.95,text='(a)',relative=True,
            size=labsize,color='white',weight='bold')

f2.add_label(x=0.9,y=0.95,text='(b)',relative=True,
            size=labsize,color='white',weight='bold')

f3.add_label(x=0.9,y=0.95,text='(c)',relative=True,
            size=labsize,color='white',weight='bold')

f4.add_label(x=0.9,y=0.95,text='(d)',relative=True,
            size=labsize,color='white',weight='bold')

# RGB

f1.add_label(x=0.1,y=0.94,text='J',relative=True,
            size=labsize,color='DodgerBlue',weight='bold')

f1.add_label(x=0.15,y=0.94,text='H',relative=True,
            size=labsize,color='lime',weight='bold')

f1.add_label(x=0.21,y=0.94,text='Ks',relative=True,
            size=labsize,color='Red',weight='bold')


# Mol Labels

#f1.add_label(x=0.15,y=0.94,text='CO (2-1)',relative=True,
#            size=labsize,color='white',weight='bold')

f2.add_label(x=0.16,y=0.94,text='HCO+ (1-0)',relative=True,
            size=labsize,color='white',weight='bold')

f3.add_label(x=0.16,y=0.94,text='HCN (1-0)',relative=True,
            size=labsize,color='white',weight='bold')

f4.add_label(x=0.16,y=0.94,text='CS (2-1)',relative=True,
            size=labsize,color='white',weight='bold')


f4.add_scalebar(2./3600., '2"=0.5 pc','bottom left')
f4.scalebar.set_color('yellow')
f4.scalebar.set(linewidth=3)
f4.scalebar.set_font(size='medium', weight='bold', \
                      stretch='normal', family='sans-serif', \
                      style='normal', variant='normal')



#f1.show_arrows(ra2deg('19:23:30.0'),dec2deg('14:19:00'),-0.1,-0.04,
#               linewidth=2,color='white',zorder=200)

#f2.show_arrows(ra2deg('19:23:30.0'),dec2deg('14:19:00'),-0.1,-0.04,
#               linewidth=2,color='white',zorder=200)

#f3.show_arrows(ra2deg('19:23:30.0'),dec2deg('14:19:00'),-0.1,-0.04,
#               linewidth=2,color='white',zorder=200)

#f4.show_arrows(ra2deg('19:23:30.0'),dec2deg('14:19:00'),-0.1,-0.04,
#               linewidth=2,color='white',zorder=200)

########################################
# Contours
########################################

color='white'
width=1
layer='mols'

# 13CO 1,2,4,8

f2.show_contour('30DorBand3.ms.HCOp.image.mom0.fits',
               levels=[0.1,0.3,0.6],
               colors=color,linewidths=width,layer=layer)

f3.show_contour('30DorBand3.ms.HCN_NOIWT.image.mom0.fits',
               levels=[0.07, 0.3],
               colors=color,linewidths=width,layer=layer)

f4.show_contour('30DorBand3.ms.CS.image.mom0.fits',
               levels=[0.06,0.25],
               colors=color,linewidths=width,layer=layer)

########################################
# axes
########################################

labsize='medium'

f1.axis_labels.set_xtext('Right Ascension (J2000)')
f1.axis_labels.set_ytext('Declination (J2000)')
f1.axis_labels.set_font(size=labsize, weight='bold', \
                       stretch='normal', family='sans-serif', \
                         style='normal', variant='normal')
f1.tick_labels.set_style('colons')
f1.tick_labels.set_font(size=labsize, weight='bold', \
                       stretch='normal', family='sans-serif', \
                         style='normal', variant='normal')

f2.axis_labels.set_xtext('Right Ascension (J2000)')
f2.axis_labels.set_ytext('Declination (J2000)')
f2.axis_labels.set_font(size=labsize, weight='bold', \
                       stretch='normal', family='sans-serif', \
                         style='normal', variant='normal')
f2.tick_labels.set_style('colons')
f2.tick_labels.set_font(size=labsize, weight='bold', \
                       stretch='normal', family='sans-serif', \
                         style='normal', variant='normal')

f3.axis_labels.set_xtext('Right Ascension (J2000)')
f3.axis_labels.set_ytext('Declination (J2000)')
f3.axis_labels.set_font(size=labsize, weight='bold', \
                       stretch='normal', family='sans-serif', \
                         style='normal', variant='normal')
f3.tick_labels.set_style('colons')
f3.tick_labels.set_font(size=labsize, weight='bold', \
                       stretch='normal', family='sans-serif', \
                         style='normal', variant='normal')

f4.axis_labels.set_xtext('Right Ascension (J2000)')
f4.axis_labels.set_ytext('Declination (J2000)')
f4.axis_labels.set_font(size=labsize, weight='bold', \
                       stretch='normal', family='sans-serif', \
                         style='normal', variant='normal')
f4.tick_labels.set_style('colons')
f4.tick_labels.set_font(size=labsize, weight='bold', \
                       stretch='normal', family='sans-serif', \
                         style='normal', variant='normal')


# fix tick problem

f1.ticks.set_linewidth(2)
f2.ticks.set_linewidth(2)
f3.ticks.set_linewidth(2)
f4.ticks.set_linewidth(2)

#f1.ticks.set_xspacing(45*15/3600.)
#f2.ticks.set_xspacing(45*15/3600.)
#f3.ticks.set_xspacing(45*15/3600.)
#f4.ticks.set_xspacing(45*15/3600.)

f1.tick_labels.set_yformat('dd:mm:ss')
f1.tick_labels.set_xformat('hh:mm:ss')
f2.tick_labels.set_yformat('dd:mm:ss')
f2.tick_labels.set_xformat('hh:mm:ss')
f3.tick_labels.set_yformat('dd:mm:ss')
f3.tick_labels.set_xformat('hh:mm:ss')
f4.tick_labels.set_yformat('dd:mm:ss')
f4.tick_labels.set_xformat('hh:mm:ss')

f1.axis_labels.set_xpad(5)
f1.axis_labels.set_ypad(-20)
f2.axis_labels.set_xpad(5)
f2.axis_labels.set_ypad(-20)
f3.axis_labels.set_xpad(5)
f3.axis_labels.set_ypad(-20)
f4.axis_labels.set_xpad(5)
f4.axis_labels.set_ypad(-20)


f1.axis_labels.hide_x()
f1.tick_labels.hide_x()

f2.axis_labels.hide_y()
f2.tick_labels.hide_y()
f2.axis_labels.hide_x()
f2.tick_labels.hide_x()

f4.axis_labels.hide_y()
f4.tick_labels.hide_y()


########################################
# save
########################################

f1.save('30Dor_JHKs_mol_zoomest.png')
#f1.save('30Dor_JHKs_mol.png')
#f1.save('30Dor_JHKs_mol.eps')