import astropy.io.fits as fits
from astropy.wcs import WCS
from astropy.wcs import WCS


def Jplus_wcs(ra_center, dec_center, crpix1, crpix2):

    wcs = WCS(naxis=2)
    wcs.wcs.crpix = [crpix1, crpix2]
    wcs.wcs.crval = [ra_center, dec_center]
    wcs.wcs.cd = [[-0.0001543333333333, 0.0],
                  [0.0, 0.0001543333333333]]
    wcs.wcs.ctype = ['RA---TAN', 'DEC--TAN']
    wcs.wcs.cunit = ['deg', 'deg']

    return wcs


def Galex_wcs(ra_center, dec_center, crpix1, crpix2):

    wcs = WCS(naxis=2)
    wcs.wcs.crpix = [crpix1, crpix2]  
    wcs.wcs.crval = [ra_center, dec_center]  

    wcs.wcs.cd = [[-0.00041666666666666, 0.0],
                 [ 0.0, 0.00041666666666666]]  
    
    wcs.wcs.ctype = ['RA---TAN', 'DEC--TAN'] 
    wcs.wcs.cunit = ['deg', 'deg'] 
    
    return wcs

def Desi_wcs(ra_center, dec_center, crpix1, crpix2):

    wcs = WCS(naxis=2)
    wcs.wcs.crpix = [crpix1, crpix2]  
    wcs.wcs.crval = [ra_center, dec_center]  

    wcs.wcs.cd = [[-7.27777777777778e-05 , 0.0],
                 [ 0.0, 7.27777777777778e-05 ]]  
    
    wcs.wcs.ctype = ['RA---TAN', 'DEC--TAN'] 
    wcs.wcs.cunit = ['deg', 'deg'] 
    
    return wcs

def Allwise_wcs(ra_center, dec_center, crpix1, crpix2):

    wcs = WCS(naxis=2)
    wcs.wcs.crpix = [crpix1, crpix2]           
    wcs.wcs.crval = [ra_center, dec_center]    
    wcs.wcs.cdelt = [-0.0003819444391411, 0.0003819444391411]  
    wcs.wcs.pc = [[1.0, 0.0],                  
                  [0.0, 1.0]]
    wcs.wcs.ctype = ['RA---SIN', 'DEC--SIN']   
    wcs.wcs.cunit = ['deg', 'deg']            
    wcs.wcs.lonpole = 180.0                   
    wcs.wcs.latpole = dec_center              
    wcs.wcs.radesys = 'FK5'                    
    wcs.wcs.equinox = 2000.0   

    return wcs

def Chandra_wcs(ra_center, dec_center, crpix1, crpix2):
 
    wcs = WCS(naxis=2)
    wcs.wcs.crpix = [crpix1, crpix2]
    wcs.wcs.crval = [ra_center, dec_center]
    wcs.wcs.ctype = ['RA---SIN', 'DEC--SIN']
    wcs.wcs.cunit = ['deg', 'deg']

    wcs.wcs.cdelt = [-0.00013666666666667, 0.00013666666666667]
    
    wcs.wcs.pc = [[1.0, 0.0], [0.0, 1.0]]
    wcs.wcs.radesys = 'FK5'     
    wcs.wcs.equinox = 2000.0     
    wcs.wcs.lonpole = 180.0     
    wcs.wcs.latpole = dec_center 

    return wcs

def VLA_wcs(ra_center, dec_center, crpix1, crpix2):

    cdelt_arcsec = 0.6  
    cdelt_deg = cdelt_arcsec / 3600.0

    wcs = WCS(naxis=2)
    wcs.wcs.crpix = [crpix1, crpix2]
    wcs.wcs.crval = [ra_center, dec_center]
    wcs.wcs.ctype = ['RA---SIN', 'DEC--SIN']
    wcs.wcs.cunit = ['deg', 'deg']
    wcs.wcs.cdelt = [-abs(cdelt_deg), abs(cdelt_deg)] 

    wcs.wcs.pc = [[1.0, 0.0],
                  [0.0, 1.0]]
    
    wcs.wcs.radesys = 'FK5'
    wcs.wcs.equinox = 2000.0
    wcs.wcs.lonpole = 180.0
    wcs.wcs.latpole = dec_center

    return wcs