contrasts.py

import numpy as np
import cv2 as cv

# TODO create generic class

class Contrast(object):
    """Base class for color contrasts"""
    def __init__(self, img):
        self._img = img
        

class LightDark(Contrast):
    """docstring for LightDark"""
    def __init__(self, img, method='luminance'):
        super(LightDark, self).__init__(img)

        meth = getattr(LightDark, method)
        self.ctrst = meth(self)

    def luminance(self):
        """Creates light/dark values using luminance quantifiers for RGB

           The array has the same dimensions as the image. However the third
           does only have the size 1 which contains the luminance value
        """
        # Luminance Faktoren nach http://introcs.cs.princeton.edu/python/31datatype/luminance.py.html
        luminance_factors = np.array([.114, .587, .299])

        # Erzeugung eines eindimensionalen Arrays für die effizientere Berechnung
        self._img = np.multiply(self._img, luminance_factors)

        # addiert alle Werte auf einer bestimmten Achse
        luminances = np.sum(self._img, axis=2)
        luminances = luminances.astype(np.uint8, copy=True)

        return luminances

    def value(self):
        """Creates light/dark values using the value channel in HSV"""
        img = cv.cvtColor(self._img, cv.COLOR_BGR2HSV)
        values = img[:, :, 2].copy()

        return values

    def lightness(self):
        """Creates light/dark values using the value channel in HSV"""
        img = cv.cvtColor(self._img, cv.COLOR_BGR2HLS)
        lightness = img[:, :, 1].copy()

        return lightness


class Saturation(Contrast):
    """docstring for Saturation"""
    def __init__(self, img, method='saturation'):
        super(Saturation, self).__init__(img)

        meth = getattr(Saturation, method)
        self.ctrst = meth(self)

    def saturation(self):
        """docstring for saturation"""
        img = cv.cvtColor(self._img, cv.COLOR_BGR2HSV)
        saturations = img[:, :, 1].copy()

        return saturations

    def chroma(self):
        """docstring for chroma"""
        pass


class Hue(Contrast):
    """docstring for Saturation"""
    def __init__(self, img, method='hue'):
        super(Hue, self).__init__(img)

        meth = getattr(Hue, method)
        self.ctrst = meth(self)

    def hue(self):
        """docstring for hue"""
        
        img = cv.cvtColor(self._img, cv.COLOR_BGR2HSV)
        hues = img[:, :, 0].copy()

        return hues