GAIA-HR 1.1.0__tar.gz

gaia_hr-1.1.0/GAIA_HR/__init__.py

@@ -0,0 +1 @@
+

gaia_hr-1.1.0/GAIA_HR/plot.py

@@ -0,0 +1,149 @@
+import matplotlib.pyplot as plt
+import pandas as pd
+import numpy as np
+from matplotlib.colors import LogNorm
+
+#Applies matplotlib's built in dark theme
+plt.style.use(["dark_background"])
+
+def plot_hr(df, property_name = "bp_rp", cmap="coolwarm", log_plot=False):
+    """
+    The plot_hr function plots the HR diagram , using the data stored in dataframe. The stars are plotted with bp-rp on x axis and absolute magnitude on y axis. 
+    Additonal properties(tempreature , surface gravity , radial velocity, metallicity,proper motion, mass , radius) of the stars can be compared using colormap.The 
+    function supports both linear and logarithmic scaling
+    
+    Args:
+        df(pandas.DataFrame):
+                        A pandas dataset containing stellar data. 
+                        The data set must contain bp-rp, absolute magnitude
+        property_name(str, optional):
+                        Default: "bp-rp"
+                        Specifies which property of the stellar data will be used for the color map 
+        cmap(str,optional):
+                        Default: "coolwarm"
+                        Specifies the matplotlib colormap used for the stars 
+        log_plot(boolean,optional):
+                        Default: False
+                        To determine whether colorscale to be logarithmic or linear
+                        True : logarithmic color scale 
+                        False : linear color scale
+    Return:
+            matplotlib.figure.Figure
+            The matplotlib figure object showing the H-R diagram using plt.show()
+    """
+
+    # Creates a new figure and axes object
+    fig, ax = plt.subplots(figsize=(12, 10))
+
+    #If log_plot is true the scatter plot uses logarithimic normalization                       
+    if log_plot:
+        scatter = ax.scatter(
+            df["bp_rp"],
+            df["mag"],
+            c=df[property_name],
+            cmap=cmap,
+            s=2,
+            norm = LogNorm(vmax=np.nanpercentile(df[property_name], 99)))
+    
+    else:
+        scatter = ax.scatter(
+            df["bp_rp"],
+            df["mag"],
+            c=df[property_name],
+            cmap=cmap,
+            s=2,
+            vmax=np.nanpercentile(df[property_name], 99)) # Gives maximum color to top outliers
+            
+    # In H-R diagram lower magnitude represents brighter stars , hence y axis is reverted
+    ax.invert_yaxis() 
+
+   # Sets X-axis label to BP-RP
+    ax.set_xlabel("BP - RP", fontsize=15)
+   # Sets Y-axis label to Absolute magnitude
+    ax.set_ylabel("Absolute Magnitude", fontsize=15)
+
+   # Creates the title for the property currently being displayed by the colormap
+    ax.set_title(
+        f"HR Diagram Colored by {property_name}",
+        fontsize=15
+    )
+
+   # Creates a color bar associated witht the scatter plot 
+    cbar = plt.colorbar(scatter)
+   # Labels the color bar according to the property 
+    cbar.set_label(property_name, fontsize=12)
+
+    # Renders the entire H-R diagram on the screen
+    plt.show()
+
+
+
+def plot_radec(df, property_name = "bp_rp", cmap="coolwarm", log_plot=False):
+    """
+    The plot_radec function plots a 2D plot of region , using the data stored in dataframe. The stars are plotted with right ascension on x axis and declination on y axis. 
+    Additonal properties(tempreature , surface gravity , radial velocity, metallicity,proper motion, mass , radius) of the stars can be compared using colormap.The 
+    function supports both linear and logarithmic scaling
+
+    Args:
+        df(pandas.DataFrame):
+                        A pandas dataset containing stellar data. 
+                        The data set must contain bp-rp, absolute magnitude
+        property_name(str, optional):
+                        Default: "bp-rp"
+                        Specifies which property of the stellar data will be used for the color map 
+        cmap(str,optional):
+                        Default: "coolwarm"
+                        Specifies the matplotlib colormap used for the stars 
+        log_plot(boolean,optional):
+                        Default: False
+                        To determine whether colorscale to be logarithmic or linear
+                        True :logarithmic color scale 
+                        False : linear color scale
+    Return:
+            matplotlib.figure.Figure
+            The matplotlib figure object showing the H-R diagram using plt.show()
+    """
+
+    # Creates a new figure and axes object
+    fig, ax = plt.subplots(figsize=(12, 10))
+
+    #If log_plot is true the scatter plot uses logarithimic normalization                       
+    if log_plot:
+        scatter = ax.scatter(
+            df["ra"],
+            df["dec"],
+            c=df[property_name],
+            cmap=cmap,
+            s=2,
+            norm = LogNorm(vmax=np.nanpercentile(df[property_name], 99)))
+    
+    else:
+        scatter = ax.scatter(
+            df["ra"],
+            df["dec"],
+            c=df[property_name],
+            cmap=cmap,
+            s=2,
+            vmax=np.nanpercentile(df[property_name], 99)) # Gives maximum color to top outliers
+            
+    # In H-R diagram lower magnitude represents brighter stars , hence y axis is reverted
+    ax.invert_yaxis() 
+
+   # Sets X-axis label to BP-RP
+    ax.set_xlabel("Right Ascension (deg)", fontsize=15)
+   # Sets Y-axis label to Absolute magnitude
+    ax.set_ylabel("Declination (deg)", fontsize=15)
+
+   # Creates the title for the property currently being displayed by the colormap
+    ax.set_title(
+        f"HR Diagram Colored by {property_name}",
+        fontsize=15
+    )
+
+   # Creates a color bar associated witht the scatter plot 
+    cbar = plt.colorbar(scatter)
+   # Labels the color bar according to the property 
+    cbar.set_label(property_name, fontsize=12)
+
+    # Renders the entire H-R diagram on the screen
+    plt.show()

gaia_hr-1.1.0/GAIA_HR/query.py

@@ -0,0 +1,409 @@
+# Importing necessary packages
+from astroquery.utils.tap.core import TapPlus
+import pandas as pd
+import numpy as np
+from astroquery.gaia import Gaia
+import pyvo
+import os, sys
+
+
+
+class HiddenPrints:
+    """Print Enabler/Disabler
+    
+    Context Manager to prevent print statement from executing within the statement setup. (Source: https://stackoverflow.com/questions/8391411)
+    """
+
+    def __enter__(self):
+        """Enter function
+        
+        Disables the print statement from executing when placed within the HiddenPrints statement.
+        """
+
+        self._original_stdout = sys.stdout
+        sys.stdout = open(os.devnull, 'w')
+
+    def __exit__(self, exc_type, exc_value, exc_traceback):
+        """Exit function
+        
+        Enables the print statement during runs when outside the HiddenPrints statement.
+
+        Args:
+            exc_type: indicates class of exception. 
+            exc_value: indicates type of exception . like divide_by_zero error, floating_point_error, which are types of arithmetic exception. 
+            exc_traceback: traceback is a report which has all of the information needed to solve the exception.
+        """
+        sys.stdout.close()
+        sys.stdout = self._original_stdout
+
+class ALL_SERVER:
+    """GAIA Server Status Checker
+
+    Checks if the GAIA servers are connecting to the system using a simple ADQL query and returns the connection status.
+    """
+    def check_gaia_server(self):
+        """ESA GAIA server checker
+
+        Function to verify the ESA GAIA server responsiveness.
+        """
+        try:
+            job = Gaia.launch_job("SELECT TOP 1 * FROM gaiadr3.gaia_source")
+            results = job.get_results()
+
+            print("GAIA Server is up and responding properly!")
+            print(f"Successfully retrieved {len(results)} row(s).")
+
+            return 1
+
+        except Exception as e:
+            print("GAIA Server is not responding")
+            print(f"Error details: {e}")
+
+    def check_vizier_server(self):
+        """TAPVizieR server checker
+
+        Function to verify the TAPVizier GAIA server responsiveness.
+        """
+        url = "http://tapvizier.u-strasbg.fr/TAPVizieR/tap"
+        print(f"Pinging {url}...")
+
+        try:
+            # Initialize connection
+            vizier_tap = TapPlus(url=url)
+
+            # Run the lightest possible query (just 1 row)
+            job = vizier_tap.launch_job("SELECT TOP 1 * FROM \"I/355/gaiadr3\"")
+            results = job.get_results()
+            
+            print("VIZIER Server is UP and responding properly!")
+            print(f"Successfully retrieved {len(results)} row(s).")
+
+            return 1
+            
+        except Exception as e:
+            print("VIZIER Server appears to be DOWN or unreachable.")
+            print(f"Error details: {e}")
+
+    def check_ari_server(self):
+        """ARI server checker
+
+        Function to verify the ARI GAIA server responsiveness.
+        """
+        url = "https://gaia.ari.uni-heidelberg.de/tap"
+        print(f"Pinging {url}...")
+
+        try:
+            # Initialize connection
+            ari_tap = TapPlus(url=url)
+            
+            # Run the lightest possible query (just 1 row)
+            job = ari_tap.launch_job("SELECT TOP 1 * FROM gaiadr3.gaia_source_lite")
+            results = job.get_results()
+            
+            print("ARI (Heidelberg) Server is UP and responding properly!")
+            print(f"Successfully retrieved {len(results)} row(s).")
+
+            return 1
+            
+        except Exception as e:
+            print("ARI Server appears to be DOWN or unreachable.")
+            print(f"Error details: {e}")
+
+    def check_aip_server(self):
+        """AIP server checker
+
+        Function to verify the AIP GAIA server responsiveness.
+        """
+        url = "https://gaia.aip.de/tap"
+        print(f"Pinging {url}...")
+
+        try:
+            # Initialize connection using pyvo
+            service = pyvo.dal.TAPService(url)
+            
+            # Run the lightest possible query (just 1 row)
+            # Force the format parameter directly on execution to bypass strict header negotiation 
+            # (asked by AIP server while using TapPlus)
+            job = service.search("SELECT TOP 1 * FROM gaiadr3.gaia_source_lite", response_format='votable')
+            
+            print("AIP (Potsdam) Server is UP and responding properly!")
+            print(f"Successfully retrieved {len(job)} row(s).")
+
+            return 1
+ 
+        except Exception as e:
+            print("AIP Server appears to be DOWN or unreachable.")
+            print(f"Error details: {e}")
+        
+    def check_all_server(self):
+        """ Check the responsiveness of all GAIA servers
+
+        Verifies whether all the servers are responding and provides exceptions if it fails.
+        """
+        self.check_gaia_server()
+        print("\n")
+        self.check_vizier_server()
+        print("\n")
+        self.check_ari_server()
+        print("\n")
+        self.check_aip_server()
+
+
+# Dictionaries to map columns from different servers to rename with a common name.
+
+column_mapping = {'source_id':'sid', 
+                  'ra':'ra', 
+                  'dec':'dec', 
+                  'parallax':'parallax',
+                  'phot_g_mean_mag':'g_mean_mag', 
+                  'bp_rp':'bp_rp', 
+                  'teff_gspphot':'teff',
+                  'logg_gspphot':'logg', 
+                  'mh_gspphot': 'mh',
+                  'pm':'pm', 
+                  'radial_velocity':'rv', 
+                  'lum_flame':'lum_flame',
+                  'radius_flame':'radius_flame', 
+                  'mass_flame':'mass_flame',}
+
+column_mapping_Vizier = {'Source':'sid', 
+                         'RA_ICRS':'ra', 
+                         'DE_ICRS':'dec', 
+                         'Plx':'parallax',
+                         'Gmag':'g_mean_mag', 
+                         'BP-RP':'bp_rp', 
+                         'Teff':'teff',
+                         'logg':'logg', 
+                         '[Fe/H]': 'mh',
+                         'PM':'pm', 
+                         'RV':'rv', 
+                         'Lum-Flame':'lum_flame',
+                         'Rad-Flame':'radius_flame', 
+                         'Mass-Flame':'mass_flame',}
+
+def fetch_gaia_data(ra, dec, radius, d_max = None, d_min = None, max_source = 10000, server = None, save_file = False, filename="gaia_data.csv"):
+    """Fetches a sample of star data from the Gaia DR3 dataset.
+    
+    Executes ADQL query to retrive sources with parallax_over_error > 20 from user specified region.
+    Joins standard astrometric data with evaluated astrophysical parameters for available sources.
+    Function utilises a fallback mechanism, attempting to connect to multiple servers until successful
+    connections and query are made.
+
+    Args:
+        ra (float): Right Ascension of region you want to query (in degrees)
+        dec (float): Declination of region you want to query (in degrees)
+        radius (float): Radius of region you want to query (in degrees)
+        d_max (float, optional): 
+                            Default: None
+                            Maximum distance to sources (in lightyears)
+        d_min (float, optional):
+                            Default: None
+                            Minimum diatance to sources (in lightyears)
+        max_sources (int, optional):
+                            Default: 10000
+                            Maximum number of sources to be queried
+        server (str, optional):
+                            Default: None, Autoselects server
+                            Name of server for querying. Available options : "gaia", "ari", "aip", "vizier"
+        save_file (bool, optional):
+                            Default: False
+                            Boolean to save fetched data as csv file
+        filename (str, optional):
+                            Default: "gaia_data.csv"
+                            Filename of saved csv file
+
+    
+    Returns:
+        pandas.DataFrame: gaia star data
+    """
+
+    # Check to verify the entered RA, DEC, radius falls within their defined range.
+    if (ra < 0) or (ra > 360):
+        raise ValueError("The specified value of RA do not exist in the celestial system bar the server issue.")
+    elif (dec < -90) or (dec > 90):
+        raise ValueError("The specified value of DEC do not exist in the celestial system bar the server issue.")
+    elif (radius < 0) or (radius > 180):
+        raise ValueError("The specified value of radius is not Queryable or the ADQL query is not accepting the assigned value.")
+    else:
+        pass
+
+    if (max_source < 0):
+        print("Invalid max_source input. Defaulting to 10000")
+        max_source = 10000
+    elif (max_source > 1000000):
+        raise ValueError("Bruh! What is wrong with you? Do you want to crash you pc?")
+
+
+    conditions = ["AND gs.parallax_over_error > 20"]
+
+    if d_max is not None:
+        min_parallax = 3261.56 / d_max
+        conditions.append(f"\n    AND gs.parallax >= {min_parallax}")
+    if d_min is not None:
+        max_parallax = 3261.56 / d_min
+        conditions.append(f"\n    AND gs.parallax <= {max_parallax}")
+
+
+    # Query construction from user input
+    query = f"""
+    SELECT TOP {max_source}
+        gs.source_id,
+        gs.ra,
+        gs.dec,
+        gs.parallax,
+        gs.phot_g_mean_mag,
+        gs.bp_rp,
+        gs.teff_gspphot,
+        gs.logg_gspphot,
+        gs.mh_gspphot,
+        gs.pm,
+        gs.radial_velocity,
+        ap.lum_flame,
+        ap.radius_flame,
+        ap.mass_flame
+    FROM gaiadr3.gaia_source AS gs
+    JOIN gaiadr3.astrophysical_parameters AS ap
+    ON gs.source_id = ap.source_id
+    WHERE 1 = CONTAINS(
+    POINT('ICRS', gs.ra, gs.dec),
+    CIRCLE('ICRS', {ra}, {dec}, {radius})
+     )
+    """
+
+    
+
+    for condition in conditions:
+        query += condition
+
+
+    # Server 1 : esa.gaia
+    if server is None or server == "gaia":
+        if ALL_SERVER().check_gaia_server() == 1 :
+
+            print("Connecting to main Gaia server")
+
+            job = Gaia.launch_job_async(query)
+            stars = job.get_results()
+            df = stars.to_pandas()
+            
+            # Rename column names to have a common format
+            df = df.rename(columns=column_mapping, errors = 'raise')
+            df["mag"] = df["g_mean_mag"] + 5 + 5 * np.log10(df["parallax"] / 1000)
+
+            if save_file:
+                df.to_csv(filename, index=False)
+
+            return df
+        
+        else:
+            raise ConnectionError("Gaia server is not responding. Kindly try again later or check another server!!")
+    
+
+    # Server 2 : gaia.ari
+    if server is None or server == "ari":
+        if ALL_SERVER().check_ari_server() == 1:
+            url = "https://gaia.ari.uni-heidelberg.de/tap"
+            ari_tap = TapPlus(url=url)
+
+            print("Connecting to Heidelberg server")
+
+            job = ari_tap.launch_job_async(query)
+            stars = job.get_results()
+            df = stars.to_pandas()
+
+            # Rename column names to have a common format
+            df = df.rename(columns=column_mapping, errors = 'raise')
+            df["mag"] = df["g_mean_mag"] + 5 + 5 * np.log10(df["parallax"] / 1000)
+
+            if save_file:
+                df.to_csv(filename, index=False)
+            
+            return df
+
+        else:
+            raise ConnectionError("ARI Heidelberg server is not responding. Kindly try again later or check another server!!")
+
+
+    # Server 3 : gaia.aip
+    if server is None or server == "aip":
+        if ALL_SERVER().check_aip_server() == 1:
+        
+            url = "https://gaia.aip.de/tap"
+            service = pyvo.dal.TAPService(url)
+
+            print("Connecting to Potsdam server")
+
+            job = service.search(query, response_format='votable')
+            astropy_table = job.to_table()
+            df = df = astropy_table.to_pandas()
+
+            # Rename column names to have a common format
+            df = df.rename(columns=column_mapping, errors = 'raise')
+            df["mag"] = df["g_mean_mag"] + 5 + 5 * np.log10(df["parallax"] / 1000)
+
+            if save_file:
+                df.to_csv(filename, index=False)
+
+            return df
+
+        else:
+            raise ConnectionError("Potsdam AIP server is not responding. Kindly try again later or check another server!!")
+    
+
+    # Server 4 : vizier 
+    if server is None or server == "vizier":
+        if ALL_SERVER().check_vizier_server() == 1:
+        
+            query = f"""
+            SELECT TOP {max_source}
+                gs.Source,
+                gs.RA_ICRS,
+                gs.DE_ICRS,
+                gs.Plx,
+                gs.Gmag,
+                gs."BP-RP",
+                gs.Teff,
+                gs.logg,
+                gs."[Fe/H]",
+                gs.PM,
+                gs.RV,
+                ap."Lum-Flame",
+                ap."Rad-Flame",
+                ap."Mass-Flame"
+            FROM "I/355/gaiadr3" AS gs
+            JOIN "I/355/paramp" AS ap
+            ON gs.Source = ap.Source
+            WHERE 1 = CONTAINS(
+                POINT('ICRS', gs.RA_ICRS, gs.DE_ICRS),
+                CIRCLE('ICRS', {ra}, {dec}, {radius})
+            )
+            AND RPlx > 20
+            """
+            
+            url = "http://tapvizier.u-strasbg.fr/TAPVizieR/tap"
+            print(f"Connecting to {url}...")
+            
+            vizier_tap = TapPlus(url=url)
+            
+            print("Executing query...")
+            job = vizier_tap.launch_job(query)
+            results = job.get_results()
+
+            df = results.to_pandas()
+
+            # Rename column names to have a common format
+            df = df.rename(columns=column_mapping_Vizier, errors = 'raise')
+            df["mag"] = df["g_mean_mag"] + 5 + 5 * np.log10(df["parallax"] / 1000)
+
+            if save_file:
+                df.to_csv(filename, index=False)
+
+            return  df
+
+        else:
+            raise ConnectionError("TAPVizier server is not responding. Kindly try again later or check another server!!")
+
+
+    # Edge case: No server response
+    else:
+        raise ConnectionError("No servers are responding. Kindly try again later!!")

gaia_hr-1.1.0/GAIA_HR.egg-info/PKG-INFO

@@ -0,0 +1,54 @@
+Metadata-Version: 2.4
+Name: GAIA_HR
+Version: 1.1.0
+Summary: Python package to plot Hertzsprung-Russel diagrams from gdr3 data
+Author-email: Midhun Goutham Murali <midhungm8101@gmail.com>, Nirmal US <nirmalus24@gmail.com>, Anansh Jain <ananshjain10@gmail.com>
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: matplotlib
+Requires-Dist: astroquery
+Requires-Dist: pyvo
+Requires-Dist: pandas
+Dynamic: license-file
+
+# GAIA_HR
+
+Library that plots HR diagram from GAIA dataset
+
+[![A rectangular badge, half black half purple containing the text made at Code Astro](https://img.shields.io/badge/Made%20at-Code/Astro-blueviolet.svg)](https://semaphorep.github.io/codeastro/)
+## Motivation
+
+This package queries data and plots Hertzsprung-Russell diagram for a user specified region of the sky (ra, dec, radius) from Gaia data release 3 [1]. It also compares  a user specified parameter of different stars in the region using a colormap.
+
+## Installation
+
+The package is installable on Python 3.x. To install the package, simply write
+
+`pip install Gaia_HR`
+
+Otherwise, clone this repo, and follow the below specified commands
+
+`cd Gaia_HR`
+
+`pip install -e .`
+
+A list of dependencies is available in requirements.txt
+
+## Additional Information
+
+Gaia HR diagrams is valid when the the relative precision on parallax is lower than 20%. This uncertainity has been included by using the parameter "parallax_over_error > 20" during the query [2].
+
+This work has made use of data from the European Space Agency (ESA) mission Gaia (https://www.cosmos.esa.int/gaia), processed by the Gaia Data Processing and Analysis Consortium (DPAC, https://www.cosmos.esa.int/web/gaia/dpac/consortium). Funding for the DPAC has been provided by national institutions, in particular the institutions participating in the Gaia Multilateral Agreement [1, 3].
+
+This research has made use of the VizieR catalogue access tool, CDS, Strasbourg, France [4]. The original description of the VizieR service was published in 2000, A&AS 143, 23.
+
+## References:
+
+1. Gaia Data Release 3 - Summary of the content and survey properties. Gaia Collaboration, A.  Vallenari, et al. A&A 674 A1 (2023). DOI: [10.1051/0004-6361/202243940](https://doi.org/10.1051/0004-6361/202243940)
+
+2. Gaia Data Release 2 - Observational Hertzsprung-Russell diagrams. Gaia Collaboration, C.  Babusiaux, et al. A&A 616 A10 (2018). DOI: [10.1051/0004-6361/201832843](https://doi.org/10.1051/0004-6361/201832843)
+
+3. The Gaia mission. Gaia Collaboration, T.  Prusti, et al. A&A 595 A1 (2016). DOI: [10.1051/0004-6361/201629272](https://doi.org/10.1051/0004-6361/201629272)
+
+4. The VizieR database of astronomical catalogues. F.  Ochsenbein, P.  Bauer, J.  Marcout. Astron. Astrophys. Suppl. Ser. 143 (1) 23-32 (2000). [DOI: 10.1051/aas:2000169](https://doi.org/10.1051/aas:2000169)

gaia_hr-1.1.0/GAIA_HR.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+LICENSE
+README.md
+pyproject.toml
+GAIA_HR/__init__.py
+GAIA_HR/plot.py
+GAIA_HR/query.py
+GAIA_HR.egg-info/PKG-INFO
+GAIA_HR.egg-info/SOURCES.txt
+GAIA_HR.egg-info/dependency_links.txt
+GAIA_HR.egg-info/requires.txt
+GAIA_HR.egg-info/top_level.txt
+test/test_GAIA_HR.py

gaia_hr-1.1.0/GAIA_HR.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

gaia_hr-1.1.0/GAIA_HR.egg-info/requires.txt

@@ -0,0 +1,5 @@
+numpy
+matplotlib
+astroquery
+pyvo
+pandas

gaia_hr-1.1.0/GAIA_HR.egg-info/top_level.txt

@@ -0,0 +1 @@
+GAIA_HR

gaia_hr-1.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 NirmalUS, AnanshJ, Midhu4
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

gaia_hr-1.1.0/PKG-INFO

@@ -0,0 +1,54 @@
+Metadata-Version: 2.4
+Name: GAIA_HR
+Version: 1.1.0
+Summary: Python package to plot Hertzsprung-Russel diagrams from gdr3 data
+Author-email: Midhun Goutham Murali <midhungm8101@gmail.com>, Nirmal US <nirmalus24@gmail.com>, Anansh Jain <ananshjain10@gmail.com>
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: matplotlib
+Requires-Dist: astroquery
+Requires-Dist: pyvo
+Requires-Dist: pandas
+Dynamic: license-file
+
+# GAIA_HR
+
+Library that plots HR diagram from GAIA dataset
+
+[![A rectangular badge, half black half purple containing the text made at Code Astro](https://img.shields.io/badge/Made%20at-Code/Astro-blueviolet.svg)](https://semaphorep.github.io/codeastro/)
+## Motivation
+
+This package queries data and plots Hertzsprung-Russell diagram for a user specified region of the sky (ra, dec, radius) from Gaia data release 3 [1]. It also compares  a user specified parameter of different stars in the region using a colormap.
+
+## Installation
+
+The package is installable on Python 3.x. To install the package, simply write
+
+`pip install Gaia_HR`
+
+Otherwise, clone this repo, and follow the below specified commands
+
+`cd Gaia_HR`
+
+`pip install -e .`
+
+A list of dependencies is available in requirements.txt
+
+## Additional Information
+
+Gaia HR diagrams is valid when the the relative precision on parallax is lower than 20%. This uncertainity has been included by using the parameter "parallax_over_error > 20" during the query [2].
+
+This work has made use of data from the European Space Agency (ESA) mission Gaia (https://www.cosmos.esa.int/gaia), processed by the Gaia Data Processing and Analysis Consortium (DPAC, https://www.cosmos.esa.int/web/gaia/dpac/consortium). Funding for the DPAC has been provided by national institutions, in particular the institutions participating in the Gaia Multilateral Agreement [1, 3].
+
+This research has made use of the VizieR catalogue access tool, CDS, Strasbourg, France [4]. The original description of the VizieR service was published in 2000, A&AS 143, 23.
+
+## References:
+
+1. Gaia Data Release 3 - Summary of the content and survey properties. Gaia Collaboration, A.  Vallenari, et al. A&A 674 A1 (2023). DOI: [10.1051/0004-6361/202243940](https://doi.org/10.1051/0004-6361/202243940)
+
+2. Gaia Data Release 2 - Observational Hertzsprung-Russell diagrams. Gaia Collaboration, C.  Babusiaux, et al. A&A 616 A10 (2018). DOI: [10.1051/0004-6361/201832843](https://doi.org/10.1051/0004-6361/201832843)
+
+3. The Gaia mission. Gaia Collaboration, T.  Prusti, et al. A&A 595 A1 (2016). DOI: [10.1051/0004-6361/201629272](https://doi.org/10.1051/0004-6361/201629272)
+
+4. The VizieR database of astronomical catalogues. F.  Ochsenbein, P.  Bauer, J.  Marcout. Astron. Astrophys. Suppl. Ser. 143 (1) 23-32 (2000). [DOI: 10.1051/aas:2000169](https://doi.org/10.1051/aas:2000169)

gaia_hr-1.1.0/README.md

@@ -0,0 +1,40 @@
+# GAIA_HR
+
+Library that plots HR diagram from GAIA dataset
+
+[![A rectangular badge, half black half purple containing the text made at Code Astro](https://img.shields.io/badge/Made%20at-Code/Astro-blueviolet.svg)](https://semaphorep.github.io/codeastro/)
+## Motivation
+
+This package queries data and plots Hertzsprung-Russell diagram for a user specified region of the sky (ra, dec, radius) from Gaia data release 3 [1]. It also compares  a user specified parameter of different stars in the region using a colormap.
+
+## Installation
+
+The package is installable on Python 3.x. To install the package, simply write
+
+`pip install Gaia_HR`
+
+Otherwise, clone this repo, and follow the below specified commands
+
+`cd Gaia_HR`
+
+`pip install -e .`
+
+A list of dependencies is available in requirements.txt
+
+## Additional Information
+
+Gaia HR diagrams is valid when the the relative precision on parallax is lower than 20%. This uncertainity has been included by using the parameter "parallax_over_error > 20" during the query [2].
+
+This work has made use of data from the European Space Agency (ESA) mission Gaia (https://www.cosmos.esa.int/gaia), processed by the Gaia Data Processing and Analysis Consortium (DPAC, https://www.cosmos.esa.int/web/gaia/dpac/consortium). Funding for the DPAC has been provided by national institutions, in particular the institutions participating in the Gaia Multilateral Agreement [1, 3].
+
+This research has made use of the VizieR catalogue access tool, CDS, Strasbourg, France [4]. The original description of the VizieR service was published in 2000, A&AS 143, 23.
+
+## References:
+
+1. Gaia Data Release 3 - Summary of the content and survey properties. Gaia Collaboration, A.  Vallenari, et al. A&A 674 A1 (2023). DOI: [10.1051/0004-6361/202243940](https://doi.org/10.1051/0004-6361/202243940)
+
+2. Gaia Data Release 2 - Observational Hertzsprung-Russell diagrams. Gaia Collaboration, C.  Babusiaux, et al. A&A 616 A10 (2018). DOI: [10.1051/0004-6361/201832843](https://doi.org/10.1051/0004-6361/201832843)
+
+3. The Gaia mission. Gaia Collaboration, T.  Prusti, et al. A&A 595 A1 (2016). DOI: [10.1051/0004-6361/201629272](https://doi.org/10.1051/0004-6361/201629272)
+
+4. The VizieR database of astronomical catalogues. F.  Ochsenbein, P.  Bauer, J.  Marcout. Astron. Astrophys. Suppl. Ser. 143 (1) 23-32 (2000). [DOI: 10.1051/aas:2000169](https://doi.org/10.1051/aas:2000169)

gaia_hr-1.1.0/pyproject.toml

@@ -0,0 +1,22 @@
+[build-system]
+requires = ["setuptools >= 61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "GAIA_HR"
+version = "1.1.0"
+description = "Python package to plot Hertzsprung-Russel diagrams from gdr3 data"
+readme = "README.md"
+
+dependencies = [
+  "numpy",
+  "matplotlib",
+  "astroquery",
+  "pyvo",
+  "pandas"
+]
+authors = [
+  {name = "Midhun Goutham Murali", email = "midhungm8101@gmail.com"},
+  {name = "Nirmal US", email = "nirmalus24@gmail.com"},
+  {name = "Anansh Jain", email = "ananshjain10@gmail.com"},
+]

gaia_hr-1.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

gaia_hr-1.1.0/test/test_GAIA_HR.py

@@ -0,0 +1,48 @@
+import pytest
+import pandas as pd
+from GAIA_HR import query
+from GAIA_HR import plot
+from unittest.mock import patch
+
+
+def test_fetch_gaia_data():
+    """
+    Unit test to evaluate correctness of fetch_gaia_data() function
+    """
+    # Pleiades
+    df = query.fetch_gaia_data(56.75, 24.12, 2, d_min=300, d_max=600, max_source=20000, server="aip")
+
+    assert df.shape[0] == 1419
+    assert df.shape[1] == 15
+
+
+def test_plot_hr():
+    """
+    Unit test to evaluate correctness of plot_hr() function
+    """
+    df = pd.read_csv("Sample_data.csv")
+    with patch("matplotlib.pyplot.show"):
+        plot.plot_hr(df, property_name="radius_flame", cmap="plasma", log_plot=True)
+
+
+def test_plot_radec():
+    """
+    Unit test to evaluate correctness of plot_radec() function
+    """
+    df = pd.read_csv("Sample_data.csv")
+    with patch("matplotlib.pyplot.show"):
+        plot.plot_radec(df, property_name="radius_flame", cmap="plasma", log_plot=True)    
+
+def end_to_end_test():
+    """
+    End-to-end test to evaluate correctness of GAIA_HR module
+    """
+    df = query.fetch_gaia_data(56.75, 24.12, 2, d_min=300, d_max=600, max_source=20000, server="aip")
+    
+    with patch("matplotlib.pyplot.show"):
+        plot.plot_hr(df, property_name="radius_flame", cmap="plasma", log_plot=True)
+    with patch("matplotlib.pyplot.show"):
+        plot.plot_radec(df, property_name="radius_flame", cmap="plasma", log_plot=True) 
+
+    assert df.shape[0] == 1419
+    assert df.shape[1] == 15