photospectra-splusdr1.py

'''
Make photo spectra from observed candidates JPLUS spectra
'''
from __future__ import print_function
import numpy as np
import glob
import json
import matplotlib.pyplot as plt
from astropy.table import Table
#import seaborn as sns
import sys
import argparse
import os
from colour import Color

Number = []
n = 1
mag_auto  = [[] for _ in range(n)]
mag_ISO_GAUSS = [[] for _ in range(n)]
mag_aper = [[] for _ in range(n)]
mag_auto_WORSTPSF = [[] for _ in range(n)]

#Error
mag_auto_err  = [[] for _ in range(n)]
mag_ISO_GAUSS_err  = [[] for _ in range(n)]

wl = [3485, 3785, 3950, 4100, 4300, 4803, 5150, 6250, 6600, 7660, 8610, 9110]
color = ["#CC00FF", "#9900FF", "#6600FF", "#0000FF", "#009999", "#006600", "#DD8000", "#FF0000", "#CC0066", "#990033", "#660033", "#330034"]
marker = ["s", "o", "o", "o", "o", "s", "o", "s", "o", "s", "o", "s"] ### tienen todos los filtros

wl1 = [ 3785, 3950, 4100, 4300, 4803, 5150, 6250, 6600, 7660, 8610, 9110]
color1 = [ "#9900FF", "#6600FF", "#0000FF", "#009999", "#006600", "#DD8000", "#FF0000", "#CC0066", "#990033", "#660033", "#330034"]
marker1 = [ "o", "o", "o", "o", "s", "o", "s", "o", "s", "o", "s"] # No tiene el primer filtro


parser = argparse.ArgumentParser(
    description="""Write wave and magnitude of a spectrum""")

parser.add_argument("source", type=str,
                    default="known-PN-jplus-idr",
                    help="Name of source, taken the prefix ")

parser.add_argument("--debug", action="store_true",
                    help="Print out verbose debugging info about each line in region file")

args = parser.parse_args()
file_ = args.source + ".tab"

data = Table.read(file_, format="ascii.tab")

for i in range(1):
    print(i)
    mag_auto[i].append(data["uJAVA_auto"][i])
    mag_auto[i].append(data["J0378_auto"][i])
    mag_auto[i].append(data["J0395_auto"][i])
    mag_auto[i].append(data["J0410_auto"][i])
    mag_auto[i].append(data["J0430_auto"][i])
    mag_auto[i].append(data["gSDSS_auto"][i])
    mag_auto[i].append(data["J0515_auto"][i]) 
    mag_auto[i].append(data["rSDSS_auto"][i]) 
    mag_auto[i].append(data["J0660_auto"][i])
    mag_auto[i].append(data["iSDSS_auto"][i]) 
    mag_auto[i].append(data["J0861_auto"][i]) 
    mag_auto[i].append(data["zSDSS_auto"][i])
    #Petro
    mag_ISO_GAUSS[i].append(data["uJAVA_ISO_GAUSS"][i])
    mag_ISO_GAUSS[i].append(data["J0378_ISO_GAUSS"][i])
    mag_ISO_GAUSS[i].append(data["J0395_ISO_GAUSS"][i])
    mag_ISO_GAUSS[i].append(data["J0410_ISO_GAUSS"][i])
    mag_ISO_GAUSS[i].append(data["J0430_ISO_GAUSS"][i])
    mag_ISO_GAUSS[i].append(data["gSDSS_ISO_GAUSS"][i])
    mag_ISO_GAUSS[i].append(data["J0515_ISO_GAUSS"][i]) 
    mag_ISO_GAUSS[i].append(data["rSDSS_ISO_GAUSS"][i]) 
    mag_ISO_GAUSS[i].append(data["J0660_ISO_GAUSS"][i])
    mag_ISO_GAUSS[i].append(data["iSDSS_ISO_GAUSS"][i]) 
    mag_ISO_GAUSS[i].append(data["J0861_ISO_GAUSS"][i]) 
    mag_ISO_GAUSS[i].append(data["zSDSS_ISO_GAUSS"][i])
   
    #ERRO AUTO
    mag_auto_err[i].append(float(data["uJAVA_auto_err"][i]))
    mag_auto_err[i].append(float(data["J0378_auto_err"][i]))
    mag_auto_err[i].append(float(data["J0395_auto_err"][i]))
    mag_auto_err[i].append(float(data["J0410_auto_err"][i]))
    mag_auto_err[i].append(float(data["J0430_auto_err"][i]))
    mag_auto_err[i].append(float(data["gSDSS_auto_err"][i]))
    mag_auto_err[i].append(float(data["J0515_auto_err"][i])) 
    mag_auto_err[i].append(float(data["rSDSS_auto_err"][i])) 
    mag_auto_err[i].append(float(data["J0660_auto_err"][i]))
    mag_auto_err[i].append(float(data["iSDSS_auto_err"][i]))
    try:
        mag_auto_err[i].append(float(data["J0861_auto_err"][i]))
    except ValueError:
        mag_auto_err[i].append(float(0.0))
    mag_auto_err[i].append(float(data["zSDSS_auto_err"][i]))

    print(mag_auto_err)
    #ERRO ISO
    # mag_ISO_GAUSS_err[i].append(data["uJAVA_ISO_GAUSS_err"][i])
    # mag_ISO_GAUSS_err[i].append(data["J0378_ISO_GAUSS_err"][i])
    # mag_ISO_GAUSS_err[i].append(data["J0395_ISO_GAUSS_err"][i])
    # mag_ISO_GAUSS_err[i].append(data["J0410_ISO_GAUSS_err"][i])
    # mag_ISO_GAUSS_err[i].append(data["J0430_ISO_GAUSS_err"][i])
    # mag_ISO_GAUSS_err[i].append(data["gSDSS_ISO_GAUSS_err"][i])
    # mag_ISO_GAUSS_err[i].append(data["J0515_ISO_GAUSS_err"][i]) 
    # mag_ISO_GAUSS_err[i].append(data["rSDSS_ISO_GAUSS_err"][i]) 
    # mag_ISO_GAUSS_err[i].append(data["J0660_ISO_GAUSS_err"][i])
    # mag_ISO_GAUSS_err[i].append(data["iSDSS_ISO_GAUSS_err"][i]) 
    # mag_ISO_GAUSS_err[i].append(data["J0861_ISO_GAUSS_err"][i]) 
    # mag_ISO_GAUSS_err[i].append(data["zSDSS_ISO_GAUSS_err"][i])

    font = {'family': 'serif',
        'color':  'black',
        'weight': 'normal',
        'size': 16,
        }

    plotfile = "photospectrum_splus_"+str(data["id"][i].split("18.")[-1].split(".g")[0])+"_auto.pdf"
    fig = plt.figure(figsize=(15.5, 9.5))
    ax = fig.add_subplot(1,1,1)
    plt.tick_params(axis='x', labelsize=42) 
    plt.tick_params(axis='y', labelsize=42)
    ax.set_xlim(xmin=3000, xmax=9700)
    #ax.set_ylim(ymin=17.5,ymax=23)
    #ax1.set_xlabel(r'$\lambda$')
    ax.set_xlabel(r'Wavelength $[\mathrm{\AA]}$', fontsize = 44)
    ax.set_ylabel(r'Magnitude [AB]', fontsize = 44)
    ax.plot(wl, mag_auto[i], '-k', alpha=0.2)#, label='Auto')
    for wl1, mag, mag_err, colors, marker_ in zip(wl, mag_auto[i], mag_auto_err[i], color, marker):
        ax.scatter(wl1, mag, color = colors, marker=marker_, s=600, zorder=10)
        ax.errorbar(wl1, mag, yerr=mag_err, marker='.', fmt='.', color=colors, ecolor=colors, elinewidth=5.9, markeredgewidth=5.2,  capsize=20)
    plt.text(0.06, 0.1, "Fr 2-21",
             transform=ax.transAxes, fontsize=48,  fontdict=font)
    #plt.subplots_adjust(bottom=0.19)
    plt.legend(fontsize=20.0)
    plt.tight_layout()
    plt.gca().invert_yaxis()
    save_path = '../../../Dropbox/JPAS/paper-phot/'
    file_save = os.path.join(save_path, plotfile)
    plt.savefig(file_save)
    plt.clf()