__init__.py 2.85 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
56

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