summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xutils/rkisp1/rkisp1_lsc_rpi_to_libcamera.py139
1 files changed, 139 insertions, 0 deletions
diff --git a/utils/rkisp1/rkisp1_lsc_rpi_to_libcamera.py b/utils/rkisp1/rkisp1_lsc_rpi_to_libcamera.py
new file mode 100755
index 00000000..e86a88ad
--- /dev/null
+++ b/utils/rkisp1/rkisp1_lsc_rpi_to_libcamera.py
@@ -0,0 +1,139 @@
+#!/usr/bin/env python3
+
+import argparse
+import json
+import numpy as np
+import scipy.ndimage
+import sys
+import yaml
+
+
+def update_yaml_config(config, reds, green, blues):
+ with open(config, 'r') as config:
+ data = yaml.safe_load(config)
+
+ xy_sizes = [0.0625, 0.0625, 0.0625,
+ 0.0625, 0.0625, 0.0625, 0.0625, 0.0625]
+
+ lsc_config = {}
+ lsc_config['x-size'] = []
+ lsc_config['y-size'] = []
+ for size in xy_sizes:
+ lsc_config['x-size'].append(size)
+ lsc_config['y-size'].append(size)
+ lsc_config['sets'] = []
+
+ for temperature in reds.keys():
+ red_table = reds[temperature].flatten()
+ blue_table = blues[temperature].flatten()
+
+ ct_set = {}
+ ct_set['ct'] = temperature
+ ct_set['r'] = []
+ ct_set['gr'] = []
+ ct_set['gb'] = []
+ ct_set['b'] = []
+
+ for f in red_table:
+ i = round(f * 1024)
+ ct_set['r'].append(i)
+
+ for f in green:
+ i = round(f * 1024)
+
+ ct_set['gr'].append(i)
+ ct_set['gb'].append(i)
+
+ for f in blue_table:
+ i = round(f * 1024)
+
+ ct_set['b'].append(i)
+
+ lsc_config['sets'].append(ct_set)
+
+ lsc = {'LensShadingCorrection': lsc_config}
+
+ # Remove any existing LensShadingCorrection section
+ algos = data['algorithms']
+ for algo in algos:
+ for k in algo.keys():
+ if k == 'LensShadingCorrection':
+ algos.remove(algo)
+ break
+
+ algos.append(lsc)
+ print(data)
+ return data
+
+
+def sort_by_temperature(calibrations):
+ data = {}
+ for entry in calibrations:
+ temp = entry['ct']
+ table = np.array(entry['table']).reshape(12, 16)
+
+ data[temp] = table
+
+ return data
+
+
+def parse_lsc_data(filename):
+ with open(filename, 'r') as tuning_file:
+ for algo in json.load(tuning_file)['algorithms']:
+ for (k, v) in algo.items():
+ if k == 'rpi.alsc':
+ return v
+
+ raise Exception("Cannot find LSC tuning data in file " + filename)
+
+
+def main():
+ parser = argparse.ArgumentParser()
+ parser.add_argument('-f', '--file', required=True,
+ help='Tuning file to parse')
+ parser.add_argument('-t', '--temperature', help='Color temperature')
+ # todo: Do not make this mandatory and default to stdio
+ parser.add_argument('-a', '--append', required=True,
+ help='RkISP configuration file to append LSC tables to')
+
+ args = parser.parse_args()
+
+ lsc_data = parse_lsc_data(args.file)
+
+ luminance_strength = lsc_data['luminance_strength']
+ luma = np.array(lsc_data['luminance_lut']).reshape(12, 16)
+ green = (luma - 1) * luminance_strength + 1
+
+ cr_data = sort_by_temperature(lsc_data['calibrations_Cr'])
+ cb_data = sort_by_temperature(lsc_data['calibrations_Cb'])
+
+ reds = {}
+ blues = {}
+
+ for temperature in cr_data.keys():
+ cr = cr_data[temperature]
+ cb = cb_data[temperature]
+
+ red = cr * green
+ red = scipy.ndimage.zoom(red, (17 / 12, 17 / 16))
+ red /= red.min()
+
+ blue = cb * green
+ blue = scipy.ndimage.zoom(blue, (17 / 12, 17 / 16))
+ blue /= blue.min()
+
+ reds[temperature] = red
+ blues[temperature] = blue
+
+ green = scipy.ndimage.zoom(green, (17 / 12, 17 / 16))
+ green /= green.min()
+
+ new_config = update_yaml_config(args.append, reds, green.flatten(), blues)
+ with open(args.append, 'w') as config:
+ yaml.safe_dump(new_config, config)
+
+ return 0
+
+
+if __name__ == '__main__':
+ sys.exit(main())