kundali-chart-mcp 0.2.1

package/python/kundali_lib/vedic/panchanga.py

@@ -0,0 +1,297 @@
+"""Panchanga: the 5 limbs of Vedic time (Tithi, Vara, Nakshatra, Yoga, Karana)."""
+
+import math
+
+# ---------------------------------------------------------------------------
+# Constant tables
+# ---------------------------------------------------------------------------
+
+TITHI_NAMES = [
+    "Pratipada",
+    "Dwitiya",
+    "Tritiya",
+    "Chaturthi",
+    "Panchami",
+    "Shashthi",
+    "Saptami",
+    "Ashtami",
+    "Navami",
+    "Dashami",
+    "Ekadashi",
+    "Dwadashi",
+    "Trayodashi",
+    "Chaturdashi",
+    "Purnima",
+    # Krishna Paksha (indices 15-29 → tithis 16-30)
+    "Pratipada",
+    "Dwitiya",
+    "Tritiya",
+    "Chaturthi",
+    "Panchami",
+    "Shashthi",
+    "Saptami",
+    "Ashtami",
+    "Navami",
+    "Dashami",
+    "Ekadashi",
+    "Dwadashi",
+    "Trayodashi",
+    "Chaturdashi",
+    "Amavasya",
+]
+
+VARA_NAMES = [
+    "Sunday",
+    "Monday",
+    "Tuesday",
+    "Wednesday",
+    "Thursday",
+    "Friday",
+    "Saturday",
+]
+VARA_LORDS = ["Sun", "Moon", "Mars", "Mercury", "Jupiter", "Venus", "Saturn"]
+
+NAKSHATRA_NAMES = [
+    "Ashwini",
+    "Bharani",
+    "Krittika",
+    "Rohini",
+    "Mrigashira",
+    "Ardra",
+    "Punarvasu",
+    "Pushya",
+    "Ashlesha",
+    "Magha",
+    "Purva Phalguni",
+    "Uttara Phalguni",
+    "Hasta",
+    "Chitra",
+    "Swati",
+    "Vishakha",
+    "Anuradha",
+    "Jyeshtha",
+    "Mula",
+    "Purva Ashadha",
+    "Uttara Ashadha",
+    "Shravana",
+    "Dhanishta",
+    "Shatabhisha",
+    "Purva Bhadrapada",
+    "Uttara Bhadrapada",
+    "Revati",
+]
+
+# Ketu, Venus, Sun, Moon, Mars, Rahu, Jupiter, Saturn, Mercury (×3)
+NAKSHATRA_LORDS = [
+    "Ketu",
+    "Venus",
+    "Sun",
+    "Moon",
+    "Mars",
+    "Rahu",
+    "Jupiter",
+    "Saturn",
+    "Mercury",
+] * 3
+
+YOGA_NAMES = [
+    "Vishkumbha",
+    "Priti",
+    "Ayushman",
+    "Saubhagya",
+    "Shobhana",
+    "Atiganda",
+    "Sukarma",
+    "Dhriti",
+    "Shoola",
+    "Ganda",
+    "Vriddhi",
+    "Dhruva",
+    "Vyaghata",
+    "Harshana",
+    "Vajra",
+    "Siddhi",
+    "Vyatipata",
+    "Variyana",
+    "Parigha",
+    "Shiva",
+    "Siddha",
+    "Sadhya",
+    "Shubha",
+    "Shukla",
+    "Brahma",
+    "Indra",
+    "Vaidhriti",
+]
+
+# 1-indexed set of inauspicious yoga numbers
+INAUSPICIOUS_YOGAS = {1, 6, 9, 10, 13, 15, 17, 19, 27}
+
+# Fixed karanas at positions 1, 58, 59, 60
+FIXED_KARANAS = {1: "Kimstughna", 58: "Shakuni", 59: "Chatushpada", 60: "Naga"}
+MOVABLE_KARANAS = ["Bava", "Balava", "Kaulava", "Taitila", "Garija", "Vanija", "Vishti"]
+
+
+# ---------------------------------------------------------------------------
+# Helper
+# ---------------------------------------------------------------------------
+
+
+def _normalize(lon: float) -> float:
+    """Bring longitude into [0, 360)."""
+    return lon % 360
+
+
+# ---------------------------------------------------------------------------
+# Main function
+# ---------------------------------------------------------------------------
+
+
+def get_panchanga(sun_lon: float, moon_lon: float, julian_day: float) -> dict:
+    """Return the five limbs of Vedic time (Panchanga).
+
+    Parameters
+    ----------
+    sun_lon : float
+        Sidereal longitude of the Sun in degrees [0, 360).
+    moon_lon : float
+        Sidereal longitude of the Moon in degrees [0, 360).
+    julian_day : float
+        Julian Day Number for the moment of interest.
+
+    Returns
+    -------
+    dict
+        Keys: tithi, vara, nakshatra, yoga, karana — each a dict with
+        relevant sub-fields.
+    """
+    sun_lon = _normalize(sun_lon)
+    moon_lon = _normalize(moon_lon)
+
+    return {
+        "tithi": _calc_tithi(sun_lon, moon_lon),
+        "vara": _calc_vara(julian_day),
+        "nakshatra": _calc_nakshatra(moon_lon),
+        "yoga": _calc_yoga(sun_lon, moon_lon),
+        "karana": _calc_karana(sun_lon, moon_lon),
+    }
+
+
+# ---------------------------------------------------------------------------
+# Tithi
+# ---------------------------------------------------------------------------
+
+
+def _calc_tithi(sun_lon: float, moon_lon: float) -> dict:
+    diff = _normalize(moon_lon - sun_lon)
+    tithi_num = math.ceil(diff / 12)
+    if tithi_num == 0:
+        tithi_num = 30  # diff == 0 edge case
+
+    percentage_elapsed = ((diff % 12) / 12) * 100
+
+    if tithi_num <= 15:
+        paksha = "Shukla"
+        name = TITHI_NAMES[tithi_num - 1]
+    else:
+        paksha = "Krishna"
+        name = TITHI_NAMES[tithi_num - 1]  # table has 30 entries
+
+    return {
+        "number": tithi_num,
+        "name": name,
+        "paksha": paksha,
+        "percentage_elapsed": round(percentage_elapsed, 2),
+    }
+
+
+# ---------------------------------------------------------------------------
+# Vara
+# ---------------------------------------------------------------------------
+
+
+def _calc_vara(julian_day: float) -> dict:
+    # JD 0.0 is Monday noon; adding 1.5 shifts so index 0 = Sunday
+    vara_index = int(julian_day + 1.5) % 7  # 0=Sunday … 6=Saturday
+    number = vara_index + 1  # 1=Sunday … 7=Saturday
+
+    return {
+        "number": number,
+        "name": VARA_NAMES[vara_index],
+        "lord": VARA_LORDS[vara_index],
+    }
+
+
+# ---------------------------------------------------------------------------
+# Nakshatra
+# ---------------------------------------------------------------------------
+
+
+def _calc_nakshatra(moon_lon: float) -> dict:
+    span = 360 / 27  # ≈ 13.333°
+    pada_span = 360 / 108  # ≈ 3.333°
+
+    nakshatra_index = int(moon_lon / span)  # 0-26
+    nakshatra_num = nakshatra_index + 1  # 1-27
+
+    lon_within = moon_lon % span
+    pada = int(lon_within / pada_span) + 1  # 1-4
+    pada = min(pada, 4)  # safety clamp
+
+    return {
+        "number": nakshatra_num,
+        "name": NAKSHATRA_NAMES[nakshatra_index],
+        "lord": NAKSHATRA_LORDS[nakshatra_index],
+        "pada": pada,
+    }
+
+
+# ---------------------------------------------------------------------------
+# Yoga
+# ---------------------------------------------------------------------------
+
+
+def _calc_yoga(sun_lon: float, moon_lon: float) -> dict:
+    combined = _normalize(sun_lon + moon_lon)
+    span = 360 / 27
+    yoga_index = int(combined / span)  # 0-26
+    yoga_num = yoga_index + 1  # 1-27
+
+    if yoga_num in INAUSPICIOUS_YOGAS:
+        nature = "Inauspicious"
+    elif yoga_num in {3, 4, 5, 7, 8, 11, 12, 14, 16, 20, 21, 22, 23, 24, 25, 26}:
+        nature = "Auspicious"
+    else:
+        nature = "Neutral"
+
+    return {
+        "number": yoga_num,
+        "name": YOGA_NAMES[yoga_index],
+        "nature": nature,
+    }
+
+
+# ---------------------------------------------------------------------------
+# Karana
+# ---------------------------------------------------------------------------
+
+
+def _calc_karana(sun_lon: float, moon_lon: float) -> dict:
+    diff = _normalize(moon_lon - sun_lon)
+    karana_num = int(diff / 6) + 1  # 1-60
+    karana_num = max(1, min(karana_num, 60))  # safety clamp
+
+    if karana_num in FIXED_KARANAS:
+        name = FIXED_KARANAS[karana_num]
+        karana_type = "Fixed"
+    else:
+        # Movable karanas: positions 2-57 (index 0 in the 7-cycle = Bava)
+        movable_index = (karana_num - 2) % 7
+        name = MOVABLE_KARANAS[movable_index]
+        karana_type = "Movable"
+
+    return {
+        "number": karana_num,
+        "name": name,
+        "type": karana_type,
+    }

package/python/kundali_lib/vedic/planets.py

@@ -0,0 +1,55 @@
+"""Planetary position calculation (sidereal)."""
+
+from typing import Optional
+
+import swisseph as swe
+
+from kundali_lib.vedic.constants import CALC_FLAGS, PLANET_IDS
+from kundali_lib.vedic.zodiac import (
+    get_nakshatra,
+    get_nakshatra_lord,
+    get_pada,
+    get_rashi,
+    get_rashi_lord,
+)
+
+
+def calc_planet_position(planet_name: str, julian_day: float) -> dict:
+    """Single planet sidereal position. Ketu derived from Rahu."""
+    if planet_name == "Ketu":
+        rahu = calc_planet_position("Rahu", julian_day)
+        lon = (rahu["longitude"] + 180) % 360
+        return _planet_row("Ketu", lon, 0, False, -rahu.get("speed_longitude", 0))
+
+    pid = PLANET_IDS.get(planet_name)
+    if pid is None:
+        raise ValueError(f"Unknown planet: {planet_name}")
+
+    xx, _ = swe.calc_ut(julian_day, pid, CALC_FLAGS)
+    lon, speed = xx[0], xx[3]
+    retro = speed < 0
+
+    return _planet_row(planet_name, lon, speed, retro)
+
+
+def _planet_row(
+    name: str,
+    longitude: float,
+    speed: float,
+    is_retro: bool,
+    speed_override: Optional[float] = None,
+) -> dict:
+    sp = speed_override if speed_override is not None else speed
+    return {
+        "name": name,
+        "longitude": longitude,
+        "rashi": get_rashi(longitude),
+        "rashi_lord": get_rashi_lord(longitude),
+        "degree": longitude % 30,
+        "nakshatra": get_nakshatra(longitude),
+        "nakshatra_lord": get_nakshatra_lord(longitude),
+        "pada": get_pada(longitude),
+        "house": 0,
+        "is_retrograde": is_retro,
+        "speed_longitude": round(sp, 4),
+    }