Commit be4a4290 authored by Martin Bauer's avatar Martin Bauer
Browse files

Import sorting using isort

parent f28c8504
[settings]
line_length=100
balanced_wrapping=True
multi_line_output=4
from lbmpy.boundaries.boundaryconditions import NoSlip, UBB, FixedDensity, NeumannByCopy, StreamInConstant from lbmpy.boundaries.boundaryconditions import (
UBB, FixedDensity, NeumannByCopy, NoSlip, StreamInConstant)
from lbmpy.boundaries.boundaryhandling import LatticeBoltzmannBoundaryHandling from lbmpy.boundaries.boundaryhandling import LatticeBoltzmannBoundaryHandling
__all__ = ['NoSlip', 'UBB', 'FixedDensity', 'NeumannByCopy', 'LatticeBoltzmannBoundaryHandling', 'StreamInConstant'] __all__ = ['NoSlip', 'UBB', 'FixedDensity', 'NeumannByCopy', 'LatticeBoltzmannBoundaryHandling', 'StreamInConstant']
import sympy as sp import sympy as sp
from pystencils import Field, Assignment
from lbmpy.boundaries.boundaryhandling import BoundaryOffsetInfo, LbmWeightInfo
from lbmpy.simplificationfactory import create_simplification_strategy
from pystencils import Assignment, Field
from pystencils.astnodes import SympyAssignment from pystencils.astnodes import SympyAssignment
from pystencils.sympyextensions import get_symmetric_part
from pystencils.data_types import create_type from pystencils.data_types import create_type
from lbmpy.simplificationfactory import create_simplification_strategy from pystencils.sympyextensions import get_symmetric_part
from lbmpy.boundaries.boundaryhandling import BoundaryOffsetInfo, LbmWeightInfo
class Boundary: class Boundary:
......
import numpy as np import numpy as np
import sympy as sp import sympy as sp
from pystencils import TypedSymbol, create_indexed_kernel, Assignment
from pystencils import Assignment, TypedSymbol, create_indexed_kernel
from pystencils.backends.cbackend import CustomCodeNode from pystencils.backends.cbackend import CustomCodeNode
from pystencils.boundaries import BoundaryHandling from pystencils.boundaries import BoundaryHandling
from pystencils.boundaries.boundaryhandling import BoundaryOffsetInfo from pystencils.boundaries.boundaryhandling import BoundaryOffsetInfo
......
from lbmpy.chapman_enskog.chapman_enskog import ChapmanEnskogAnalysis, remove_higher_order_u, \ from lbmpy.chapman_enskog.chapman_enskog import (
take_moments, chapman_enskog_ansatz, CeMoment CeMoment, ChapmanEnskogAnalysis, chapman_enskog_ansatz, remove_higher_order_u, take_moments)
from lbmpy.chapman_enskog.chapman_enskog_steady_state import SteadyStateChapmanEnskogAnalysis from lbmpy.chapman_enskog.chapman_enskog_steady_state import SteadyStateChapmanEnskogAnalysis
__all__ = ['ChapmanEnskogAnalysis', 'remove_higher_order_u', 'take_moments', 'chapman_enskog_ansatz', 'CeMoment', __all__ = ['ChapmanEnskogAnalysis', 'remove_higher_order_u', 'take_moments', 'chapman_enskog_ansatz', 'CeMoment',
......
import sympy as sp
from collections import namedtuple from collections import namedtuple
import sympy as sp
from sympy.core.cache import cacheit from sympy.core.cache import cacheit
from lbmpy.chapman_enskog.derivative import (
chapman_enskog_derivative_expansion, chapman_enskog_derivative_recombination)
from lbmpy.moments import (
discrete_moment, get_moment_indices, moment_matrix, non_aliased_moment,
polynomial_to_exponent_representation)
from pystencils.cache import disk_cache from pystencils.cache import disk_cache
from pystencils.fd import expand_diff_full, Diff, DiffOperator, expand_diff_linear, \ from pystencils.fd import (
expand_diff_products, normalize_diff_order Diff, DiffOperator, expand_diff_full, expand_diff_linear, expand_diff_products,
normalize_diff_order)
from pystencils.sympyextensions import normalize_product, symmetric_product from pystencils.sympyextensions import normalize_product, symmetric_product
from lbmpy.moments import discrete_moment, moment_matrix, polynomial_to_exponent_representation, get_moment_indices, \
non_aliased_moment
from lbmpy.chapman_enskog.derivative import chapman_enskog_derivative_recombination, chapman_enskog_derivative_expansion
class ChapmanEnskogAnalysis: class ChapmanEnskogAnalysis:
......
from collections import OrderedDict, namedtuple
import sympy as sp import sympy as sp
from lbmpy.chapman_enskog.chapman_enskog import CeMoment, Diff, expanded_symbol, take_moments
from lbmpy.moments import moments_of_order, moments_up_to_order
from pystencils.fd import expand_diff_full from pystencils.fd import expand_diff_full
from lbmpy.chapman_enskog.chapman_enskog import CeMoment, Diff, take_moments
from lbmpy.chapman_enskog.chapman_enskog import expanded_symbol
from lbmpy.moments import moments_up_to_order, moments_of_order
from collections import OrderedDict, namedtuple
def poly_moments(order, dim): def poly_moments(order, dim):
......
import sympy as sp
import functools import functools
from pystencils.fd import Diff, DiffOperator, expand_diff_linear, normalize_diff_order, \
collect_diffs import sympy as sp
from pystencils.sympyextensions import normalize_product, multidimensional_sum, kronecker_delta
from lbmpy.chapman_enskog.chapman_enskog import LbMethodEqMoments, CeMoment, take_moments, insert_moments from lbmpy.chapman_enskog.chapman_enskog import (
from lbmpy.chapman_enskog.chapman_enskog import expanded_symbol, chapman_enskog_ansatz, remove_higher_order_u CeMoment, LbMethodEqMoments, chapman_enskog_ansatz, expanded_symbol, insert_moments,
remove_higher_order_u, take_moments)
from pystencils.fd import (
Diff, DiffOperator, collect_diffs, expand_diff_linear, normalize_diff_order)
from pystencils.sympyextensions import kronecker_delta, multidimensional_sum, normalize_product
class SteadyStateChapmanEnskogAnalysis: class SteadyStateChapmanEnskogAnalysis:
......
import sympy as sp import sympy as sp
from pystencils.fd import Diff from pystencils.fd import Diff
......
...@@ -163,31 +163,32 @@ For example, to modify the AST one can run:: ...@@ -163,31 +163,32 @@ For example, to modify the AST one can run::
func = create_lb_function(ast=ast, ...) func = create_lb_function(ast=ast, ...)
""" """
import sympy as sp
from copy import copy from copy import copy
import sympy as sp
import lbmpy.forcemodels as forcemodels
from lbmpy.fieldaccess import (
AAEvenTimeStepAccessor, AAOddTimeStepAccessor, CollideOnlyInplaceAccessor,
EsoTwistEvenTimeStepAccessor, EsoTwistOddTimeStepAccessor, PdfFieldAccessor,
PeriodicTwoFieldsAccessor, StreamPullTwoFieldsAccessor, StreamPushTwoFieldsAccessor)
from lbmpy.methods import (
create_mrt3, create_mrt_orthogonal, create_mrt_raw, create_srt, create_trt, create_trt_kbc)
from lbmpy.methods.creationfunctions import create_generic_mrt from lbmpy.methods.creationfunctions import create_generic_mrt
from lbmpy.methods.cumulantbased import CumulantBasedLbMethod from lbmpy.methods.cumulantbased import CumulantBasedLbMethod
from pystencils.cache import disk_cache_no_fallback from lbmpy.methods.entropic import add_entropy_condition, add_iterative_entropy_condition
from pystencils.data_types import collate_types
from pystencils import AssignmentCollection
from pystencils.field import get_layout_of_array, Field
from pystencils import create_kernel, Assignment
from lbmpy.turbulence_models import add_smagorinsky_model
from lbmpy.methods import create_srt, create_trt, create_mrt_orthogonal, create_trt_kbc, \
create_mrt_raw, create_mrt3
from lbmpy.methods.entropic import add_iterative_entropy_condition, add_entropy_condition
from lbmpy.methods.entropic_eq_srt import create_srt_entropic from lbmpy.methods.entropic_eq_srt import create_srt_entropic
from lbmpy.relaxationrates import relaxation_rate_from_magic_number from lbmpy.relaxationrates import relaxation_rate_from_magic_number
from lbmpy.simplificationfactory import create_simplification_strategy
from lbmpy.stencils import get_stencil from lbmpy.stencils import get_stencil
from lbmpy.turbulence_models import add_smagorinsky_model
from lbmpy.updatekernels import create_lbm_kernel, create_stream_pull_with_output_kernel
from pystencils import Assignment, AssignmentCollection, create_kernel
from pystencils.cache import disk_cache_no_fallback
from pystencils.data_types import collate_types
from pystencils.field import Field, get_layout_of_array
from pystencils.simp import add_subexpressions_for_field_reads from pystencils.simp import add_subexpressions_for_field_reads
from pystencils.stencil import have_same_entries from pystencils.stencil import have_same_entries
import lbmpy.forcemodels as forcemodels
from lbmpy.simplificationfactory import create_simplification_strategy
from lbmpy.fieldaccess import StreamPullTwoFieldsAccessor, PeriodicTwoFieldsAccessor, CollideOnlyInplaceAccessor, \
EsoTwistEvenTimeStepAccessor, EsoTwistOddTimeStepAccessor, AAEvenTimeStepAccessor, AAOddTimeStepAccessor, \
StreamPushTwoFieldsAccessor, PdfFieldAccessor
from lbmpy.updatekernels import create_lbm_kernel, create_stream_pull_with_output_kernel
def create_lb_function(ast=None, optimization={}, **kwargs): def create_lb_function(ast=None, optimization={}, **kwargs):
......
...@@ -8,8 +8,7 @@ import sympy as sp ...@@ -8,8 +8,7 @@ import sympy as sp
from lbmpy.continuous_distribution_measures import multi_differentiation from lbmpy.continuous_distribution_measures import multi_differentiation
from lbmpy.moments import moments_up_to_component_order from lbmpy.moments import moments_up_to_component_order
from pystencils.cache import memorycache from pystencils.cache import memorycache
from pystencils.sympyextensions import fast_subs from pystencils.sympyextensions import fast_subs, scalar_product
from pystencils.sympyextensions import scalar_product
def __get_indexed_symbols(passed_symbols, prefix, indices): def __get_indexed_symbols(passed_symbols, prefix, indices):
......
import sympy as sp
import abc import abc
from pystencils.stencil import inverse_direction
from pystencils import Field
import sympy as sp
from pystencils import Field
# ------------------------------------------------ Interface -----------------------------------------------------------
from pystencils.astnodes import LoopOverCoordinate
from pystencils.stencil import inverse_direction
__all__ = ['PdfFieldAccessor', 'CollideOnlyInplaceAccessor', 'StreamPullTwoFieldsAccessor', __all__ = ['PdfFieldAccessor', 'CollideOnlyInplaceAccessor', 'StreamPullTwoFieldsAccessor',
'AAEvenTimeStepAccessor', 'AAOddTimeStepAccessor', 'AAEvenTimeStepAccessor', 'AAOddTimeStepAccessor',
...@@ -10,10 +13,6 @@ __all__ = ['PdfFieldAccessor', 'CollideOnlyInplaceAccessor', 'StreamPullTwoField ...@@ -10,10 +13,6 @@ __all__ = ['PdfFieldAccessor', 'CollideOnlyInplaceAccessor', 'StreamPullTwoField
'visualize_pdf_field_accessor', 'visualize_field_mapping'] 'visualize_pdf_field_accessor', 'visualize_field_mapping']
# ------------------------------------------------ Interface -----------------------------------------------------------
from pystencils.astnodes import LoopOverCoordinate
class PdfFieldAccessor: class PdfFieldAccessor:
""" """
Defines how data is read and written in an LBM time step. Defines how data is read and written in an LBM time step.
...@@ -24,6 +23,7 @@ class PdfFieldAccessor: ...@@ -24,6 +23,7 @@ class PdfFieldAccessor:
- esoteric twist single field update - esoteric twist single field update
- -
""" """
@abc.abstractmethod @abc.abstractmethod
def read(self, field, stencil): def read(self, field, stencil):
"""Returns sequence of field accesses for all stencil values where pdfs are read from""" """Returns sequence of field accesses for all stencil values where pdfs are read from"""
......
...@@ -87,6 +87,7 @@ second moment nonzero :class:`Luo` :class:`Guo` ...@@ -87,6 +87,7 @@ second moment nonzero :class:`Luo` :class:`Guo`
""" """
import sympy as sp import sympy as sp
from lbmpy.relaxationrates import get_shear_relaxation_rate from lbmpy.relaxationrates import get_shear_relaxation_rate
......
import numpy as np import numpy as np
from lbmpy.boundaries import NoSlip, UBB from lbmpy.boundaries import UBB, NoSlip
from pystencils.slicing import normalize_slice, shift_slice, slice_intersection, slice_from_direction from pystencils.slicing import (
normalize_slice, shift_slice, slice_from_direction, slice_intersection)
def add_box_boundary(boundary_handling, boundary=NoSlip(), replace=True): def add_box_boundary(boundary_handling, boundary=NoSlip(), replace=True):
......
import sympy as sp
from collections import defaultdict from collections import defaultdict
from pystencils import Field
import sympy as sp
from lbmpy.methods.abstractlbmethod import LbmCollisionRule from lbmpy.methods.abstractlbmethod import LbmCollisionRule
from pystencils import Field
def create_lbm_split_groups(cr: LbmCollisionRule, opposing_directions=True): def create_lbm_split_groups(cr: LbmCollisionRule, opposing_directions=True):
......
from types import MappingProxyType from types import MappingProxyType
import numpy as np import numpy as np
from lbmpy.boundaries.boundaryhandling import LatticeBoltzmannBoundaryHandling from lbmpy.boundaries.boundaryhandling import LatticeBoltzmannBoundaryHandling
from lbmpy.creationfunctions import switch_to_symbolic_relaxation_rates_for_omega_adapting_methods, \ from lbmpy.creationfunctions import (
create_lb_function, update_with_default_parameters create_lb_function, switch_to_symbolic_relaxation_rates_for_omega_adapting_methods,
from lbmpy.macroscopic_value_kernels import create_advanced_velocity_setter_collision_rule, \ update_with_default_parameters)
pdf_initialization_assignments from lbmpy.macroscopic_value_kernels import (
create_advanced_velocity_setter_collision_rule, pdf_initialization_assignments)
from lbmpy.simplificationfactory import create_simplification_strategy from lbmpy.simplificationfactory import create_simplification_strategy
from lbmpy.stencils import get_stencil from lbmpy.stencils import get_stencil
from pystencils import create_kernel, make_slice, create_data_handling from pystencils import create_data_handling, create_kernel, make_slice
from pystencils.slicing import SlicedGetter from pystencils.slicing import SlicedGetter
from pystencils.timeloop import TimeLoop from pystencils.timeloop import TimeLoop
......
import functools import functools
from copy import deepcopy from copy import deepcopy
from pystencils.field import Field, get_layout_of_array
from lbmpy.simplificationfactory import create_simplification_strategy from lbmpy.simplificationfactory import create_simplification_strategy
from pystencils.field import Field, get_layout_of_array
def pdf_initialization_assignments(lb_method, density, velocity, pdfs): def pdf_initialization_assignments(lb_method, density, velocity, pdfs):
......
...@@ -23,9 +23,10 @@ Examples: ...@@ -23,9 +23,10 @@ Examples:
""" """
import numpy as np
import warnings import warnings
import numpy as np
# Optional packages cpuinfo, pycuda and psutil for hardware queries # Optional packages cpuinfo, pycuda and psutil for hardware queries
try: try:
from cpuinfo import get_cpu_info from cpuinfo import get_cpu_info
......
from lbmpy.methods.momentbased import RelaxationInfo, AbstractLbMethod, AbstractConservedQuantityComputation, \ from lbmpy.methods.creationfunctions import (
MomentBasedLbMethod create_mrt3, create_mrt_orthogonal, create_mrt_raw, create_srt, create_trt, create_trt_kbc,
create_trt_with_magic_number, create_with_continuous_maxwellian_eq_moments,
create_with_discrete_maxwellian_eq_moments)
from lbmpy.methods.momentbased import (
AbstractConservedQuantityComputation, AbstractLbMethod, MomentBasedLbMethod, RelaxationInfo)
from .conservedquantitycomputation import DensityVelocityComputation from .conservedquantitycomputation import DensityVelocityComputation
from lbmpy.methods.creationfunctions import create_srt, create_trt, create_trt_with_magic_number, create_trt_kbc, \
create_mrt_orthogonal, create_mrt_raw, create_mrt3, \
create_with_continuous_maxwellian_eq_moments, create_with_discrete_maxwellian_eq_moments
__all__ = ['RelaxationInfo', 'AbstractLbMethod', __all__ = ['RelaxationInfo', 'AbstractLbMethod',
'AbstractConservedQuantityComputation', 'DensityVelocityComputation', 'MomentBasedLbMethod', 'AbstractConservedQuantityComputation', 'DensityVelocityComputation', 'MomentBasedLbMethod',
......
import abc import abc
import sympy as sp
from collections import namedtuple from collections import namedtuple
from pystencils import AssignmentCollection
import sympy as sp
from pystencils import AssignmentCollection
RelaxationInfo = namedtuple('RelaxationInfo', ['equilibrium_value', 'relaxation_rate']) RelaxationInfo = namedtuple('RelaxationInfo', ['equilibrium_value', 'relaxation_rate'])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment