noether.data.datasets.cfd.dataset

Classes

AeroDataset

Dataset implementation for aerodynamic datasets with volume and surface fields.

Module Contents

class noether.data.datasets.cfd.dataset.AeroDataset(dataset_config, filemap)

Bases: noether.data.Dataset

Dataset implementation for aerodynamic datasets with volume and surface fields. This unified dataset class provides an interface for aerodynamics dataset with volume and surface fields. The dataset behavior such as the dataset choice, train/val/test split IDs, etc. is configured through constructor parameters, allowing for easy extension to new datasets.

Parameters:
filemap
getitem_surface_position(idx)

Retrieves surface positions (num_surface_points, 3)

Parameters:

idx (int)

Return type:

torch.Tensor

getitem_surface_pressure(idx)

Retrieves surface pressures (num_surface_points, 1)

Parameters:

idx (int)

Return type:

torch.Tensor

getitem_surface_friction(idx)

Retrieves surface friction (=wallshearstress) (num_surface_points, 3)

Parameters:

idx (int)

Return type:

torch.Tensor

getitem_volume_position(idx)

Retrieves volume position (num_volume_points, 3)

Parameters:

idx (int)

Return type:

torch.Tensor

getitem_volume_pressure(idx)

Retrieves volume pressures (num_volume_points, 1)

Parameters:

idx (int)

Return type:

torch.Tensor

getitem_volume_velocity(idx)

Retrieves volume velocity (num_volume_points, 3)

Parameters:

idx (int)

Return type:

torch.Tensor

getitem_volume_vorticity(idx)

Retrieves volume vorticity (num_volume_points, 3)

Parameters:

idx (int)

Return type:

torch.Tensor

getitem_volume_sdf(idx)

Retrieve signed distance field at volume points.

Parameters:

idx (int)

Return type:

torch.Tensor

getitem_volume_normals(idx)

Retrieve normal vectors at volume points.

Parameters:

idx (int)

Return type:

torch.Tensor

getitem_surface_normals(idx)

Retrieve surface normal vectors.

Parameters:

idx (int)

Return type:

torch.Tensor