AstroToolkit 1.0.0__py3-none-any.whl

AstroToolkit/Overlays/Overlay_GALEX.py

@@ -0,0 +1,148 @@
+import pandas as pd
+import math
+from astropy.time import Time
+
+from ..Surveys.Gaia import GaiaQueryCoords
+from ..Surveys.GALEX import GALEXQueryCoords
+from ..Miscellaneous.ProperMotionCorrection import PMCorrection
+
+#Creates Galex overlay
+def getGalexOverlayNUV(plot,ra,dec,sizeAS,mjd,border,pmra,pmdec):
+	#checks if GALEX search was successful
+	gaiaTime=[2016,0]
+	imageTime=Time(mjd,format='mjd').to_datetime()
+	imageTime=[imageTime.year,imageTime.month]
+	galexTime=[2007,0]
+	
+	if pmra!=None and pmdec!=None:
+		sizeAS=PMCorrection(gaiaTime,imageTime,ra,dec,pmra,pmdec,radius=sizeAS)[2]
+	
+	nearby=GALEXQueryCoords(ra,dec,sizeAS)
+	
+	detections_made=False
+	if isinstance(nearby,pd.DataFrame):
+		nearby_gaia=GaiaQueryCoords(ra,dec,sizeAS)
+		
+		#Plotting of Galex detections + brightness/radius scaling
+		nearbyRA=nearby.loc[:,'RAJ2000'].tolist()
+		nearbyDEC=nearby.loc[:,'DEJ2000'].tolist()
+		nearbyMagUV=nearby.loc[:,'NUVmag'].tolist()
+		
+		#checks if any NUV magnitudes exist
+		magCheck=False
+		for i in range(0,len(nearbyMagUV)):
+			if not math.isnan(nearbyMagUV[i]):
+				magCheck=True
+		
+		if magCheck==False:
+			return plot, detections_made
+			
+		#fetches gaia data
+		if isinstance(nearby_gaia,pd.DataFrame):
+			nearbyRA_gaia=nearby_gaia.loc[:,'ra'].tolist()
+			nearbyDEC_gaia=nearby_gaia.loc[:,'dec'].tolist()
+			nearbyPMRA=nearby_gaia.loc[:,'pmra'].tolist()
+			nearbyPMDEC=nearby_gaia.loc[:,'pmdec'].tolist()
+			
+			#transforms gaia detections back to GALEX time
+			for i in range(0,len(nearbyRA_gaia)):
+				if not math.isnan(nearbyPMRA[i]) and not math.isnan(nearbyPMDEC[i]):
+					RA_corrected,DEC_corrected=PMCorrection(gaiaTime,galexTime,nearbyRA_gaia[i],nearbyDEC_gaia[i],nearbyPMRA[i],nearbyPMDEC[i])
+					nearbyRA_gaia[i]=RA_corrected
+					nearbyDEC_gaia[i]=DEC_corrected
+			
+			#checks if there are any GALEX objects within a radius, if there are, applies gaia pm corrections to matching GALEX detection
+			correctedIndices=[]
+			for i in range(0,len(nearbyRA)):
+				for j in range(0,len(nearbyRA_gaia)):
+					delta=math.sqrt((nearbyRA[i]-nearbyRA_gaia[j])**2+(nearbyDEC[i]-nearbyDEC_gaia[j])**2)*3600
+					if delta<5:
+						if not math.isnan(nearbyPMRA[j]) and not math.isnan(nearbyPMDEC[j]):
+							RA_corrected,DEC_corrected=PMCorrection(galexTime,imageTime,nearbyRA[i],nearbyDEC[i],nearbyPMRA[j],nearbyPMDEC[j])
+							nearbyRA[i]=RA_corrected
+							nearbyDEC[i]=DEC_corrected
+							correctedIndices.append(i)
+							
+		#draw detections
+		for i in range(0,len(nearby)):
+			radiusMultiplier=(nearbyMagUV[i]/20.7)
+			radius=(border/50+(border/75)**radiusMultiplier)*0.75
+			
+			if i in correctedIndices and not math.isnan(nearbyMagUV[i]):
+				plot.circle(x=nearbyRA[i],y=nearbyDEC[i],radius=radius,line_color='blue',fill_color=None,legend_label='GALEX NUV detections (corrected)')
+				detections_made=True
+			elif i not in correctedIndices and not math.isnan(nearbyMagUV[i]):
+				plot.circle(x=nearbyRA[i],y=nearbyDEC[i],radius=radius,line_color='purple',fill_color=None,line_dash='dotted',legend_label='GALEX NUV detections (uncorrected)')
+				detections_made=True
+		
+	return plot, detections_made
+	
+#Creates Galex overlay
+def getGalexOverlayFUV(plot,ra,dec,sizeAS,mjd,border,pmra,pmdec):
+	gaiaTime=[2016,0]
+	imageTime=Time(mjd,format='mjd').to_datetime()
+	imageTime=[imageTime.year,imageTime.month]
+	galexTime=[2007,0]
+
+	if pmra!=None and pmdec!=None:
+		sizeAS=PMCorrection(gaiaTime,imageTime,ra,dec,pmra,pmdec,radius=sizeAS)[2]
+
+	nearby=GALEXQueryCoords(ra,dec,sizeAS)
+
+	detections_made=False
+	if isinstance(nearby,pd.DataFrame):
+		nearby_gaia=GaiaQueryCoords(ra,dec,sizeAS)
+		
+		#Plotting of Galex detections + brightness/radius scaling
+		nearbyRA=nearby.loc[:,'RAJ2000'].tolist()
+		nearbyDEC=nearby.loc[:,'DEJ2000'].tolist()
+		nearbyMagUV=nearby.loc[:,'FUVmag'].tolist()
+		
+		#checks if any FUV magnitudes exist
+		magCheck=False
+		for i in range(0,len(nearbyMagUV)):
+			if not math.isnan(nearbyMagUV[i]):
+				magCheck=True
+		
+		if magCheck==False:
+			return plot, detections_made
+		
+		#fetches gaia data
+		if isinstance(nearby_gaia,pd.DataFrame):
+			nearbyRA_gaia=nearby_gaia.loc[:,'ra'].tolist()
+			nearbyDEC_gaia=nearby_gaia.loc[:,'dec'].tolist()
+			nearbyPMRA=nearby_gaia.loc[:,'pmra'].tolist()
+			nearbyPMDEC=nearby_gaia.loc[:,'pmdec'].tolist()
+			
+			#transforms gaia detections back to GALEX time
+			for i in range(0,len(nearbyRA_gaia)):
+				if not math.isnan(nearbyPMRA[i]) and not math.isnan(nearbyPMDEC[i]):
+					RA_corrected,DEC_corrected=PMCorrection(gaiaTime,galexTime,nearbyRA_gaia[i],nearbyDEC_gaia[i],nearbyPMRA[i],nearbyPMDEC[i])
+					nearbyRA_gaia[i]=RA_corrected
+					nearbyDEC_gaia[i]=DEC_corrected
+			
+			#checks if there are any GALEX objects within a radius, if there are, applies gaia pm corrections to matching GALEX detection
+			correctedIndices=[]
+			for i in range(0,len(nearbyRA)):
+				for j in range(0,len(nearbyRA_gaia)):
+					delta=math.sqrt((nearbyRA[i]-nearbyRA_gaia[j])**2+(nearbyDEC[i]-nearbyDEC_gaia[j])**2)*3600
+					if delta<5:
+						if not math.isnan(nearbyPMRA[j]) and not math.isnan(nearbyPMDEC[j]):
+							RA_corrected,DEC_corrected=PMCorrection(galexTime,imageTime,nearbyRA[i],nearbyDEC[i],nearbyPMRA[j],nearbyPMDEC[j])
+							nearbyRA[i]=RA_corrected
+							nearbyDEC[i]=DEC_corrected
+							correctedIndices.append(i)
+		
+		#draw detections
+		for i in range(0,len(nearby)):
+			radiusMultiplier=(nearbyMagUV[i]/20.7)
+			radius=(border/50+(border/75)**radiusMultiplier/2)*0.75
+			
+			if i in correctedIndices and not math.isnan(nearbyMagUV[i]):
+				plot.circle(x=nearbyRA[i],y=nearbyDEC[i],radius=radius,line_color='blue',fill_color=None,legend_label='GALEX FUV detections (corrected)')
+				detections_made=True
+			elif i not in correctedIndices and not math.isnan(nearbyMagUV[i]):
+				plot.circle(x=nearbyRA[i],y=nearbyDEC[i],radius=radius,line_color='purple',fill_color=None,line_dash='dotted',legend_label='GALEX FUV detections (uncorrected)')
+				detections_made=True
+		
+	return plot, detections_made

AstroToolkit/Overlays/Overlay_Gaia.py

@@ -0,0 +1,63 @@
+import pandas as pd
+import math
+from astropy.time import Time
+
+from ..Surveys.Gaia import GaiaQueryCoords
+from ..Miscellaneous.ProperMotionCorrection import PMCorrection
+
+#Creates Gaia overlay
+def getGaiaOverlay(plot,ra,dec,sizeAS,mjd,border,pmra=None,pmdec=None):
+	gaiaTime=[2016,0]
+	imageTime=Time(mjd,format='mjd').to_datetime()
+	imageTime=[imageTime.year,imageTime.month]
+	
+	if pmra==None and pmdec==None:
+		pass
+	else:
+		sizeAS=PMCorrection(gaiaTime,imageTime,ra,dec,pmra,pmdec,radius=sizeAS)[2]
+
+	detections_made=False
+	nearby=GaiaQueryCoords(ra,dec,sizeAS,'dr3')
+	#Check if any data was returned by Gaia search
+	if isinstance(nearby,pd.DataFrame):
+		dataAvailability=True
+		nearbyRA=nearby.loc[:,'ra'].tolist()
+		nearbyDEC=nearby.loc[:,'dec'].tolist()
+		nearbyMagG=nearby.loc[:,'phot_g_mean_mag'].tolist()
+		nearbyPMRA=nearby.loc[:,'pmra'].tolist()
+		nearbyPMDEC=nearby.loc[:,'pmdec'].tolist()
+		
+		#Check if any g magnitudes exist
+		magCheck=False
+		for i in range(0,len(nearbyMagG)):
+			if not math.isnan(nearbyMagG[i]):
+				magCheck=True
+		
+		if magCheck==False:
+			return plot, detections_made
+		
+		#correct detection coordinates and list those that were corrected
+		correctedIndices=[]
+		for i in range(0,len(nearbyRA)):
+			if not math.isnan(nearbyPMRA[i]) and not math.isnan(nearbyPMDEC[i]):
+				RA_corrected,DEC_corrected=PMCorrection(gaiaTime,imageTime,nearbyRA[i],nearbyDEC[i],nearbyPMRA[i],nearbyPMDEC[i])
+				nearbyRA[i]=RA_corrected
+				nearbyDEC[i]=DEC_corrected
+				correctedIndices.append(i)
+		
+		#draw detections
+		for i in range(0,len(nearby)):
+			radiusMultiplier=(nearbyMagG[i]/20.7)
+			radius=(border/50+(border/75)**radiusMultiplier)*0.75
+			
+			if i in correctedIndices and not math.isnan(nearbyMagG[i]):
+				plot.circle(x=nearbyRA[i],y=nearbyDEC[i],radius=radius,line_color='red',fill_color=None,legend_label='Gaia g detections (corrected)')
+				detections_made=True
+			elif i not in correctedIndices and not math.isnan(nearbyMagG[i]):
+				plot.circle(x=nearbyRA[i],y=nearbyDEC[i],radius=radius,line_color='red',fill_color=None,line_dash='dotted',legend_label='Gaia g detections (uncorrected)')
+				detections_made=True
+	else:		
+		print('[PanSTARRS: getGaiaOverlay] Note: no Gaia detections found. This will limit accuracy of other detection overlays.')
+		return plot, detections_made
+	
+	return plot, detections_made

AstroToolkit/Overlays/Overlay_ROSAT.py

@@ -0,0 +1,20 @@
+import pandas as pd
+
+from ..Surveys.ROSAT import ROSATQueryCoords
+
+#Creates ROSAT overlay
+def getROSATOverlay(plot,ra,dec,sizeAS):
+	nearby=ROSATQueryCoords(ra,dec,sizeAS)
+	#checks if ROSAT query was successful
+
+	detections_made=False
+	if isinstance(nearby,pd.DataFrame):
+		nearbyRA=nearby.loc[:,'RAJ2000'].tolist()
+		nearbyDEC=nearby.loc[:,'DEJ2000'].tolist()
+		
+		#draws detections
+		for i in range(0,len(nearby)):
+			plot.cross(x=nearbyRA[i],y=nearbyDEC[i],color='orange',size=10,legend_label='ROSAT detections (uncorrected)')
+			detections_made=True
+	
+	return plot, detections_made

AstroToolkit/Overlays/Overlay_Selection.py

@@ -0,0 +1,42 @@
+from .Overlay_Gaia import getGaiaOverlay
+from .Overlay_GALEX import getGalexOverlayFUV
+from .Overlay_GALEX import getGalexOverlayNUV
+from .Overlay_ROSAT import getROSATOverlay
+from .Overlay_ZTF import getZTFOverlay
+
+def overlaySelection(plot,ra,dec,overlay,mjd,sizeAS,border,pmra=None,pmdec=None):
+	plot.cross(x=ra,y=dec,color='lime',size=10)
+	
+	overlay=['gaia']+overlay
+	
+	for i in range(0,len(overlay)):
+		overlay[i]=overlay[i].lower()
+	for i in range(0,len(overlay)):
+		if overlay[i] not in ['gaia','galex_nuv','rosat','ztf','galex_fuv']:
+			raise Exception('invalid overlay')
+
+	adaptive_correction=True
+	if pmra==None and pmdec==None:
+		adapative_correction=False
+
+	if 'gaia' in overlay:
+		if adaptive_correction==True:
+			plot,detections_made=getGaiaOverlay(plot=plot,ra=ra,dec=dec,sizeAS=sizeAS,mjd=mjd,border=border,pmra=pmra,pmdec=pmdec)
+		else:
+			plot,detections_made=getGaiaOverlay(plot=plot,ra=ra,dec=dec,sizeAS=sizeAS,mjd=mjd,border=border)
+	if 'galex_fuv' in overlay:
+		if adaptive_correction==True:
+			plot,detections_made=getGalexOverlayFUV(plot=plot,ra=ra,dec=dec,sizeAS=sizeAS,mjd=mjd,border=border,pmra=pmra,pmdec=pmdec)
+		else:
+			plot,detections_made=getGalexOverlayFUV(plot=plot,ra=ra,dec=dec,sizeAS=sizeAS,mjd=mjd,border=border)
+	if 'galex_nuv' in overlay:
+		if adaptive_correction==True:
+			plot,detections_made=getGalexOverlayNUV(plot=plot,ra=ra,dec=dec,sizeAS=sizeAS,mjd=mjd,border=border,pmra=pmra,pmdec=pmdec)
+		else:
+			plot,detections_made=getGalexOverlayNUV(plot=plot,ra=ra,dec=dec,sizeAS=sizeAS,mjd=mjd,border=border)
+	if 'rosat' in overlay:
+		plot,detections_made=getROSATOverlay(plot,ra,dec,sizeAS)
+	if 'ztf' in overlay:
+		plot,detections_made=getZTFOverlay(plot,ra,dec,sizeAS)
+	
+	return plot, detections_made

AstroToolkit/Overlays/Overlay_ZTF.py

@@ -0,0 +1,46 @@
+from ..Surveys.ZTF import getData
+import pandas as pd
+
+#Creates ZTF overlay
+def getZTFOverlay(plot,ra,dec,sizeAS):
+	detections_made=False
+	data=getData(ra,dec,sizeAS/2,'gri')
+	#Checks if data was returned and if so, splits into bands
+	if data!=None:
+		gData,rData,iData=data[0],data[1],data[2]
+	else:
+		return plot, detections_made
+	
+	#checks which bands have data
+	do_g,do_r,do_i=False,False,False
+	if isinstance(gData,pd.DataFrame):
+		do_g=True
+	elif isinstance(rData,pd.DataFrame):
+		do_r=True
+	elif isinstance(iData,pd.DataFrame):
+		do_i=True
+	
+	#draws detections
+	if do_g==True:
+		nearbyRA=gData.loc[:,'ra'].tolist()
+		nearbyDEC=gData.loc[:,'dec'].tolist()
+
+		for i in range(0,len(gData)):
+			plot.cross(x=nearbyRA[i],y=nearbyDEC[i],color='aqua',legend_label='ZTF detections (g) (uncorrected)')
+			detections_made=True
+	elif do_r==True:
+		nearbyRA=rData.loc[:,'ra'].tolist()
+		nearbyDEC=rData.loc[:,'dec'].tolist()
+		
+		for i in range(0,len(gData)):
+			plot.cross(x=nearbyRA[i],y=nearbyDEC[i],color='aqua',legend_label='ZTF detections (r) (uncorrected)')
+			detections_made=True
+	elif do_i==True:
+		nearbyRA=iData.loc[:,'ra'].tolist()
+		nearbyDEC=iData.loc[:,'dec'].tolist()
+		
+		for i in range(0,len(gData)):
+			plot.cross(x=nearbyRA[i],y=nearbyDEC[i],color='aqua',legend_label='ZTF detections (i) (uncorrected)')
+			detections_made=True
+	
+	return plot, detections_made

AstroToolkit/Surveys/DSS.py

@@ -0,0 +1,153 @@
+from astropy.io import fits
+from astropy.visualization import PercentileInterval, AsinhStretch
+from io import BytesIO
+import requests
+import numpy as np
+from astropy.wcs import WCS
+import warnings
+from astropy.utils.exceptions import AstropyWarning
+from astropy.time import Time
+import astropy
+from bokeh.plotting import figure
+from bokeh.models import Range1d
+
+
+#Ignore warning about deprecated header
+warnings.simplefilter('ignore',category=AstropyWarning)
+
+# DSS Query
+def get_image(ra,dec,size=30):
+	service='http://archive.stsci.edu/cgi-bin/dss_search'
+	url=f'{service}?ra={ra}&d={dec}&v=3&e=J2000&f=fits&h={size}&w={size}'
+	
+	try:
+		r=requests.get(url,timeout=15)
+		if r.status_code!=200:
+			print('[DSS: get_image_data] Error: experiencing issues with DSS.')
+			return None, None
+	except:
+		print('[DSS: get_image_data] Error: experiencing issues with DSS.')
+		return None, None
+	
+	try:
+		fh=fits.open(BytesIO(r.content))
+		fitsImage=fh[0].data
+		image_header=fh[0].header
+		fitsImage[np.isnan(fitsImage)]=0.0
+		transform=AsinhStretch()+PercentileInterval(95)
+		image=transform(fitsImage)
+		
+		if not isinstance(image,np.ndarray):
+			return None, None
+
+		return image,image_header
+	except:
+		print('[DSS: get_image_data] Error: no DSS images found using given fields')
+		return None, None
+
+# Filters inputs and performs any necessary arithmetic
+def input_management(ra,dec,size=30):
+	if not isinstance(ra,float):
+		try:
+			ra=float(ra)
+		except:
+			raise Exception('object RA must be a float or an integer')
+	if not isinstance(dec,float):
+		try:
+			dec=float(dec)
+		except:
+			raise Exception('object DEC must be a float or an integer')
+	
+	if not isinstance(size,int):
+		try:
+			size=int(size)
+		except:
+			raise Exception('size must take an integer value')
+	
+	sizeAS=size
+	size=size/60
+	
+	if size>60:
+		raise Exception('maximum size allowed by DSS is 60 arcminutes.')
+	
+	return ra, dec, size, sizeAS
+
+# Returns only the wcs projection for a retrieved image
+def get_info(ra,dec,size=30):
+	ra,dec,size,sizeAS=input_management(ra=ra,dec=dec,size=size)
+
+	image_data=get_image(ra=ra,dec=dec,size=size)[1]
+	
+	if not isinstance(image_data,astropy.io.fits.header.Header):
+		return None, None
+	
+	wcs=WCS(image_data)
+	
+	time=image_data['DATE-OBS']
+	mins=int(time[14:16])
+	hours=int(time[11:13])
+	if mins>=60:
+		mins=mins-60
+		hours+=1
+		
+	mins=str(mins).zfill(2)
+	hours=str(hours).zfill(2)
+	
+	time=time[0:10]+'T'+str(hours)+':'+str(mins)+':'+time[17:20]
+	
+	imageTime=Time(time,format='fits')
+	
+	mjd=imageTime.mjd
+	
+	return wcs, mjd
+
+# Sets up axis and casts image onto them
+def get_plot(ra,dec,size=30):
+	ra,dec,size,sizeAS=input_management(ra=ra,dec=dec,size=size)
+
+	# Create axes and plot image
+	plot=figure(width=400,height=400,title=f'DSS Image ({sizeAS}")')
+	# Fixes scaling issue
+	plot.min_border=75
+
+	plot.xgrid.grid_line_color=None
+	plot.ygrid.grid_line_color=None	
+
+	image,image_data=get_image(ra=ra,dec=dec,size=size)
+	
+	if not isinstance(image,np.ndarray) or not isinstance(image_data,astropy.io.fits.header.Header):
+		return None
+
+	# Get x and y limit (in pixels) from image header
+	xlim=image_data['NAXIS1']
+	ylim=image_data['NAXIS2']
+
+	# DSS images have one less pixel in one axis sometimes, so set both axes limits to that of the smaller one
+	lower_limit=min([xlim,ylim])
+	if lower_limit==xlim:
+		ylim=xlim
+	elif lower_limit==ylim:
+		xlim=ylim
+	
+	# Get points on axes in world coordinates (deg)
+	points=[]
+	wcs=WCS(image_data)
+	
+	# Get array of pixels in x and y
+	x_points=np.arange(start=0,stop=xlim+1,step=1)
+	y_points=np.arange(start=0,stop=ylim+1,step=1)
+
+	# Apply transformation to coordinates
+	coords=wcs.all_pix2world(x_points,y_points,1)
+	x_points,y_points=coords[0],coords[1]
+	
+	# Get image size in degrees
+	x_range=max(x_points)-min(x_points)
+	y_range=max(y_points)-min(y_points)
+
+	plot.x_range=Range1d(max(x_points),min(x_points))
+	plot.y_range=Range1d(min(y_points),max(y_points))
+
+	plot.image(image=[image],x=x_points[0],y=y_points[0],dw=x_range,dh=y_range,palette='Greys256',level="image",origin='bottom_right',anchor='bottom_right')
+	
+	return plot

AstroToolkit/Surveys/GALEX.py

@@ -0,0 +1,45 @@
+import pandas as pd
+import astropy.coordinates as coord
+from astroquery.vizier import Vizier
+import astropy.units as u
+
+catalogue_name='II/335/galex_ais'
+
+Vizier.ROW_LIMIT = -1
+
+def GALEXQueryCoords(ra,dec,radius=3):
+	data=[]
+	
+	if not isinstance(radius,int) and not isinstance(radius,float):
+		raise Exception('search radius must be a float or an integer')
+	
+	if not isinstance(ra,float):
+		try:
+			ra=float(ra)
+		except:
+			raise Exception('object RA must be a float or an integer')
+	if not isinstance(dec,float):
+		try:
+			dec=float(dec)
+		except:
+			raise Exception('object DEC must be a float or an integer')
+	
+	v=Vizier(columns=['**'])
+	
+	#Sends Vizier query, can return multiple objects in given search radius
+	data.append(v.query_region(coord.SkyCoord(ra=ra,dec=dec,unit=(u.deg,u.deg),frame='icrs'),width=radius*u.arcsec,catalog=catalogue_name))
+	if len(data[0])==0:
+		print('[GALEX: GALEXQueryCoords] Error: no GALEX data found at given coordinates')
+		return None
+	data=data[0][0].to_pandas()
+	data=data.reset_index(drop=True)
+	return data
+	
+def GALEXGetPhotometryCoords(ra,dec,radius=3):
+	GALEXdata=GALEXQueryCoords(ra,dec,radius=radius)
+	if not isinstance(GALEXdata,pd.DataFrame) or GALEXdata.empty:
+		return None
+	
+	photometry=GALEXdata[['RAJ2000','DEJ2000','objid','NUVmag','e_NUVmag','FUVmag','e_FUVmag']].copy()
+
+	return photometry