Skip to content
Snippets Groups Projects
Commit 0d79bb47 authored by Timo Koch's avatar Timo Koch
Browse files

[python][io] Improve style, add some, use wrapper helpers

parent 338f84df
No related branches found
No related tags found
1 merge request!2681Feature/python main file
"""DuMux input-output library"""
from dune.generator.generator import SimpleGenerator from dune.generator.generator import SimpleGenerator
from dune.common.hashit import hashIt from dune.common.hashit import hashIt
from dumux.wrapping import cppWrapperCreator, cppWrapperClassAlias
@cppWrapperCreator
def _createVtkOutputModule(*, gridVariables, solutionVector, name):
"""Construct a VtkOutputModule"""
# construct a VtkOutputModule
# the grid geometry is JIT compiled
def VtkOutputModule(*, gridVariables, solutionVector, name):
includes = gridVariables._includes + solutionVector._includes includes = gridVariables._includes + solutionVector._includes
includes += ["dumux/python/io/vtkoutputmodule.hh", "dumux/io/vtkoutputmodule.hh"] includes += ["dumux/python/io/vtkoutputmodule.hh", "dumux/io/vtkoutputmodule.hh"]
typeName = "Dumux::VtkOutputModule<{}, {}>".format( typeName = "Dumux::VtkOutputModule<{}, {}>".format(
...@@ -11,7 +16,10 @@ def VtkOutputModule(*, gridVariables, solutionVector, name): ...@@ -11,7 +16,10 @@ def VtkOutputModule(*, gridVariables, solutionVector, name):
) )
moduleName = "vtkoutputmodule_" + hashIt(typeName) moduleName = "vtkoutputmodule_" + hashIt(typeName)
generator = SimpleGenerator("VtkOutputModule", "Dumux::Python") generator = SimpleGenerator("VtkOutputModule", "Dumux::Python")
module = generator.load( module = generator.load(includes, typeName, moduleName, preamble=gridVariables.model.cppHeader)
includes, typeName, moduleName, preamble=gridVariables._model.getProperties()
)
return module.VtkOutputModule(gridVariables, solutionVector, name) return module.VtkOutputModule(gridVariables, solutionVector, name)
@cppWrapperClassAlias(creator=_createVtkOutputModule)
class VtkOutputModule:
"""Class alias used to instantiate a VtkOutputModule"""
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment