SYMBOL INDEX (827 symbols across 133 files) FILE: examples/advanced_flows/PartiallySaturatedObstacle.py class ObstaclePartially (line 22) | class ObstaclePartially(lt.Obstacle): method __init__ (line 23) | def __init__(self, resolution, domain_length_x): method post_boundaries (line 29) | def post_boundaries(self): method initial_solution (line 48) | def initial_solution(self, x): class Show2D (line 60) | class Show2D: method __init__ (line 61) | def __init__(self, mask, outdir: str, **kwargs): method __call__ (line 75) | def __call__(self, data, title: str, name: str, vlim=None): FILE: examples/advanced_projects/efficient_bounce_back_obstacle/auxiliary_code/data_processing_and_plotting.py function plot_force_coefficient (line 23) | def plot_force_coefficient(data_array: np.ndarray, ylabel: str, ylim: tu... function analyze_periodic_timeseries (line 105) | def analyze_periodic_timeseries(data_array: np.ndarray, periodic_start_r... function draw_circular_mask (line 262) | def draw_circular_mask(flow: 'Flow', gridpoints_per_diameter: int, outpu... class ProfilePlotter (line 430) | class ProfilePlotter: method __init__ (line 439) | def __init__(self, flow: 'Flow', output_path: str, reference_data_path... method import_profile_reference_data (line 453) | def import_profile_reference_data(self, data_path: str): method process_data (line 579) | def process_data(self, save: bool = False): method save_timeseries_to_files (line 666) | def save_timeseries_to_files(self, basepath: str): method plot_velocity_profiles (line 679) | def plot_velocity_profiles(self, show_reference: bool = False, method plot_reynolds_stress_profiles (line 802) | def plot_reynolds_stress_profiles(self, show_reference: bool = False, FILE: examples/advanced_projects/efficient_bounce_back_obstacle/auxiliary_code/helperCode.py class Logger (line 5) | class Logger(object): method __init__ (line 7) | def __init__(self, outdir): method write (line 11) | def write(self, message): method flush (line 17) | def flush(self): FILE: examples/advanced_projects/efficient_bounce_back_obstacle/boundary/fullway_bounce_back_boundary.py class FullwayBounceBackBoundary (line 9) | class FullwayBounceBackBoundary(Boundary): method __init__ (line 20) | def __init__(self, context: 'Context', flow: 'Flow', mask: np.ndarray ... method __call__ (line 135) | def __call__(self, flow: 'Flow'): method calc_force_on_boundary (line 161) | def calc_force_on_boundary(self, f: torch.Tensor): method make_no_collision_mask (line 174) | def make_no_collision_mask(self, f_shape: List[int], context: 'Context... method make_no_streaming_mask (line 177) | def make_no_streaming_mask(self, shape: List[int], context: 'Context')... method native_available (line 181) | def native_available(self) -> bool: method native_generator (line 184) | def native_generator(self, index: int): FILE: examples/advanced_projects/efficient_bounce_back_obstacle/boundary/halfway_bounce_back_boundary.py class HalfwayBounceBackBoundary (line 10) | class HalfwayBounceBackBoundary(Boundary): method __init__ (line 19) | def __init__(self, context: 'Context', flow: 'Flow', solid_boundary_da... method __call__ (line 170) | def __call__(self, flow: 'Flow'): method make_no_collision_mask (line 192) | def make_no_collision_mask(self, f_shape: List[int], context: 'Context' method make_no_streaming_mask (line 203) | def make_no_streaming_mask(self, f_shape: List[int], context: 'Context' method calc_force_on_boundary (line 211) | def calc_force_on_boundary(self): method store_f_collided (line 220) | def store_f_collided(self, f_collided: torch.Tensor): method initialize_f_collided (line 244) | def initialize_f_collided(self): method native_available (line 250) | def native_available(self) -> bool: method native_generator (line 253) | def native_generator(self, index: int): FILE: examples/advanced_projects/efficient_bounce_back_obstacle/boundary/linear_interpolated_bounce_back_boundary.py class LinearInterpolatedBounceBackBoundary (line 9) | class LinearInterpolatedBounceBackBoundary(Boundary): method __init__ (line 20) | def __init__(self, context: 'Context', flow: 'Flow', solid_boundary_da... method __call__ (line 56) | def __call__(self, flow: 'Flow'): method make_no_streaming_mask (line 104) | def make_no_streaming_mask(self, f_shape: List[int], context: Context): method make_no_collision_mask (line 111) | def make_no_collision_mask(self, f_shape: List[int], context: Context): method calc_force_on_boundary (line 118) | def calc_force_on_boundary(self, f_bounced: torch.Tensor): method store_f_collided (line 151) | def store_f_collided(self, f_collided: torch.Tensor): method initialize_f_collided (line 193) | def initialize_f_collided(self): method native_available (line 207) | def native_available(self) -> bool: method native_generator (line 210) | def native_generator(self, index: int): FILE: examples/advanced_projects/efficient_bounce_back_obstacle/boundary/solid_boundary_data.py class SolidBoundaryData (line 8) | class SolidBoundaryData(dict): FILE: examples/advanced_projects/efficient_bounce_back_obstacle/flow/obstacle_cylinder.py class ObstacleCylinder (line 15) | class ObstacleCylinder(ExtFlow): method __init__ (line 38) | def __init__(self, context: Context, resolution: Union[int, List[int]], method make_units (line 185) | def make_units(self, reynolds_number: float, mach_number: float, resol... method make_resolution (line 194) | def make_resolution(self, resolution: Union[int, List[int]], method obstacle_mask (line 202) | def obstacle_mask(self): method obstacle_mask (line 206) | def obstacle_mask(self, m: np.ndarray | torch.Tensor): method initial_pu (line 210) | def initial_pu(self): method make_solid_boundary_data (line 287) | def make_solid_boundary_data(self, x_center: float, y_center: float, r... method make_ibb_index_lists (line 293) | def make_ibb_index_lists(self, x_center: float, y_center: float, radiu... method grid (line 489) | def grid(self): method post_boundaries (line 495) | def post_boundaries(self): method post_streaming_boundaries (line 515) | def post_streaming_boundaries(self): method _unit_vector (line 590) | def _unit_vector(self, i=0): FILE: examples/advanced_projects/efficient_bounce_back_obstacle/reporter/observables_force_coefficients.py class DragCoefficient (line 15) | class DragCoefficient(Observable): method __init__ (line 22) | def __init__(self, flow: 'Flow', obstacle_boundary: method __call__ (line 37) | def __call__(self, f: torch.Tensor = None): class LiftCoefficient (line 51) | class LiftCoefficient(Observable): method __init__ (line 58) | def __init__(self, flow: 'Flow', obstacle_boundary: method __call__ (line 73) | def __call__(self, f: torch.Tensor = None): FILE: examples/advanced_projects/efficient_bounce_back_obstacle/reporter/reporter_ProfileReporter.py class ProfileReporter (line 10) | class ProfileReporter(Reporter): method __init__ (line 16) | def __init__(self, interval: int, flow: 'Flow', position_lu: float, i_... method __call__ (line 39) | def __call__(self, simulation: 'Simulation'): FILE: examples/advanced_projects/efficient_bounce_back_obstacle/reporter/reporter_advanced_vtk_reporter.py function write_vtk (line 15) | def write_vtk(point_dict, id=0, filename_base="./data/output", class VTKReporterAdvanced (line 26) | class VTKReporterAdvanced(Reporter): method __init__ (line 33) | def __init__(self, flow: 'Flow', interval: int = 50, filename_base: st... method __call__ (line 59) | def __call__(self, simulation: Simulation): method output_mask (line 100) | def output_mask(self, mask: np.ndarray | torch.Tensor, outdir: Optiona... class VTKsliceReporter (line 145) | class VTKsliceReporter(Reporter): method __init__ (line 149) | def __init__(self, flow: 'Flow', interval: int = 50, filename_base: st... method __call__ (line 195) | def __call__(self, simulation: Simulation): method output_mask (line 242) | def output_mask(self, mask: np.ndarray | torch.Tensor, outdir: Optiona... FILE: examples/advanced_projects/efficient_bounce_back_obstacle/simulation/ebb_simulation.py class EbbSimulation (line 11) | class EbbSimulation(Simulation): method __init__ (line 17) | def __init__(self, flow: 'Flow', collision: 'Collision', method _post_streaming_boundaries (line 81) | def _post_streaming_boundaries(self): method _pass_f_collided (line 87) | def _pass_f_collided(self): method __call__ (line 93) | def __call__(self, num_steps: int) -> float: FILE: examples/development/manually_generate_cuda_native.py function ensure_empty_dir (line 8) | def ensure_empty_dir(path): function main (line 26) | def main(verbose=False, install=False): FILE: lettuce/_context.py class Context (line 9) | class Context: method __init__ (line 10) | def __init__(self, device: Optional[torch.device | str] = None, method synchronize (line 64) | def synchronize(self): method empty_tensor (line 72) | def empty_tensor(self, size: Union[List[int], torch.Size], *args, method zero_tensor (line 77) | def zero_tensor(self, size: Union[List[int], torch.Size], *args, method full_tensor (line 82) | def full_tensor(self, size: Union[List[int], torch.Size], value, *args, method one_tensor (line 87) | def one_tensor(self, size: Union[List[int], torch.Size], *args, dtype=... method convert_to_tensor (line 92) | def convert_to_tensor(self, array, *args, method convert_to_ndarray (line 118) | def convert_to_ndarray(tensor: Union[torch.Tensor, List]) -> np.ndarray: FILE: lettuce/_flow.py class Equilibrium (line 17) | class Equilibrium(ABC): method __call__ (line 19) | def __call__(self, flow: 'Flow', rho=None, u=None) -> torch.Tensor: method native_available (line 23) | def native_available(self) -> bool: method native_generator (line 27) | def native_generator(self) -> 'NativeEquilibrium': class Boundary (line 31) | class Boundary(ABC): method __call__ (line 33) | def __call__(self, flow: 'Flow'): method make_no_collision_mask (line 37) | def make_no_collision_mask(self, shape: List[int], context: 'Context' method make_no_streaming_mask (line 42) | def make_no_streaming_mask(self, shape: List[int], context: 'Context' method native_available (line 47) | def native_available(self) -> bool: method native_generator (line 51) | def native_generator(self, index: int) -> 'NativeBoundary': class Flow (line 55) | class Flow(ABC): method __init__ (line 81) | def __init__(self, context: 'Context', resolution: List[int], method pre_boundaries (line 99) | def pre_boundaries(self) -> List['Boundary']: method post_boundaries (line 105) | def post_boundaries(self) -> List['Boundary']: method boundaries (line 118) | def boundaries(self) -> List['Boundary']: method initial_pu (line 123) | def initial_pu(self) -> (float, Union[np.array, torch.Tensor]): method initialize (line 127) | def initialize(self): method f_next (line 146) | def f_next(self) -> torch.Tensor: method f_next (line 154) | def f_next(self, f_next_: torch.Tensor): method rho (line 157) | def rho(self, f: Optional[torch.Tensor] = None) -> torch.Tensor: method rho_pu (line 162) | def rho_pu(self) -> torch.Tensor: method p_pu (line 166) | def p_pu(self) -> torch.Tensor: method u_pu (line 170) | def u_pu(self): method j (line 173) | def j(self, f: Optional[torch.Tensor] = None) -> torch.Tensor: method u (line 178) | def u(self, f: Optional[torch.Tensor] = None, rho=None, acceleration=None method velocity (line 196) | def velocity(self, f: Optional[torch.Tensor] = None): method incompressible_energy (line 200) | def incompressible_energy(self, f: Optional[torch.Tensor] = None method entropy (line 206) | def entropy(self, f: Optional[torch.Tensor] = None) -> torch.Tensor: method pseudo_entropy_global (line 213) | def pseudo_entropy_global(self, f: Optional[torch.Tensor] = None method pseudo_entropy_local (line 220) | def pseudo_entropy_local(self, f: Optional[torch.Tensor] = None method shear_tensor (line 228) | def shear_tensor(self, f: Optional[torch.Tensor] = None) -> torch.Tensor: method einsum (line 236) | def einsum(self, equation, fields, *args) -> torch.Tensor: method dump (line 258) | def dump(self, filename): method load (line 262) | def load(self, filename): function pressure_poisson (line 271) | def pressure_poisson(units: 'UnitConversion', u, rho0, tol_abs=1e-10, function initialize_pressure_poisson (line 323) | def initialize_pressure_poisson(flow: 'Flow', function initialize_f_neq (line 341) | def initialize_f_neq(flow: 'Flow'): FILE: lettuce/_simulation.py class Collision (line 17) | class Collision(ABC): method __call__ (line 19) | def __call__(self, flow: 'Flow'): method native_available (line 23) | def native_available(self) -> bool: method native_generator (line 27) | def native_generator(self, index: int) -> 'NativeCollision': class Reporter (line 31) | class Reporter(ABC): method __init__ (line 34) | def __init__(self, interval: int): method __call__ (line 38) | def __call__(self, simulation: 'Simulation'): class Simulation (line 42) | class Simulation: method __init__ (line 53) | def __init__(self, flow: 'Flow', collision: 'Collision', method step (line 231) | def step(self, num_steps: int): method units (line 238) | def units(self): method __stream (line 242) | def __stream(f, i, e, d): method _stream (line 245) | def _stream(self): method _collide (line 258) | def _collide(self): method _report (line 307) | def _report(self): method __call__ (line 311) | def __call__(self, num_steps): class BreakableSimulation (line 325) | class BreakableSimulation(Simulation): method __init__ (line 326) | def __init__(self, flow: 'Flow', collision: 'Collision', method __call__ (line 330) | def __call__(self, num_steps: int) -> float: FILE: lettuce/_stencil.py class Stencil (line 14) | class Stencil(ABC): method d (line 22) | def d(self) -> int: method q (line 27) | def q(self) -> int: class TorchStencil (line 31) | class TorchStencil: method __init__ (line 37) | def __init__(self, stencil: 'Stencil', context: 'Context'): method d (line 44) | def d(self) -> Literal[1, 2, 3]: method q (line 49) | def q(self) -> int: FILE: lettuce/_unit.py class UnitConversion (line 13) | class UnitConversion: method __init__ (line 21) | def __init__(self, reynolds_number, mach_number=0.05, method characteristic_velocity_lu (line 35) | def characteristic_velocity_lu(self): method characteristic_pressure_pu (line 39) | def characteristic_pressure_pu(self): method characteristic_pressure_lu (line 44) | def characteristic_pressure_lu(self): method viscosity_lu (line 49) | def viscosity_lu(self): method viscosity_pu (line 54) | def viscosity_pu(self): method relaxation_parameter_lu (line 59) | def relaxation_parameter_lu(self): method convert_velocity_to_pu (line 62) | def convert_velocity_to_pu(self, velocity_in_lu): method convert_velocity_to_lu (line 66) | def convert_velocity_to_lu(self, velocity_in_pu): method convert_acceleration_to_pu (line 70) | def convert_acceleration_to_pu(self, acceleration_in_lu): method convert_acceleration_to_lu (line 77) | def convert_acceleration_to_lu(self, acceleration_in_pu): method convert_time_to_pu (line 84) | def convert_time_to_pu(self, time_in_lu): method convert_time_to_lu (line 89) | def convert_time_to_lu(self, time_in_pu): method convert_density_lu_to_pressure_pu (line 94) | def convert_density_lu_to_pressure_pu(self, density_lu): method convert_pressure_pu_to_density_lu (line 99) | def convert_pressure_pu_to_density_lu(self, pressure_pu): method convert_density_to_pu (line 103) | def convert_density_to_pu(self, density_lu): method convert_density_to_lu (line 107) | def convert_density_to_lu(self, density_pu): method convert_pressure_to_pu (line 111) | def convert_pressure_to_pu(self, pressure_lu): method convert_pressure_to_lu (line 115) | def convert_pressure_to_lu(self, pressure_pu): method convert_length_to_pu (line 119) | def convert_length_to_pu(self, length_lu): method convert_length_to_lu (line 123) | def convert_length_to_lu(self, length_pu): method convert_energy_to_pu (line 127) | def convert_energy_to_pu(self, energy_lu): method convert_energy_to_lu (line 132) | def convert_energy_to_lu(self, energy_pu): method convert_incompressible_energy_to_pu (line 137) | def convert_incompressible_energy_to_pu(self, energy_lu): method convert_incompressible_energy_to_lu (line 142) | def convert_incompressible_energy_to_lu(self, energy_pu): FILE: lettuce/_version.py function get_keywords (line 20) | def get_keywords(): class VersioneerConfig (line 33) | class VersioneerConfig: function get_config (line 37) | def get_config(): class NotThisMethod (line 51) | class NotThisMethod(Exception): function register_vcs_handler (line 59) | def register_vcs_handler(vcs, method): # decorator function run_command (line 72) | def run_command(commands, args, cwd=None, verbose=False, hide_stderr=False, function versions_from_parentdir (line 107) | def versions_from_parentdir(parentdir_prefix, root, verbose): function git_get_keywords (line 132) | def git_get_keywords(versionfile_abs): function git_versions_from_keywords (line 160) | def git_versions_from_keywords(keywords, tag_prefix, verbose): function git_pieces_from_vcs (line 224) | def git_pieces_from_vcs(tag_prefix, root, verbose, runner=run_command): function plus_or_dot (line 352) | def plus_or_dot(pieces): function render_pep440 (line 359) | def render_pep440(pieces): function render_pep440_branch (line 384) | def render_pep440_branch(pieces): function pep440_split_post (line 414) | def pep440_split_post(ver): function render_pep440_pre (line 424) | def render_pep440_pre(pieces): function render_pep440_post (line 448) | def render_pep440_post(pieces): function render_pep440_post_branch (line 475) | def render_pep440_post_branch(pieces): function render_pep440_old (line 504) | def render_pep440_old(pieces): function render_git_describe (line 526) | def render_git_describe(pieces): function render_git_describe_long (line 546) | def render_git_describe_long(pieces): function render (line 566) | def render(pieces, style): function get_versions (line 602) | def get_versions(): FILE: lettuce/base.py class LatticeBase (line 4) | class LatticeBase: method __init__ (line 18) | def __init__(self, lattice): method native_available (line 30) | def native_available(self) -> bool: method create_native (line 40) | def create_native(self) -> Optional['NativeLatticeBase']: FILE: lettuce/cli.py function main (line 39) | def main(ctx, cuda, gpu_id, precision): function benchmark (line 87) | def benchmark(ctx, steps, resolution, profile_out, flow, vtk_out, function convergence (line 139) | def convergence(ctx, use_cuda_native): FILE: lettuce/cuda_native/_default_code_gen.py class StreamingStrategy (line 13) | class StreamingStrategy(Enum): method pre_streaming (line 20) | def pre_streaming(self) -> bool: method post_streaming (line 24) | def post_streaming(self) -> bool: class DefaultCodeGeneration (line 28) | class DefaultCodeGeneration(Registry): method __init__ (line 38) | def __init__(self, stencil: 'Stencil', method d (line 60) | def d(self) -> str: method q (line 64) | def q(self) -> str: method e (line 68) | def e(self, q: int, d: int) -> str: method w (line 74) | def w(self, q: int) -> str: method opposite (line 79) | def opposite(self, q: int) -> str: method support_no_collision_mask (line 84) | def support_no_collision_mask(self) -> str: method support_no_streaming_mask (line 88) | def support_no_streaming_mask(self) -> str: method thread_count (line 92) | def thread_count(self, d: Optional[int] = None) -> str: method cuda_block_count (line 102) | def cuda_block_count(self, d: Optional[int] = None) -> str: method kernel_block_count (line 111) | def kernel_block_count(self, d: int) -> str: method cuda_size (line 117) | def cuda_size(self, d: int) -> str: method kernel_size (line 124) | def kernel_size(self, d: int) -> str: method kernel_index (line 130) | def kernel_index(self, d: int) -> str: method cuda_stride (line 138) | def cuda_stride(self, d: int) -> str: method kernel_stride (line 143) | def kernel_stride(self, d: int) -> str: method kernel_offset (line 147) | def kernel_offset(self, d: int, v: int = 0) -> str: method kernel_base_index (line 170) | def kernel_base_index(self) -> str: method assert_ncm (line 174) | def assert_ncm(self): method assert_nsm (line 179) | def assert_nsm(self): method cuda_no_collision_mask (line 185) | def cuda_no_collision_mask(self) -> str: method cuda_ncm (line 189) | def cuda_ncm(self) -> str: method kernel_no_collision_mask (line 193) | def kernel_no_collision_mask(self) -> str: method kernel_ncm (line 197) | def kernel_ncm(self) -> str: method cuda_no_streaming_mask (line 201) | def cuda_no_streaming_mask(self) -> str: method cuda_nsm (line 205) | def cuda_nsm(self) -> str: method kernel_no_streaming_mask (line 209) | def kernel_no_streaming_mask(self) -> str: method kernel_nsm (line 214) | def kernel_nsm(self) -> str: method cuda_f (line 218) | def cuda_f(self) -> str: method kernel_f (line 222) | def kernel_f(self) -> str: method kernel_stream_offset (line 227) | def kernel_stream_offset(self, q: int) -> str: method f_reg (line 233) | def f_reg(self, q: int) -> str: method kernel_rho (line 257) | def kernel_rho(self) -> str: method kernel_rho_inv (line 261) | def kernel_rho_inv(self) -> str: method kernel_u (line 265) | def kernel_u(self, d: int) -> str: method cuda_f_next (line 277) | def cuda_f_next(self) -> str: method kernel_f_next (line 281) | def kernel_f_next(self) -> str: method generate (line 285) | def generate(self): FILE: lettuce/cuda_native/_generator.py class Generator (line 18) | class Generator(DefaultCodeGeneration): method __init__ (line 21) | def __init__(self, method version (line 33) | def version(self): method name (line 37) | def name(self): method generate (line 46) | def generate(self) -> Dict[str, str]: method format (line 54) | def format(self, generate_dir: Optional[str] = None) -> str: method _resolve (line 86) | def _resolve(self): method resolve (line 99) | def resolve(self, install: bool = True): method install (line 112) | def install(directory: str): FILE: lettuce/cuda_native/_registry.py class Parameter (line 9) | class Parameter: method __str__ (line 13) | def __str__(self): class RegistryList (line 17) | class RegistryList(list): method __init__ (line 21) | def __init__(self, dimensions: int): method registered (line 26) | def registered(self, guard: Optional[Hashable] = None, cond: bool = Tr... method register (line 37) | def register(self, guard: Optional[Hashable] = None, cond: bool = True... method append (line 49) | def append(self, item: Any, guard: Optional[Hashable] = None, cond: bo... method extend (line 53) | def extend(self, other, guard: Optional[Hashable] = None, cond: bool =... method __str__ (line 57) | def __str__(self): class CodeRegistryList (line 61) | class CodeRegistryList(RegistryList): method mutable (line 62) | def mutable(self, variable_type: str, variable: str, value: str) -> str: method variable (line 66) | def variable(self, variable_type: str, variable: str, value: str) -> str: method __str__ (line 70) | def __str__(self): class ParameterRegistryList (line 74) | class ParameterRegistryList(RegistryList): method __str__ (line 75) | def __str__(self): class Registry (line 79) | class Registry: method __init__ (line 82) | def __init__(self, d: int): method cuda_hook (line 95) | def cuda_hook(self, py_value: str, cuda_param: Parameter, cond: bool =... method kernel_hook (line 100) | def kernel_hook(self, cuda_value: str, kernel_param: Parameter, cond: ... method python_pre (line 106) | def python_pre(self) -> CodeRegistryList: method python_post (line 110) | def python_post(self) -> CodeRegistryList: method cuda (line 114) | def cuda(self) -> CodeRegistryList: method pipes (line 118) | def pipes(self) -> CodeRegistryList: method pipe (line 122) | def pipe(self) -> CodeRegistryList: method joined_buffer (line 125) | def joined_buffer(self) -> Dict[str, str]: FILE: lettuce/cuda_native/_transformer.py class NativeTransformer (line 14) | class NativeTransformer(ABC): method __init__ (line 17) | def __init__(self, index: int): method create (line 22) | def create(index: int): method generate (line 26) | def generate(self, reg: 'Registry'): class NativeBoundary (line 30) | class NativeBoundary(NativeTransformer, ABC): class NativeEquilibrium (line 34) | class NativeEquilibrium(ABC): method f_eq (line 36) | def f_eq(self, reg: 'Registry', q: int, rho: Optional[str] = None, u: ... class NativeCollision (line 40) | class NativeCollision(NativeTransformer, ABC): FILE: lettuce/cuda_native/_util.py function lettuce_hash (line 6) | def lettuce_hash(value: Union[str, List[str]]) -> str: FILE: lettuce/cuda_native/ext/_boundary/bounce_back_boundary.py class NativeBounceBackBoundary (line 6) | class NativeBounceBackBoundary(NativeBoundary): method __init__ (line 8) | def __init__(self, index): method create (line 12) | def create(index): method generate (line 15) | def generate(self, reg: 'DefaultCodeGeneration'): FILE: lettuce/cuda_native/ext/_boundary/equilibrium_pu.py class NativeEquilibriumBoundaryPu (line 6) | class NativeEquilibriumBoundaryPu(NativeBoundary): method __init__ (line 8) | def __init__(self, index): method create (line 12) | def create(index): method cuda_velocity (line 15) | def cuda_velocity(self, reg: 'DefaultCodeGeneration'): method cuda_velocity_size (line 20) | def cuda_velocity_size(self, reg: 'DefaultCodeGeneration', d: int): method kernel_velocity_size (line 26) | def kernel_velocity_size(self, reg: 'DefaultCodeGeneration', d: int): method kernel_velocity (line 31) | def kernel_velocity(self, reg: 'DefaultCodeGeneration', d: int): method cuda_density (line 55) | def cuda_density(self, reg: 'DefaultCodeGeneration'): method cuda_density_size (line 60) | def cuda_density_size(self, reg: 'DefaultCodeGeneration', d: int): method kernel_density_size (line 66) | def kernel_density_size(self, reg: 'DefaultCodeGeneration', d: int): method kernel_density (line 71) | def kernel_density(self, reg: 'DefaultCodeGeneration'): method generate (line 89) | def generate(self, reg: 'DefaultCodeGeneration'): FILE: lettuce/cuda_native/ext/_boundary/no_boundary.py class NativeNoBoundary (line 6) | class NativeNoBoundary(NativeBoundary): method __init__ (line 7) | def __init__(self, index): method create (line 11) | def create(index): method generate (line 14) | def generate(self, generator: 'Generator'): FILE: lettuce/cuda_native/ext/_collision/bgk_collision.py class NativeBGKCollision (line 9) | class NativeBGKCollision(NativeCollision): method __init__ (line 11) | def __init__(self, index: int, force: Optional['NativeForce'] = None): method create (line 16) | def create(index: int, force: Optional['NativeForce'] = None): method cuda_tau_inv (line 21) | def cuda_tau_inv(self, reg: 'DefaultCodeGeneration'): method kernel_tau_inv (line 25) | def kernel_tau_inv(self, reg: 'DefaultCodeGeneration'): method generate (line 29) | def generate(self, reg: 'DefaultCodeGeneration'): FILE: lettuce/cuda_native/ext/_collision/no_collision.py class NativeNoCollision (line 8) | class NativeNoCollision(NativeCollision): method create (line 11) | def create(index: int, force: None = None): method generate (line 15) | def generate(self, _): FILE: lettuce/cuda_native/ext/_equilibrium/quadratic_equilibrium.py class NativeQuadraticEquilibrium (line 8) | class NativeQuadraticEquilibrium(NativeEquilibrium): method uxu (line 11) | def uxu(self, reg: 'DefaultCodeGeneration', u: [str] = None) -> str: method exu (line 19) | def exu(self, reg: 'DefaultCodeGeneration', q: int, u: [str] = None) -... method cs_pow_two (line 27) | def cs_pow_two(self, _) -> str: method two_cs_pow_two (line 31) | def two_cs_pow_two(self, _) -> str: method f_eq (line 34) | def f_eq(self, reg: 'DefaultCodeGeneration', q: int, rho: str = None, ... FILE: lettuce/cuda_native/ext/_force/_force.py class NativeForce (line 1) | class NativeForce: FILE: lettuce/ext/_boundary/anti_bounce_back_outlet.py class AntiBounceBackOutlet (line 13) | class AntiBounceBackOutlet(Boundary): method __init__ (line 22) | def __init__(self, direction: [List[int]], flow: 'Flow', method __call__ (line 71) | def __call__(self, flow: 'Flow'): method make_no_streaming_mask (line 93) | def make_no_streaming_mask(self, f_shape, context: 'Context'): method make_no_collision_mask (line 100) | def make_no_collision_mask(self, shape: List[int], context: 'Context'): method native_available (line 105) | def native_available(self) -> bool: method native_generator (line 108) | def native_generator(self, index: int) -> 'NativeBoundary': FILE: lettuce/ext/_boundary/bounce_back_boundary.py class BounceBackBoundary (line 10) | class BounceBackBoundary(Boundary): method __init__ (line 13) | def __init__(self, mask: torch.Tensor): method __call__ (line 17) | def __call__(self, flow: 'Flow'): method make_no_streaming_mask (line 20) | def make_no_streaming_mask(self, shape: List[int], context: 'Context' method make_no_collision_mask (line 24) | def make_no_collision_mask(self, shape: List[int], context: 'Context' method native_available (line 28) | def native_available(self) -> bool: method native_generator (line 31) | def native_generator(self, index: int) -> 'NativeBoundary': FILE: lettuce/ext/_boundary/equilibrium_boundary_pu.py class EquilibriumBoundaryPU (line 15) | class EquilibriumBoundaryPU(Boundary): method checked_tensor (line 24) | def checked_tensor(t: Union[torch.Tensor, np.ndarray, list, tuple, num... method __init__ (line 71) | def __init__(self, context: 'Context', flow: 'Flow', mask, velocity, p... method __call__ (line 79) | def __call__(self, flow: 'Flow'): method make_no_collision_mask (line 86) | def make_no_collision_mask(self, shape: List[int], context: 'Context' method make_no_streaming_mask (line 90) | def make_no_streaming_mask(self, shape: List[int], context: 'Context' method native_available (line 94) | def native_available(self) -> bool: method native_generator (line 97) | def native_generator(self, index: int) -> 'NativeBoundary': FILE: lettuce/ext/_boundary/equilibrium_outlet_p.py class EquilibriumOutletP (line 12) | class EquilibriumOutletP(AntiBounceBackOutlet): method __init__ (line 16) | def __init__(self, direction: List[int], flow: 'Flow', method __call__ (line 63) | def __call__(self, flow: 'Flow'): method make_no_streaming_mask (line 75) | def make_no_streaming_mask(self, shape: List[int], context: 'Context' method make_no_collision_mask (line 82) | def make_no_collision_mask(self, shape: List[int], context: 'Context'): method native_available (line 87) | def native_available(self) -> bool: method native_generator (line 90) | def native_generator(self, index: int): FILE: lettuce/ext/_boundary/partially_saturated_boundary.py class PartiallySaturatedBC (line 6) | class PartiallySaturatedBC(Boundary): method __init__ (line 15) | def __init__(self, mask: torch.Tensor, tau: float, saturation: float): method __call__ (line 22) | def __call__(self, flow: 'Flow'): method make_no_collision_mask (line 39) | def make_no_collision_mask(self, shape: List[int], context: 'Context' method make_no_streaming_mask (line 43) | def make_no_streaming_mask(self, shape: List[int], context: 'Context' method native_available (line 47) | def native_available(self) -> bool: method native_generator (line 50) | def native_generator(self, index: int) -> 'NativeBoundary': FILE: lettuce/ext/_collision/bgk_collision.py class BGKCollision (line 12) | class BGKCollision(Collision): method __init__ (line 13) | def __init__(self, tau, force: Optional['Force'] = None): method __call__ (line 17) | def __call__(self, flow: 'Flow') -> torch.Tensor: method name (line 24) | def name(self) -> str: method native_available (line 29) | def native_available(self) -> bool: method native_generator (line 32) | def native_generator(self, index: int) -> 'NativeCollision': FILE: lettuce/ext/_collision/kbc_collision.py class KBCCollision (line 11) | class KBCCollision(Collision): method __init__ (line 17) | def __init__(self, tau: float = None): method kbc_moment_transform (line 22) | def kbc_moment_transform(self, f): method kbc_moment_transform_3d (line 25) | def kbc_moment_transform_3d(self, f): method kbc_moment_transform_2d (line 33) | def kbc_moment_transform_2d(self, f): method compute_s_seq_from_m (line 41) | def compute_s_seq_from_m(self, f, m): method compute_s_seq_from_m_3d (line 44) | def compute_s_seq_from_m_3d(self, f, m): method compute_s_seq_from_m_2d (line 76) | def compute_s_seq_from_m_2d(self, f, m): method __call__ (line 96) | def __call__(self, flow: 'Flow') -> torch.Tensor: method native_available (line 162) | def native_available(self) -> bool: method native_generator (line 165) | def native_generator(self, index: int) -> 'NativeCollision': class KBCCollision2D (line 169) | class KBCCollision2D(KBCCollision): method __init__ (line 170) | def __init__(self, tau: float = None): class KBCCollision3D (line 176) | class KBCCollision3D(KBCCollision): method __init__ (line 177) | def __init__(self, tau: float = None): FILE: lettuce/ext/_collision/mrt_collision.py class MRTCollision (line 8) | class MRTCollision(Collision): method __init__ (line 17) | def __init__(self, transform: 'Transform', relaxation_parameters: list, method __call__ (line 23) | def __call__(self, flow: 'Flow'): method native_available (line 31) | def native_available(self) -> bool: method native_generator (line 34) | def native_generator(self, index: int) -> 'NativeCollision': FILE: lettuce/ext/_collision/no_collision.py class NoCollision (line 9) | class NoCollision(Collision): method __call__ (line 10) | def __call__(self, flow: 'Flow') -> torch.Tensor: method native_available (line 13) | def native_available(self) -> bool: method native_generator (line 16) | def native_generator(self, index: int) -> 'NativeCollision': FILE: lettuce/ext/_collision/regularized_collision.py class RegularizedCollision (line 8) | class RegularizedCollision(Collision): method __init__ (line 13) | def __init__(self, tau: float = None): method __call__ (line 17) | def __call__(self, flow: 'Flow'): method native_available (line 46) | def native_available(self) -> bool: method native_generator (line 49) | def native_generator(self, index: int) -> 'NativeCollision': FILE: lettuce/ext/_collision/smagorinsky_collision.py class SmagorinskyCollision (line 9) | class SmagorinskyCollision(Collision): method __init__ (line 14) | def __init__(self, tau, smagorinsky_constant=0.17, force: 'Force' = No... method __call__ (line 21) | def __call__(self, flow: 'Flow'): method native_available (line 40) | def native_available(self) -> bool: method native_generator (line 43) | def native_generator(self, index: int) -> 'NativeCollision': FILE: lettuce/ext/_collision/trt_collision.py class TRTCollision (line 6) | class TRTCollision(Collision): method __init__ (line 12) | def __init__(self, tau, tau_minus=1.0): method __call__ (line 16) | def __call__(self, flow: 'Flow'): method native_available (line 29) | def native_available(self) -> bool: method native_generator (line 32) | def native_generator(self, index: int) -> 'NativeCollision': FILE: lettuce/ext/_equilibrium/incompressible_quadratic_equilibrium.py class IncompressibleQuadraticEquilibrium (line 8) | class IncompressibleQuadraticEquilibrium(Equilibrium): method __init__ (line 9) | def __init__(self, rho0=1.0): method __call__ (line 12) | def __call__(self, flow: 'Flow', rho=None, u=None): FILE: lettuce/ext/_equilibrium/quadratic_equilibrium.py class QuadraticEquilibrium (line 10) | class QuadraticEquilibrium(Equilibrium): method __call__ (line 11) | def __call__(self, flow: 'Flow', rho=None, u=None): method native_available (line 26) | def native_available(self) -> bool: method native_generator (line 29) | def native_generator(self) -> 'NativeEquilibrium': FILE: lettuce/ext/_equilibrium/quadratic_equilibrium_less_memory.py class QuadraticEquilibriumLessMemory (line 8) | class QuadraticEquilibriumLessMemory(Equilibrium): method __call__ (line 14) | def __call__(self, flow: 'Flow', rho=None, u=None): method native_available (line 31) | def native_available(self) -> bool: method native_generator (line 34) | def native_generator(self) -> 'NativeEquilibrium': FILE: lettuce/ext/_flows/_ext_flow.py class ExtFlow (line 8) | class ExtFlow(Flow, ABC): method __init__ (line 16) | def __init__(self, context: 'Context', resolution: Union[int, List[int]], method make_resolution (line 35) | def make_resolution(self, resolution: Union[int, List[int]], method make_units (line 40) | def make_units(self, reynolds_number, mach_number, resolution: List[int] FILE: lettuce/ext/_flows/couette.py class CouetteFlow2D (line 16) | class CouetteFlow2D(ExtFlow): method __init__ (line 18) | def __init__(self, context: 'Context', resolution: Union[int, List[int]], method make_resolution (line 26) | def make_resolution(self, resolution: Union[int, List[int]], method make_units (line 33) | def make_units(self, reynolds_number, mach_number, resolution: List[int] method analytic_solution (line 43) | def analytic_solution(self): method initial_pu (line 49) | def initial_pu(self): method grid (line 56) | def grid(self): method post_boundaries (line 64) | def post_boundaries(self): FILE: lettuce/ext/_flows/decayingturbulence.py class DecayingTurbulence (line 23) | class DecayingTurbulence(ExtFlow): method __init__ (line 25) | def __init__(self, context: 'Context', resolution: Union[int, List[int]], method make_resolution (line 47) | def make_resolution(self, resolution: Union[int, List[int]], method make_units (line 54) | def make_units(self, reynolds_number, mach_number, resolution method analytic_solution (line 64) | def analytic_solution(self, x, t=0): method _generate_wavenumbers (line 67) | def _generate_wavenumbers(self): method _generate_spectrum (line 78) | def _generate_spectrum(self): method _generate_initial_velocity (line 88) | def _generate_initial_velocity(self, ek, wavenumber): method _compute_initial_pressure (line 157) | def _compute_initial_pressure(self): method initial_pu (line 162) | def initial_pu(self): method energy_spectrum (line 173) | def energy_spectrum(self): method grid (line 177) | def grid(self) -> (torch.Tensor, ...): method post_boundaries (line 188) | def post_boundaries(self) -> List['Boundary']: FILE: lettuce/ext/_flows/doublyshear.py class DoublyPeriodicShear2D (line 19) | class DoublyPeriodicShear2D(ExtFlow): method __init__ (line 21) | def __init__(self, context: 'Context', resolution: Union[int, List[int]], method make_resolution (line 35) | def make_resolution(self, resolution: Union[int, List[int]], method make_units (line 43) | def make_units(self, reynolds_number, mach_number, method analytic_solution (line 51) | def analytic_solution(self, t=0): method initial_pu (line 54) | def initial_pu(self) -> (float, Union[np.array, torch.Tensor]): method grid (line 68) | def grid(self) -> (torch.Tensor, torch.Tensor): method post_boundaries (line 79) | def post_boundaries(self): FILE: lettuce/ext/_flows/lamboseenvortex.py class LambOseenVortex2D (line 19) | class LambOseenVortex2D(ExtFlow): method __init__ (line 20) | def __init__(self, context: 'Context', resolution: Union[int, List[int]], method make_resolution (line 50) | def make_resolution(self, resolution: Union[int, List[int]], method make_units (line 58) | def make_units(self, reynolds_number, mach_number, method grid (line 67) | def grid(self): method initial_pu (line 75) | def initial_pu(self) -> (torch.Tensor, torch.Tensor): method initial_lamboseenvortex (line 79) | def initial_lamboseenvortex(self) -> (torch.Tensor, torch.Tensor): method post_boundaries (line 133) | def post_boundaries(self) -> List['Boundary']: FILE: lettuce/ext/_flows/liddrivencavity.py class Cavity2D (line 14) | class Cavity2D(ExtFlow): method __init__ (line 16) | def __init__(self, context: 'Context', resolution, reynolds_number, method make_resolution (line 20) | def make_resolution(self, resolution: Union[int, List[int]], method make_units (line 28) | def make_units(self, reynolds_number, mach_number, method initial_pu (line 36) | def initial_pu(self): method grid (line 41) | def grid(self) -> (torch.Tensor, torch.Tensor): method post_boundaries (line 52) | def post_boundaries(self): FILE: lettuce/ext/_flows/obstacle.py class Obstacle (line 16) | class Obstacle(ExtFlow): method __init__ (line 54) | def __init__(self, context: Context, resolution: Union[int, List[int]], method make_units (line 67) | def make_units(self, reynolds_number, mach_number, resolution: List[int] method make_resolution (line 76) | def make_resolution(self, resolution: Union[int, List[int]], method mask (line 84) | def mask(self): method mask (line 88) | def mask(self, m): method initial_pu (line 94) | def initial_pu(self) -> (float, Union[np.array, torch.Tensor]): method grid (line 102) | def grid(self): method post_boundaries (line 108) | def post_boundaries(self): method _unit_vector (line 124) | def _unit_vector(self, i=0): function Obstacle2D (line 128) | def Obstacle2D(context: 'Context', resolution: Union[int, List[int]], function Obstacle3D (line 141) | def Obstacle3D(context: 'Context', resolution: Union[int, List[int]], FILE: lettuce/ext/_flows/poiseuille.py class PoiseuilleFlow2D (line 18) | class PoiseuilleFlow2D(ExtFlow): method __init__ (line 20) | def __init__(self, context: 'Context', resolution: Union[int, List[int]], method analytic_solution (line 30) | def analytic_solution(self, t=0) -> (torch.Tensor, torch.Tensor): method initial_pu (line 42) | def initial_pu(self): method make_units (line 51) | def make_units(self, reynolds_number, mach_number, method make_resolution (line 60) | def make_resolution(self, resolution: Union[int, List[int]], method grid (line 69) | def grid(self): method post_boundaries (line 78) | def post_boundaries(self): method acceleration (line 85) | def acceleration(self): FILE: lettuce/ext/_flows/taylorgreen.py class TaylorGreenVortex (line 16) | class TaylorGreenVortex(ExtFlow): method __init__ (line 17) | def __init__(self, context: 'Context', resolution: Union[int, List[int]], method make_resolution (line 33) | def make_resolution(self, resolution: Union[int, List[int]], method make_units (line 43) | def make_units(self, reynolds_number, mach_number, method grid (line 53) | def grid(self): method initial_pu (line 63) | def initial_pu(self) -> (torch.Tensor, torch.Tensor): method analytic_solution (line 66) | def analytic_solution(self, t: float) -> (torch.Tensor, torch.Tensor): method post_boundaries (line 97) | def post_boundaries(self) -> List['Boundary']: function TaylorGreenVortex3D (line 101) | def TaylorGreenVortex3D(context: 'Context', resolution: Union[int, List[... function TaylorGreenVortex2D (line 113) | def TaylorGreenVortex2D(context: 'Context', resolution: Union[int, List[... FILE: lettuce/ext/_force/_force.py class Force (line 6) | class Force(ABC): method __init__ (line 8) | def __init__(self, flow: 'Flow', tau, acceleration): method source_term (line 12) | def source_term(self, u): method u_eq (line 16) | def u_eq(self, flow: 'Flow'): method ueq_scaling_factor (line 21) | def ueq_scaling_factor(self): method native_available (line 25) | def native_available(self) -> bool: method native_generator (line 29) | def native_generator(self) -> 'NativeForce': FILE: lettuce/ext/_force/guo.py class Guo (line 9) | class Guo(Force): method __init__ (line 11) | def __init__(self, flow, tau, acceleration): method source_term (line 16) | def source_term(self, u): method u_eq (line 29) | def u_eq(self, flow: 'Flow' = None): method ueq_scaling_factor (line 36) | def ueq_scaling_factor(self): method native_available (line 39) | def native_available(self) -> bool: method native_generator (line 42) | def native_generator(self) -> 'NativeForce': FILE: lettuce/ext/_force/shan_chen.py class ShanChen (line 7) | class ShanChen(Force): method __init__ (line 9) | def __init__(self, flow, tau, acceleration): method source_term (line 13) | def source_term(self, u): method u_eq (line 16) | def u_eq(self, flow: 'Flow'): method ueq_scaling_factor (line 22) | def ueq_scaling_factor(self): method native_available (line 25) | def native_available(self) -> bool: method native_generator (line 28) | def native_generator(self) -> 'NativeForce': FILE: lettuce/ext/_reporter/error_reporter.py class ErrorReporter (line 9) | class ErrorReporter(Reporter): method __init__ (line 12) | def __init__(self, analytical_solution, interval=1, out=sys.stdout): method __call__ (line 19) | def __call__(self, simulation: 'Simulation'): FILE: lettuce/ext/_reporter/failure_reporter.py class FailureReporterBase (line 12) | class FailureReporterBase(Reporter, ABC): method __init__ (line 25) | def __init__(self, interval: int, k: int = 100, outdir: Optional[str]=... method __call__ (line 34) | def __call__(self, simulation: 'BreakableSimulation'): method _get_top_failures (line 57) | def _get_top_failures(self, mask: torch.Tensor, values: torch.Tensor) ... method _write_log (line 78) | def _write_log(self, simulation: 'BreakableSimulation', results: List[... method save_vtk (line 99) | def save_vtk(self, simulation: 'BreakableSimulation'): method locations_string (line 116) | def locations_string(self, simulation: 'BreakableSimulation') -> List[... method is_failed (line 120) | def is_failed(self, simulation: 'BreakableSimulation') -> torch.Tensor: method get_results (line 125) | def get_results(self, simulation: 'BreakableSimulation') -> List[Tuple]: class NaNReporter (line 131) | class NaNReporter(FailureReporterBase): method __init__ (line 133) | def __init__(self, interval: int, k: int =100, outdir: Optional[str]=N... method is_failed (line 138) | def is_failed(self, simulation: 'BreakableSimulation') -> torch.Tensor: method get_results (line 141) | def get_results(self, simulation: 'BreakableSimulation') -> List[Tuple]: method locations_string (line 145) | def locations_string(self, simulation: 'BreakableSimulation') -> List[... class HighMaReporter (line 156) | class HighMaReporter(FailureReporterBase): method __init__ (line 157) | def __init__(self, interval: int, threshold: float =0.3, k: int =100, method is_failed (line 164) | def is_failed(self, simulation: 'BreakableSimulation') -> torch.Tensor: method get_results (line 169) | def get_results(self, simulation: 'BreakableSimulation') -> List[Tuple]: method locations_string (line 175) | def locations_string(self, simulation: 'BreakableSimulation') -> List[... FILE: lettuce/ext/_reporter/observable_reporter.py class Observable (line 17) | class Observable(ABC): method __init__ (line 18) | def __init__(self, flow: 'Flow'): method __call__ (line 23) | def __call__(self, f: Optional[torch.Tensor] = None): class MaximumVelocity (line 27) | class MaximumVelocity(Observable): method __call__ (line 30) | def __call__(self, f: Optional[torch.Tensor] = None): class IncompressibleKineticEnergy (line 34) | class IncompressibleKineticEnergy(Observable): method __call__ (line 37) | def __call__(self, f: Optional[torch.Tensor] = None): class Enstrophy (line 45) | class Enstrophy(Observable): method __call__ (line 53) | def __call__(self, f: Optional[torch.Tensor] = None): class EnergySpectrum (line 71) | class EnergySpectrum(Observable): method __init__ (line 74) | def __init__(self, flow: Flow): method __call__ (line 99) | def __call__(self, f: Optional[torch.Tensor] = None): method spectrum_from_u (line 103) | def spectrum_from_u(self, u): method _ekin_spectrum (line 110) | def _ekin_spectrum(self, u): method _ekin_spectrum_torch_lt_18 (line 119) | def _ekin_spectrum_torch_lt_18(self, u): method _ekin_spectrum_torch_ge_18 (line 131) | def _ekin_spectrum_torch_ge_18(self, u): class Mass (line 140) | class Mass(Observable): method __init__ (line 150) | def __init__(self, flow: Flow, no_mass_mask=None): method __call__ (line 154) | def __call__(self, f: Optional[torch.Tensor] = None): class ObservableReporter (line 161) | class ObservableReporter(Reporter): method __init__ (line 177) | def __init__(self, observable, interval=1, out=sys.stdout): method __call__ (line 185) | def __call__(self, simulation: 'Simulation'): FILE: lettuce/ext/_reporter/progress_reporter.py function append_txt_file (line 6) | def append_txt_file(filename, line: str): class ProgressReporter (line 12) | class ProgressReporter(Reporter): method __init__ (line 30) | def __init__(self, interval=1000, t_max=0, i_target=0, i_start=0, method __call__ (line 55) | def __call__(self, simulation: 'Simulation'): method start_timer (line 110) | def start_timer(self): FILE: lettuce/ext/_reporter/vtk_reporter.py function write_vtk (line 10) | def write_vtk(point_dict, id=0, filename_base="./data/output"): class VTKReporter (line 18) | class VTKReporter(Reporter): method __init__ (line 21) | def __init__(self, interval=50, filename_base="./data/output"): method __call__ (line 29) | def __call__(self, simulation: 'Simulation'): method output_mask (line 49) | def output_mask(self, simulation: 'Simulation'): FILE: lettuce/ext/_reporter/write_image.py function write_image (line 4) | def write_image(filename, array2d): FILE: lettuce/ext/_stencil/d1q3.py class D1Q3 (line 6) | class D1Q3(Stencil): method __init__ (line 7) | def __init__(self): FILE: lettuce/ext/_stencil/d2q9.py class D2Q9 (line 6) | class D2Q9(Stencil): method __init__ (line 7) | def __init__(self): FILE: lettuce/ext/_stencil/d3q15.py class D3Q15 (line 6) | class D3Q15(Stencil): method __init__ (line 7) | def __init__(self): FILE: lettuce/ext/_stencil/d3q19.py class D3Q19 (line 6) | class D3Q19(Stencil): method __init__ (line 7) | def __init__(self): FILE: lettuce/ext/_stencil/d3q27.py class D3Q27 (line 6) | class D3Q27(Stencil): method __init__ (line 7) | def __init__(self): FILE: lettuce/util/datautils.py class HDF5Reporter (line 17) | class HDF5Reporter(Reporter): method __init__ (line 49) | def __init__(self, flow, collision, interval, filebase='./output', met... method __call__ (line 66) | def __call__(self, simulation: 'Simulation'): # i, t, f): method _pickle_to_h5 (line 76) | def _pickle_to_h5(instance): class LettuceDataset (line 83) | class LettuceDataset(data.Dataset): method __init__ (line 112) | def __init__(self, filebase, transform=None, target=False, skip_idx_to... method __str__ (line 123) | def __str__(self): method __len__ (line 131) | def __len__(self): method __getitem__ (line 134) | def __getitem__(self, idx): method __del__ (line 145) | def __del__(self): method get_data (line 148) | def get_data(self, idx): method get_attr (line 151) | def get_attr(self, attr): method _unpickle_from_h5 (line 155) | def _unpickle_from_h5(byte_str): FILE: lettuce/util/moments.py function moment_tensor (line 34) | def moment_tensor(e: List[List[int]], multiindex): function get_default_moment_transform (line 41) | def get_default_moment_transform(stencil: 'Stencil', context: 'Context'): class Moments (line 51) | class Moments: method __init__ (line 52) | def __init__(self, lattice): class Transform (line 60) | class Transform: method __init__ (line 65) | def __init__(self, stencil: 'Stencil', context: 'Context', names=None): method __getitem__ (line 71) | def __getitem__(self, moment_names): method transform (line 76) | def transform(self, f): method inverse_transform (line 79) | def inverse_transform(self, m): method equilibrium (line 82) | def equilibrium(self, m: torch.Tensor, flow: 'Flow'): method einsum (line 96) | def einsum(self, equation, fields, *args) -> torch.Tensor: method mv (line 112) | def mv(self, m, v) -> torch.Tensor: class D1Q3Transform (line 117) | class D1Q3Transform(Transform): method __init__ (line 131) | def __init__(self, stencil: 'Stencil', context: 'Context'): method transform (line 136) | def transform(self, f): method inverse_transform (line 139) | def inverse_transform(self, m): class D2Q9Dellar (line 147) | class D2Q9Dellar(Transform): method __init__ (line 177) | def __init__(self, stencil: 'Stencil', context: 'Context'): method transform (line 182) | def transform(self, f): method inverse_transform (line 185) | def inverse_transform(self, m): method equilibrium (line 188) | def equilibrium(self, m, flow: 'Flow'): class D2Q9Lallemand (line 207) | class D2Q9Lallemand(Transform): method __init__ (line 233) | def __init__(self, stencil: 'Stencil', context: 'Context'): method transform (line 238) | def transform(self, f): method inverse_transform (line 241) | def inverse_transform(self, m): method equilibrium (line 244) | def equilibrium(self, m, flow: 'Flow'): class D3Q27Hermite (line 369) | class D3Q27Hermite(Transform): method __init__ (line 535) | def __init__(self, stencil: 'Stencil', context: 'Context'): method transform (line 540) | def transform(self, f): method inverse_transform (line 543) | def inverse_transform(self, m): method equilibrium (line 546) | def equilibrium(self, m, flow: 'Flow'): FILE: lettuce/util/utility.py function get_subclasses (line 15) | def get_subclasses(cls, module): class LettuceException (line 21) | class LettuceException(Exception): class LettuceWarning (line 25) | class LettuceWarning(UserWarning): class InefficientCodeWarning (line 29) | class InefficientCodeWarning(LettuceWarning): class ExperimentalWarning (line 33) | class ExperimentalWarning(LettuceWarning): function torch_gradient (line 37) | def torch_gradient(f, dx=1, order=2): function grid_fine_to_coarse (line 102) | def grid_fine_to_coarse(flow: 'Flow', f_fine, tau_fine, tau_coarse): function torch_jacobi (line 119) | def torch_jacobi(f, p, dx, dim, tol_abs=1e-10, max_num_steps=100000): function append_axes (line 159) | def append_axes(array, n): FILE: tests/boundary/test_antibounceback_outlet_bc.py function test_anti_bounce_back_outlet (line 6) | def test_anti_bounce_back_outlet(fix_configuration, fix_stencil): FILE: tests/boundary/test_bc_masks.py function test_masks (line 4) | def test_masks(fix_configuration): FILE: tests/boundary/test_bounceback_bc.py function test_bounce_back_boundary (line 6) | def test_bounce_back_boundary(fix_stencil, fix_configuration): function test_bounce_back_boundary_not_applied_if_mask_empty (line 22) | def test_bounce_back_boundary_not_applied_if_mask_empty(fix_stencil, FILE: tests/boundary/test_equilibrium_bc_outlet_p.py function test_equilibrium_outlet_p_algorithm (line 6) | def test_equilibrium_outlet_p_algorithm(fix_stencil, fix_configuration): FILE: tests/boundary/test_equilibrium_bc_pu.py function moment_dims_params (line 4) | def moment_dims_params(): function moment_dims_ids (line 11) | def moment_dims_ids(): function fix_stencil_x_moment_dims (line 20) | def fix_stencil_x_moment_dims(request): class DummyEquilibriumBoundary (line 24) | class DummyEquilibriumBoundary(EquilibriumBoundaryPU): method make_no_streaming_mask (line 32) | def make_no_streaming_mask(self, shape: List[int], context: 'Context')... function test_equilibrium_boundary_pu_algorithm (line 37) | def test_equilibrium_boundary_pu_algorithm(fix_stencil, fix_configuration): function test_equilibrium_boundary_pu_tgv (line 84) | def test_equilibrium_boundary_pu_tgv(fix_stencil, fix_configuration): function test_equilibrium_boundary_pu_native (line 125) | def test_equilibrium_boundary_pu_native(fix_stencil_x_moment_dims, fix_d... FILE: tests/boundary/test_equilibrium_pressure_outlet.py function test_equilibrium_pressure_outlet (line 4) | def test_equilibrium_pressure_outlet(fix_configuration, fix_stencil): FILE: tests/boundary/test_partiallysaturated_bc.py function test_partiallysaturated_boundary (line 6) | def test_partiallysaturated_boundary(fix_stencil, fix_configuration): function test_fullysaturated_like_neq_bounceback (line 24) | def test_fullysaturated_like_neq_bounceback(fix_stencil, fix_configurati... function test_partiallysaturated_boundary_not_applied_if_mask_empty (line 55) | def test_partiallysaturated_boundary_not_applied_if_mask_empty(fix_stencil, FILE: tests/collision/test_collision_conserves_mass.py function test_collision_conserves_mass (line 4) | def test_collision_conserves_mass(fix_conserving_collision, FILE: tests/collision/test_collision_conserves_momentum.py function test_collision_conserves_momentum (line 4) | def test_collision_conserves_momentum(fix_conserving_collision, FILE: tests/collision/test_collision_fixpoint_2x.py function test_collision_fixpoint_2x (line 5) | def test_collision_fixpoint_2x(Collision, FILE: tests/collision/test_collision_fixpoint_2x_MRT.py function test_collision_fixpoint_2x_MRT (line 6) | def test_collision_fixpoint_2x_MRT(Transform, fix_configuration): FILE: tests/collision/test_collision_optimizes_pseudo_entropy.py function test_collision_optimizes_pseudo_entropy (line 4) | def test_collision_optimizes_pseudo_entropy(fix_configuration, FILE: tests/collision/test_collision_relaxes_shear_moments.py function test_collision_relaxes_shear_moments (line 7) | def test_collision_relaxes_shear_moments(Collision, FILE: tests/collision/test_force.py function test_force (line 6) | def test_force(ForceType, fix_device): FILE: tests/conftest.py function dtype_params (line 10) | def dtype_params(): function dtype_ids (line 14) | def dtype_ids(): function stencil1d_params (line 18) | def stencil1d_params(): function stencil2d_params (line 22) | def stencil2d_params(): function stencil3d_params (line 26) | def stencil3d_params(): function stencil_params (line 30) | def stencil_params(): function stencil1d_ids (line 34) | def stencil1d_ids(): function stencil2d_ids (line 38) | def stencil2d_ids(): function stencil3d_ids (line 42) | def stencil3d_ids(): function stencil_ids (line 46) | def stencil_ids(): function device_params (line 50) | def device_params(): function device_ids (line 54) | def device_ids(): function native_params (line 58) | def native_params(): function native_ids (line 62) | def native_ids(): function configuration_params (line 66) | def configuration_params(): function configuration_ids (line 74) | def configuration_ids(): function transform_params (line 87) | def transform_params(): function transform_ids (line 103) | def transform_ids(): function fix_transform (line 108) | def fix_transform(request): function fix_collision (line 116) | def fix_collision(request): function conserving_collision_params (line 120) | def conserving_collision_params(): function conserving_collision_ids (line 130) | def conserving_collision_ids(): function fix_dtype (line 141) | def fix_dtype(request): function fix_stencil1d (line 146) | def fix_stencil1d(request): function fix_stencil2d (line 151) | def fix_stencil2d(request): function fix_stencil3d (line 156) | def fix_stencil3d(request): function fix_stencil (line 161) | def fix_stencil(request): function fix_device (line 166) | def fix_device(request): function fix_native (line 174) | def fix_native(request): function fix_configuration (line 182) | def fix_configuration(request): function fix_conserving_collision (line 191) | def fix_conserving_collision(request): class TestFlow (line 195) | class TestFlow(ExtFlow): method __init__ (line 198) | def __init__(self, context: 'Context', resolution: Union[int, List[int]], method make_resolution (line 206) | def make_resolution(self, resolution: List[int], method make_units (line 216) | def make_units(self, reynolds_number, mach_number, method initial_pu (line 221) | def initial_pu(self) -> (float, Union[np.array, torch.Tensor]): method boundaries (line 227) | def boundaries(self) -> List['Boundary']: method boundaries (line 231) | def boundaries(self, boundaries: List['Boundary']): function DummyTGV (line 235) | def DummyTGV(context: 'Context', resolution: Union[int, List[int]], class DummyFlow (line 243) | class DummyFlow(ExtFlow): method __init__ (line 245) | def __init__(self, context: Context, resolution: int = 16): method make_resolution (line 248) | def make_resolution(self, resolution: Union[int, List[int]], method make_units (line 253) | def make_units(self, reynolds_number, mach_number, _: List[int] method initial_pu (line 258) | def initial_pu(self) -> (float, List[float]): method initialize (line 261) | def initialize(self): method boundaries (line 265) | def boundaries(self) -> List['Boundary']: FILE: tests/flow/test_divergence.py function test_divergence (line 5) | def test_divergence(stencil2d3d, fix_configuration): FILE: tests/flow/test_einsum.py class EinsumFlow (line 5) | class EinsumFlow(TestFlow): method __init__ (line 6) | def __init__(self, context, res): method j (line 11) | def j(self, f: Optional[torch.Tensor] = None) -> torch.Tensor: method incompressible_energy (line 16) | def incompressible_energy(self, f: Optional[torch.Tensor] = None method entropy (line 22) | def entropy(self, f: Optional[torch.Tensor] = None) -> torch.Tensor: method pseudo_entropy_global (line 29) | def pseudo_entropy_global(self, f: Optional[torch.Tensor] = None method pseudo_entropy_local (line 36) | def pseudo_entropy_local(self, f: Optional[torch.Tensor] = None method shear_tensor (line 44) | def shear_tensor(self, f: Optional[torch.Tensor] = None) -> torch.Tensor: method einsum (line 52) | def einsum(self, equation, fields, *args) -> torch.Tensor: function initialize_f_neq_floweinsum (line 68) | def initialize_f_neq_floweinsum(flow: 'EinsumFlow'): function initialize_f_neq_torcheinsum (line 97) | def initialize_f_neq_torcheinsum(flow: 'Flow'): class RegularizedEinsum (line 126) | class RegularizedEinsum(RegularizedCollision): method __call__ (line 127) | def __call__(self, flow: 'Flow'): class GuoEinsum (line 156) | class GuoEinsum(Guo): method source_term (line 157) | def source_term(self, u): function test_einsum (line 172) | def test_einsum(fix_dim): FILE: tests/flow/test_flow.py function test_flow (line 5) | def test_flow(flowname, fix_configuration): FILE: tests/flow/test_initialize_fneq.py function test_initialize_fneq (line 10) | def test_initialize_fneq(fix_configuration, fix_stencil, Case): FILE: tests/flow/test_initialize_pressure.py function test_initialize_pressure (line 5) | def test_initialize_pressure(fix_dtype): FILE: tests/flow/test_obstacle.py function test_obstacle (line 5) | def test_obstacle(stencil2d3d, fix_configuration): FILE: tests/flow/test_pressure_poisson.py function test_pressure_poisson (line 6) | def test_pressure_poisson(fix_configuration, Stencil): FILE: tests/moments/test_conserved_moments_d2q9.py function test_conserved_moments_d2q9 (line 6) | def test_conserved_moments_d2q9(MomentSet): FILE: tests/moments/test_getitem.py function test_getitem (line 4) | def test_getitem(fix_device, fix_dtype): FILE: tests/moments/test_inverse_transform.py function test_inverse_transform (line 4) | def test_inverse_transform(fix_transform): FILE: tests/moments/test_moment_equilibrium_D3Q27Hermite.py function test_moment_equilibrium_D3Q27Hermite (line 4) | def test_moment_equilibrium_D3Q27Hermite(fix_device, fix_dtype): FILE: tests/moments/test_moment_equilibrium_dellar.py function test_moment_equilibrium_dellar (line 4) | def test_moment_equilibrium_dellar(fix_device, fix_dtype): FILE: tests/moments/test_moment_equilibrium_lallemand.py function test_moment_equilibrium_lallemand (line 4) | def test_moment_equilibrium_lallemand(fix_device, fix_dtype): FILE: tests/moments/test_moments_density.py function test_moments_density_array (line 5) | def test_moments_density_array(fix_stencil): function test_more_moments_density_array (line 10) | def test_more_moments_density_array(fix_stencil): function test_moments_density_tensor (line 15) | def test_moments_density_tensor(fix_stencil, fix_device): function test_more_moments_density_tensor (line 25) | def test_more_moments_density_tensor(fix_stencil, fix_device): FILE: tests/moments/test_orthogonality.py function test_orthogonality (line 4) | def test_orthogonality(fix_device, fix_dtype, fix_transform): FILE: tests/native/test_native_bgk_collision.py class DummyBGK (line 9) | class DummyBGK(ExtFlow): method __init__ (line 10) | def __init__(self, context: Context): method make_resolution (line 13) | def make_resolution(self, resolution: Union[int, List[int]], method make_units (line 18) | def make_units(self, reynolds_number, mach_number, _: List[int] method initial_pu (line 23) | def initial_pu(self) -> (float, List[float]): method initialize (line 26) | def initialize(self): method boundaries (line 31) | def boundaries(self) -> List['Boundary']: function test_native_bgk_collision (line 35) | def test_native_bgk_collision(): FILE: tests/native/test_native_bounce_back.py class MyBounceBackBoundary (line 11) | class MyBounceBackBoundary(BounceBackBoundary): method make_no_collision_mask (line 12) | def make_no_collision_mask(self, shape: List[int], context: 'Context' class DummyBBBC (line 22) | class DummyBBBC(DummyFlow): method initialize (line 24) | def initialize(self): method boundaries (line 29) | def boundaries(self) -> List['Boundary']: function test_native_bounce_back (line 33) | def test_native_bounce_back(): FILE: tests/native/test_native_equilibrium_pu.py class my_equilibrium_boundary_mask (line 12) | class my_equilibrium_boundary_mask(EquilibriumBoundaryPU): method make_no_collision_mask (line 14) | def make_no_collision_mask(self, shape: List[int], context: 'Context' method make_no_streaming_mask (line 21) | def make_no_streaming_mask(self, shape: List[int], context: 'Context' function test_equilibrium_boundary_pu_native (line 27) | def test_equilibrium_boundary_pu_native(): FILE: tests/native/test_native_no_streaming_mask.py function test_native_no_streaming_mask (line 4) | def test_native_no_streaming_mask(): FILE: tests/native/test_native_streaming.py class DummyStreaming (line 9) | class DummyStreaming(DummyFlow): method initialize (line 11) | def initialize(self): function test_native_streaming (line 24) | def test_native_streaming(): FILE: tests/native/test_native_streaming_strategy.py function test_native_streaming_strategy (line 11) | def test_native_streaming_strategy(streaming_strategy): FILE: tests/reporter/test_HDF5Reporter.py function test_HDF5Reporter (line 5) | def test_HDF5Reporter(tmpdir): FILE: tests/reporter/test_energy_spectrum.py function test_energy_spectrum (line 5) | def test_energy_spectrum(tmpdir, flowname): FILE: tests/reporter/test_generic_reporters.py function test_generic_reporters (line 10) | def test_generic_reporters(Observable, Case, fix_configuration): FILE: tests/reporter/test_high_ma_reporter.py function test_high_ma_reporter (line 5) | def test_high_ma_reporter(tmpdir): FILE: tests/reporter/test_nan_reporter.py function test_nan_reporter (line 5) | def test_nan_reporter(tmpdir): FILE: tests/reporter/test_vtk_reporter_mask.py function test_vtk_reporter_mask (line 6) | def test_vtk_reporter_mask(tmpdir): FILE: tests/reporter/test_vtk_reporter_no_mask.py function test_vtk_reporter_no_mask (line 5) | def test_vtk_reporter_no_mask(tmpdir): FILE: tests/reporter/test_write_image.py function test_write_image (line 5) | def test_write_image(tmpdir): FILE: tests/reporter/test_write_vtk.py function test_write_vtk (line 6) | def test_write_vtk(tmpdir): FILE: tests/stencil/test_first_zero.py function test_first_zero (line 4) | def test_first_zero(fix_stencil): FILE: tests/stencil/test_opposite.py function test_opposite (line 3) | def test_opposite(fix_stencil): FILE: tests/stencil/test_symmetry.py function test_symmetry (line 4) | def test_symmetry(fix_stencil): FILE: tests/stencil/test_weights.py function test_weights (line 4) | def test_weights(fix_stencil): FILE: tests/test_checkpoint.py function test_checkpoint (line 5) | def test_checkpoint(tmpdir): FILE: tests/test_cli.py function response (line 14) | def response(): function test_content (line 23) | def test_content(response): function test_command_line_interface (line 29) | def test_command_line_interface(): FILE: tests/test_equilibrium.py function test_equilibrium_conserves_mass (line 5) | def test_equilibrium_conserves_mass(fix_equilibrium, function test_equilibrium_conserves_momentum (line 24) | def test_equilibrium_conserves_momentum(fix_equilibrium, FILE: tests/unit/__init__.py function create_default_unit_conversion (line 4) | def create_default_unit_conversion(): FILE: tests/unit/test_consistency.py function test_consistency (line 5) | def test_consistency(): FILE: tests/unit/test_conversion_reversible.py function test_conversion_reversible (line 5) | def test_conversion_reversible(): FILE: tests/unit/test_reynolds_number_consistent.py function test_reynolds_number_consistent (line 5) | def test_reynolds_number_consistent(): FILE: tests/util/test_grid_fine_to_coarse.py function test_grid_fine_to_coarse (line 5) | def test_grid_fine_to_coarse(dims): FILE: tests/util/test_torch_gradient.py function test_torch_gradient (line 6) | def test_torch_gradient(dims, order): FILE: versioneer.py class VersioneerConfig (line 293) | class VersioneerConfig: function get_root (line 297) | def get_root(): function get_config_from_root (line 336) | def get_config_from_root(root): class NotThisMethod (line 364) | class NotThisMethod(Exception): function register_vcs_handler (line 373) | def register_vcs_handler(vcs, method): # decorator function run_command (line 382) | def run_command(commands, args, cwd=None, verbose=False, hide_stderr=False, function git_get_keywords (line 1065) | def git_get_keywords(versionfile_abs): function git_versions_from_keywords (line 1093) | def git_versions_from_keywords(keywords, tag_prefix, verbose): function git_pieces_from_vcs (line 1157) | def git_pieces_from_vcs(tag_prefix, root, verbose, runner=run_command): function do_vcs_install (line 1285) | def do_vcs_install(manifest_in, versionfile_source, ipy): function versions_from_parentdir (line 1322) | def versions_from_parentdir(parentdir_prefix, root, verbose): function versions_from_file (line 1364) | def versions_from_file(filename): function write_to_version_file (line 1381) | def write_to_version_file(filename, versions): function plus_or_dot (line 1392) | def plus_or_dot(pieces): function render_pep440 (line 1399) | def render_pep440(pieces): function render_pep440_branch (line 1424) | def render_pep440_branch(pieces): function pep440_split_post (line 1454) | def pep440_split_post(ver): function render_pep440_pre (line 1464) | def render_pep440_pre(pieces): function render_pep440_post (line 1488) | def render_pep440_post(pieces): function render_pep440_post_branch (line 1515) | def render_pep440_post_branch(pieces): function render_pep440_old (line 1544) | def render_pep440_old(pieces): function render_git_describe (line 1566) | def render_git_describe(pieces): function render_git_describe_long (line 1586) | def render_git_describe_long(pieces): function render (line 1606) | def render(pieces, style): class VersioneerBadRootError (line 1642) | class VersioneerBadRootError(Exception): function get_versions (line 1646) | def get_versions(verbose=False): function get_version (line 1722) | def get_version(): function get_cmdclass (line 1727) | def get_cmdclass(cmdclass=None): function do_setup (line 1978) | def do_setup(): function scan_setup_py (line 2066) | def scan_setup_py():