Add histograms

editor/utils/__init__.py

@@ -2,3 +2,4 @@ from .interpolate import interpolate
 from .random import random
 from .apply_pixel_shader import apply_pixel_shader
 from .get_colour_lut import get_colour_lut
+from .compute_histogram import compute_histogram

editor/utils/compute_histogram.py

@@ -0,0 +1,14 @@
+from PIL import Image
+import numpy as np
+
+
+def compute_histogram(image_path, bins: int, value_range=(0, 256)):
+    image = Image.open(image_path)
+    image = np.array(image)
+
+    histogram, _ = np.histogramdd(
+        image.reshape(-1, 3), bins=bins, range=[value_range, value_range, value_range]
+    )
+    histogram = histogram / np.sum(histogram)
+
+    return histogram.astype(np.float32)