732642856
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
"""Shared Swiss Ephemeris ayanamsa helpers.
|
|
|
|
Swiss Ephemeris stores sidereal mode globally inside the process. Any helper
|
|
that calls calc_ut(..., FLG_SIDEREAL) must set the intended mode explicitly.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
try:
|
|
import swisseph as swe
|
|
except ImportError: # pragma: no cover
|
|
swe = None
|
|
|
|
|
|
AYANAMSA_MODES = {
|
|
'lahiri': getattr(swe, 'SIDM_LAHIRI', 1) if swe else 1,
|
|
'raman': getattr(swe, 'SIDM_RAMAN', 3) if swe else 3,
|
|
'kp': getattr(swe, 'SIDM_KRISHNAMURTI', 5) if swe else 5,
|
|
'krishnamurti': getattr(swe, 'SIDM_KRISHNAMURTI', 5) if swe else 5,
|
|
'fagan_bradley': getattr(swe, 'SIDM_FAGAN_BRADLEY', 0) if swe else 0,
|
|
'djwhal_khul': getattr(swe, 'SIDM_DJWHAL_KHUL', 6) if swe else 6,
|
|
'sassanian': getattr(swe, 'SIDM_SASSANIAN', 16) if swe else 16,
|
|
'true_citra': getattr(swe, 'SIDM_TRUE_CITRA', 27) if swe else 27,
|
|
}
|
|
|
|
AYANAMSA_DISPLAY_NAMES = {
|
|
'lahiri': 'Lahiri',
|
|
'raman': 'Raman',
|
|
'kp': 'Krishnamurti/KP',
|
|
'krishnamurti': 'Krishnamurti/KP',
|
|
'fagan_bradley': 'Fagan-Bradley',
|
|
'djwhal_khul': 'Djwhal Khul',
|
|
'sassanian': 'Sassanian',
|
|
'true_citra': 'True Citra',
|
|
}
|
|
|
|
ACTIVE_AYANAMSA_NAME = 'lahiri'
|
|
|
|
|
|
def normalize_ayanamsa_name(name: Any = None) -> str:
|
|
key = str(name or 'lahiri').strip().lower().replace('-', '_')
|
|
if key in ('krishnamurti_paddhati', 'krishnamurti/kp'):
|
|
key = 'kp'
|
|
return key if key in AYANAMSA_MODES else 'lahiri'
|
|
|
|
|
|
def ayanamsa_display_name(name: Any = None) -> str:
|
|
key = normalize_ayanamsa_name(name)
|
|
return AYANAMSA_DISPLAY_NAMES.get(key, key.title())
|
|
|
|
|
|
def current_ayanamsa_name(args: Any = None) -> str:
|
|
name = getattr(args, 'ayanamsa', None) if args is not None else None
|
|
return normalize_ayanamsa_name(name or ACTIVE_AYANAMSA_NAME)
|
|
|
|
|
|
def apply_ayanamsa(name: Any = 'lahiri', swe_module: Any = None) -> bool:
|
|
"""Set Swiss Ephemeris sidereal mode explicitly."""
|
|
global ACTIVE_AYANAMSA_NAME
|
|
module = swe_module or swe
|
|
if module is None:
|
|
return False
|
|
key = normalize_ayanamsa_name(name)
|
|
try:
|
|
module.set_sid_mode(AYANAMSA_MODES[key])
|
|
except Exception:
|
|
return False
|
|
ACTIVE_AYANAMSA_NAME = key
|
|
return True
|
|
|
|
|
|
def sidereal_flags(swe_module: Any, ayanamsa_name: Any = 'lahiri') -> int:
|
|
apply_ayanamsa(ayanamsa_name, swe_module)
|
|
return getattr(swe_module, 'FLG_SWIEPH', 2) | getattr(swe_module, 'FLG_SIDEREAL', 65536)
|