from .version import __version__

from . import dobj

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 .domain_tuple import DomainTuple
from .multi_domain import MultiDomain
from .field import Field
from .multi_field import MultiField

from .operators.operator import Operator
from .operators.adder import Adder
from .operators.log1p import Log1p
from .operators.diagonal_operator import DiagonalOperator
from .operators.distributors import DOFDistributor, PowerDistributor
from .operators.domain_tuple_field_inserter import DomainTupleFieldInserter
from .operators.contraction_operator import ContractionOperator
from .operators.linear_interpolation import LinearInterpolator
from .operators.endomorphic_operator import EndomorphicOperator
from .operators.harmonic_operators import (
    FFTOperator, HartleyOperator, SHTOperator, HarmonicTransformOperator,
    HarmonicSmoothingOperator)
from .operators.field_zero_padder import FieldZeroPadder
from .operators.inversion_enabler import InversionEnabler
from .operators.linear_operator import LinearOperator
from .operators.mask_operator import MaskOperator
from .operators.regridding_operator import RegriddingOperator
from .operators.sampling_enabler import SamplingEnabler
from .operators.sandwich_operator import SandwichOperator
from .operators.scaling_operator import ScalingOperator
from .operators.block_diagonal_operator import BlockDiagonalOperator
from .operators.outer_product_operator import OuterProduct
from .operators.simple_linear_operators import (
    VdotOperator, ConjugationOperator, Realizer,
    FieldAdapter, ducktape, GeometryRemover, NullOperator,
    MatrixProductOperator, PartialExtractor)
from .operators.value_inserter import ValueInserter
from .operators.energy_operators import (
    EnergyOperator, GaussianEnergy, PoissonianEnergy, InverseGammaLikelihood,
    BernoulliEnergy, StandardHamiltonian, AveragedEnergy, QuadraticFormOperator,
    Squared2NormOperator, StudentTEnergy)
from .operators.convolution_operators import FuncConvolutionOperator

from .probing import probe_with_posterior_samples, probe_diagonal, \
    StatCalculator, approximation2endo

from .minimization.line_search import LineSearch
from .minimization.iteration_controllers import (
    IterationController, GradientNormController, DeltaEnergyController,
    GradInfNormController, AbsDeltaEnergyController)
from .minimization.minimizer import Minimizer
from .minimization.conjugate_gradient import ConjugateGradient
from .minimization.nonlinear_cg import NonlinearCG
from .minimization.descent_minimizers import (
    DescentMinimizer, SteepestDescent, VL_BFGS, L_BFGS, RelaxedNewton,
    NewtonCG)
from .minimization.scipy_minimizer import L_BFGS_B
from .minimization.energy import Energy
from .minimization.quadratic_energy import QuadraticEnergy
from .minimization.energy_adapter import EnergyAdapter
from .minimization.metric_gaussian_kl import MetricGaussianKL

from .sugar import *
from .plot import Plot

from .library.inverse_gamma_operator import InverseGammaOperator
from .library.los_response import LOSResponse
from .library.dynamic_operator import (dynamic_operator,
                                       dynamic_lightcone_operator)
from .library.light_cone_operator import LightConeOperator

from .library.wiener_filter_curvature import WienerFilterCurvature
from .library.adjust_variances import (make_adjust_variances_hamiltonian,
                                       do_adjust_variances)
from .library.gridder import GridderMaker
from .library.correlated_fields import CorrelatedFieldMaker

from . import extra

from .utilities import memo, frozendict

from .logger import logger

from .linearization import Linearization

from .operator_spectrum import operator_spectrum

from . import internal_config
_scheme = internal_config.parallelization_scheme()
if _scheme == "Samples":
    from .minimization.metric_gaussian_kl_mpi import MetricGaussianKL_MPI

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