summaryrefslogtreecommitdiffstats
path: root/src/curve.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/curve.py')
-rw-r--r--src/curve.py41
1 files changed, 29 insertions, 12 deletions
diff --git a/src/curve.py b/src/curve.py
index c69bdce..4ef52a4 100644
--- a/src/curve.py
+++ b/src/curve.py
@@ -298,24 +298,36 @@ def cie_brightness(level):
(r_curve[i], g_curve[i], b_curve[i]) = ciexyy_to_srgb(x, y, Y * level)
-def linearise():
+def linearise(r = True, g = None, b = None):
'''
Convert the curves from formatted in standard RGB to linear RGB
+
+ @param r:bool Whether to convert the red colour curve
+ @param g:bool? Whether to convert the green colour curve, defaults to `r` if `None`
+ @param b:bool? Whether to convert the blue colour curve, defaults to `r` if `None`
'''
+ if g is None: g = r
+ if b is None: b = r
for i in range(i_size):
- r, g, b = r_curve[i], g_curve[i], b_curve[i]
- (r, g, b) = standard_to_linear(r, g, b)
- r_curve[i], g_curve[i], b_curve[i] = r, g, b
+ sr, sg, sb = r_curve[i], g_curve[i], b_curve[i]
+ (lr, lg, lb) = standard_to_linear(sr, sg, sb)
+ r_curve[i], g_curve[i], b_curve[i] = (lr if r else sr), (lg if g else sg), (lb if b else sb)
-def standardise():
+def standardise(r = True, g = None, b = None):
'''
Convert the curves from formatted in linear RGB to standard RGB
+
+ @param r:bool Whether to convert the red colour curve
+ @param g:bool? Whether to convert the green colour curve, defaults to `r` if `None`
+ @param b:bool? Whether to convert the blue colour curve, defaults to `r` if `None`
'''
+ if g is None: g = r
+ if b is None: b = r
for i in range(i_size):
- r, g, b = r_curve[i], g_curve[i], b_curve[i]
- (r, g, b) = linear_to_standard(r, g, b)
- r_curve[i], g_curve[i], b_curve[i] = r, g, b
+ lr, lg, lb = r_curve[i], g_curve[i], b_curve[i]
+ (sr, sg, sb) = linear_to_standard(lr, lg, lb)
+ r_curve[i], g_curve[i], b_curve[i] = (sr if r else lr), (sg if g else lg), (sb if b else lb)
def gamma(r, g = None, b = None):
@@ -509,11 +521,16 @@ def start_over():
b_curve[i] = v
-def clip():
+def clip(r = True, g = None, b = None):
'''
Clip all values below the actual minimum and above actual maximums
+
+ @param r:bool Whether to clip the red colour curve
+ @param g:bool? Whether to clip the green colour curve, defaults to `r` if `None`
+ @param b:bool? Whether to clip the blue colour curve, defaults to `r` if `None`
'''
- for curve in (r_curve, g_curve, b_curve):
- for i in range(i_size):
- curve[i] = min(max(0.0, curve[i]), 1.0)
+ for curve, action in curves(r, r if g is None else g, r if b is None else b):
+ if action:
+ for i in range(i_size):
+ curve[i] = min(max(0.0, curve[i]), 1.0)