Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
pycodegen
lbmpy_walberla
Commits
a11fc764
Commit
a11fc764
authored
Jul 11, 2019
by
Martin Bauer
Browse files
Import sorting with isort
parent
2c3ca968
Pipeline
#16411
passed with stage
in 1 minute and 49 seconds
Changes
6
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
.isort.cfg
0 → 100644
View file @
a11fc764
[settings]
line_length=100
balanced_wrapping=True
multi_line_output=4
lbmpy_walberla/__init__.py
View file @
a11fc764
from
.walberla_lbm_generation
import
generate_lattice_model
,
RefinementScaling
from
.boundary
import
generate_boundary
from
.walberla_lbm_generation
import
RefinementScaling
,
generate_lattice_model
__all__
=
[
'generate_lattice_model'
,
'RefinementScaling'
,
'generate_boundary'
]
lbmpy_walberla/boundary.py
View file @
a11fc764
import
numpy
as
np
from
jinja2
import
Environment
,
PackageLoader
from
lbmpy.boundaries.boundaryhandling
import
create_lattice_boltzmann_boundary_kernel
from
lbmpy_walberla.walberla_lbm_generation
import
KernelInfo
from
pystencils
import
Field
,
FieldType
from
pystencils.data_types
import
create_type
,
TypedSymbol
from
pystencils.boundaries.createindexlist
import
(
boundary_index_array_coordinate_names
,
direction_member_name
,
numpy_data_type_for_boundary_object
)
from
pystencils.data_types
import
TypedSymbol
,
create_type
from
pystencils_walberla.codegen
import
default_create_kernel_parameters
from
pystencils_walberla.jinja_filters
import
add_pystencils_filters_to_jinja_env
from
lbmpy.boundaries.boundaryhandling
import
create_lattice_boltzmann_boundary_kernel
from
pystencils.boundaries.createindexlist
import
numpy_data_type_for_boundary_object
,
\
boundary_index_array_coordinate_names
,
direction_member_name
from
lbmpy_walberla.walberla_lbm_generation
import
KernelInfo
def
generate_boundary
(
generation_context
,
class_name
,
boundary_object
,
lb_method
,
**
create_kernel_params
):
struct_name
=
"IndexInfo"
...
...
lbmpy_walberla/sparse.py
View file @
a11fc764
import
numpy
as
np
from
lbmpy.sparse
import
create_symbolic_list
,
create_lb_update_rule_sparse
,
create_macroscopic_value_getter_sparse
,
\
create_macroscopic_value_setter_sparse
from
lbmpy.sparse
import
(
create_lb_update_rule_sparse
,
create_macroscopic_value_getter_sparse
,
create_macroscopic_value_setter_sparse
,
create_symbolic_list
)
from
pystencils
import
TypedSymbol
,
create_kernel
...
...
lbmpy_walberla/walberla_lbm_generation.py
View file @
a11fc764
import
numpy
as
np
import
sympy
as
sp
from
sympy.tensor
import
IndexedBase
from
jinja2
import
Environment
,
PackageLoader
,
Template
import
numpy
as
np
from
sympy.tensor
import
IndexedBase
from
lbmpy.creationfunctions
import
create_lb_update_rule
,
update_with_default_parameters
from
lbmpy.relaxationrates
import
relaxation_rate_scaling
from
lbmpy.stencils
import
get_stencil
from
pystencils
import
AssignmentCollection
from
lbmpy.updatekernels
import
create_stream_pull_only_kernel
from
pystencils
import
AssignmentCollection
,
create_kernel
from
pystencils.astnodes
import
SympyAssignment
from
pystencils.sympyextensions
import
get_symmetric_part
from
pystencils.field
import
Field
from
pystencils.stencil
import
offset_to_direction_string
,
have_same_entries
from
pystencils.backends.cbackend
import
CustomSympyPrinter
,
CBackend
,
get_headers
from
pystencils.backends.cbackend
import
CBackend
,
CustomSympyPrinter
,
get_headers
from
pystencils.data_types
import
TypedSymbol
from
pystencils.field
import
Field
from
pystencils.stencil
import
have_same_entries
,
offset_to_direction_string
from
pystencils.sympyextensions
import
get_symmetric_part
from
pystencils.transformations
import
add_types
from
pystencils_walberla.codegen
import
KernelInfo
,
default_create_kernel_parameters
from
pystencils_walberla.jinja_filters
import
add_pystencils_filters_to_jinja_env
from
pystencils
import
create_kernel
from
lbmpy.relaxationrates
import
relaxation_rate_scaling
from
lbmpy.creationfunctions
import
update_with_default_parameters
,
create_lb_update_rule
from
lbmpy.updatekernels
import
create_stream_pull_only_kernel
cpp_printer
=
CustomSympyPrinter
(
dialect
=
'c'
)
REFINEMENT_SCALE_FACTOR
=
sp
.
Symbol
(
"level_scale_factor"
)
...
...
lbmpy_walberla_tests/test_walberla_codegen.py
View file @
a11fc764
import
unittest
import
sympy
as
sp
import
pystencils
as
ps
from
lbmpy.boundaries
import
NoSlip
,
UBB
from
lbmpy.boundaries
import
UBB
,
NoSlip
from
lbmpy.creationfunctions
import
create_lb_method
,
create_lb_update_rule
from
lbmpy_walberla
import
generate_lattice_model
,
RefinementScaling
,
generate_boundary
from
lbmpy_walberla
import
RefinementScaling
,
generate_boundary
,
generate_lattice_model
from
lbmpy_walberla.sparse
import
ListLbGenerator
from
pystencils_walberla.cmake_integration
import
ManualCodeGenerationContext
from
pystencils_walberla
import
generate_pack_info_for_field
,
generate_pack_info_from_kernel
import
unit
te
s
t
from
pystencils_walberla.cmake_integration
import
ManualCodeGenerationCon
te
x
t
class
WalberlaLbmpyCodegenTest
(
unittest
.
TestCase
):
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment