fluxfem 0.1.1a0__py3-none-any.whl

fluxfem/__init__.py

@@ -0,0 +1,343 @@
+from .core import (
+    FESpaceBase,
+    FESpace,
+    FESpacePytree,
+    Expr,
+    FieldRef,
+    ParamRef,
+    trial_ref,
+    test_ref,
+    unknown_ref,
+    Params,
+    LinearForm,
+    BilinearForm,
+    ResidualForm,
+    MixedWeakForm,
+    compile_bilinear,
+    compile_linear,
+    compile_residual,
+    compile_surface_linear,
+    compile_mixed_residual,
+    sdot,
+    dOmega,
+    FormContext,
+    MixedFormContext,
+    FieldPair,
+    ElementVector,
+    vector_load_form,
+    make_space,
+    make_space_pytree,
+    make_hex_basis,
+    make_hex_basis_pytree,
+    make_hex_space,
+    make_hex_space_pytree,
+    make_hex20_basis,
+    make_hex20_basis_pytree,
+    make_hex20_space,
+    make_hex20_space_pytree,
+    make_hex27_basis,
+    make_hex27_basis_pytree,
+    make_hex27_space,
+    make_hex27_space_pytree,
+    make_tet_basis,
+    make_tet_basis_pytree,
+    make_tet_space,
+    make_tet_space_pytree,
+    make_tet10_basis,
+    make_tet10_basis_pytree,
+    make_tet10_space,
+    make_tet10_space_pytree,
+    make_element_residual_kernel,
+    make_element_jacobian_kernel,
+    element_residual,
+    element_jacobian,
+    make_sparsity_pattern,
+    assemble_functional,
+    assemble_mass_matrix,
+    scalar_body_force_form,
+    make_scalar_body_force_form,
+    constant_body_force_form,
+    vector_body_force_form,
+    BaseMeshPytree,
+    bbox_predicate,
+    plane_predicate,
+    axis_plane_predicate,
+    slab_predicate,
+    HexMesh,
+    HexMeshPytree,
+    StructuredHexBox,
+    SurfaceMesh,
+    SurfaceMeshPytree,
+    load_gmsh_mesh,
+    load_gmsh_hex_mesh,
+    load_gmsh_tet_mesh,
+    make_surface_from_facets,
+    TetMesh,
+    TetMeshPytree,
+    StructuredTetBox,
+    StructuredTetTensorBox,
+    tag_axis_minmax_facets,
+    HexTriLinearBasis,
+    HexTriLinearBasisPytree,
+    HexSerendipityBasis20,
+    HexSerendipityBasis20Pytree,
+    HexTriQuadraticBasis27,
+    HexTriQuadraticBasis27Pytree,
+    TetLinearBasis,
+    TetLinearBasisPytree,
+    TetQuadraticBasis10,
+    TetQuadraticBasis10Pytree,
+    MeshData,
+    BasisData,
+    SpaceData,
+    MeshData,
+    BasisData,
+    SpaceData,
+    lame_parameters,
+    isotropic_3d_D,
+    spdirect_solve_cpu,
+    spdirect_solve_gpu,
+    spdirect_solve_jax,
+    coo_to_csr,
+    SparsityPattern,
+    FluxSparseMatrix,
+    LinearSolver,
+    NonlinearSolver,
+    enforce_dirichlet_dense,
+    enforce_dirichlet_sparse,
+    free_dofs,
+    condense_dirichlet_fluxsparse,
+    condense_dirichlet_dense,
+    expand_dirichlet_solution,
+    cg_solve,
+    cg_solve_jax,
+    NonlinearAnalysis,
+    NewtonLoopConfig,
+    LoadStepResult,
+    NewtonSolveRunner,
+    solve_nonlinear,
+    LinearAnalysis,
+    LinearSolveConfig,
+    LinearStepResult,
+    LinearSolveRunner,
+    newton_solve,
+    SurfaceFormField,
+    SurfaceFormContext,
+    vector_surface_load_form,
+    make_vector_surface_load_form,
+    assemble_surface_linear_form,
+)
+from .tools.visualizer import write_vtu, write_displacement_vtu
+from .tools.jit import make_jitted_residual, make_jitted_jacobian
+from .physics import (
+    linear_elasticity_form,
+    vector_body_force_form,
+    constant_body_force_vector_form,
+    diffusion_form,
+    dot,
+    ddot,
+    transpose_last2,
+    sym_grad,
+    sym_grad_u,
+    right_cauchy_green,
+    green_lagrange_strain,
+    deformation_gradient,
+    pk2_neo_hookean,
+    neo_hookean_residual_form,
+    make_elastic_point_data,
+    write_elastic_vtu,
+    make_point_data_displacement,
+    write_point_data_vtu,
+    interpolate_at_points,
+    principal_stresses,
+    principal_sum,
+    max_shear_stress,
+    von_mises_stress,
+)
+
+__all__ = [
+    "FESpaceBase",
+    "FESpace",
+    "FESpacePytree",
+    "Expr",
+    "FieldRef",
+    "ParamRef",
+    "trial_ref",
+    "test_ref",
+    "unknown_ref",
+    "Params",
+    "LinearForm",
+    "BilinearForm",
+    "ResidualForm",
+    "MixedWeakForm",
+    "compile_bilinear",
+    "compile_linear",
+    "compile_residual",
+    "compile_surface_linear",
+    "compile_mixed_residual",
+    "sdot",
+    "dOmega",
+    "FormContext",
+    "MixedFormContext",
+    "FieldPair",
+    "ElementVector",
+    "vector_load_form",
+    "make_space",
+    "make_space_pytree",
+    "make_hex_basis",
+    "make_hex_basis_pytree",
+    "make_hex_space",
+    "make_hex_space_pytree",
+    "make_hex20_basis",
+    "make_hex20_basis_pytree",
+    "make_hex20_space",
+    "make_hex20_space_pytree",
+    "make_hex27_basis",
+    "make_hex27_basis_pytree",
+    "make_hex27_space",
+    "make_hex27_space_pytree",
+    "make_tet_basis",
+    "make_tet_basis_pytree",
+    "make_tet_space",
+    "make_tet_space_pytree",
+    "make_tet10_basis",
+    "make_tet10_basis_pytree",
+    "make_tet10_space",
+    "make_tet10_space_pytree",
+    "MeshData",
+    "BasisData",
+    "SpaceData",
+    "make_element_residual_kernel",
+    "make_element_jacobian_kernel",
+    "element_residual",
+    "element_jacobian",
+    "make_sparsity_pattern",
+    "assemble_functional",
+    "assemble_mass_matrix",
+    "scalar_body_force_form",
+    "make_scalar_body_force_form",
+    "vector_body_force_form",
+    "linear_elasticity_form",
+    "constant_body_force_form",
+    "constant_body_force_vector_form",
+    "diffusion_form",
+    "dot",
+    "ddot",
+    "transpose_last2",
+    "sym_grad",
+    "sym_grad_u",
+    "right_cauchy_green",
+    "green_lagrange_strain",
+    "deformation_gradient",
+    "pk2_neo_hookean",
+    "neo_hookean_residual_form",
+    "StokesSpaces",
+    "make_stokes_spaces",
+    "HexMesh",
+    "HexMeshPytree",
+    "StructuredHexBox",
+    "SurfaceMesh",
+    "SurfaceMeshPytree",
+    "SurfaceFormField",
+    "SurfaceFormContext",
+    "vector_surface_load_form",
+    "make_vector_surface_load_form",
+    "assemble_surface_linear_form",
+    "load_gmsh_mesh",
+    "load_gmsh_hex_mesh",
+    "load_gmsh_tet_mesh",
+    "make_surface_from_facets",
+    "TetMesh",
+    "TetMeshPytree",
+    "StructuredTetBox",
+    "StructuredTetTensorBox",
+    "BaseMeshPytree",
+    "bbox_predicate",
+    "plane_predicate",
+    "axis_plane_predicate",
+    "slab_predicate",
+    "HexTriLinearBasis",
+    "HexTriLinearBasisPytree",
+    "HexSerendipityBasis20",
+    "HexSerendipityBasis20Pytree",
+    "HexTriQuadraticBasis27",
+    "HexTriQuadraticBasis27Pytree",
+    "TetLinearBasis",
+    "TetLinearBasisPytree",
+    "TetQuadraticBasis10",
+    "TetQuadraticBasis10Pytree",
+    "lame_parameters",
+    "isotropic_3d_D",
+    "spdirect_solve_cpu",
+    "spdirect_solve_gpu",
+    "spdirect_solve_jax",
+    "coo_to_csr",
+    "SparsityPattern",
+    "FluxSparseMatrix",
+    "LinearSolver",
+    "NonlinearSolver",
+    "enforce_dirichlet_dense",
+    "enforce_dirichlet_sparse",
+    "free_dofs",
+    "condense_dirichlet_fluxsparse",
+    "condense_dirichlet_dense",
+    "expand_dirichlet_solution",
+    "cg_solve",
+    "NonlinearAnalysis",
+    "NewtonLoopConfig",
+    "LoadStepResult",
+    "NewtonSolveRunner",
+    "solve_nonlinear",
+    "LinearAnalysis",
+    "LinearSolveConfig",
+    "LinearStepResult",
+    "LinearSolveRunner",
+    "newton_solve",
+    "write_vtu",
+    "write_displacement_vtu",
+    "make_jitted_residual",
+    "make_jitted_jacobian",
+    "make_elastic_point_data",
+    "write_elastic_vtu",
+    "make_point_data_displacement",
+    "write_point_data_vtu",
+    "principal_stresses",
+    "principal_sum",
+    "max_shear_stress",
+    "von_mises_stress",
+]
+
+
+try:
+    from importlib.metadata import version, PackageNotFoundError
+except ImportError:
+    # old python
+    from importlib_metadata import version, PackageNotFoundError
+
+
+def read_version_from_pyproject():
+    import pathlib
+    import re
+
+    root = pathlib.Path(__file__).resolve().parent.parent
+    # while root != root.parent and depth < max_depth:
+    pyproject_path = root / "pyproject.toml"
+    if pyproject_path.exists():
+        content = pyproject_path.read_text()
+        match = re.search(r'version\s*=\s*"(.*?)"', content)
+        if match:
+            return match.group(1)
+    # root = root.parent
+    #     depth += 1
+    return "0.0.0"
+
+
+def get_version(package_name):
+    try:
+        # print("version-", version(package_name))
+        return version(package_name)
+    except PackageNotFoundError:
+        return read_version_from_pyproject()
+
+
+__version__ = get_version("fluxfem")

fluxfem/core/__init__.py

@@ -0,0 +1,316 @@
+from .forms import FormContext, MixedFormContext, FieldPair, ElementVector, vector_load_form
+from .space import (
+    FESpaceBase,
+    FESpace,
+    FESpacePytree,
+    make_space,
+    make_space_pytree,
+    make_tet_space,
+    make_tet_space_pytree,
+    make_tet10_space,
+    make_tet10_space_pytree,
+    make_hex_space,
+    make_hex_space_pytree,
+    make_hex20_space,
+    make_hex20_space_pytree,
+    make_hex27_space,
+    make_hex27_space_pytree,
+)
+from .data import MeshData, BasisData, SpaceData
+from .basis import (
+    make_hex_basis,
+    make_hex_basis_pytree,
+    make_hex20_basis,
+    make_hex20_basis_pytree,
+    make_hex27_basis,
+    make_hex27_basis_pytree,
+    make_tet_basis,
+    make_tet_basis_pytree,
+    make_tet10_basis,
+    make_tet10_basis_pytree,
+)
+from .assembly import (
+    assemble_bilinear_form,
+    assemble_linear_form,
+    assemble_functional,
+    assemble_residual,
+    assemble_jacobian,
+    assemble_residual_scatter,
+    assemble_jacobian_scatter,
+    assemble_residual_elementwise_xla,
+    assemble_jacobian_elementwise_xla,
+    make_element_residual_kernel,
+    make_element_jacobian_kernel,
+    element_residual,
+    element_jacobian,
+    make_sparsity_pattern,
+    assemble_jacobian_values,
+    assemble_mass_matrix,
+    scalar_body_force_form,
+    make_scalar_body_force_form,
+    constant_body_force_form,
+)
+from ..physics.elasticity.linear import vector_body_force_form, constant_body_force_vector_form
+from .interp import eval_shape_functions_hex8, interpolate_field_at_element_points
+from .weakform import (
+    Expr,
+    FieldRef,
+    ParamRef,
+    trial_ref,
+    test_ref,
+    unknown_ref,
+    Params,
+    LinearForm,
+    BilinearForm,
+    ResidualForm,
+    compile_surface_linear,
+    MixedWeakForm,
+    compile_bilinear,
+    compile_linear,
+    compile_residual,
+    compile_mixed_residual,
+    grad,
+    sym_grad,
+    dot,
+    sdot,
+    ddot,
+    inner,
+    action,
+    gaction,
+    normal,
+    ds,
+    dOmega,
+    I,
+    det,
+    inv,
+    transpose,
+    log,
+    transpose_last2,
+    einsum,
+)
+from ..mesh import (
+    BaseMeshPytree,
+    bbox_predicate,
+    plane_predicate,
+    axis_plane_predicate,
+    slab_predicate,
+    HexMesh,
+    HexMeshPytree,
+    StructuredHexBox,
+    SurfaceMesh,
+    SurfaceMeshPytree,
+    tag_axis_minmax_facets,
+    TetMesh,
+    TetMeshPytree,
+    StructuredTetBox,
+    StructuredTetTensorBox,
+    load_gmsh_mesh,
+    load_gmsh_hex_mesh,
+    load_gmsh_tet_mesh,
+    make_surface_from_facets,
+)
+from .basis import (
+    HexTriLinearBasis,
+    HexTriLinearBasisPytree,
+    HexSerendipityBasis20,
+    HexSerendipityBasis20Pytree,
+    HexTriQuadraticBasis27,
+    HexTriQuadraticBasis27Pytree,
+    TetLinearBasis,
+    TetLinearBasisPytree,
+    TetQuadraticBasis10,
+    TetQuadraticBasis10Pytree,
+)
+from ..physics import lame_parameters, isotropic_3d_D
+from .solver import spdirect_solve_cpu, spdirect_solve_gpu, spdirect_solve_jax, coo_to_csr
+from ..solver import (
+    SparsityPattern,
+    FluxSparseMatrix,
+    LinearSolver,
+    NonlinearSolver,
+    enforce_dirichlet_dense,
+    enforce_dirichlet_sparse,
+    free_dofs,
+    condense_dirichlet_fluxsparse,
+    condense_dirichlet_dense,
+    expand_dirichlet_solution,
+    cg_solve,
+    cg_solve_jax,
+    NonlinearAnalysis,
+    NewtonLoopConfig,
+    LoadStepResult,
+    NewtonSolveRunner,
+    solve_nonlinear,
+    LinearAnalysis,
+    LinearSolveConfig,
+    LinearStepResult,
+    LinearSolveRunner,
+    newton_solve,
+)
+from ..solver.bc import (
+    SurfaceFormField,
+    SurfaceFormContext,
+    vector_surface_load_form,
+    make_vector_surface_load_form,
+    assemble_surface_linear_form,
+)
+
+__all__ = [
+    "FESpace",
+    "FESpaceBase",
+    "FESpacePytree",
+    "FormContext",
+    "MixedFormContext",
+    "FieldPair",
+    "ElementVector",
+    "vector_load_form",
+    "make_space",
+    "make_space_pytree",
+    "make_hex_basis",
+    "make_hex_basis_pytree",
+    "make_hex_space",
+    "make_hex_space_pytree",
+    "make_hex20_basis",
+    "make_hex20_basis_pytree",
+    "make_hex20_space",
+    "make_hex20_space_pytree",
+    "make_hex27_basis",
+    "make_hex27_basis_pytree",
+    "make_hex27_space",
+    "make_hex27_space_pytree",
+    "make_tet_basis",
+    "make_tet_basis_pytree",
+    "make_tet_space",
+    "make_tet_space_pytree",
+    "make_tet10_basis",
+    "make_tet10_basis_pytree",
+    "make_tet10_space",
+    "make_tet10_space_pytree",
+    "assemble_bilinear_form",
+    "assemble_linear_form",
+    "assemble_functional",
+    "assemble_residual",
+    "assemble_jacobian",
+    "assemble_residual_scatter",
+    "assemble_jacobian_scatter",
+    "assemble_residual_elementwise_xla",
+    "assemble_jacobian_elementwise_xla",
+    "make_element_residual_kernel",
+    "make_element_jacobian_kernel",
+    "element_residual",
+    "element_jacobian",
+    "make_sparsity_pattern",
+    "assemble_jacobian_values",
+    "assemble_residual_elementwise",
+    "assemble_jacobian_elementwise",
+    "assemble_mass_matrix",
+    "scalar_body_force_form",
+    "make_scalar_body_force_form",
+    "constant_body_force_form",
+    "vector_body_force_form",
+    "constant_body_force_vector_form",
+    "eval_shape_functions_hex8",
+    "interpolate_field_at_element_points",
+    "Expr",
+    "FieldRef",
+    "ParamRef",
+    "trial_ref",
+    "test_ref",
+    "unknown_ref",
+    "Params",
+    "LinearForm",
+    "BilinearForm",
+    "ResidualForm",
+    "MixedWeakForm",
+    "compile_bilinear",
+    "compile_linear",
+    "compile_residual",
+    "compile_surface_linear",
+    "compile_mixed_residual",
+    "grad",
+    "sym_grad",
+    "dot",
+    "sdot",
+    "ddot",
+    "inner",
+    "action",
+    "gaction",
+    "normal",
+    "ds",
+    "dOmega",
+    "I",
+    "det",
+    "inv",
+    "transpose",
+    "log",
+    "transpose_last2",
+    "einsum",
+    "HexMesh",
+    "HexMeshPytree",
+    "StructuredHexBox",
+    "SurfaceMesh",
+    "SurfaceMeshPytree",
+    "SurfaceFormField",
+    "SurfaceFormContext",
+    "vector_surface_load_form",
+    "make_vector_surface_load_form",
+    "assemble_surface_linear_form",
+    "load_gmsh_mesh",
+    "load_gmsh_hex_mesh",
+    "load_gmsh_tet_mesh",
+    "make_surface_from_facets",
+    "TetMesh",
+    "TetMeshPytree",
+    "StructuredTetBox",
+    "StructuredTetTensorBox",
+    "tag_axis_minmax_facets",
+    "BaseMeshPytree",
+    "bbox_predicate",
+    "plane_predicate",
+    "axis_plane_predicate",
+    "slab_predicate",
+    "HexTriLinearBasis",
+    "HexTriLinearBasisPytree",
+    "HexSerendipityBasis20",
+    "HexSerendipityBasis20Pytree",
+    "HexTriQuadraticBasis27",
+    "HexTriQuadraticBasis27Pytree",
+    "TetLinearBasis",
+    "TetLinearBasisPytree",
+    "TetQuadraticBasis10",
+    "TetQuadraticBasis10Pytree",
+    "lame_parameters",
+    "isotropic_3d_D",
+    "spdirect_solve_cpu",
+    "spdirect_solve_gpu",
+    "spdirect_solve_jax",
+    "coo_to_csr",
+    "SparsityPattern",
+    "FluxSparseMatrix",
+    "LinearSolver",
+    "NonlinearSolver",
+    "enforce_dirichlet_dense",
+    "enforce_dirichlet_sparse",
+    "free_dofs",
+    "condense_dirichlet_fluxsparse",
+    "condense_dirichlet_dense",
+    "expand_dirichlet_solution",
+    "cg_solve",
+    "cg_solve_jax",
+    "NonlinearAnalysis",
+    "NewtonLoopConfig",
+    "LoadStepResult",
+    "NewtonSolveRunner",
+    "solve_nonlinear",
+    "LinearAnalysis",
+    "LinearSolveConfig",
+    "LinearStepResult",
+    "LinearSolveRunner",
+    "newton_solve",
+    "jacobian_sparse",
+    "eval_shape_functions_hex8",
+    "interpolate_field_at_element_points",
+    "MeshData",
+    "BasisData",
+    "SpaceData",
+]