Graph Neural Network Datapipes

The VortexSheddingDataset processes flow field data around bluff bodies, capturing vortex shedding patterns and flow structures for graph-based learning. The VortexSheddingDataset is used in the VortexShedding CFD examples.

The AhmedBodyDataset manages flow field data around Ahmed bodies, supporting aerodynamic analysis and drag prediction tasks. The AhmedBodyDataset is used in the AeroGraphNet CFD External Aerodynamics example.

The DrivAerNetDataset handles automotive aerodynamics surface data, providing access to surface pressure and wall shear stress distributions. The DrivAerNetDataset is used in the AeroGraphNet and FIGConvNet CFD External Aerodynamics examples.

The StokesDataset processes Stokes flow simulations in pipe domains obstructed by random polygons, supporting various boundary conditions and geometry configurations. The StokesDataset is used in the Stokes MeshGraphNet CFD example.

The GNN utilities provide helper functions for reading VTP files and saving/loading JSON-serialized statistics used by the GNN datapipes. The GNN utilities are used by the GNN dataset classes and in the structural mechanics examples.