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

from . import dobj
Martin Reinecke's avatar
Martin Reinecke committed
4 5 6 7 8 9 10 11 12 13 14 15

from .domains.domain import Domain
from .domains.structured_domain import StructuredDomain
from .domains.unstructured_domain import UnstructuredDomain
from .domains.rg_space import RGSpace
from .domains.lm_space import LMSpace
from .domains.gl_space import GLSpace
from .domains.hp_space import HPSpace
from .domains.power_space import PowerSpace
from .domains.dof_space import DOFSpace
from .domains.log_rg_space import LogRGSpace

16
from .domain_tuple import DomainTuple
Martin Reinecke's avatar
Martin Reinecke committed
17
from .multi_domain import MultiDomain
18
from .field import Field
Martin Reinecke's avatar
Martin Reinecke committed
19
from .multi_field import MultiField
Philipp Arras's avatar
Fixups  
Philipp Arras committed
20

Martin Reinecke's avatar
Martin Reinecke committed
21
from .operators.operator import Operator
22
from .operators.central_zero_padder import CentralZeroPadder
Martin Reinecke's avatar
Martin Reinecke committed
23 24 25 26 27 28
from .operators.diagonal_operator import DiagonalOperator
from .operators.dof_distributor import DOFDistributor
from .operators.domain_distributor import DomainDistributor
from .operators.endomorphic_operator import EndomorphicOperator
from .operators.exp_transform import ExpTransform
from .operators.fft_operator import FFTOperator
Martin Reinecke's avatar
Martin Reinecke committed
29
from .operators.field_zero_padder import FieldZeroPadder
Martin Reinecke's avatar
Martin Reinecke committed
30 31
from .operators.hartley_operator import HartleyOperator
from .operators.harmonic_smoothing_operator import HarmonicSmoothingOperator
Martin Reinecke's avatar
Martin Reinecke committed
32 33 34 35 36 37 38 39 40 41 42 43 44
from .operators.harmonic_transform_operator import HarmonicTransformOperator
from .operators.inversion_enabler import InversionEnabler
from .operators.laplace_operator import LaplaceOperator
from .operators.linear_operator import LinearOperator
from .operators.mask_operator import MaskOperator
from .operators.power_distributor import PowerDistributor
from .operators.qht_operator import QHTOperator
from .operators.sampling_enabler import SamplingEnabler
from .operators.sandwich_operator import SandwichOperator
from .operators.scaling_operator import ScalingOperator
from .operators.slope_operator import SlopeOperator
from .operators.smoothness_operator import SmoothnessOperator
from .operators.symmetrizing_operator import SymmetrizingOperator
Martin Reinecke's avatar
Martin Reinecke committed
45
from .operators.block_diagonal_operator import BlockDiagonalOperator
Martin Reinecke's avatar
Martin Reinecke committed
46 47 48
from .operators.simple_linear_operators import (
    VdotOperator, SumReductionOperator, ConjugationOperator, Realizer,
    FieldAdapter, GeometryRemover, NullOperator)
Martin Reinecke's avatar
Martin Reinecke committed
49 50 51
from .operators.energy_operators import (
    EnergyOperator, GaussianEnergy, PoissonianEnergy, BernoulliEnergy,
    Hamiltonian, SampledKullbachLeiblerDivergence)
Martin Reinecke's avatar
Martin Reinecke committed
52

53 54
from .probing.utils import probe_with_posterior_samples, probe_diagonal, \
    StatCalculator
Martin Reinecke's avatar
Martin Reinecke committed
55 56 57 58 59 60 61 62 63 64 65 66 67

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.l_bfgs import L_BFGS
from .minimization.relaxed_newton import RelaxedNewton
Martin Reinecke's avatar
Martin Reinecke committed
68 69
from .minimization.scipy_minimizer import (ScipyMinimizer, NewtonCG, L_BFGS_B,
                                           ScipyCG)
Martin Reinecke's avatar
Martin Reinecke committed
70 71 72
from .minimization.energy import Energy
from .minimization.quadratic_energy import QuadraticEnergy
from .minimization.line_energy import LineEnergy
Martin Reinecke's avatar
Martin Reinecke committed
73
from .minimization.energy_adapter import EnergyAdapter
Martin Reinecke's avatar
Martin Reinecke committed
74

Martin Reinecke's avatar
Martin Reinecke committed
75
from .sugar import *
Martin Reinecke's avatar
Martin Reinecke committed
76
from .plotting.plot import plot, plot_finish
Philipp Arras's avatar
Fixups  
Philipp Arras committed
77

Martin Reinecke's avatar
Martin Reinecke committed
78
from .library.amplitude_model import AmplitudeModel
Martin Reinecke's avatar
Martin Reinecke committed
79
from .library.los_response import LOSResponse
Martin Reinecke's avatar
merge  
Martin Reinecke committed
80 81 82

#from .library.inverse_gamma_model import InverseGammaModel

Philipp Arras's avatar
Fixup  
Philipp Arras committed
83
from .library.wiener_filter_curvature import WienerFilterCurvature
84
from .library.correlated_fields import CorrelatedField
Martin Reinecke's avatar
Martin Reinecke committed
85
#                                         make_mf_correlated_field)
Martin Reinecke's avatar
Martin Reinecke committed
86

Philipp Arras's avatar
Move it  
Philipp Arras committed
87
from . import extra
88

Martin Reinecke's avatar
Martin Reinecke committed
89
from .utilities import memo, frozendict
Martin Reinecke's avatar
Martin Reinecke committed
90

Martin Reinecke's avatar
Martin Reinecke committed
91 92
from .logger import logger

Martin Reinecke's avatar
Martin Reinecke committed
93
from .linearization import Linearization
Martin Reinecke's avatar
Martin Reinecke committed
94

95
# We deliberately don't set __all__ here, because we don't want people to do a
Philipp Arras's avatar
Philipp Arras committed
96
# "from nifty5 import *"; that would swamp the global namespace.