Check and upgrade precision as needed

Ray9T · Jun 22, 2018 · 41e1ce9 · 41e1ce9
1 parent 3d2c8e1
commit 41e1ce9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/cr2_replace.py b/cr2_replace.py
@@ -69,6 +69,13 @@ def main():
 
    # read sensor image file
    sensor = jbimage.image_file.read(args.sensor).squeeze()
+   # determine precision
+   precision_sensor = jbimage.get_precision(sensor.max())
+   print "Sensor image required precision: %d bits" % precision_sensor
+   # decide if we need to upgrade encoding precision
+   if precision_sensor > precision:
+      print "Upgrading from precision: %d bits" % precision
+      precision = precision_sensor
 
    # check input image parameters
    assert len(sensor.shape) == 2 # must be a one-channel image