from typing import List
import random


class LotterySpecification:
    def __init__(self, alpha: int, beta: int, c: int, epsilon: int):
        assert(epsilon <= beta)
        self.alpha = alpha
        self.beta = beta
        self.epsilon = epsilon
        self.c = c


class Lottery:
    def __init__(self, lottery_specification: LotterySpecification, treatment: str, with_signal=True):
        self.alpha = lottery_specification.alpha
        self.beta = lottery_specification.beta
        self.epsilon = lottery_specification.epsilon
        self.c = lottery_specification.c

        self.random_value = random.randint(0, 100)
        if treatment == "cv":
            self.p = self.c
            self.value = random.randint(self.alpha, self.beta)

            self.signal = random.randint(self.value - self.epsilon, self.value + self.epsilon)

            if self.random_value <= self.c:
                self.outcome = self.value
            else:
                self.outcome = 0
        # cp
        else:
            if with_signal:
                self.p = random.randint(self.alpha, self.beta)
            else:
                self.p = int((self.alpha + self.beta)/2)
            self.value = self.c

            self.signal = random.randint(self.p - self.epsilon, self.p + self.epsilon)

            if self.random_value <= self.p:
                self.outcome = self.c
            else:
                self.outcome = 0



class RedBlueLottery:
    HIGH = 0
    LOW = 1
    VALUE = 0
    NUMBER = 1
    ALL_KNOWN = 1
    COMPOUND_RISK = 2
    AMBIGUITY = 3
    BET_HIGH_RED = 0
    BET_HIGH_BLUE = 1

    def __init__(self, lid: int, ltype: int, total: int, matrix: List, min_cutoff: int, max_cutoff: int):
        self.lid = lid
        self.ltype = ltype
        self.total = total
        self.matrix = matrix
        self.max_cutoff = max_cutoff
        self.min_cutoff = min_cutoff

    @property
    def high_value(self):
        return self.matrix[RedBlueLottery.VALUE][RedBlueLottery.HIGH]

    @property
    def low_value(self):
        return self.matrix[RedBlueLottery.VALUE][RedBlueLottery.LOW]

    @property
    def number_red(self):
        assert self.ltype == RedBlueLottery.ALL_KNOWN or self.ltype == RedBlueLottery.AMBIGUITY
        return self.matrix[RedBlueLottery.NUMBER][RedBlueLottery.HIGH]

    @property
    def number_blue(self):
        assert self.ltype == RedBlueLottery.ALL_KNOWN
        return self.matrix[RedBlueLottery.NUMBER][RedBlueLottery.LOW]

    def has_number_blue(self):
        return self.ltype == RedBlueLottery.ALL_KNOWN

    def has_number_red(self):
        return self.ltype == RedBlueLottery.ALL_KNOWN or self.ltype == RedBlueLottery.AMBIGUITY

    def __str__(self):
        return 'RedBlueLottery ' % self.lid

    def __rpr__(self):
        return "RedBlueLottery: {}, Type: {}".format(self.lid, self.ltype)