exocat 0.1.0__py3-none-any.whl

exocat/__init__.py

@@ -0,0 +1,4 @@
+from .bulk_density import BulkDensity
+from .temperature import Temperature
+
+__version__ = "0.1.0"

exocat/bulk_density.py

@@ -0,0 +1,69 @@
+#%%
+import numpy as np
+import matplotlib.pyplot as plt
+import pandas as pd
+
+class BulkDensity:
+    """A class to represent and classify bulk density values."""
+
+    def __init__(self, df: pd.DataFrame|np.ndarray|list) -> None:
+
+        if isinstance(df, pd.DataFrame):
+            self.df = df
+            self.density = np.atleast_1d(np.asarray(df["rho_b"]))
+        else:
+            self.df = None
+            self.density = np.atleast_1d(np.asarray(df))
+
+        self.classifications = None
+
+    def classify(self) -> np.ndarray|str:
+
+        density_jup = 1.326
+        density_gcm3 = self.density * density_jup
+        
+        output = np.empty_like(density_gcm3, dtype=object)
+        output[density_gcm3 < 0.5] = "gaseous"
+        output[(density_gcm3 >= 0.5) & (density_gcm3 < 2)] = "gaseous/water-rich"
+        output[(density_gcm3 >= 2) & (density_gcm3 < 7)] = "rocky"
+        output[(density_gcm3 >= 7) & (density_gcm3 < 13)] = "metallic"
+        output[density_gcm3 >= 13] = "super dense"
+        self.classifications = output
+
+        return self.classifications
+
+    def plot(self, bins:int=50) -> None:
+
+        plt.figure(figsize=(10, 6))
+        ax = plt.gca()
+
+        ax.axvspan(xmin=-999, xmax=0.5, color='green', linestyle='--', label='gaseous', alpha=0.2)
+        ax.axvspan(xmin=0.5, xmax=2, color='blue', linestyle='--', label='gaseous/water-rich', alpha=0.2)
+        ax.axvspan(xmin=2, xmax=7, color='orange', linestyle='--', label='rocky', alpha=0.2)
+        ax.axvspan(xmin=7, xmax=13, color='red', linestyle='--', label='metallic', alpha=0.2)
+        ax.axvspan(xmin=13, xmax=999, color='purple', linestyle='--', label='super dense', alpha=0.2)
+
+        ax.hist(self.density, bins=bins, color='skyblue', edgecolor='black', align="mid", alpha=0.7)
+
+        xmin = np.min(self.density) - 1
+        xmax = np.max(self.density) + 1
+
+        gas_mid = (xmin + 0.5) / 2
+        water_mid = (0.5 + 2) / 2
+        rocky_mid = (2 + 7) / 2
+        metallic_mid = (7 + 13) / 2
+        super_dense_mid = (13 + xmax) / 2
+        y_coord1 = ax.get_ylim()[1] * 0.9
+        y_coord2 = ax.get_ylim()[1] * 0.95
+        ax.text(gas_mid, y_coord1, 'Gaseous', color='green', fontsize=12, ha='center')
+        ax.text(water_mid, y_coord2, 'Gaseous/Water-Rich', color='blue', fontsize=12, ha='center')
+        ax.text(rocky_mid, y_coord1, 'Rocky', color='orange', fontsize=12, ha='center')
+        ax.text(metallic_mid, y_coord2, 'Metallic', color='red', fontsize=12, ha='center')
+        ax.text(super_dense_mid, y_coord1, 'Super Dense', color='purple', fontsize=12, ha='center')
+
+        ax.set_xlim(xmin, xmax)
+        ax.set_title('Bulk Density Distribution')
+        ax.set_xlabel('Bulk Density (kg/m³)')
+        ax.set_ylabel('Frequency')
+        ax.grid(axis='y', alpha=0.5)
+        plt.show()

exocat/plots.py

@@ -0,0 +1,48 @@
+import matplotlib.pyplot as plt
+import numpy as np
+
+class Plot:
+    """A class to create scatter plots with color mapping based on a DataFrame."""
+
+    def __init__(self, df: np.ndarray) -> None:
+        """
+        Initialize the Plot class with a DataFrame.
+
+        Parameters
+        ----------
+        df : np.ndarray
+            A DataFrame containing the data to be plotted.
+        """
+
+        self.df = df
+    
+    def plot_cmap(self, x:str, y:str, c:str, cmap:str='viridis', title:str=None) -> None:
+        """
+        Create a scatter plot with color mapping based on a DataFrame.
+        
+        Parameters
+        ----------
+        x : str
+            The column name for the x-axis values.
+        y : str
+            The column name for the y-axis values.
+        c : str
+            The column name for the color mapping values.
+        cmap : str, optional
+            The colormap to use for the scatter plot (default is 'viridis').
+        title : str, optional
+            The title of the plot (default is None).
+
+        Returns
+        -------
+        None
+        """
+        plt.figure(figsize=(10, 6))
+        scatter = plt.scatter(self.df[x], self.df[y], c=self.df[c], cmap=cmap)
+        plt.colorbar(scatter, label=c)
+        plt.xlabel(x)
+        plt.ylabel(y)
+        if title:
+            plt.title(title)
+        plt.grid()
+        plt.show()

exocat/read_tepcat.py

@@ -0,0 +1,22 @@
+#%%
+import requests
+from astropy.io import ascii
+
+link = "https://www.astro.keele.ac.uk/jkt/tepcat/allinfo-ascii.txt"
+
+txt = requests.get(link).text
+
+lines = txt.splitlines()
+# Strip the leading comment marker from the header line
+lines[0] = lines[0].removeprefix("#").strip()
+
+ascii_dat = ascii.read(
+    lines,
+    format="fixed_width",
+    delimiter=" ",
+    comment=None,
+)
+
+dat = ascii_dat.to_pandas()
+# Strip leading and trailing hyphens from column names
+dat.columns = dat.columns.str.strip("-")

exocat/temperature.py

@@ -0,0 +1,76 @@
+import numpy as np
+import matplotlib.pyplot as plt
+import pandas as pd
+
+class Temperature:
+    """A class to represent and classify temperature values."""
+
+    def __init__(self, df: pd.DataFrame|np.ndarray|list) -> None:
+        """
+        Initialize the Temperature class with a DataFrame or an array of temperature values.
+        
+        Parameters
+        ----------
+        df : pd.DataFrame, np.ndarray, or list
+            A DataFrame containing a column named 'Teq' or an array/list of temperature values
+        """
+
+        if isinstance(df, pd.DataFrame):
+            self.df = df
+            self.temperature = np.atleast_1d(np.asarray(df["Teq"]))
+        else:
+            self.df = None
+            self.temperature = np.atleast_1d(np.asarray(df))
+        
+        self.classifications = None
+
+    def classify(self) -> np.ndarray|str:
+        """
+        Classify the temperature values into categories: 'cold', 'warm', or 'hot'.
+
+        Returns
+        -------
+        np.ndarray or str
+            An array of classifications corresponding to the temperature values.
+        """
+        
+        output = np.empty_like(self.temperature, dtype=object)
+        output[self.temperature > 373] = "hot"
+        output[(self.temperature <= 373) & (self.temperature > 273)] = "warm"
+        output[self.temperature <= 273] = "cold"
+
+        self.classifications = output
+        return self.classifications
+    
+    def plot(self, bins:int=50) -> None:
+        """
+        Plot the distribution of temperature values with color-coded regions for 'cold', 'warm', and 'hot' categories.
+
+        Parameters
+        ----------
+        bins : int, optional
+            The number of bins to use for the histogram (default is 50).
+        """
+
+        plt.figure(figsize=(10, 6))
+        ax = plt.gca()
+
+        ax.axvspan(xmin=self.temperature.min(), xmax=273, color='blue', linestyle='--', label='cold', alpha=0.2)
+        ax.axvspan(xmin=273, xmax=373, color='orange', linestyle='--', label='warm', alpha=0.2)
+        ax.axvspan(xmin=373, xmax=self.temperature.max(), color='red', linestyle='--', label='hot', alpha=0.2)
+
+        ax.hist(self.temperature, bins=bins, color='skyblue', edgecolor='black', align="mid", alpha=0.7)
+
+        cold_mid = (self.temperature.min() + 273) / 2
+        warm_mid = (273 + 373) / 2
+        hot_mid = (373 + self.temperature.max()) / 2
+        y_coord = ax.get_ylim()[1] * 0.9
+        ax.text(cold_mid, y_coord, 'Cold', color='blue', fontsize=12, ha='center')
+        ax.text(warm_mid, y_coord, 'Warm', color='orange', fontsize=12, ha='center')
+        ax.text(hot_mid, y_coord, 'Hot', color='red', fontsize=12, ha='center')
+
+        ax.set_title('Temperature Distribution')
+        ax.set_xlabel('Temperature (K)')
+        ax.set_ylabel('Frequency')
+        ax.grid(axis='y', alpha=0.5)
+        plt.show()

exocat-0.1.0.dist-info/METADATA

@@ -0,0 +1,31 @@
+Metadata-Version: 2.4
+Name: exocat
+Version: 0.1.0
+Summary: Classifies exoplanet parameters using the TEPCat catalogue
+Author: Ben Cadell, Mélissa Azombo, Khang Nguyen
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/Benjamin-Cadell/exocat
+Keywords: tepcat,classification
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: POSIX
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: matplotlib
+Requires-Dist: astropy
+Requires-Dist: requests
+Requires-Dist: pandas
+Dynamic: license-file
+
+# ExoCat
+
+This is a project created and developed during the 2026 Code/Astro course at the Manchester local hub.
+
+The authors are:
+
+    Benjamin Cadell
+    Mélissa Azombo
+    Khang Nguyen
+
+The code will classify known transiting exoplanets using the Transiting ExoPlanets Catalogue (TEPCat) using their studied parameters. As of 23/06/2026 the code is still in active development.

exocat-0.1.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+exocat/__init__.py,sha256=Jck2am-Kc7yY3vua1BUYPvdKyFvEdm-MaoImtqSekZI,97
+exocat/bulk_density.py,sha256=VINkKl8XqRxvAzcpNqVPDBeJucYp7240iGlXBUJJk3Y,2809
+exocat/plots.py,sha256=GffLLLyTtH10tjc3QgMcbk9CSbHva3fOeShM3dTUt6M,1378
+exocat/read_tepcat.py,sha256=pPMJTvn-VLyX8ZT5yk6-MCdcygUIulVhoXMFnNH0PUE,503
+exocat/temperature.py,sha256=-BIh6nE5GWUnJuflyZ1U2ocAsPn8dmqa8WUdqV_bXhQ,2852
+exocat-0.1.0.dist-info/licenses/LICENSE,sha256=iMzcYTs_w1MWoT1K6CsTTsNdjxf5oVscJfhl6sj9mA4,1072
+exocat-0.1.0.dist-info/METADATA,sha256=Vg9lYF3euvDd4ytRdLPvHlAi34e_R0Ef5SkFKxT57gI,981
+exocat-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+exocat-0.1.0.dist-info/top_level.txt,sha256=7UiqTDhRL1RQyIf-PvrQdwLuTYSisGQhKGWLbi1gknU,7
+exocat-0.1.0.dist-info/RECORD,,

exocat-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

exocat-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Benjamin-Cadell
+
+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.

exocat-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+exocat