Refactoring, use odm_version(), add seconds

opendm/log.py

@@ -6,6 +6,7 @@ import datetime
 import dateutil.parser
 import shutil
 import multiprocessing
+from repoze.lru import lru_cache
 
 from opendm.arghelpers import double_quote, args_to_dict
 from vmem import virtual_memory
@@ -30,6 +31,7 @@ else:
 
 lock = threading.Lock()
 
+@lru_cache(maxsize=None)
 def odm_version():
     with open(os.path.join(os.path.dirname(__file__), "..", "VERSION")) as f:
         return f.read().split("\n")[0].strip()

opendm/photo.py

@@ -21,6 +21,17 @@ from opensfm.geo import ecef_from_lla
 
 projections = ['perspective', 'fisheye', 'fisheye_opencv', 'brown', 'dual', 'equirectangular', 'spherical']
 
+def find_mean_utc_time(photos):
+    utc_times = []
+    for p in photos: 
+        if p.utc_time is not None:
+            utc_times.append(p.utc_time / 1000.0)
+    if len(utc_times) == 0:
+        return None
+    
+    return np.mean(utc_times)
+
+
 def find_largest_photo_dims(photos):
     max_mp = 0
     max_dims = None

stages/odm_postprocess.py

@@ -8,7 +8,7 @@ from osgeo import gdal
 from opendm import io
 from opendm import log
 from opendm import types
-from opendm import context
+from opendm import photo
 from opendm.utils import copy_paths, get_processing_results_paths
 from opendm.ogctiles import build_3dtiles
 
@@ -19,56 +19,25 @@ class ODMPostProcess(types.ODM_Stage):
 
         log.ODM_INFO("Post Processing")
 
-        # -- TIFFTAG information - add datetime and software version to .tif metadata 
-
-        # Gather information 
-        # ODM version ...
-        with open(os.path.join(context.root_path, 'VERSION')) as version_file:
-            version = version_file.read().strip()
-        # Datetimes  ...
-        shots_file = tree.path("odm_report", "shots.geojson")
-        if not args.skip_report and os.path.isfile(shots_file): 
-            # Open file
-            with open(shots_file, 'r') as f:
-                odm_shots = json.loads(f.read())
-            # Compute mean time 
-            cts = []
-            for feat in odm_shots["features"]:
-                ct = feat["properties"]["capture_time"]
-                cts.append(ct)
-            mean_dt = datetime.fromtimestamp(np.mean(cts)) 
-            # Format it
-            CAPTURE_DATETIME = mean_dt.strftime('%Y:%m:%d %H:%M') + '+00:00' #UTC
-        else:
-            #try instead with images.json file
-            images_file = tree.path('images.json') 
-            if os.path.isfile(images_file): 
-                # Open file
-                with open(images_file, 'r') as f:
-                    imgs = json.loads(f.read())
-                # Compute mean time 
-                cts = []
-                for img in imgs: 
-                    ct = img["utc_time"]/1000. #ms to s
-                    cts.append(ct) 
-                mean_dt = datetime.fromtimestamp(np.mean(cts)) 
-                # Format it
-                CAPTURE_DATETIME = mean_dt.strftime('%Y:%m:%d %H:%M') + '+00:00' #UTC
-            else:
-                CAPTURE_DATETIME = None
-
-        # Add it 
-        for product in [tree.odm_orthophoto_tif,
+        rasters = [tree.odm_orthophoto_tif,
                     tree.path("odm_dem", "dsm.tif"),
-                        tree.path("odm_dem", "dtm.tif")]:
-            for pdt in [product, product.replace('.tif', '.original.tif')]:
-                if os.path.isfile(pdt):
-                    log.ODM_INFO("Adding TIFFTAGs to {} ...".format(pdt))
-                    with rasterio.open(pdt, 'r+') as rst:
-                        rst.update_tags(TIFFTAG_DATETIME=CAPTURE_DATETIME)
-                        rst.update_tags(TIFFTAG_SOFTWARE='OpenDroneMap {}'.format(version))
+                    tree.path("odm_dem", "dtm.tif")]
 
-        #  -- GCP info
+        mean_capture_time = photo.find_mean_utc_time(reconstruction.photos)
+        mean_capture_dt = None
+        if mean_capture_time is not None:
+            mean_capture_dt = datetime.fromtimestamp(mean_capture_time).strftime('%Y:%m:%d %H:%M:%S') + '+00:00'
+
+        # Add TIFF tags
+        for product in rasters:
+            if os.path.isfile(product):
+                log.ODM_INFO("Adding TIFFTAGs to {}".format(product))
+                with rasterio.open(product, 'r+') as rst:
+                    if mean_capture_dt is not None:
+                        rst.update_tags(TIFFTAG_DATETIME=mean_capture_dt)
+                    rst.update_tags(TIFFTAG_SOFTWARE='ODM {}'.format(log.odm_version()))
+
+        # GCP info
         if not outputs['large']:
             # TODO: support for split-merge?
 
@@ -83,9 +52,7 @@ class ODMPostProcess(types.ODM_Stage):
                 with open(gcp_gml_export_file) as f:
                     gcp_xml = f.read()
 
-                for product in [tree.odm_orthophoto_tif,
-                                tree.path("odm_dem", "dsm.tif"),
-                                tree.path("odm_dem", "dtm.tif")]:
+                for product in rasters:
                     if os.path.isfile(product):
                         ds = gdal.Open(product)
                         if ds is not None:
@@ -100,11 +67,9 @@ class ODMPostProcess(types.ODM_Stage):
                         else:
                             log.ODM_WARNING("Cannot open %s for writing, skipping GCP embedding" % product)
 
-        # -- 3D tiles
         if getattr(args, '3d_tiles'):
             build_3dtiles(args, tree, reconstruction, self.rerun())
 
-        # -- Copy to 
         if args.copy_to:
             try:
                 copy_paths([os.path.join(args.project_path, p) for p in get_processing_results_paths()], args.copy_to, self.rerun())