/
location.py
104 lines (70 loc) · 2.24 KB
/
location.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
'''location related functions'''
import pandas as pd
import swisseph as swe
import math
def _fit360(n):
while n >= 360:
n = n - 360
if n >= 0:
return n
else:
n = n + 360
return n
#
def loc_of_planet(planet, start, end, freq='1D', scale=1, fit360=False):
"""Calculate the locations of planet within a time span.
parameters:
planet: the planet variable in swisseph
start, end: the time span
freq: the calculation freq
scale: mulitply the planet location
return a pandas Series with planet location
"""
results = []
drange = pd.date_range(start, end, freq=freq, tz='utc')
for date in drange:
year = date.year
month = date.month
day = date.day
hour = date.hour
minute = date.minute
second = date.second
jd = swe.utc_to_jd(year, month, day, hour, minute, second, 1)
ut = jd[1]
loc = swe.calc_ut(ut, planet)
results.append(loc[0]*scale)
res = pd.Series(results, drange, name=swe.get_planet_name(planet))
if scale > 1 and fit360:
return res.apply(_fit360)
return res
#
def lon_to_text(lon):
"""return the text represetation of the location"""
signs = ['Ari', 'Tau', 'Gem', 'Can',
'Leo', 'Vir', 'Lib', 'Sco',
'Sag', 'Cap', 'Aqu', 'Pis'
]
unifiedLon = _fit360(lon)
signum = int(unifiedLon // 30)
deg_float = unifiedLon % 30
minutef, deg = math.modf(deg_float)
minute = round(minutef * 60)
string = str(int(deg)) + signs[signum] + str(int(minute))
return string
#
def _fit180(n):
if abs(n) >= 180:
return 360 - abs(n)
else:
return abs(n)
def location_diff(planet1, planet2, start, end, freq="1D", scale=1, fit180=True):
"""Calculate the difference of location between 2 planets in given time span"""
loc_planet1 = loc_of_planet(planet1, start, end, freq, scale, fit360=True)
loc_planet2 = loc_of_planet(planet2, start, end, freq, scale, fit360=True)
#name1 = swe.get_planet_name(planet1)
#name2 = swe.get_planet_name(planet2)
diff = loc_planet1 - loc_planet2
if fit180:
return diff.apply(_fit180)
return diff
#