kundali-chart-mcp 0.2.1

package/python/kundali_lib/vedic/nabhasha.py

@@ -0,0 +1,349 @@
+"""Nabhasha Yogas: special yogas based on overall chart patterns and planetary distributions."""
+
+from __future__ import annotations
+
+from kundali_lib.vedic.constants import RASHIS
+
+# ---------------------------------------------------------------------------
+# Reference data
+# ---------------------------------------------------------------------------
+
+MOVABLE_SIGNS = {"Aries", "Cancer", "Libra", "Capricorn"}
+FIXED_SIGNS = {"Taurus", "Leo", "Scorpio", "Aquarius"}
+DUAL_SIGNS = {"Gemini", "Virgo", "Sagittarius", "Pisces"}
+
+KENDRA_HOUSES = {1, 4, 7, 10}
+APOKLIMA_HOUSES = {3, 6, 9, 12}
+
+BENEFIC_PLANETS = {"Moon", "Mercury", "Jupiter", "Venus"}
+MALEFIC_PLANETS = {"Sun", "Mars", "Saturn", "Rahu", "Ketu"}
+
+# ---------------------------------------------------------------------------
+# Ashrayas
+# ---------------------------------------------------------------------------
+
+ASHRAYAS: dict[str, dict] = {
+    "Rajju": {
+        "condition": "All planets in movable signs (Aries, Cancer, Libra, Capricorn)",
+        "effect": "Person travels frequently, unsettled",
+    },
+    "Musala": {
+        "condition": "All planets in fixed signs (Taurus, Leo, Scorpio, Aquarius)",
+        "effect": "Determined, stubborn, famous",
+    },
+    "Nala": {
+        "condition": "All planets in dual/common signs (Gemini, Virgo, Sagittarius, Pisces)",
+        "effect": "Skilled in arts, skilled in work with hands",
+    },
+}
+
+# ---------------------------------------------------------------------------
+# Dala
+# ---------------------------------------------------------------------------
+
+DALA: dict[str, dict] = {
+    "Mala": {
+        "condition": "Benefics (Moon, Mercury, Jupiter, Venus) all in kendras (1,4,7,10)",
+        "effect": "Happiness, wealth, honour",
+    },
+    "Sarpa": {
+        "condition": "Only malefics (Sun, Mars, Saturn, Rahu, Ketu) in kendras",
+        "effect": "Wicked, poor, dependent",
+    },
+}
+
+# ---------------------------------------------------------------------------
+# Akriti patterns (simplified — primary 10)
+# ---------------------------------------------------------------------------
+
+AKRITI_YOGAS: dict[int, dict] = {
+    2: {
+        "name": "Gada",
+        "description": "Planets in two adjacent houses",
+        "effect": "Wealthy, skilled",
+    },
+    3: {
+        "name": "Shakata",
+        "description": "Planets in houses 1 and 7",
+        "effect": "Variable fortune",
+    },
+    4: {
+        "name": "Vihaga",
+        "description": "Planets in houses 4 and 10",
+        "effect": "Messenger, traveller",
+    },
+    5: {
+        "name": "Shringataka",
+        "description": "Planets in trikonas (1,5,9)",
+        "effect": "Pleasure seeker",
+    },
+    6: {
+        "name": "Hala",
+        "description": "Planets in consecutive houses",
+        "effect": "Farmer, works hard",
+    },
+    7: {
+        "name": "Vajra",
+        "description": "Benefics in 1 and 7, malefics in 4 and 10",
+        "effect": "Rich, happy in youth and old age",
+    },
+    8: {
+        "name": "Yava",
+        "description": "Benefics in 1 and 7, malefics in 4 and 10 (reverse of Vajra)",
+        "effect": "Rich in middle age",
+    },
+    9: {
+        "name": "Kamala",
+        "description": "Planets in all 4 kendras",
+        "effect": "Virtuous, wealthy, famous",
+    },
+    10: {
+        "name": "Vapi",
+        "description": "All planets in Apoklimas (3,6,9,12)",
+        "effect": "Wealthy but hidden",
+    },
+    18: {
+        "name": "Ardha Chandra",
+        "description": "Planets in 7 consecutive houses",
+        "effect": "Commander",
+    },
+}
+
+
+# ---------------------------------------------------------------------------
+# Internal helpers
+# ---------------------------------------------------------------------------
+
+
+def _sign_type(rashi: str) -> str:
+    if rashi in MOVABLE_SIGNS:
+        return "movable"
+    if rashi in FIXED_SIGNS:
+        return "fixed"
+    return "dual"
+
+
+def _occupied_houses(planetary_positions: list[dict]) -> list[int]:
+    """Return sorted list of houses that have at least one planet."""
+    seen: set[int] = set()
+    for p in planetary_positions:
+        h = p.get("house", 0)
+        if 1 <= h <= 12:
+            seen.add(h)
+    return sorted(seen)
+
+
+def _houses_are_consecutive(houses: list[int], span: int) -> bool:
+    """True if all houses lie within a consecutive arc of ``span`` houses."""
+    if not houses:
+        return False
+    for start in range(1, 13):
+        arc = set((start + i - 1) % 12 + 1 for i in range(span))
+        if set(houses).issubset(arc):
+            return True
+    return False
+
+
+# ---------------------------------------------------------------------------
+# Yoga checkers
+# ---------------------------------------------------------------------------
+
+
+def _check_ashrayas(planetary_positions: list[dict]) -> str | None:
+    """Return the name of the Ashrayas yoga present, or None."""
+    rashis = [p.get("rashi", "") for p in planetary_positions]
+    types = [_sign_type(r) for r in rashis if r]
+    if not types:
+        return None
+    if all(t == "movable" for t in types):
+        return "Rajju"
+    if all(t == "fixed" for t in types):
+        return "Musala"
+    if all(t == "dual" for t in types):
+        return "Nala"
+    return None
+
+
+def _check_dala(planetary_positions: list[dict]) -> str | None:
+    """Return 'Mala', 'Sarpa', or None."""
+    kendra_planets = [
+        p for p in planetary_positions if p.get("house", 0) in KENDRA_HOUSES
+    ]
+    if not kendra_planets:
+        return None
+
+    kendra_names = {p["name"] for p in kendra_planets}
+
+    # Mala: all 4 benefics in kendras
+    benefics_in_kendra = BENEFIC_PLANETS.intersection(kendra_names)
+    if benefics_in_kendra == BENEFIC_PLANETS:
+        return "Mala"
+
+    # Sarpa: only malefics in kendras (no benefic in kendra at all)
+    if not kendra_names.intersection(BENEFIC_PLANETS) and kendra_names:
+        return "Sarpa"
+
+    return None
+
+
+def _check_akriti(occupied: list[int], planetary_positions: list[dict]) -> dict | None:
+    """Return the Akriti yoga dict or None based on occupied house patterns."""
+    occ_set = set(occupied)
+    n = len(occ_set)
+
+    # Kamala: planets in all 4 kendras
+    if KENDRA_HOUSES.issubset(occ_set):
+        return AKRITI_YOGAS[9]
+
+    # Vapi: all in Apoklimas
+    if occ_set and occ_set.issubset(APOKLIMA_HOUSES):
+        return AKRITI_YOGAS[10]
+
+    # Ardha Chandra: planets in 7 consecutive houses
+    if _houses_are_consecutive(occupied, 7) and 5 <= n <= 8:
+        return AKRITI_YOGAS[18]
+
+    # Shringataka: only trikonas occupied
+    if occ_set and occ_set.issubset({1, 5, 9}):
+        return AKRITI_YOGAS[5]
+
+    # Vihaga: only 4 and 10 occupied
+    if occ_set and occ_set.issubset({4, 10}):
+        return AKRITI_YOGAS[4]
+
+    # Shakata: only 1 and 7 occupied
+    if occ_set and occ_set.issubset({1, 7}):
+        return AKRITI_YOGAS[3]
+
+    # Gada: two adjacent houses occupied only
+    if n == 2 and len(occupied) == 2:
+        diff = abs(occupied[0] - occupied[1])
+        if diff == 1 or diff == 11:
+            return AKRITI_YOGAS[2]
+
+    # Hala: all in 6 consecutive houses
+    if _houses_are_consecutive(occupied, 6) and n <= 7:
+        return AKRITI_YOGAS[6]
+
+    # Vajra / Yava (both use houses 1, 4, 7, 10)
+    if occ_set and occ_set.issubset({1, 4, 7, 10}):
+        benefics = {
+            p["name"] for p in planetary_positions if p["name"] in BENEFIC_PLANETS
+        }
+        malefics = {
+            p["name"] for p in planetary_positions if p["name"] in MALEFIC_PLANETS
+        }
+
+        def _in_house(names: set, houses: set) -> bool:
+            return all(
+                any(
+                    p["name"] in names and p.get("house", 0) in houses
+                    for p in planetary_positions
+                )
+                for _ in [1]
+            )
+
+        benefic_houses = {
+            p.get("house", 0)
+            for p in planetary_positions
+            if p["name"] in BENEFIC_PLANETS
+        }
+        malefic_houses = {
+            p.get("house", 0)
+            for p in planetary_positions
+            if p["name"] in MALEFIC_PLANETS
+        }
+
+        if benefic_houses.issubset({1, 7}) and malefic_houses.issubset({4, 10}):
+            return AKRITI_YOGAS[7]  # Vajra
+        if benefic_houses.issubset({4, 10}) and malefic_houses.issubset({1, 7}):
+            return AKRITI_YOGAS[8]  # Yava
+
+    return None
+
+
+# ---------------------------------------------------------------------------
+# Public API
+# ---------------------------------------------------------------------------
+
+
+def get_nabhasha_yogas(planetary_positions: list[dict]) -> dict:
+    """Identify Nabhasha Yogas from a chart's planetary positions.
+
+    Args:
+        planetary_positions: List of planet dicts as produced by ``build_chart()``
+                             (each dict must have ``name``, ``rashi``, and ``house``).
+
+    Returns:
+        dict with Ashrayas, Dala, Akriti yogas found, along with supporting
+        statistics and a plain-language summary.
+    """
+    occupied = _occupied_houses(planetary_positions)
+
+    # ── Sign-type counts ──────────────────────────────────────────────────────
+    type_counts: dict[str, int] = {"movable": 0, "fixed": 0, "dual": 0}
+    for p in planetary_positions:
+        r = p.get("rashi", "")
+        if r:
+            type_counts[_sign_type(r)] += 1
+
+    # ── Yoga checks ───────────────────────────────────────────────────────────
+    ashrayas_name = _check_ashrayas(planetary_positions)
+    dala_name = _check_dala(planetary_positions)
+    akriti_data = _check_akriti(occupied, planetary_positions)
+
+    yogas_present: list[dict] = []
+
+    if ashrayas_name:
+        info = ASHRAYAS[ashrayas_name]
+        yogas_present.append(
+            {
+                "name": ashrayas_name,
+                "category": "Ashrayas",
+                "description": info["condition"],
+                "effect": info["effect"],
+            }
+        )
+
+    if dala_name:
+        info = DALA[dala_name]
+        yogas_present.append(
+            {
+                "name": dala_name,
+                "category": "Dala",
+                "description": info["condition"],
+                "effect": info["effect"],
+            }
+        )
+
+    if akriti_data:
+        yogas_present.append(
+            {
+                "name": akriti_data["name"],
+                "category": "Akriti",
+                "description": akriti_data["description"],
+                "effect": akriti_data["effect"],
+            }
+        )
+
+    # ── Summary ───────────────────────────────────────────────────────────────
+    if yogas_present:
+        names = ", ".join(y["name"] for y in yogas_present)
+        summary = f"Nabhasha yogas present: {names}. Planets occupy houses: {occupied}."
+    else:
+        summary = (
+            f"No primary Nabhasha yoga detected. "
+            f"Planets occupy houses: {occupied}. "
+            f"Sign distribution — movable: {type_counts['movable']}, "
+            f"fixed: {type_counts['fixed']}, dual: {type_counts['dual']}."
+        )
+
+    return {
+        "ashrayas_yoga": ashrayas_name,
+        "dala_yoga": dala_name,
+        "akriti_yoga": akriti_data["name"] if akriti_data else None,
+        "occupied_houses": occupied,
+        "occupied_signs_type": type_counts,
+        "yogas_present": yogas_present,
+        "summary": summary,
+    }