import astropy.units as u
import numpy as np

SM1N={'image_name' : 'SM1N_continuum.image.fits',
      'image_label' : 'SM1N',
      'rc' : 246.613333,
      'dc' : -24.389556,
      'scalebar_size' : 500*u.au, 'scalebar_pos' : 'bottom right',
      'beam_loc': 'top left',
      'distance' : 137.3*u.pc, 'beam_pos' : 'bottom left',
      'im_width' : 0.01, 
      'im_height' : 0.01,
      'vmin' : -0.00015,
      'vmax' : 1.6e-3,
      'scale' : 'linear',
      'clev_mult' : 5.,
      'co_mult_r' : 3.,
      'co_mult_b' : 5.,
      'cbar_loc' : 'bottom right',
      'cbar_ticks' : [0,0.0004,0.0008,0.0012,0.0016],
      'sigma' : 4.6e-5,
      'uv_amp_max' : 30.,
      'uv_amp_min' : -5.}

SM1={'image_name' : 'SM1_continuum_selfcal_3.pbcor.fits',
     'image_label' : 'SM1',
     'rc' : 246.616083,
     'dc' : -24.399861,
     'scalebar_size' : 500*u.au, 'scalebar_pos' : 'left',
     'beam_loc': 'bottom left',
     'distance' : 137.3*u.pc, 'beam_pos' : 'top left',
     'im_width' : 0.012, 
     'im_height' : 0.012,
     'vmin' : -0.00015,
     'vmax' : 0.1,
     'vmid' : -0.0003,
     'scale' : 'log',
     'clev_mult' : 5.,
     'co_mult_r' : 3.,
     'co_mult_b' : 5.,
     'cbar_loc' : 'bottom right',
     'cbar_ticks' : [0,0.01,0.03,0.06,0.09],
     'sigma' : 1e-4,
     'uv_amp_max' : 150.,
     'uv_amp_min' : -5.}

AN6={'image_name' : 'AN6_continuum.pbcor.fits',
     'image_label' : 'N6',
     'rc' : 246.631667,
     'dc' : -24.414444,
     'scalebar_size' : 100*u.au, 'scalebar_pos' : 'bottom right',
     'beam_loc': 'top left',
     'distance' : 137.3*u.pc, 'beam_pos' : 'bottom left',
     'im_width' : 0.005, 
     'im_height' : 0.005,
     'vmin' : -0.00015,
     'vmax' : 0.00055,
     'scale' : 'linear',
     'clev_mult' : 3.,
     'sigma' : 4.e-5,
     'uv_amp_max' : 8.,
     'uv_amp_min' : -8.}

L16263_2422={'image_name' : '16263_2422_continuum_selfcal_2.pbcor.fits',
             'image_label' : 'GSS 30 IRS3',
             'rc' : 246.590167,
             'dc' : -24.381139,
             'scalebar_size' : 500*u.au, 'scalebar_pos' : 'left',
             'beam_loc': 'top left',
             'distance' : 137.3*u.pc, 'beam_pos' : 'bottom left',
             'im_width' : 0.012, 
             'im_height' : 0.012,
             'vmin' : -0.00015,
             'vmax' : 0.1,
             'scale' : 'log',
             'vmid' : -0.0003,
             'sigma' : 8.1e-5,
             'uv_amp_max' : 170.,
             'uv_amp_min' : -5.}

L16263_2={'image_name' : '16263_2422_continuum_selfcal_2.pbcor.fits',
      'image_label' : 'GSS 30 IRS1',
      'rc' : 246.590167,
      'dc' : -24.381139,
      'scalebar_size' : 500*u.au, 'scalebar_pos' : 'left',
      'beam_loc': 'top left',
      'distance' : 137.3*u.pc, 'beam_pos' : 'bottom left',
      'im_width' : 0.012, 
      'im_height' : 0.012,
      'vmin' : -0.00015,
      'vmax' : 0.1,
      'scale' : 'log',
      'vmid' : -0.0003,
      'sigma' : 8.1e-5,
      'uv_amp_max' : 170.,
      'uv_amp_min' : -5.}

N3mm={'image_name' : 'SM1_continuum_selfcal_3.pbcor.fits',
      'image_label' : 'N3-mm',
      'rc' : 246.616083,
      'dc' : -24.399861,
      'scalebar_size' : 500*u.au, 'scalebar_pos' : 'left',
      'beam_loc': 'top left',
      'distance' : 137.3*u.pc, 'beam_pos' : 'bottom left',
      'im_width' : 0.012, 
      'im_height' : 0.012,
      'vmin' : -0.00015,
      'vmax' : 0.1,
      'scale' : 'log',
      'vmid' : -0.0003,
      'sigma' : 8.1e-5,
      'uv_amp_max' : 170.,
      'uv_amp_min' : -5.}

B2A7={'image_name' : 'B2-A7_continuum_taper.image.fits',
      'image_label' : 'B2-A7',
      'rc' : 246.8854167,
      'dc' : -24.4491667,
      'scalebar_size' : 100*u.au, 'scalebar_pos' : 'bottom right',
      'beam_loc': 'top left',
      'distance' : 137.3*u.pc, 'beam_pos' : 'bottom left',
      'im_width' : 0.01, 
      'im_height' : 0.01,
      'vmin' : -0.00015,
      'vmax' : 0.00055,
      'scale' : 'linear',
      'clev_mult' : 3.,
      'co_sigma' : 9.e-3,
      'co_mult_r' : 3.,
      'co_mult_b' : 5.,
      'cbar_loc' : 'bottom right',
      'cbar_ticks' : [-0.0001,0,0.0001,0.0002,0.0003,0.0004,0.0005],
      'sigma' : 6.3e-5,
      'uv_amp_max' : 8.,
      'uv_amp_min' : -8.}

AN6t={'image_name' : 'AN6_continuum_taper.image.fits',
      'image_label' : 'N6',
      'rc' : 246.631667,
      'dc' : -24.414444,
      'scalebar_size' : 100*u.au, 'scalebar_pos' : 'bottom right',
      'beam_loc': 'top left',
      'distance' : 137.3*u.pc, 'beam_pos' : 'bottom left',
      'im_width' : 0.01, 
      'im_height' : 0.01,
      'vmin' : -0.00015,
      'vmax' : 0.00055,
      'scale' : 'linear',
      'clev_mult' : 3.,
      'co_sigma'  : 7.e-3,
      'co_mult_r' : 3.,
      'co_mult_b' : 5.,
      'cbar_loc' : 'bottom right',
      'cbar_ticks' : [-0.0001,0,0.0001,0.0002,0.0003,0.0004,0.0005,0.0006],     
      'sigma' : 6.5e-5,
      'uv_amp_max' : 8.,
      'uv_amp_min' : -8.}

plottingDictionary = {"SM1N" : SM1N, "SM1" : SM1, "AN6" : AN6, "16263_2422" : L16263_2422, "16263_2422_1" : L16263_2, "B2A7" : B2A7, "AN6t": AN6t, "N3-mm" : N3mm}

