Add functionality to create photoscan object from absolute filepath info (#198)

Author: sadhana-r

src/openlifu/photoscan.py

@@ -87,6 +87,23 @@ def to_file(self, filename):
         with open(filename, 'w') as file:
             file.write(self.to_json(compact = False))
 
+    @staticmethod
+    def from_absolute_filepaths_info(photoscan_absolute_filepath_info: Dict) -> "Photoscan":
+        """
+        Create a photoscan from a photoscan dictionary containing absolute filepath information for
+        the affiliated data (keys: id, name, model_abspath, texture_abspath, photoscan_approved).
+        A photoscan info dictionary can be obtained from a Database using `get_photoscan_absolute_filepaths_info`.
+        """
+        photoscan_dict = {"id": photoscan_absolute_filepath_info["id"],\
+                    "name": photoscan_absolute_filepath_info["name"],\
+                    "model_filename": Path(photoscan_absolute_filepath_info["model_abspath"]).name,
+                    "texture_filename": Path(photoscan_absolute_filepath_info["texture_abspath"]).name,
+                    "photoscan_approved": photoscan_absolute_filepath_info["photoscan_approved"]}
+        if "mtl_abspath" in photoscan_absolute_filepath_info:
+            photoscan_dict["mtl_filename"] = Path(photoscan_absolute_filepath_info["mtl_abspath"]).name
+
+        return Photoscan.from_dict(photoscan_dict)
+
 def load_data_from_filepaths(model_abspath: str, texture_abspath: str) -> Tuple[vtk.vtkPolyData, vtk.vtkImageData]:
     """
     This function returns the data directly from the model and texture filepaths without requiring a photoscan object.

tests/test_photoscans.py

@@ -35,7 +35,7 @@ def test_load_photoscan(example_database:Database, tmp_path:Path):
     subject_id = "example_subject"
     session_id = "example_session"
     photoscan_id = "example_photoscan"
-    photoscan_info = example_database.get_photoscan_info(subject_id, session_id, photoscan_id)
+    photoscan_info = example_database.get_photoscan_absolute_filepaths_info(subject_id, session_id, photoscan_id)
     [model, texture] = load_data_from_filepaths(photoscan_info["model_abspath"], photoscan_info["texture_abspath"])
     assert model is not None
     assert texture is not None
@@ -51,6 +51,10 @@ def test_load_photoscan(example_database:Database, tmp_path:Path):
     assert isinstance(model, vtkPolyData)
     assert isinstance(texture,vtkImageData)
 
+    # From photoscan info
+    photoscan_from_info = Photoscan.from_absolute_filepaths_info(photoscan_info)
+    assert photoscan == photoscan_from_info
+
     bogus_file = Path(tmp_path/"test_db_files/bogus_photoscan.pdf")
     bogus_file.parent.mkdir(parents=True, exist_ok=True)
     bogus_file.touch()
@@ -64,6 +68,7 @@ def test_load_photoscan(example_database:Database, tmp_path:Path):
     with pytest.raises(FileNotFoundError, match="does not exist"):
         load_data_from_filepaths(bogus_file, photoscan_info["texture_abspath"])
 
+
 def test_convert_between_ras_and_lps():
 
     # Create example data