starplot 0.13.0__py2.py3-none-any.whl → 0.15.0__py2.py3-none-any.whl

starplot/cli.py

@@ -0,0 +1,33 @@
+import sys
+
+from starplot.styles import fonts
+
+COMMANDS = ["setup"]
+
+
+def setup(options):
+    from starplot import settings
+    from starplot.data import db, bigsky
+
+    print("Installing DuckDB spatial extension...")
+
+    con = db.connect()
+    con.load_extension("spatial")
+
+    fonts.load()
+
+    print(f"Installed to: {settings.DUCKDB_EXTENSION_PATH}")
+
+    if "--install-big-sky" in options:
+        bigsky.download_if_not_exists()
+        print(f"Big Sky Catalog downloaded and installed to: {settings.DOWNLOAD_PATH}")
+
+
+def main():
+    command = sys.argv[1].lower()
+
+    if command not in COMMANDS:
+        print(f"Unrecognized command: {command}")
+
+    if command == "setup":
+        setup(sys.argv[2:])

starplot/coordinates.py

@@ -0,0 +1,6 @@
+class CoordinateSystem:
+    RA_DEC = "radec"
+    AZ_ALT = "azalt"
+    AXES = "axes"
+    PROJECTED = "projected"
+    DISPLAY = "display"

starplot/data/__init__.py

@@ -1,31 +1,13 @@
-import os
-
-from enum import Enum
-from pathlib import Path
-
 from skyfield.api import Loader
 
-HERE = Path(__file__).resolve().parent
-DATA_PATH = HERE / "library"
-
-load = Loader(DATA_PATH)
+from starplot import settings
 
+load = Loader(settings.DATA_PATH)
 
-def env(name, default):
-    return os.environ.get(name) or default
 
+class DataFiles:
+    BIG_SKY = settings.DOWNLOAD_PATH / "bigsky.0.4.0.stars.parquet"
 
-class DataFiles(str, Enum):
-    # Built-In Files
-    CONSTELLATION_LINES = DATA_PATH / "constellation_lines_inv.gpkg"
-    CONSTELLATION_LINES_HIP = DATA_PATH / "constellation_lines_hips.json"
-    CONSTELLATION_BORDERS = DATA_PATH / "constellation_borders_inv.gpkg"
-    MILKY_WAY = DATA_PATH / "milkyway.gpkg"
-    HIPPARCOS = DATA_PATH / "stars.hipparcos.parquet"
-    BIG_SKY_MAG11 = DATA_PATH / "stars.bigsky.mag11.parquet"
-    ONGC = DATA_PATH / "ongc.gpkg.zip"
-    CONSTELLATIONS = DATA_PATH / "constellations.gpkg"
+    BIG_SKY_MAG11 = settings.DATA_PATH / "bigsky.0.4.0.stars.mag11.parquet"
 
-    # Downloaded Files
-    _DOWNLOAD_PATH = Path(env("STARPLOT_DOWNLOAD_PATH", str(DATA_PATH)))
-    BIG_SKY = _DOWNLOAD_PATH / "stars.bigsky.parquet"
+    DATABASE = settings.DATA_PATH / "sky.db"

starplot/data/bigsky.py

@@ -1,56 +1,58 @@
 import os
+from pathlib import Path
 
 import pandas as pd
 
-from starplot.data import DATA_PATH, DataFiles, utils
+from starplot import settings
+from starplot.data import DataFiles, utils
 
 
-BIG_SKY_VERSION = "0.1.0"
+BIG_SKY_VERSION = "0.4.0"
+BIG_SKY_FILENAME = f"bigsky.{BIG_SKY_VERSION}.stars.csv.gz"
+BIG_SKY_PQ_FILENAME = f"bigsky.{BIG_SKY_VERSION}.stars.parquet"
 
-BIG_SKY_FILENAME = "bigsky.stars.csv.gz"
+BIG_SKY_MAG11_FILENAME = f"bigsky.{BIG_SKY_VERSION}.stars.mag11.csv.gz"
+BIG_SKY_MAG11_PQ_FILENAME = f"bigsky.{BIG_SKY_VERSION}.stars.mag11.parquet"
 
-BIG_SKY_URL = f"https://github.com/steveberardi/bigsky/releases/download/v{BIG_SKY_VERSION}/{BIG_SKY_FILENAME}"
 
-DOWNLOADED_PATH = DATA_PATH / BIG_SKY_FILENAME
-
-DIGITS = 4
-
-BIG_SKY_ASSETS = {
-    DataFiles.BIG_SKY: "bigsky.stars.csv.gz",
-    DataFiles.BIG_SKY_MAG11: "bigsky.stars.mag11.csv.gz",
-}
-
-
-def url(filename: str, version: str):
+def get_url(version: str = BIG_SKY_VERSION, filename: str = BIG_SKY_FILENAME):
     return f"https://github.com/steveberardi/bigsky/releases/download/v{version}/{filename}"
 
 
 def download(
-    filename: str = BIG_SKY_FILENAME,
-    version: str = BIG_SKY_VERSION,
-    download_path: str = None,
-    digits: int = 4,
+    url: str = None,
+    download_path: str = settings.DOWNLOAD_PATH,
+    download_filename: str = BIG_SKY_FILENAME,
+    build_file: str = DataFiles.BIG_SKY,
 ):
-    download_path = download_path or str(DATA_PATH / filename)
+    url = url or get_url()
+    download_path = Path(download_path)
+
+    if not os.path.exists(download_path):
+        os.makedirs(download_path)
+
+    full_download_path = download_path / download_filename
     utils.download(
-        url(filename, version),
-        download_path,
+        url,
+        full_download_path,
         "Big Sky Star Catalog",
     )
     to_parquet(
-        download_path,
-        DataFiles.BIG_SKY,
-        digits,
+        full_download_path,
+        build_file,
     )
 
 
-def to_parquet(source_path: str, destination_path: str, digits: int = DIGITS):
+def to_parquet(source_path: str, destination_path: str):
+    import pyarrow as pa
+    import pyarrow.parquet as pq
+
     print("Preparing Big Sky Catalog for Starplot...")
 
     df = pd.read_csv(
         source_path,
         header=0,
-        names=[
+        usecols=[
             "tyc_id",
             "hip_id",
             "ccdm",
@@ -61,14 +63,11 @@ def to_parquet(source_path: str, destination_path: str, digits: int = DIGITS):
             "ra_mas_per_year",
             "dec_mas_per_year",
             "parallax_mas",
+            "constellation",
         ],
         compression="gzip",
     )
 
-    df["ra_hours"] = df.apply(
-        lambda row: round(row.ra_degrees_j2000 / 15, digits), axis=1
-    )
-
     df = df.assign(epoch_year=2000)
 
     df = df.rename(
@@ -79,19 +78,38 @@ def to_parquet(source_path: str, destination_path: str, digits: int = DIGITS):
         }
     )
 
-    df.to_parquet(destination_path, compression="gzip")
-
-    print(f"Done! {destination_path.value}")
+    df = df.sort_values(["magnitude"])
 
+    table = pa.Table.from_pandas(df)
+    table = table.drop_columns("__index_level_0__")
 
-def load(path):
-    if not exists(path):
-        download(filename=BIG_SKY_ASSETS.get(path))
-
-    df = pd.read_parquet(path)
+    pq.write_table(
+        table,
+        destination_path,
+        compression="none",
+        sorting_columns=[
+            pq.SortingColumn(df.columns.get_loc("magnitude")),
+        ],
+    )
 
-    return df.set_index("tyc_id")
+    print(f"Done! {destination_path}")
 
 
 def exists(path) -> bool:
     return os.path.isfile(path)
+
+
+def download_if_not_exists(
+    filename: str = DataFiles.BIG_SKY,
+    url: str = None,
+    download_path: str = settings.DOWNLOAD_PATH,
+    download_filename: str = BIG_SKY_FILENAME,
+    build_file: str = DataFiles.BIG_SKY,
+):
+    if not exists(filename):
+        download(
+            url=url,
+            download_path=download_path,
+            download_filename=download_filename,
+            build_file=build_file,
+        )