1 import numpy as np
2 from contracts import contract, check, new_contract
3 from .. import assert_allclose
4
5 from .exceptions import DoesNotBelong
6
7 from .differentiable_manifold import DifferentiableManifold, RandomManifold
8 from .tangent_bundle import TangentBundle
9 new_contract('DifferentiableManifold', DifferentiableManifold)
10
11 from .product_manifold import ProductManifold
12
13 from .group import *
14 from .matrix_linear_space import *
15 from .matrix_lie_algebra import *
16 from .matrix_lie_group import *
17 from .matrix_lie_group_tangent import *
18
19 from .sphere import Sphere, Sphere1
20 S1 = Sphere1()
21 S2 = Sphere(2)
22 S = {1: S1, 2: S2}
23
24 from .torus import *
25 T1 = Torus(1)
26 T2 = Torus(2)
27 T3 = Torus(3)
28 T = {1: T1, 2: T2, 3: T3}
29
30 from .euclidean import Euclidean
31 R1 = Euclidean(1)
32 R2 = Euclidean(2)
33 R3 = Euclidean(3)
34 R = {1: R1, 2: R2, 3: R3}
35
36 from .translation_algebra import tran
37 tran1 = tran(1)
38 tran2 = tran(2)
39 tran3 = tran(3)
40 tran = {1: tran1, 2: tran2, 3: tran3}
41
42 from .translation_group import Tran
43 Tran1 = Tran(1)
44 Tran2 = Tran(2)
45 Tran3 = Tran(3)
46 Tran = {1: Tran1, 2: Tran2, 3: Tran3}
47
48 from .special_orthogonal_algebra import so_algebra
49 so2 = so_algebra(2)
50 so3 = so_algebra(3)
51 so = {2: so2, 3: so3}
52
53 from .special_orthogonal_group import SO_group
54 SO2 = SO_group(2)
55 SO3 = SO_group(3)
56 SO = {2: SO2, 3: SO3}
57
58 from .special_euclidean_algebra import se_algebra
59 se2 = se_algebra(2, alpha=1)
60 se3 = se_algebra(3, alpha=1)
61 se = {2: se2, 3: se3}
62
63 from .special_euclidean_group import SE_group
64 SE2 = SE_group(2)
65 SE3 = SE_group(3)
66 SE = {2: SE2, 3: SE3}
67
68 TSE2 = SE2.tangent_bundle()
69 TSE3 = SE3.tangent_bundle()
70 TSE = {2: TSE2, 3: TSE3}
71
72 all_manifolds = [
73 SO3, SO2,
74 R1, R2, R3,
75 T1, T2, T3,
76 Tran1, Tran2, Tran3,
77 SE2, SE3,
78 S1, S2,
79 se2, se3,
80 so2, so3,
81 tran1, tran2, tran3,
82 TSE2, TSE3
83 ]
84
85 from .embedding_relations import *
86 from .manifold_embedding_propagation import compute_manifold_relations
87 compute_manifold_relations(all_manifolds)
88