Simplify syntax to get data from PTDATA

disruption_py/inout/mds.py

@@ -194,6 +194,8 @@ def get_data(
         np.ndarray
             Returns the node data.
         """
+        if tree_name and tree_name.lower() == "ptdata":
+            path, tree_name = self._resolve_d3d_ptdata_path(path)
 
         if tree_name is not None:
             self.open_tree(tree_name)
@@ -238,6 +240,9 @@ def get_data_with_dims(
 
         dim_nums = dim_nums or [0]
 
+        if tree_name and tree_name.lower() == "ptdata":
+            path, tree_name = self._resolve_d3d_ptdata_path(path)
+
         if tree_name is not None:
             self.open_tree(tree_name)
 
@@ -284,6 +289,9 @@ def get_dims(
 
         dim_nums = dim_nums or [0]
 
+        if tree_name and tree_name.lower() == "ptdata":
+            path, tree_name = self._resolve_d3d_ptdata_path(path)
+
         if tree_name is not None:
             self.open_tree(tree_name)
 
@@ -320,3 +328,12 @@ def tree_name(self, for_name: str) -> str:
         The tree name for for_name, whether it is a nickname or tree name itself
         """
         return self.get_tree_name_of_nickname(for_name) or for_name
+
+    def _resolve_d3d_ptdata_path(self, path: str) -> Tuple[str, None]:
+        """
+        This function allows a user to specify tree = 'ptdata' in get methods.
+        It only works with simple expressions (i.e. no additional mathematical operations).
+
+        Return the ptdata path and tree=None
+        """
+        return [f'ptdata("{path}", {self.shot_id})', None]