Extend crash datapipe, readers, and update README

examples/structural_mechanics/crash/conf/config.yaml

@@ -25,6 +25,7 @@ experiment_desc: "unified training recipe for crash models"
 run_desc: "unified training recipe for crash models"
 
 defaults:
+  - reader: vtp #d3plot
   - datapipe: point_cloud   # will be overridden by model configs
   - model: transolver_autoregressive_rollout_training
   - training: default

examples/structural_mechanics/crash/conf/datapipe/graph.yaml

@@ -15,9 +15,10 @@
 # limitations under the License.
 
 _target_: datapipe.CrashGraphDataset
+_convert_: all
 data_dir: ${training.raw_data_dir}
 name: crash_train
 split: train
 num_samples: ${training.num_training_samples}
 num_steps: ${training.num_time_steps}
-wall_node_disp_threshold: 1.0
+features: [thickness]

examples/structural_mechanics/crash/conf/datapipe/point_cloud.yaml

@@ -15,7 +15,8 @@
 # limitations under the License.
 
 _target_: datapipe.CrashPointCloudDataset
+_convert_: all
 data_dir: ${training.raw_data_dir}
 num_samples: ${training.num_training_samples}
 num_steps: ${training.num_time_steps}
-wall_node_disp_threshold: 1.0
+features: [thickness]

examples/structural_mechanics/crash/conf/model/mgn_autoregressive_rollout_training.yaml

@@ -15,6 +15,7 @@
 # limitations under the License.
 
 _target_: rollout.MeshGraphNetAutoregressiveRolloutTraining
+_convert_: all
 
 input_dim_nodes: 7                 # pos(3) + vel(3) + thickness(1)
 input_dim_edges: 4                 # dx, dy, dz, distance

examples/structural_mechanics/crash/conf/model/mgn_one_step_rollout.yaml

@@ -15,6 +15,7 @@
 # limitations under the License.
 
 _target_: rollout.MeshGraphNetAutoregressiveRolloutTraining
+_convert_: all
 
 input_dim_nodes: 7                 # pos(3) + vel(3) + thickness(1)
 input_dim_edges: 4                 # dx, dy, dz, distance

examples/structural_mechanics/crash/conf/model/mgn_time_conditional.yaml

@@ -15,6 +15,7 @@
 # limitations under the License.
 
 _target_: rollout.MeshGraphNetTimeConditionalRollout
+_convert_: all
 
 input_dim_nodes: 5                  # pos(3) + thickness(1) + time(1)
 input_dim_edges: 4                  # dx, dy, dz, distance

examples/structural_mechanics/crash/conf/model/transolver_autoregressive_rollout_training.yaml

@@ -15,6 +15,7 @@
 # limitations under the License.
 
 _target_: rollout.TransolverAutoregressiveRolloutTraining
+_convert_: all
 
 functional_dim: 5
 embedding_dim: 3

examples/structural_mechanics/crash/conf/model/transolver_one_step_rollout.yaml

@@ -15,6 +15,7 @@
 # limitations under the License.
 
 _target_: rollout.TransolverOneStepRollout
+_convert_: all
 
 functional_dim: 4
 embedding_dim: 3

examples/structural_mechanics/crash/conf/model/transolver_time_conditional.yaml

@@ -15,6 +15,7 @@
 # limitations under the License.
 
 _target_: rollout.TransolverTimeConditionalRollout
+_convert_: all
 
 functional_dim: 1                # thickness
 embedding_dim: 3                 # position

examples/structural_mechanics/crash/conf/reader/d3plot.yaml

@@ -0,0 +1,20 @@
+# SPDX-FileCopyrightText: Copyright (c) 2023 - 2025 NVIDIA CORPORATION & AFFILIATES.
+# SPDX-FileCopyrightText: All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+_target_: d3plot_reader.Reader
+_convert_: all
+wall_node_disp_threshold: 1.0
+

examples/structural_mechanics/crash/conf/reader/vtp.yaml

@@ -0,0 +1,18 @@
+# SPDX-FileCopyrightText: Copyright (c) 2023 - 2025 NVIDIA CORPORATION & AFFILIATES.
+# SPDX-FileCopyrightText: All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+_target_: vtp_reader.Reader
+_convert_: all

examples/structural_mechanics/crash/d3plot_reader.py

@@ -19,10 +19,10 @@
 import pyvista as pv
 
 from lasso.dyna import D3plot, ArrayType
-from typing import Dict, List, Optional
+from typing import Optional
 
 
-def find_run_folders(base_data_dir: str) -> List[str]:
+def find_run_folders(base_data_dir: str) -> list[str]:
     """
     Find run directories containing LS-DYNA d3plot files.
 
@@ -43,7 +43,7 @@ def find_run_folders(base_data_dir: str) -> List[str]:
     return run_dirs
 
 
-def parse_k_file(k_file_path: str) -> Dict[int, float]:
+def parse_k_file(k_file_path: str) -> dict[int, float]:
     """
     Parse LS-DYNA keyword (.k) file to extract part thickness values.
 
@@ -53,8 +53,8 @@ def parse_k_file(k_file_path: str) -> Dict[int, float]:
     Returns:
         Dictionary mapping part ID -> thickness.
     """
-    part_to_section: Dict[int, int] = {}
-    section_thickness: Dict[int, float] = {}
+    part_to_section: dict[int, int] = {}
+    section_thickness: dict[int, float] = {}
 
     with open(k_file_path, "r") as f:
         lines = [
@@ -187,7 +187,7 @@ def build_edges_from_mesh_connectivity(mesh_connectivity) -> set:
 def compute_node_thickness(
     mesh_connectivity,
     part_ids,
-    part_thickness_map: Dict[int, float],
+    part_thickness_map: dict[int, float],
     actual_part_ids: Optional[np.ndarray] = None,
 ) -> np.ndarray:
     """
@@ -404,8 +404,34 @@ def process_d3plot_data(
             write_vtp,
             logger,
         )
-        point_data_all.append(
-            {"mesh_pos": mesh_pos_all, "thickness": filtered_thickness}
-        )
+        point_data_all.append({"coords": mesh_pos_all, "thickness": filtered_thickness})
 
     return srcs, dsts, point_data_all
+
+
+class Reader:
+    """
+    Reader for LS-DYNA d3plot files.
+
+    Args:
+        wall_node_disp_threshold: threshold for filtering wall nodes
+    """
+
+    def __init__(self, wall_node_disp_threshold: float = 1.0):
+        self.wall_node_disp_threshold = wall_node_disp_threshold
+
+    def __call__(
+        self,
+        data_dir: str,
+        num_samples: int,
+        split: str,
+        logger=None,
+    ):
+        write_vtp = False if split == "train" else True
+        return process_d3plot_data(
+            data_dir=data_dir,
+            num_samples=num_samples,
+            wall_node_disp_threshold=self.wall_node_disp_threshold,
+            write_vtp=write_vtp,
+            logger=logger,
+        )