starplot 0.16.4__py2.py3-none-any.whl → 0.17.0__py2.py3-none-any.whl

starplot/data/dsos.py

@@ -2,32 +2,13 @@ from functools import cache
 
 from ibis import _, row_number, coalesce
 
+from starplot.config import settings
 from starplot.data import db
-
-
-class DsoLabelMaker(dict):
-    """
-    This is pretty hacky, but it helps keep a consistent interface for plotting labels and any overrides.
-
-    Basically this is just a dictionary that returns the key itself for any get() call, unless
-    the key is present in the 'overrides' dict that's passed on init
-    """
-
-    def __init__(self, *args, **kwargs):
-        self._overrides = kwargs.get("overrides") or {}
-
-    def __getitem__(self, key):
-        return self.get(key)
-
-    def get(self, key):
-        return self._overrides.get(key) or key
-
-
-DSO_LABELS_DEFAULT = DsoLabelMaker()
+from starplot.data.translations import language_name_column
 
 
 @cache
-def table():
+def table(language):
     con = db.connect()
     dsos = con.table("deep_sky_objects")
 
@@ -37,6 +18,9 @@ def table():
         constellation_id=_.constellation,
         magnitude=coalesce(_.mag_v, _.mag_b, None),
         size=_.size_deg2,
+        common_names=getattr(
+            dsos, language_name_column(language, column_prefix="common_names")
+        ),
         rowid=row_number(),
         sk=row_number(),
     )
@@ -44,7 +28,7 @@ def table():
 
 def load(extent=None, filters=None, sql=None):
     filters = filters or []
-    dsos = table()
+    dsos = table(language=settings.language)
 
     if extent:
         dsos = dsos.filter(_.geometry.intersects(extent))

starplot/data/stars.py

@@ -2,7 +2,9 @@ from functools import cache
 
 from ibis import _, row_number
 
+from starplot.config import settings
 from starplot.data import bigsky, DataFiles, db
+from starplot.data.translations import language_name_column
 
 
 class StarCatalog:
@@ -31,7 +33,11 @@ class StarCatalog:
 
 
 @cache
-def table(catalog: StarCatalog = StarCatalog.BIG_SKY_MAG11, table_name="stars"):
+def table(
+    catalog: StarCatalog = StarCatalog.BIG_SKY_MAG11,
+    table_name="stars",
+    language: str = "en-us",
+):
     con = db.connect()
 
     if catalog == StarCatalog.BIG_SKY_MAG11:
@@ -56,6 +62,10 @@ def table(catalog: StarCatalog = StarCatalog.BIG_SKY_MAG11, table_name="stars"):
         sk=row_number(),
     )
 
+    designations = designations.mutate(
+        name=getattr(designations, language_name_column(language))
+    )
+
     stars = stars.join(
         designations,
         [
@@ -75,7 +85,7 @@ def load(
     sql=None,
 ):
     filters = filters or []
-    stars = table(catalog)
+    stars = table(catalog, language=settings.language)
 
     if extent:
         stars = stars.filter(stars.geometry.intersects(extent))

starplot/data/translations.py

@@ -0,0 +1,161 @@
+LABELS = {
+    "en-us": {
+        "legend": "legend",
+        "star magnitude": "star magnitude",
+        "star": "star",
+        "deep sky object": "deep sky object",
+        "open cluster": "open cluster",
+        "globular cluster": "globular cluster",
+        "nebula": "nebula",
+        "galaxy": "galaxy",
+        "dark nebula": "dark nebula",
+        "association of stars": "association of stars",
+        "double star": "double star",
+        "emission nebula": "emission nebula",
+        "galaxy pair": "galaxy pair",
+        "galaxy triplet": "galaxy triplet",
+        "galaxy cluster": "galaxy cluster",
+        "group of galaxies": "group of galaxies",
+        "hii ionized region": "hii ionized region",
+        "nova star": "nova star",
+        "planetary nebula": "planetary nebula",
+        "reflection nebula": "reflection nebula",
+        "star cluster nebula": "star cluster nebula",
+        "supernova remnant": "supernova remnant",
+        "unknown": "unknown",
+        "planet": "planet",
+        "mercury": "mercury",
+        "venus": "venus",
+        "mars": "mars",
+        "jupiter": "jupiter",
+        "saturn": "saturn",
+        "uranus": "uranus",
+        "neptune": "neptune",
+        "pluto": "pluto",
+        "sun": "sun",
+        "moon": "moon",
+        "north": "north",
+        "east": "east",
+        "south": "south",
+        "west": "west",
+        "ecliptic": "ecliptic",
+        "celestial equator": "celestial equator",
+        "n": "n",
+        "e": "e",
+        "s": "s",
+        "w": "w",
+        "milky way": "milky way",
+    },
+    "fr": {
+        "legend": "légende",
+        "star magnitude": "magnitude des étoiles",
+        "star": "étoile",
+        "deep sky object": "objet du ciel profond",
+        "open cluster": "amas ouvert",
+        "globular cluster": "amas globulaire",
+        "nebula": "nébuleuse",
+        "galaxy": "galaxie",
+        "dark nebula": "nébuleuse obscure",
+        "association of stars": "amas d'étoiles",
+        "double star": "étoile double",
+        "emission nebula": "nébuleuse en émission",
+        "galaxy pair": "paire de galaxies",
+        "galaxy triplet": "triplet de galaxies",
+        "galaxy cluster": "amas de galaxies",
+        "group of galaxies": "groupes de galaxies",
+        "hii ionized region": "région d'hydrogène ionisé",
+        "nova star": "nova",
+        "planetary nebula": "nébuleuse planétaire",
+        "reflection nebula": "nébuleuse par réflexion",
+        "star cluster nebula": "nébuleuse d’amas d’étoiles",
+        "supernova remnant": "rémanent de supernova",
+        "unknown": "inconnu",
+        "planet": "planète",
+        "mercury": "mercure",
+        "venus": "venus",
+        "mars": "mars",
+        "jupiter": "jupiter",
+        "saturn": "saturne",
+        "uranus": "uranus",
+        "neptune": "neptune",
+        "pluto": "pluton",
+        "sun": "soleil",
+        "moon": "lune",
+        "north": "nord",
+        "east": "est",
+        "south": "sud",
+        "west": "ouest",
+        "ecliptic": "écliptique",
+        "celestial equator": "équateur céleste",
+        "n": "n",
+        "e": "e",
+        "s": "s",
+        "w": "o",
+        "milky way": "voie lactée",
+    },
+    "zh-cn": {
+        "legend": "图例",
+        "star magnitude": "星等",
+        "star": "恒星",
+        "deep sky object": "深空天体",
+        "open cluster": "疏散星团",
+        "globular cluster": "球状星团",
+        "nebula": "星云",
+        "galaxy": "星系",
+        "dark nebula": "暗星云",
+        "association of stars": "星协",
+        "double star": "双星",
+        "emission nebula": "发射星云",
+        "galaxy pair": "星系对",
+        "galaxy triplet": "三重星系",
+        "galaxy cluster": "星系团",
+        "group of galaxies": "星系群",
+        "hii ionized region": "hii电离氢区",
+        "nova star": "新星",
+        "planetary nebula": "行星状星云",
+        "reflection nebula": "反射星云",
+        "star cluster nebula": "星团星云",
+        "supernova remnant": "超新星遗迹",
+        "unknown": "未知天体",
+        "planet": "行星",
+        "mercury": "水星",
+        "venus": "金星",
+        "mars": "火星",
+        "jupiter": "木星",
+        "saturn": "土星",
+        "uranus": "天王星",
+        "neptune": "海王星",
+        "pluto": "冥王星",
+        "sun": "太阳",
+        "moon": "月球",
+        "north": "北",
+        "east": "东",
+        "south": "南",
+        "west": "西",
+        "ecliptic": "黄道",
+        "celestial equator": "天赤道",
+        "n": "北",
+        "e": "东",
+        "s": "南",
+        "w": "西",
+        "milky way": "银河",
+    },
+}
+
+
+def language_name_column(language: str, column_prefix: str = "name") -> str:
+    language_name = language.replace("-", "_").lower()
+
+    return f"{column_prefix}_{language_name}"
+
+
+def translate(text: str, language: str) -> str:
+    if not text:
+        return text
+
+    translation = LABELS[language.lower()].get(text.lower())
+
+    if not translation:
+        return text
+
+    return translation.title()

starplot/geometry.py

@@ -5,7 +5,7 @@ from typing import Union
 from shapely import transform
 from shapely.geometry import Point, Polygon, MultiPolygon
 
-from starplot import geod, utils
+from starplot import geod
 
 GLOBAL_EXTENT = Polygon(
     [
@@ -18,17 +18,18 @@ GLOBAL_EXTENT = Polygon(
 )
 
 
-def circle(center, diameter_degrees):
+def circle(center, diameter_degrees, num_pts=100):
     points = geod.ellipse(
         center,
         diameter_degrees,
         diameter_degrees,
         angle=0,
-        num_pts=100,
+        num_pts=num_pts,
     )
-    points = [
-        (round(24 - utils.lon_to_ra(lon), 4), round(dec, 4)) for lon, dec in points
-    ]
+    # points = [
+    #     (round(24 - utils.lon_to_ra(lon), 4), round(dec, 4)) for lon, dec in points
+    # ]
+    points = [(round(lon, 4), round(dec, 4)) for lon, dec in points]
     return Polygon(points)
 
 

starplot/horizon.py

@@ -3,9 +3,6 @@ import math
 from functools import cache
 from typing import Callable
 
-# import pandas as pd
-# import geopandas as gpd
-
 from cartopy import crs as ccrs
 from matplotlib import pyplot as plt, patches
 from matplotlib.ticker import FixedLocator, FuncFormatter
@@ -14,13 +11,14 @@ from shapely import Point, Polygon, MultiPolygon
 from starplot.coordinates import CoordinateSystem
 from starplot.base import BasePlot, DPI
 from starplot.mixins import ExtentMaskMixin
-from starplot.observer import Observer
+from starplot.models.observer import Observer
 from starplot.plotters import (
     ConstellationPlotterMixin,
     StarPlotterMixin,
     DsoPlotterMixin,
     MilkyWayPlotterMixin,
     GradientBackgroundMixin,
+    LegendPlotterMixin,
 )
 from starplot.styles import (
     PlotStyle,
@@ -30,8 +28,6 @@ from starplot.styles import (
     GradientDirection,
 )
 
-# pd.options.mode.chained_assignment = None  # default='warn'
-
 DEFAULT_HORIZON_STYLE = PlotStyle().extend(extensions.MAP)
 
 DEFAULT_HORIZON_LABELS = {
@@ -55,6 +51,7 @@ class HorizonPlot(
     DsoPlotterMixin,
     MilkyWayPlotterMixin,
     GradientBackgroundMixin,
+    LegendPlotterMixin,
 ):
     """Creates a new horizon plot.
 
@@ -378,6 +375,11 @@ class HorizonPlot(
                 label,
                 (x, patch_y + 0.027),
                 xycoords=self.ax.transAxes,
+                xytext=(
+                    style.label.offset_x * self.scale,
+                    style.label.offset_y * self.scale,
+                ),
+                textcoords="offset points",
                 **style.label.matplot_kwargs(self.scale),
                 clip_on=False,
             )
@@ -432,7 +434,6 @@ class HorizonPlot(
         label_style_kwargs.pop("ha")
 
         line_style_kwargs = style.line.matplot_kwargs()
-
         gridlines = self.ax.gridlines(
             draw_labels=show_labels,
             x_inline=False,
@@ -497,7 +498,7 @@ class HorizonPlot(
                     "|",
                     (x, -0.011 * self.scale),
                     xycoords=self.ax.transAxes,
-                    **self.style.gridlines.label.matplot_kwargs(self.scale / 2),
+                    **style.label.matplot_kwargs(self.scale / 2),
                 )
 
     @cache

starplot/map.py

@@ -14,7 +14,7 @@ from starplot.coordinates import CoordinateSystem
 from starplot import geod
 from starplot.base import BasePlot, DPI
 from starplot.mixins import ExtentMaskMixin
-from starplot.observer import Observer
+from starplot.models.observer import Observer
 from starplot.plotters import (
     ConstellationPlotterMixin,
     StarPlotterMixin,

starplot/models/__init__.py

@@ -1,7 +1,19 @@
-from .constellation import Constellation  # noqa: F401,F403
-from .dso import DSO, DsoType  # noqa: F401,F403
-from .star import Star  # noqa: F401,F403
-from .planet import Planet  # noqa: F401,F403
-from .moon import Moon  # noqa: F401,F403
-from .sun import Sun  # noqa: F401,F403
-from .objects import ObjectList  # noqa: F401,F403
+# ruff: noqa: F401,F403
+from .constellation import Constellation
+from .comet import Comet
+from .dso import DSO, DsoType
+from .star import Star
+from .planet import Planet
+from .moon import Moon
+from .sun import Sun
+from .optics import (
+    Optic,
+    Scope,
+    Reflector,
+    Refractor,
+    Binoculars,
+    Camera,
+)
+from .objects import ObjectList
+from .satellite import Satellite
+from .observer import Observer

starplot/models/base.py

@@ -83,7 +83,7 @@ class SkyObject(BaseModel, CreateMapMixin, CreateOpticMixin):
     def constellation_id(self) -> str | None:
         """Identifier of the constellation that contains this object. The ID is the three-letter (all lowercase) abbreviation from the International Astronomical Union (IAU)."""
         if not self._constellation_id:
-            pos = position_of_radec(self.ra, self.dec)
+            pos = position_of_radec(self.ra / 15, self.dec)
             self._constellation_id = constellation_at()(pos).lower()
         return self._constellation_id