#!/usr/bin/env python
# coding: utf-8

# # Schwarzschild spacetime
# 
# This worksheet demonstrates a few capabilities of <a href="http://sagemanifolds.obspm.fr/">SageManifolds</a> (version 1.0, as included in SageMath 7.5) in computations regarding Schwarzschild spacetime.
# 
# Click [here](https://raw.githubusercontent.com/sagemanifolds/SageManifolds/master/Worksheets/v1.0/SM_Schwarzschild.ipynb) to download the worksheet file (ipynb format). To run it, you must start SageMath with the Jupyter notebook, via the command `sage -n jupyter`
# 

# *NB:* a version of SageMath at least equal to 7.5 is required to run this worksheet:

# In[1]:


version()


# First we set up the notebook to display mathematical objects using LaTeX rendering:

# In[2]:


get_ipython().run_line_magic('display', 'latex')


# ## Spacetime manifold
# 
# We declare the Schwarzschild spacetime as a 4-dimensional differentiable manifold:

# In[3]:


M = Manifold(4, 'M', r'\mathcal{M}') ; M


# In[4]:


print(M)


# <p>The spacetime manifold $\mathcal{M}$ can be split into 4 regions, corresponding to the 4 quadrants in the Kruskal diagram.Let us denote by $\mathcal{R}_{\mathrm{I}}$ to $\mathcal{R}_{\mathrm{IV}}$ the interiors of these 4 regions. $\mathcal{R}_{\mathrm{I}}$ and  $\mathcal{R}_{\mathrm{III}}$  are asymtotically flat regions outside the event horizon;  $\mathcal{R}_{\mathrm{II}}$ is inside the future event horizon and $\mathcal{R}_{\mathrm{IV}}$ is inside the past event horizon.</p>

# In[5]:


regI = M.open_subset('R_I', r'\mathcal{R}_{\mathrm{I}}')
regII = M.open_subset('R_II', r'\mathcal{R}_{\mathrm{II}}')
regIII = M.open_subset('R_III', r'\mathcal{R}_{\mathrm{III}}')
regIV = M.open_subset('R_IV', r'\mathcal{R}_{\mathrm{IV}}')
regI, regII, regIII, regIV


# <p><span id="cell_outer_8">The parameter $m$ of </span><span id="cell_outer_8">the Schwarzschild spacetime is declared as a symbolic variable:</span></p>

# In[6]:


m = var('m') ; assume(m>=0)


# <h2>Boyer-Lindquist coordinates</h2>
# <p>The standard Boyer-Lindquist coordinates (also called Schwarzschild coordinates) are defined on $\mathcal{R}_{\mathrm{I}}\cup \mathcal{R}_{\mathrm{II}}$</p>

# In[7]:


regI_II = regI.union(regII) ; regI_II


# In[8]:


X.<t,r,th,ph> = regI_II.chart(r't r:(0,+oo) th:(0,pi):\theta ph:(0,2*pi):\phi')
print(X)


# In[9]:


X


# <p>We naturally introduce two subcharts as the restrictions of the chart $X$ to regions $\mathcal{R}_{\mathrm{I}}$ and $\mathcal{R}_{\mathrm{II}}$ respectively. Since, in terms of the Boyer-Lindquist coordinates, $\mathcal{R}_{\mathrm{I}}$ (resp. $\mathcal{R}_{\mathrm{II}}$) is defined by $r>2m$ (resp. $r<2m$), we set</p>

# In[10]:


X_I = X.restrict(regI, r>2*m) ; X_I


# In[11]:


X_II = X.restrict(regII, r<2*m) ; X_II


# <p>At this stage, the manifold's atlas has 3 charts:</p>

# In[12]:


M.atlas()


# In[13]:


M.default_chart()


# <p>Three vector frames have been defined on the manifold: the three coordinate frames:</p>

# In[14]:


M.frames()


# In[15]:


print(M.default_frame())


# In[16]:


M.default_frame().domain()


# <h2>Metric tensor</h2>
# <p>The metric tensor is defined as follows:</p>

# In[17]:


g = M.lorentzian_metric('g')
print(g)


# <p><span>The metric tensor is set by its components in the coordinate frame associated with Schwarzschild coordinates, which is the current manifold's default frame:</span></p>

# In[18]:


g[0,0], g[1,1] = -(1-2*m/r), 1/(1-2*m/r)
g[2,2], g[3,3] = r^2, (r*sin(th))^2


# In[19]:


g.display()


# <p>As an example, let us consider a vector field defined only on $\mathcal{R}_{\mathrm{I}}$:</p>

# In[20]:


v = regI.vector_field('v')
v[0] = 1
v[1] = 1 - 2*m/r
# unset components are zero   
v.display()


# In[21]:


v.domain()


# In[22]:


g.domain()


# <p>Since $\mathcal{R}_{\mathrm{I}}\subset \mathcal{M}$, it is possible to apply $g$ to $v$:</p>

# In[23]:


s = g(v,v) ; print(s)


# In[24]:


s.display() # v is indeed a null vector


# <div id="cell_text_59" class="text_cell">
# <h2>Levi-Civita Connection</h2>
# </div>
# <div id="insert_new_cell_before27" class="insert_new_cell"> </div>
# <div id="cell_text_27" class="text_cell">
# <p>The Levi-Civita connection $\nabla$ associated with $g$:</p>
# </div>

# In[25]:


nab = g.connection() ; print(nab)


# <p>Let us verify that the covariant derivative of $g$ with respect to $\nabla$ vanishes identically:</p>

# In[26]:


nab(g) == 0


# In[27]:


nab(g).display()


# <p>The nonzero Christoffel symbols of $g$ with respect to Schwarzschild coordinates, skipping those that can be deduced by symmetry:</p>

# In[28]:


g.christoffel_symbols_display()


# <h2>Curvature</h2>
# 
# <p>The Riemann curvature tensor associated with $g$:</p>

# In[29]:


R = g.riemann() ; print(R)


# <p>The Weyl conformal tensor associated with $g$:</p>

# In[30]:


C = g.weyl() ; print(C)


# In[31]:


C.display()


# <p>The Ricci tensor associated with $g$:</p>

# In[32]:


Ric = g.ricci() ; print(Ric)


# <h3>Einstein equation</h3>
# <p>Let us check that the Schwarzschild metric is a solution of the vacuum Einstein equation:</p>

# In[33]:


Ric == 0


# In[34]:


Ric.display() # another view of the above


# <p>Contrary to the Ricci tensor, the Riemann tensor does not vanish:</p>

# In[35]:


R[:]


# In[36]:


R.display()


# <p>The nonzero components of the Riemann tensor, skipping those that can be deduced by antisymmetry:</p>

# In[37]:


R.display_comp(only_nonredundant=True)


# In[38]:


Ric[:]


# <p>Since the Ricci tensor is zero, the Weyl tensor is of course equal to the Riemann tensor:</p>

# In[39]:


C == R


# <h3>Bianchi identity</h3>
# 
# <p>Let us check the Bianchi identity $\nabla_p R^i_{\ \, j kl} + \nabla_k R^i_{\ \, jlp} + \nabla_l R^i_{\ \, jpk} = 0$:</p>

# In[40]:


DR = nab(R) ; print(DR)


# In[41]:


for i in M.irange():
    for j in M.irange():
        for k in M.irange():
            for l in M.irange():
                for p in M.irange():
                    print DR[i,j,k,l,p] + DR[i,j,l,p,k] + DR[i,j,p,k,l] ,


# <p>Let us check that if we turn the first $+$ sign into a $-$ one, the Bianchi identity does no longer hold:</p>

# In[42]:


for i in M.irange():
    for j in M.irange():
        for k in M.irange():
            for l in M.irange():
                for p in M.irange():
                    print DR[i,j,k,l,p] - DR[i,j,l,p,k] + DR[i,j,p,k,l] ,


# <h3>Kretschmann scalar</h3>
# <p>Let us first introduce tensor $R^\flat$, of components $R_{ijkl} = g_{ip} R^p_{\ \, jkl}$:</p>

# In[43]:


dR = R.down(g) ; print(dR)


# <p>and tensor $R^\sharp$, of components $R^{ijkl} = g^{jp} g^{kq} g^{lr} R^i_{\ \, pqr}$:</p>

# In[44]:


uR = R.up(g) ; print(uR)


# <p>The Kretschmann scalar is $K := R^{ijkl} R_{ijkl}$:</p>

# In[45]:


Kr = 0
for i in M.irange():
    for j in M.irange():
        for k in M.irange():
            for l in M.irange():
                Kr += uR[i,j,k,l]*dR[i,j,k,l]
Kr


# <p>Instead of the above loops, the Kretschmann scalar can also be computed by means of the contract() method, asking that the contraction takes place on all indices (positions 0, 1, 2, 3):</p>

# In[46]:


Kr = uR.contract(0, 1, 2, 3, dR, 0, 1, 2, 3)
Kr.expr()


# <p>The contraction can also be performed by means of index notations:</p>

# In[47]:


Kr = uR['^{ijkl}']*dR['_{ijkl}']
Kr.expr()


# <h2>Eddington-Finkelstein coordinates</h2>
# <p>Let us introduce new coordinates on the spacetime manifold: the ingoing Eddington-Finkelstein ones:</p>

# In[48]:


X_EF.<v,r,th,ph> = regI_II.chart(r'v r:(0,+oo) th:(0,pi):\theta ph:(0,2*pi):\varphi')
print(X_EF) ; X_EF


# <p>The change from Schwarzschild (Boyer-Lindquist) coordinates to the ingoing Eddington-Finkelstein ones:</p>

# In[49]:


ch_BL_EF_I = X_I.transition_map(X_EF, [t+r+2*m*ln(r/(2*m)-1), r, th, ph], restrictions2=r>2*m)


# In[50]:


print(ch_BL_EF_I) ; ch_BL_EF_I


# In[51]:


ch_BL_EF_I.display()


# In[52]:


X_EF_I = X_EF.restrict(regI) ; X_EF_I


# In[53]:


ch_BL_EF_II = X_II.transition_map(X_EF, [t+r+2*m*ln(1-r/(2*m)), r, th, ph], restrictions2=r<2*m)


# In[54]:


print(ch_BL_EF_II) ; ch_BL_EF_II


# In[55]:


ch_BL_EF_II.display()


# In[56]:


X_EF_II = X_EF.restrict(regII) ; X_EF_II


# <p>The manifold's atlas has now 6 charts:</p>

# In[57]:


M.atlas()


# <p>The default chart is 'BL':</p>

# In[58]:


M.default_chart()


# <p>The change from Eddington-Finkelstein coordinates to the Schwarzschild (Boyer-Lindquist) ones, computed as the inverse of ch_BL_EF:</p>

# In[59]:


ch_EF_BL_I = ch_BL_EF_I.inverse() ; print(ch_EF_BL_I)


# In[60]:


ch_EF_BL_I.display()


# In[61]:


ch_EF_BL_II = ch_BL_EF_II.inverse() ; print(ch_EF_BL_II)


# In[62]:


ch_EF_BL_II.display()


# <p>At this stage, 6 vector frames have been defined on the manifold: the 6 coordinate frames associated with the various charts:</p>

# In[63]:


M.frames()


# <p>The default frame is:</p>

# In[64]:


M.default_frame()


# <p>The coframes are the duals of the defined vector frames:</p>

# In[65]:


M.coframes()


# <p>If not specified, tensor components are assumed to refer to the manifold's default frame. For instance, for the metric tensor:</p>

# In[66]:


g.display()


# In[67]:


g[:]


# <p>The tensor components in the frame associated with Eddington-Finkelstein coordinates in Region I are obtained by providing the frame to the function <span style="font-family: courier new,courier;">display()</span>:</p>

# In[68]:


g.display(X_EF_I.frame())


# <p>They are also returned by the method comp(), with the frame as argument:</p>

# In[69]:


g.comp(X_EF_I.frame())[:]


# <p>or, as a schortcut,</p>

# In[70]:


g[X_EF_I.frame(),:]


# <p>Similarly, the metric<span id="cell_outer_53"> components in the frame associated with Eddington-Finkelstein coordinates in Region II are obtained by<br /></span></p>

# In[71]:


g.display(X_EF_II.frame())


# <p><span id="cell_outer_56">Note that their form is identical to that in Region I.</span></p>
# 
# <h3>Plot of the Boyer-Lindquist coordinates in terms of the Eddington-Finkelstein ones</h3>
# <p>Let us perform the plot in Region I:</p>

# In[72]:


X_I.plot(X_EF_I, ranges={t:(0,8), r:(2.1,10)}, fixed_coords={th:pi/2,ph:0}, ambient_coords=(r,v), style={t:'--', r:'-'}, parameters={m:1})


# <h2>Tetrad of the static observer</h2>
# <p>Let us introduce the orthonormal tetrad $(e_\alpha)$ associated with the static observers in Schwarzschild spacetime, i.e. the observers whose worldlines are parallel to the timelike Killing vector in the Region I.</p>
# <p>The orthonormal tetrad is defined via a tangent-space automorphism that relates it to the Boyer-Lindquist coordinate frame in Region I:</p>

# In[73]:


ch_to_stat = regI.automorphism_field()
ch_to_stat[0,0], ch_to_stat[1,1] = 1/sqrt(1-2*m/r), sqrt(1-2*m/r)
ch_to_stat[2,2], ch_to_stat[3,3] = 1/r, 1/(r*sin(th))
ch_to_stat[:]


# In[74]:


e = X_I.frame().new_frame(ch_to_stat, 'e') ; print(e)


# <p>At this stage, 7 vector frames have been defined on the manifold $\mathcal{M}$:</p>

# In[75]:


M.frames()


# <p>The first vector of the tetrad is the static observer 4-velocity:</p>

# In[76]:


print(e[0])


# In[77]:


e[0].display()


# <p>As any 4-velocity, it is a unit timelike vector:</p>

# In[78]:


g(e[0],e[0]).expr()


# <p>Let us check that the tetrad $(e_\alpha)$ is orthonormal:</p>

# In[79]:


for i in M.irange():
    for j in M.irange():
        print g(e[i],e[j]).expr() , 
    print " "


# <p>Another view of the above result:</p>

# In[80]:


g[e,:]


# <p>or, equivalently,</p>

# In[81]:


g.display(e)


# <p>The expression of the 4-velocity $e_0$ and the vector $e_1$ in terms of the frame associated with Eddington-Finkelstein coordinates:</p>

# In[82]:


e[0].display(X_EF_I.frame())


# In[83]:


e[1].display(X_EF_I.frame())


# <p>Contrary to vectors of a coordinate frame, the vectors of the tetrad $e$ do not commute: their structure coefficients are not identically zero:</p>

# In[84]:


e.structure_coeff()[:]


# <p>Equivalently, the Lie derivative of one vector along another one is not necessarily zero:</p>

# In[85]:


e[0].lie_der(e[1]).display(e)


# <p>The curvature 2-form $\Omega^0_{\ \, 1}$ associated with the tetrad $(e_\alpha)$:</p>

# In[86]:


g.connection().curvature_form(0,1,e).display(X_I.frame())


# <h2>Kruskal-Szekeres coordinates</h2>
# <p>Let us now introduce the Kruskal-Szekeres coordinates $(U,V,\theta,\varphi)$ on the spacetime manifold, via the standard transformation expressing them in terms of the Boyer-Lindquist coordinates $(t,r,\theta,\varphi)$:</p>

# In[87]:


M0 = regI.union(regII).union(regIII).union(regIV) ; M0


# In[88]:


X_KS.<U,V,th,ph> = M0.chart(r'U V th:(0,pi):\theta ph:(0,2*pi):\varphi')
X_KS.add_restrictions(V^2 < 1 + U^2)
X_KS


# In[89]:


X_KS_I = X_KS.restrict(regI, [U>0, V<U, V>-U]) ; X_KS_I


# In[90]:


ch_BL_KS_I = X_I.transition_map(X_KS_I, [sqrt(r/(2*m)-1)*exp(r/(4*m))*cosh(t/(4*m)), 
                                         sqrt(r/(2*m)-1)*exp(r/(4*m))*sinh(t/(4*m)), th, ph])
print(ch_BL_KS_I)
ch_BL_KS_I.display()


# In[91]:


X_KS_II = X_KS.restrict(regII, [V>0, V>U, V>-U]) ; X_KS_II


# In[92]:


ch_BL_KS_II = X_II.transition_map(X_KS_II, [sqrt(1-r/(2*m))*exp(r/(4*m))*sinh(t/(4*m)), 
                                            sqrt(1-r/(2*m))*exp(r/(4*m))*cosh(t/(4*m)), th, ph])
print(ch_BL_KS_II)
ch_BL_KS_II.display()


# <h3>Plot of the Boyer-Lindquist coordinates in terms of the Kruskal ones</h3>
# <p>We draw the Boyer-Lindquist chart in Region I (red) and Region II (green), with lines of constant $r$ being dashed:</p>

# In[93]:


graphI = X_I.plot(X_KS, ranges={t:(-12,12), r:(2.001,5)}, number_values={t:17, r:9}, 
                  fixed_coords={th:pi/2,ph:0}, ambient_coords=(U,V), 
                  style={t:'--', r:'-'}, parameters={m:1})
graphII = X_II.plot(X_KS, ranges={t:(-12,12), r:(0.001,1.999)}, number_values={t:17, r:9}, 
                    fixed_coords={th:pi/2,ph:0}, ambient_coords=(U,V), 
                    style={t:'--', r:'-'}, color='green', parameters={m:1})
show(graphI+graphII, xmin=-3, xmax=3, ymin=-3, ymax=3, axes_labels=['$U$', '$V$'])


# <p>We may add to the graph the singularity $r=0$ as a Boyer-Lindquist chart plot with $(r,\theta,\phi)$ fixed at $(0, \pi/2, 0)$.  Similarly, we add the event horizon $r=2m$ as a Boyer-Lindquist chart plot with $(r,\theta,\phi)$ fixed at $(2.00001m, \pi/2, 0)$:</p>

# In[94]:


graph_r0 = X_II.plot(X_KS, fixed_coords={r:0, th:pi/2, ph:0}, ambient_coords=(U,V),
                     color='yellow', thickness=3, parameters={m:1})
graph_r2 = X_I.plot(X_KS, ranges={t:(-40,40)}, fixed_coords={r:2.00001, th:pi/2, ph:0},
                    ambient_coords=(U,V), color='black', thickness=2, parameters={m:1})
show(graphI+graphII+graph_r0+graph_r2, xmin=-3, xmax=3, ymin=-3, ymax=3, 
     axes_labels=['$U$', '$V$'])


# <h3>Plot of the Eddington-Finkelstein coordinates in terms of the Kruskal ones</h3>
# <p>We first get the change of coordinates $(v,r,\theta,\phi) \mapsto (U,V,\theta,\phi)$ by composing the change $(v,r,\theta,\phi) \mapsto (t,r,\theta,\phi)$ with $(t,r,\theta,\phi) \mapsto (U,V,\theta,\phi)$: </p>

# In[95]:


ch_EF_KS_I = ch_BL_KS_I * ch_EF_BL_I
ch_EF_KS_I


# In[96]:


ch_EF_KS_I.display()


# In[97]:


ch_EF_KS_II = ch_BL_KS_II * ch_EF_BL_II
ch_EF_KS_II


# In[98]:


graphI_EF = X_EF_I.plot(X_KS, ranges={v:(-12,12), r:(2.001,5)}, number_values={v:17, r:9},
                        fixed_coords={th:pi/2,ph:0}, ambient_coords=(U,V),
                        style={v:'--', r:'-'}, parameters={m:1})
graphII_EF = X_EF_II.plot(X_KS, ranges={v:(-12,12), r:(0.001,1.999)}, number_values={v:17, r:9},
                          fixed_coords={th:pi/2,ph:0}, ambient_coords=(U,V),
                          style={v:'--', r:'-'}, color='green', parameters={m:1})
show(graphI_EF+graphII_EF+graph_r0+graph_r2, xmin=-3, xmax=3, ymin=-3, ymax=3,
     axes_labels=['$U$', '$V$'])


# <p>There are now 9 charts defined on the spacetime manifold:</p>

# In[99]:


M.atlas()


# In[100]:


len(M.atlas())


# <p>There are 8 explicit coordinate changes (the coordinate change KS $\rightarrow$ BL is not known in explicit form):</p>

# In[101]:


M.coord_changes()


# In[102]:


len(M.coord_changes())


# <p>There are 10 vector frames (among which 9 coordinate frames):</p>

# In[103]:


M.frames()


# In[104]:


len(M.frames())


# <p>There are 14 fields of tangent space automorphisms expressing the changes of coordinate bases and tetrad:</p>

# In[105]:


len(M.changes_of_frame())


# <p>Thanks to these changes of frames, the components of the metric tensor with respect to the Kruskal-Szekeres can be computed by the method <span style="font-family: courier new,courier;">display()</span> and are found to be:</p>

# In[106]:


g.display(X_KS_I.frame())


# In[107]:


g[X_KS_I.frame(),:]


# In[108]:


g.display(X_KS_II.frame())


# <p>The first vector of the orthonormal tetrad $e$ expressed on the <span id="cell_outer_86"><span id="cell_outer_121">Kruskal-Szekeres</span></span> frame:</p>

# In[109]:


e[0].display(X_KS_I.frame())


# <p>The Riemann curvature tensor in terms of the <span id="cell_outer_121">Kruskal-Szekeres</span> coordinates:</p>

# In[110]:


g.riemann().display(X_KS_I.frame())


# In[111]:


g.riemann().display_comp(X_KS_I.frame(), only_nonredundant=True)


# <p>The curvature 2-form $\Omega^0_{\ \, 1}$ associated to the Kruskal-Szekeres coordinate frame:</p>

# In[112]:


om = g.connection().curvature_form(0,1, X_KS_I.frame()) ; print(om)


# In[113]:


om.display(X_KS_I.frame())


# <h2>Isotropic coordinates</h2>
# <p>Let us now introduce isotropic coordinates $(t,\bar{r},\theta,\varphi)$ on the spacetime manifold:</p>

# In[114]:


regI_III = regI.union(regIII) ; regI_III


# In[115]:


X_iso.<t,ri,th,ph> = regI_III.chart(r't ri:(0,+oo):\bar{r} th:(0,pi):\theta ph:(0,2*pi):\varphi')
print(X_iso) ; X_iso


# In[116]:


X_iso_I = X_iso.restrict(regI, ri>m/2) ; X_iso_I


# <p>The transformation from the isotropic coordinates to the Boyer-Lindquist ones:</p>

# In[117]:


assume(2*ri>m) # we consider only region I
ch_iso_BL_I = X_iso_I.transition_map(X_I, [t, ri*(1+m/(2*ri))^2, th, ph])
print(ch_iso_BL_I)
ch_iso_BL_I.display()


# In[118]:


assume(r>2*m) # we consider only region I
ch_iso_BL_I.set_inverse(t, (r-m+sqrt(r*(r-2*m)))/2, th, ph, verbose=True)


# In[119]:


ch_iso_BL_I.inverse().display()


# <p><span id="cell_outer_90">At this stage, 11 charts have been defined on the manifold $\mathcal{M}$:</span></p>

# In[120]:


M.atlas()


# In[121]:


len(M.atlas())


# <p>12 vector frames have been defined on $\mathcal{M}$: 11 coordinate bases and the tetrad $(e_\alpha)$:</p>

# In[122]:


M.frames()


# In[123]:


len(M.frames())


# <p>The components of the metric tensor in terms of the isotropic coordinates are given by</p>

# In[124]:


g.display(X_iso_I.frame(), X_iso_I)


# <p>The $g_{00}$ component can be factorized:</p>

# In[125]:


g[X_iso_I.frame(), 0,0, X_iso_I]


# In[126]:


g[X_iso_I.frame(), 0,0, X_iso_I].factor()


# <p>Let us also factorize the other components:</p>

# In[127]:


for i in range(1,4):
    g[X_iso_I.frame(), i,i, X_iso_I].factor()


# <p>The output of the <span style="font-family: courier new,courier;">display()</span> command looks nicer:</p>

# In[128]:


g.display(X_iso_I.frame(), X_iso_I)


# <p>Expression of the tetrad associated with the static observer in terms of the isotropic coordinate basis:</p>

# In[129]:


e[0].display(X_iso_I.frame(), X_iso_I)


# In[130]:


e[1].display(X_iso_I.frame(), X_iso_I)


# In[131]:


e[2].display(X_iso_I.frame(), X_iso_I)


# In[132]:


e[3].display(X_iso_I.frame(), X_iso_I)


# In[ ]: