__init__.py 2.88 KB
Newer Older
1
from .version import __version__
Martin Reinecke's avatar
Martin Reinecke committed
2
3
4

from . import dobj

Martin Reinecke's avatar
Martin Reinecke committed
5
6
7
from .spaces.domain import Domain
from .spaces.unstructured_domain import UnstructuredDomain
from .spaces.structured_domain import StructuredDomain
Martin Reinecke's avatar
Martin Reinecke committed
8
9
10
11
12
13
14
from .spaces.rg_space import RGSpace
from .spaces.lm_space import LMSpace
from .spaces.hp_space import HPSpace
from .spaces.gl_space import GLSpace
from .spaces.dof_space import DOFSpace
from .spaces.power_space import PowerSpace

15
16
from .domain_tuple import DomainTuple

Martin Reinecke's avatar
Martin Reinecke committed
17
18
19
20
from .operators.linear_operator import LinearOperator
from .operators.endomorphic_operator import EndomorphicOperator
from .operators.scaling_operator import ScalingOperator
from .operators.diagonal_operator import DiagonalOperator
21
from .operators.harmonic_transform_operator import HarmonicTransformOperator
Martin Reinecke's avatar
Martin Reinecke committed
22
23
24
from .operators.fft_operator import FFTOperator
from .operators.fft_smoothing_operator import FFTSmoothingOperator
from .operators.direct_smoothing_operator import DirectSmoothingOperator
Martin Reinecke's avatar
Martin Reinecke committed
25
from .operators.geometry_remover import GeometryRemover
Martin Reinecke's avatar
Martin Reinecke committed
26
27
28
29
30
31
32
33
34
from .operators.laplace_operator import LaplaceOperator
from .operators.power_projection_operator import PowerProjectionOperator
from .operators.inversion_enabler import InversionEnabler

from .field import Field, sqrt, exp, log

from .probing.prober import Prober
from .probing.diagonal_prober_mixin import DiagonalProberMixin
from .probing.trace_prober_mixin import TraceProberMixin
35
from .probing.utils import probe_with_posterior_samples
Martin Reinecke's avatar
Martin Reinecke committed
36
37
38
39
40
41
42
43
44
45
46
47

from .minimization.line_search import LineSearch
from .minimization.line_search_strong_wolfe import LineSearchStrongWolfe
from .minimization.iteration_controller import IterationController
from .minimization.gradient_norm_controller import GradientNormController
from .minimization.minimizer import Minimizer
from .minimization.conjugate_gradient import ConjugateGradient
from .minimization.nonlinear_cg import NonlinearCG
from .minimization.descent_minimizer import DescentMinimizer
from .minimization.steepest_descent import SteepestDescent
from .minimization.vl_bfgs import VL_BFGS
from .minimization.relaxed_newton import RelaxedNewton
Martin Reinecke's avatar
Martin Reinecke committed
48
from .minimization.scipy_minimizer import NewtonCG, L_BFGS_B
Martin Reinecke's avatar
Martin Reinecke committed
49
50
51
52
from .minimization.energy import Energy
from .minimization.quadratic_energy import QuadraticEnergy
from .minimization.line_energy import LineEnergy

Martin Reinecke's avatar
Martin Reinecke committed
53
from .sugar import *
Martin Reinecke's avatar
Martin Reinecke committed
54
from .plotting.plot import plot
Martin Reinecke's avatar
Martin Reinecke committed
55
from . import library
Philipp Arras's avatar
Move it    
Philipp Arras committed
56
from . import extra
57

Martin Reinecke's avatar
Martin Reinecke committed
58
59
60
__all__ = ["Domain", "UnstructuredDomain", "StructuredDomain",
           "RGSpace", "LMSpace", "HPSpace", "GLSpace", "DOFSpace",
           "PowerSpace", "DomainTuple",
61
62
           "LinearOperator", "EndomorphicOperator", "ScalingOperator",
           "DiagonalOperator", "FFTOperator", "FFTSmoothingOperator",
Martin Reinecke's avatar
Martin Reinecke committed
63
           "DirectSmoothingOperator", "LaplaceOperator",
64
65
66
           "PowerProjectionOperator", "InversionEnabler",
           "Field", "sqrt", "exp", "log",
           "Prober", "DiagonalProberMixin", "TraceProberMixin"]