import stilts

table = stilts.tread('/data1/rcsed_v2/photometry/results/for_ukidss_offset.fits')
table = table.cmd_addcol('delta', 'YPETROMAG_ukidss > 0 ? (aper3mag_z_sdss-petromag_z_sdss)-(aper3mag_Y_ukidss-YPETROMAG_ukidss) : (JPETROMAG_ukidss > 0 ? (aper3mag_z_sdss-petromag_z_sdss)-(aper3mag_J_ukidss-JPETROMAG_ukidss) : (HPETROMAG_ukidss > 0 ? (aper3mag_z_sdss-petromag_z_sdss)-(aper3mag_H_ukidss-HPETROMAG_ukidss) : (aper3mag_z_sdss-petromag_z_sdss)-(aper3mag_K_ukidss-KPETROMAG_ukidss)) )')
table = table.cmd_addcol('petromag_Y_ukidss', 'YPETROMAG_ukidss-delta-A_Y_ukidss').cmd_delcols('YPETROMAG_ukidss')
table = table.cmd_addcol('petromag_J_ukidss', 'JPETROMAG_ukidss-delta-A_J_ukidss').cmd_delcols('JPETROMAG_ukidss')
table = table.cmd_addcol('petromag_H_ukidss', 'HPETROMAG_ukidss-delta-A_H_ukidss').cmd_delcols('HPETROMAG_ukidss')
table = table.cmd_addcol('petromag_K_ukidss', 'KPETROMAG_ukidss-delta-A_K_ukidss').cmd_delcols('KPETROMAG_ukidss')

table = table.cmd_replacecol('petromag_Y_ukidss', 'petromag_Y_ukidss+0.634')
table = table.cmd_replacecol('petromag_J_ukidss', 'petromag_J_ukidss+0.938')
table = table.cmd_replacecol('petromag_H_ukidss', 'petromag_H_ukidss+1.379')
table = table.cmd_replacecol('petromag_K_ukidss', 'petromag_K_ukidss+1.9')


print('writing')
table.write('/data1/rcsed_v2/photometry/results/rcsed_v2_ukidssoffset.fits')
print('done')

table = table.cmd_addcol('z', '! NULL_z_sdss ? z_sdss : ( ! NULL_z_2df ? z_2df : ( ! NULL_z_6df ? z_6df : ( ! NULL_z_uzc ? z_uzc : ( ! NULL_z_hectospec ? z_hectospec : ( ! NULL_z_cfa ? z_cfa : ( ! NULL_z_lamost ? z_lamost : ( ! NULL_z_lega_c ? z_lega_c : ( ! NULL_Z_deep2 ? Z_deep2 : ( ! NULL_Z_deep3 ? Z_deep3 : ( ! NULL_z_wigglez ? z_wigglez : ( ! NULL_Z_gama ? Z_gama : z_2dFLenS )))))))))))')
kcorr_table = table.cmd_select('(z_sdss< 0.5) || (z_2df< 0.5) || (z_6df< 0.5) || (z_uzc< 0.5) || (z_hectospec< 0.5) || (z_cfa< 0.5) || (z_lamost< 0.5) || (z_lega_c< 0.5) || (Z_deep2< 0.5) || (Z_deep3< 0.5) || (z_wigglez< 0.5) || (Z_gama< 0.5) || (z_2dFLenS < 0.5)')
kcorr_table = kcorr_table.cmd_addcol('kcorr_petromag_Y_ukidss', 'kCorr(KCF_Y, z, KCC_YH, petromag_Y_ukidss-petromag_H_ukidss)')
kcorr_table = kcorr_table.cmd_addcol('kcorr_petromag_J_ukidss', 'kCorr(KCF_J, z, KCC_JH, petromag_J_ukidss-petromag_H_ukidss)')
kcorr_table = kcorr_table.cmd_addcol('kcorr_petromag_H_ukidss', 'kCorr(KCF_H, z, KCC_JH, petromag_J_ukidss-petromag_H_ukidss)')
kcorr_table = kcorr_table.cmd_addcol('kcorr_petromag_K_ukidss', 'kCorr(KCF_K, z, KCC_JK, petromag_J_ukidss-petromag_K_ukidss)')


print('writing kcorr')
kcorr_table.write('rcsed_v2_ukidssoffset_kcorr.fits')
print('done')