numericalRelativity/riem.py at master · ghyls/numericalRelativity · GitHub

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
from sympy import diff, simplify
from christ import christoffel


def riemman(ds, g_mn, abcd):

    # calcula R^a_{bcd}
    # ds se refiere al orden de las variabels en el intervalo
    # g_mn es la métrica
    # abcd = los índices de la componente de R que queremos calcular


    orden = []
    lenDS = 0
    for i, e in enumerate(ds):
        #str -> sym
        if e != '':
            lenDS += 1
            exec("var"+str(i)+"=sp.symbols('"+e+"')")
            exec("orden.append(var"+str(i)+")")


    # Creamos un array con los índices de los símbolos de Christoffel
    # \Gamma_ab^c que quedemos calcular,
    ABC = []

    # R^a_{bcd} tiene cuatro términos. Para cada término,
    ABC.append([abcd[3],  abcd[1],  abcd[0]])   #t1
    ABC.append([abcd[2],  abcd[1],  abcd[0]])   #t2
    for i in range(lenDS):
        ABC += [[abcd[2], ds[i], abcd[0]], [abcd[3], abcd[1], ds[i]]]    #t3
        ABC += [[abcd[3], ds[i], abcd[0]], [abcd[2], abcd[1], ds[i]]]    #t4

    symbols = christoffel(ds, g_mn, ABC=ABC)

    if type(symbols[0]) == str: return symbols # in case we dounf an error
    else: symbols = symbols[0]


    # cada termino de la expresión de R:

    r1 = diff(symbols[0], abcd[2])
    r2 = diff(symbols[1], abcd[3])
    r3 = r4 = 0

    i = 2   #rellenamos r3 y r4
    while i < len(symbols):
        r3 += symbols[i] * symbols[i+1]
        i += 2
        r4 += symbols[i] * symbols[i+1]
        i += 2


    S = simplify(r1 - r2 + r3 - r4)   #esto es R^a_{bcd}


    return([S])


#abcd = ["t", "r", "t", "r"]
#ds = ['t', 'r', 'th', 'fi']
#g_mn = [['-(1-2*G*M/r)', '', '', ''], ['', '1/(1-2*G*M/r)', '', ''], ['', '', 'r**2', ''], ['', '', '', 'r**2*sin(th)**2']]
#
#R = riemman(ds, g_mn, abcd)[0]
#print(R)
#R.subs('t', 'th')
#print(R)


'''
a simple but powerful application for symbolic calculus on Numerical Relativity


Riemman tensor
Cristophel symbol
symbolic calculator
Metric Simple


R T C S S C

Simple S
'''