import tensorflow as tf

import math

class Simulator():
    def __init__(self, type) -> None:
        if type == 'D':
            # deuteranope
            self.color_matrix = tf.convert_to_tensor([[1, 0, 0], [0.494207, 0, 1.24827], [0, 0, 1]])
        elif type == 'P':
            # protanope
            self.color_matrix = tf.convert_to_tensor([[0, 2.02344, -2.52581], [0, 1, 0], [0, 0, 1]])
        elif type == 'T':
            # tritanope
