diff-vbd/src/diff_vbd/__init__.py at main · vindou/diff-vbd · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
"""Library-first API for the differentiable VBD solver."""

from diff_vbd.config import load_config, load_problem_from_yaml
from diff_vbd.export import (
    build_export_metadata,
    export_simulation_npz,
    extract_surface_mesh,
    surface_trajectory_from_history,
)
from diff_vbd.model import (
    AccelerationOptions,
    BoundaryConditions,
    DirichletSpec,
    LineSearchOptions,
    MaterialParams,
    MeshData,
    RigidRegionSpec,
    SimulationProblem,
    SimulationState,
    SolverOptions,
    TopologyData,
)
from diff_vbd.problems import build_cantilever_problem
from diff_vbd.setup.problem_builder import assemble_problem, initial_state
from diff_vbd.setup.boundary_conditions import (
    assemble_boundary_conditions,
    assemble_dirichlet_boundary_conditions,
    evaluate_dirichlet_targets,
)
from diff_vbd.solver import (
    elastic_potential,
    inertia_potential,
    potential_energy,
    simulate,
    step,
    variational_energy,
)

__all__ = [
    "BoundaryConditions",
    "AccelerationOptions",
    "DirichletSpec",
    "LineSearchOptions",
    "MaterialParams",
    "MeshData",
    "RigidRegionSpec",
    "SimulationProblem",
    "SimulationState",
    "SolverOptions",
    "TopologyData",
    "assemble_problem",
    "assemble_boundary_conditions",
    "assemble_dirichlet_boundary_conditions",
    "build_export_metadata",
    "build_cantilever_problem",
    "elastic_potential",
    "evaluate_dirichlet_targets",
    "export_simulation_npz",
    "extract_surface_mesh",
    "inertia_potential",
    "initial_state",
    "load_config",
    "load_problem_from_yaml",
    "potential_energy",
    "simulate",
    "step",
    "surface_trajectory_from_history",
    "variational_energy",
]