demo.py

#!/usr/bin/env python
'''
Created on 9 Oct 2020

@author: Christoph Minz
@license: BSD 3-Clause
'''
from __future__ import annotations
from typing import List, Tuple  # @UnusedImport
from causets.sprinkledcauset import SprinkledCauset  # @UnresolvedImport
from causets.spacetimes import deSitterSpacetime  # @UnresolvedImport
from causets.shapes import CoordinateShape  # @UnresolvedImport
from causets.causetevent import CausetEvent  # @UnresolvedImport @UnusedImport
import causets.causetplotting as cplt  # @UnresolvedImport

# Create a sprinkle from de Sitter spacetime with cosmological horizon at
# radius 1.0. Coordinates range over a hollow cylinder with height 3.0. 30% of
# the cylinder interior is hollow.
S: CoordinateShape = CoordinateShape(3, 'cylinder', duration=3.0, hollow=0.3)
C: SprinkledCauset = SprinkledCauset(intensity=100.0,
                                     spacetime=deSitterSpacetime(3), shape=S)
e: CausetEvent = C.CentralAntichain().pop()  # pick one event

# Plotting setup:
cplt.setDefaultColors('UniYork')  # using University of York brand colours
dims: List[int] = [1, 2, 0]  # choose the (order of) plot dimensions
if len(dims) > 2:
    cplt.figure(figsize=(8.0, 8.0))
S.plot(dims)  # plot the embedding shape
# Add causet plots and show result:
cplt.plot(C, dims=dims, events={'alpha': 0.05},
          links={'alpha': 0.1, 'linewidth': 0.5}, labels=False)
cplt.plot(list(e.Cone), dims=dims, spacetime=C.Spacetime,
          events={'markerfacecolor': 'cs:darkblue'},
          links={'alpha': 0.6, 'linewidth': 1.5}, labels=False)
# end times for past and future light-cone:
timeslices: Tuple[float, float] = S.Limits(0)
cplt.plot(e, dims=dims, spacetime=C.Spacetime,
          events={'markerfacecolor': 'cs:red'},
          pastcones={'alpha': 1.0}, futurecones={'alpha': 1.0},
          time=timeslices)
ax: cplt.Axes = cplt.gca()
ax.set_xlabel('space' if dims[0] > 0 else 'time')
ax.set_ylabel('space' if dims[1] > 0 else 'time')
if len(dims) > 2:
    ax.set_zlabel('space' if dims[2] > 0 else 'time')
    ax.grid(False)
cplt.show()