NvidiaWarp-GarmentCode/warp/fem/__init__.py

from .geometry import Geometry, Grid2D, Trimesh2D, Quadmesh2D, Grid3D, Tetmesh, Hexmesh
from .geometry import GeometryPartition, LinearGeometryPartition, ExplicitGeometryPartition

from .space import FunctionSpace, make_polynomial_space, ElementBasis
from .space import BasisSpace, PointBasisSpace, make_polynomial_basis_space, make_collocated_function_space
from .space import DofMapper, SkewSymmetricTensorMapper, SymmetricTensorMapper
from .space import SpaceTopology, SpacePartition, SpaceRestriction, make_space_partition, make_space_restriction

from .domain import GeometryDomain, Cells, Sides, BoundarySides, FrontierSides
from .quadrature import Quadrature, RegularQuadrature, NodalQuadrature, ExplicitQuadrature, PicQuadrature
from .polynomial import Polynomial

from .field import FieldLike, DiscreteField, make_test, make_trial, make_restriction

from .integrate import integrate, interpolate

from .operator import integrand
from .operator import position, normal, lookup, measure, measure_ratio, deformation_gradient
from .operator import inner, grad, div, outer, grad_outer, div_outer
from .operator import degree, at_node
from .operator import D, curl, jump, average, grad_jump, grad_average

from .types import Sample, Field, Domain, Coords, ElementIndex

from .dirichlet import project_linear_system, normalize_dirichlet_projector

from .cache import TemporaryStore, set_default_temporary_store, borrow_temporary, borrow_temporary_like