import random
import math
import time

from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants

from selective_learning1.payoffmatrix import PayoffMatrix

class TimerPage(Page):
    def is_displayed(self):
        return self.round_number == 1

    def vars_for_template(self):
        return{"poor_performance": self.player.participant.vars["poor_performance"]}

    def before_next_page(self):
        self.participant.vars['expiry'] = time.time() + Constants.time_limit*60

class InstructionTwoPage(Page):
    def is_displayed(self):
        return (self.participant.vars['expiry']-time.time()>3) and (self.player.participant.vars["poor_performance"] == False)

    def vars_for_template(self):
        return{ 'totalpayoff': self.participant.payoff}

    def before_next_page(self):
        solution_list=[-1]*PayoffMatrix.N
        row_solution_list =[-1]*Constants.num_rows
        col_solution_list =[-1]*Constants.num_rows
        problem_type = [-1]*(Constants.num_cells + 2*Constants.num_rows)
        row_options= [i for i in range (0, Constants.num_rows+2) ]
        column_options=[i for i in range (0, Constants.num_rows+2) ]
        row_choices = [0]*(Constants.num_choices)
        column_choices = [0]*(Constants.num_choices)
        choice_list = [0]*(Constants.num_cells + 1)
        row_choice_list = [0]*Constants.num_rows
        column_choice_list = [0]*Constants.num_rows
        problem_no = ''
        treatment_type = 0
        if self.player.treat_group == 1 :
            treatment_random = random.randint(1,4)
            if (treatment_random == 1) & (self.player.cell_reveal_round <= Constants.cell_reveal_limit) & (self.round_number > 2):
                treatment_type = 1
                self.player.treat_type = 1
                self.player.cell_reveal_round = self.player.cell_reveal_round + 1
            elif (treatment_random == 2) & (self.player.both_reveal_round <= Constants.both_reveal_limit) & (self.round_number >2):
                treatment_type = 3
                self.player.treat_type = 3
                self.player.both_reveal_round = self.player.both_reveal_round + 1
        elif self.player.treat_group == 2 :
            treatment_random = random.randint(1,4)
            if (treatment_random == 1) & (self.player.avg_reveal_round <= Constants.avg_reveal_limit) & (self.round_number > 2):
                treatment_type = 2
                self.player.treat_type = 2
                self.player.avg_reveal_round = self.player.avg_reveal_round + 1
            elif (treatment_random == 2) & (self.player.both_reveal_round <= Constants.both_reveal_limit) & (self.round_number > 2):
                treatment_type = 3
                self.player.treat_type = 3
                self.player.both_reveal_round = self.player.both_reveal_round + 1
        self.player.participant.vars.update(PayoffMatrix.PayoffList(self))
        self.player.participant.vars['solution_list'] = solution_list
        self.player.participant.vars['row_solution_list'] = row_solution_list
        self.player.participant.vars['col_solution_list'] = col_solution_list
        self.player.participant.vars['problem_type'] = problem_type
        self.player.participant.vars['row_options'] = row_options
        self.player.participant.vars['column_options'] = column_options
        self.player.participant.vars['row_choices'] = row_choices
        self.player.participant.vars['column_choices'] = column_choices
        self.player.participant.vars['choice_list'] = choice_list
        self.player.participant.vars['row_choice_list'] = row_choice_list
        self.player.participant.vars['column_choice_list'] = column_choice_list
        self.player.participant.vars['problem_no'] = problem_no
        self.player.participant.vars['treatment_type'] = treatment_type




    timer_text = 'Time left to finish this round:'

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()


class PriorPage(Page):
    def vars_for_template(self):
        return{'prior_type': self.player.participant.vars['prior_type'],
                'zeros': self.player.participant.vars['zeros']}

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

    def is_displayed(self):
        return (self.participant.vars['expiry']-time.time()>3) and (self.player.participant.vars["poor_performance"] == False)

    def before_next_page(self):
        self.player.prior = self.player.participant.vars['prior_type']
        self.player.zeroes = self.player.participant.vars['zeros']

class TreatmentTypePage(Page):
    def is_displayed(self):
        return self.player.cont_learning and (self.player.participant.vars['expiry'] - time.time()>3) & (self.player.treat_type != 0) and (self.player.participant.vars["poor_performance"] == False)

    def vars_for_template(self):
        return{'treat_type': self.player.participant.vars["treatment_type"]}

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

class SelectProblemPage(Page):
    form_model = 'player'
    form_fields = ['problem','row_problem','col_problem', 'no_learning']

    def vars_for_template(self):
        if self.player.no_learning == 0:
            self.player.tot_prob += 1
        return {'zeros': self.player.participant.vars['zeros'],
            'cutoff': self.player.participant.vars['random_cutoff']}

    def is_displayed(self):
            return self.player.cont_learning and (self.participant.vars['expiry'] - time.time()>3) and (self.player.no_learning != 1) and (self.player.participant.vars["poor_performance"] == False)

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()



class ChooseToSolvePage(Page):
    form_model ='player'
    form_fields = ['solution', 'reveal_cell', 'reveal_avg', 'reveal_both']

    def is_displayed(self):
        return self.player.cont_learning and (self.participant.vars['expiry'] - time.time()>3) and (self.player.no_learning != 1) and (self.player.participant.vars["poor_performance"] == False)

    def vars_for_template(self):
        pid = self.player.problem
        pid_row = self.player.row_problem
        pid_col = self.player.col_problem
        if pid != None:
            pid_row= -1
            pid_col= -1
            self.player.participant.vars['problem_no'] = "P"+str(self.player.problem + 1)
            self.player.participant.vars['solution_list'][pid] = self.player.solution
            self.player.participant.vars['problem_type'][self.player.tot_prob-1] = 1
            self.player.cell_open = self.player.cell_open + 1
        elif pid_row != None:
            pid = -1
            pid_col= -1
            self.player.participant.vars['problem_no'] = "C"+str(self.player.row_problem + 1)
            self.player.participant.vars['row_solution_list'][pid_row] = self.player.solution
            self.player.participant.vars['problem_type'][self.player.tot_prob-1] = 2
            self.player.avg_open = self.player.avg_open + 1
        elif pid_col != None:
            pid = -1
            pid_row= -1
            self.player.participant.vars['problem_no'] = "R"+str(self.player.col_problem + 1)
            self.player.participant.vars['col_solution_list'][pid_col] = self.player.solution
            self.player.participant.vars['problem_type'][self.player.tot_prob-1] = 2
            self.player.avg_open = self.player.avg_open + 1
        return {'prob' : self.player.participant.vars['problem_list'][pid],
                'row_avg': self.player.participant.vars['row_average'][pid_row],
                'col_avg': self.player.participant.vars['column_average'][pid_col],
                'prob_type':self.player.participant.vars['problem_type'],
                'rand_cutoff': self.player.participant.vars['random_cutoff'],
                'zeros': self.player.participant.vars['zeros'],
                'problem_no': self.player.participant.vars['problem_no'],
                'treatment_type': self.player.participant.vars["treatment_type"]}

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

    def before_next_page(self):
        self.player.reveal_cell_total = self.player.reveal_cell_total + self.player.reveal_cell
        self.player.reveal_avg_total = self.player.reveal_avg_total + self.player.reveal_avg
        self.player.reveal_both_total = self.player.reveal_both_total + self.player.reveal_both
        if self.player.tot_prob > 0 :
            self.player.info_first = self.player.participant.vars["problem_type"][0]
            self.player.info_last = self.player.participant.vars["problem_type"][self.player.tot_prob-1]
        if self.player.tot_prob > 1:
            if (self.player.participant.vars["problem_type"][self.player.tot_prob-1] == 2) & (self.player.participant.vars["problem_type"][self.player.tot_prob-2] == 1) :
                self.player.ca_switch += 1
            elif (self.player.participant.vars["problem_type"][self.player.tot_prob-1] == 1) & (self.player.participant.vars["problem_type"][self.player.tot_prob-2] == 2) :
                self.player.ac_switch += 1


class CellOnlyRevealPage(Page):
    form_model ='player'
    form_fields = ['solution']

    def is_displayed(self):
        return self.player.cont_learning and (self.participant.vars['expiry'] - time.time()>3) and (self.player.reveal_cell == 1) and (self.player.reveal_cell_total <= Constants.cell_reveal) and (self.player.participant.vars["poor_performance"] == False)

    def vars_for_template(self):

        return{ 'problem_no': self.player.participant.vars['problem_no'],
                'rand_cutoff': self.player.participant.vars['random_cutoff'],
                'check_payoff': self.player.participant.vars['check_payoff'][self.player.problem],
                'payoff_cell': self.player.participant.vars['payoff_matrix'][self.player.problem//Constants.num_rows][self.player.problem % Constants.num_rows]}

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

    def before_next_page(self):
        self.player.reveal_cell = 0

class AvgOnlyRevealPage(Page):
    form_model ='player'
    form_fields = ['solution']

    def is_displayed(self):
        return self.player.cont_learning and (self.participant.vars['expiry'] - time.time()>3) and (self.player.reveal_avg == 1) and (self.player.reveal_avg_total <= Constants.avg_reveal) and (self.player.participant.vars["poor_performance"] == False)

    def vars_for_template(self):
        if self.player.row_problem != None:
            one_row_index = self.player.row_problem
        else:
            one_row_index = -1
        if self.player.col_problem != None:
            one_col_index = self.player.col_problem
        else:
            one_col_index = -1
        return{ 'problem_no': self.player.participant.vars['problem_no'],
                'one_row': self.player.participant.vars['ones_row'][one_row_index],
                'one_col': self.player.participant.vars['ones_column'][one_col_index]
                }

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

    def before_next_page(self):
        self.player.reveal_avg = 0

class BothRevealPage(Page):
    form_model ='player'
    form_fields = ['solution']

    def is_displayed(self):
        return self.player.cont_learning and (self.participant.vars['expiry'] - time.time()>3) and (self.player.reveal_both == 1) and (self.player.reveal_both_total <= Constants.both_reveal) and (self.player.participant.vars["poor_performance"] == False)

    def vars_for_template(self):
        if self.player.problem != None:
            prob_index = self.player.problem
        else :
            prob_index = 0
        if self.player.row_problem != None:
            one_row_index = self.player.row_problem
        else:
            one_row_index = -1
        if self.player.col_problem != None:
            one_col_index = self.player.col_problem
        else:
            one_col_index = -1
        return{ 'problem_no': self.player.participant.vars['problem_no'],
                'rand_cutoff': self.player.participant.vars['random_cutoff'],
                'payoff_cell': self.player.participant.vars['payoff_matrix'][prob_index//Constants.num_rows][prob_index % Constants.num_rows],
                'one_row': self.player.participant.vars['ones_row'][one_row_index],
                'one_col': self.player.participant.vars['ones_column'][one_col_index]
                }


    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

    def before_next_page(self):
        if self.player.problem != None:
            self.player.reveal_cell_total_in_both = self.player.reveal_cell_total_in_both + self.player.reveal_both
        elif self.player.row_problem != None:
            self.player.reveal_avg_total_in_both = self.player.reveal_avg_total_in_both + self.player.reveal_both
        elif self.player.col_problem != None:
            self.player.reveal_avg_total_in_both = self.player.reveal_avg_total_in_both + self.player.reveal_both

        self.player.reveal_both = 0

class ContinueLearningPage(Page):
    form_model ='player'
    form_fields = ['cont_learning']

    def is_displayed(self):
        return self.player.cont_learning and (self.participant.vars['expiry']-time.time()>3) and (self.player.no_learning != 1) and (self.player.participant.vars["poor_performance"] == False)

    def vars_for_template(self):
        pid = self.player.problem
        pid_row = self.player.row_problem
        pid_col = self.player.col_problem
        if pid != None:
            pid_row= -1
            pid_col= -1
            self.player.participant.vars['solution_list'][pid] = self.player.solution
        elif pid_row != None:
            pid = -1
            pid_col= -1
            self.player.participant.vars['row_solution_list'][pid_row] = self.player.solution
        elif pid_col != None:
            pid = -1
            pid_row= -1
            self.player.participant.vars['col_solution_list'][pid_col] = self.player.solution
        return{'sol_list' : self.player.participant.vars['solution_list'],
                'row_sol_list': self.player.participant.vars['row_solution_list'],
                'col_sol_list':self.player.participant.vars['col_solution_list']}

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

class FinalChoiceOnePage(Page):
    form_model = 'player'
    form_fields = ['row_choice', 'column_choice']

    def row_choice_choices(self):
        choices = [[1,'Square'],
                [2,'Triangle'],
                [3, 'Circle'],
                [4, 'Diamond'],
                [5, 'Pentagon'],
                [6, 'Spade'],
                [7, 'Heart'],
                [8, 'Rectangle'],
                [9, 'Ellipse'],
                [0, 'Average Row']]
        # random.shuffle(choices)
        return choices

    def column_choice_choices(self):
        choices=[[1, 'Red'],
                [2, 'Blue'],
                [3, 'Orange'],
                [4, 'Purple'],
                [5, 'Pink'],
                [6, 'Green'],
                [7, 'Gray'],
                [8, 'Brown'],
                [9, 'Violet'],
                [0, 'Average Column'] ]
        # random.shuffle(choices)
        return choices

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

    def is_displayed(self):
        return (self.participant.vars['expiry']-time.time()>3) and (self.player.participant.vars["poor_performance"] == False)

    def before_next_page(self):
        self.player.choice_counter += 1


class FinalChoiceTwoPage(Page):
    def vars_for_template(self):
        rid = self.player.row_choice
        cid = self.player.column_choice
        if self.player.choice_counter == 1:
             self.player.row_choice_1 = self.player.row_choice
             self.player.col_choice_1 = self.player.column_choice
        elif self.player.choice_counter == 2:
            self.player.row_choice_2 = self.player.row_choice
            self.player.col_choice_2 = self.player.column_choice
        elif self.player.choice_counter == 3:
            self.player.row_choice_3 = self.player.row_choice
            self.player.col_choice_3 = self.player.column_choice
        elif self.player.choice_counter == 4:
            self.player.row_choice_4 = self.player.row_choice
            self.player.col_choice_4 = self.player.column_choice
        elif self.player.choice_counter == 5:
            self.player.row_choice_5 = self.player.row_choice
            self.player.col_choice_5 = self.player.column_choice
        elif self.player.choice_counter == 6:
            self.player.row_choice_6 = self.player.row_choice
            self.player.col_choice_6 = self.player.column_choice
        elif self.player.choice_counter == 7:
            self.player.row_choice_7 = self.player.row_choice
            self.player.col_choice_7 = self.player.column_choice
        elif self.player.choice_counter == 8:
            self.player.row_choice_8 = self.player.row_choice
            self.player.col_choice_8 = self.player.column_choice
        elif self.player.choice_counter == 9:
            self.player.row_choice_9 = self.player.row_choice
            self.player.col_choice_9 = self.player.column_choice
        if rid != 0 and cid != 0:
            self.player.participant.vars['choice_list'][(rid-1)*Constants.num_rows + (cid-1)] = 1
        elif rid == 0 and cid != 0:
            self.player.participant.vars['column_choice_list'][cid-1] = 1
        elif rid!=0 and cid == 0:
            self.player.participant.vars['row_choice_list'][rid-1] = 1
        else:
            self.player.participant.vars['choice_list'][-1] = 1
            if self.player.choice_counter == 1:
                self.player.payoff_choice_1 = 0
            elif self.player.choice_counter == 2:
                self.player.payoff_choice_2 = 0
            elif self.player.choice_counter == 3:
                self.player.payoff_choice_3 = 0
            elif self.player.choice_counter == 3:
                self.player.payoff_choice_3 = 0
            elif self.player.choice_counter == 4:
                self.player.payoff_choice_4 = 0
            elif self.player.choice_counter == 5:
                self.player.payoff_choice_5 = 0
            elif self.player.choice_counter == 6:
                self.player.payoff_choice_6 = 0
            elif self.player.choice_counter == 7:
                self.player.payoff_choice_7 = 0
            elif self.player.choice_counter == 8:
                self.player.payoff_choice_8 = 0
            elif self.player.choice_counter == 9:
                self.player.payoff_choice_9 = 0
        for i in range(Constants.num_cells):
            if self.player.participant.vars['choice_list'][i] == 1:
                if self.player.choice_counter == 1:
                    self.player.payoff_choice_1 = self.player.participant.vars['check_payoff'][i]
                elif self.player.choice_counter == 2:
                    self.player.payoff_choice_2 = self.player.participant.vars['check_payoff'][i]
                elif self.player.choice_counter == 3:
                    self.player.payoff_choice_3 = self.player.participant.vars['check_payoff'][i]
                elif self.player.choice_counter == 4:
                    self.player.payoff_choice_4 = self.player.participant.vars['check_payoff'][i]
                elif self.player.choice_counter == 5:
                    self.player.payoff_choice_5 = self.player.participant.vars['check_payoff'][i]
                elif self.player.choice_counter == 6:
                    self.player.payoff_choice_6 = self.player.participant.vars['check_payoff'][i]
                elif self.player.choice_counter == 7:
                    self.player.payoff_choice_7 = self.player.participant.vars['check_payoff'][i]
                elif self.player.choice_counter == 8:
                    self.player.payoff_choice_8 = self.player.participant.vars['check_payoff'][i]
                elif self.player.choice_counter == 9:
                    self.player.payoff_choice_9 = self.player.participant.vars['check_payoff'][i]
        for i in range(Constants.num_rows):
            if  self.player.participant.vars['row_choice_list'][i] == 1:
                payoff_row = self.player.participant.vars['check_payoff'][i*Constants.num_rows : (i+1)*Constants.num_rows]
                rand_row= random.randint(1, len(payoff_row))
                if self.player.choice_counter == 1:
                    self.player.payoff_choice_1 = payoff_row[rand_row-1]
                elif self.player.choice_counter == 2:
                    self.player.payoff_choice_2 = payoff_row[rand_row-1]
                elif self.player.choice_counter == 3:
                    self.player.payoff_choice_3 = payoff_row[rand_row-1]
                elif self.player.choice_counter == 4:
                    self.player.payoff_choice_4 = payoff_row[rand_row-1]
                elif self.player.choice_counter == 5:
                    self.player.payoff_choice_5 = payoff_row[rand_row-1]
                elif self.player.choice_counter == 6:
                    self.player.payoff_choice_6 = payoff_row[rand_row-1]
                elif self.player.choice_counter == 7:
                    self.player.payoff_choice_7 = payoff_row[rand_row-1]
                elif self.player.choice_counter == 8:
                    self.player.payoff_choice_8 = payoff_row[rand_row-1]
                elif self.player.choice_counter == 9:
                    self.player.payoff_choice_9 = payoff_row[rand_row-1]
        for i in range(Constants.num_rows):
            if self.player.participant.vars['column_choice_list'][i] == 1:
                payoff_col = self.player.participant.vars['check_payoff'][i: Constants.num_rows*(Constants.num_rows -1) + i+1: Constants.num_rows]
                rand_col = random.randint(1, len(payoff_col))
                if self.player.choice_counter == 1:
                    self.player.payoff_choice_1 = payoff_col[rand_col-1]
                elif self.player.choice_counter == 2:
                    self.player.payoff_choice_2 = payoff_col[rand_col-1]
                elif self.player.choice_counter == 3:
                    self.player.payoff_choice_3 = payoff_col[rand_col-1]
                elif self.player.choice_counter == 4:
                    self.player.payoff_choice_4 = payoff_col[rand_col-1]
                elif self.player.choice_counter == 5:
                    self.player.payoff_choice_5 = payoff_col[rand_col-1]
                elif self.player.choice_counter == 6:
                    self.player.payoff_choice_6 = payoff_col[rand_col-1]
                elif self.player.choice_counter == 7:
                    self.player.payoff_choice_7 = payoff_col[rand_col-1]
                elif self.player.choice_counter == 8:
                    self.player.payoff_choice_8 = payoff_col[rand_col-1]
                elif self.player.choice_counter == 9:
                    self.player.payoff_choice_9 = payoff_col[rand_col-1]
        if rid == 0:
            self.player.participant.vars['row_choices'][self.player.choice_counter-1] = "average row"
        else:
            self.player.participant.vars['row_choices'][self.player.choice_counter-1] = 'row '+ str(self.player.participant.vars['row_options'][rid])
        if cid == 0:
            self.player.participant.vars['column_choices'][self.player.choice_counter-1] = 'average column'
        else:
            self.player.participant.vars['column_choices'][self.player.choice_counter-1] = 'column '+ str(self.player.participant.vars['column_options'][cid])

        return {'row_choices': self.player.participant.vars['row_choices'],
                'column_choices': self.player.participant.vars['column_choices'],
                'choice_list': self.player.participant.vars['choice_list'],
                'row_choice_list': self.player.participant.vars['row_choice_list'],
                'column_choice_list': self.player.participant.vars['column_choice_list'] }

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

    def is_displayed(self):
        return (self.participant.vars['expiry']-time.time()>3) and (self.player.participant.vars["poor_performance"] == False)

    def before_next_page(self):
        rand_choice = random.randint(1, Constants.num_choices)
        payoff_list = [self.player.payoff_choice_1, self.player.payoff_choice_2,self.player.payoff_choice_3, self.player.payoff_choice_4,self.player.payoff_choice_5,self.player.payoff_choice_6,self.player.payoff_choice_7,self.player.payoff_choice_8,self.player.payoff_choice_9]
        self.player.payoff = payoff_list[rand_choice-1]

class PayoffInfoPage(Page):
    def vars_for_template(self):
         payoff =  self.player.in_round(self.round_number).payoff
         return{'finalpayoff' : payoff}

    def get_timeout_seconds(self):
        return self.participant.vars['expiry'] - time.time()

    def is_displayed(self):
        return (self.participant.vars['expiry']-time.time()>3) and (self.player.participant.vars["poor_performance"] == False)




num_of_tries= PayoffMatrix.N + 2*PayoffMatrix.n
learning_page_sequence = [SelectProblemPage, ChooseToSolvePage,  CellOnlyRevealPage, AvgOnlyRevealPage, BothRevealPage, ContinueLearningPage]*num_of_tries
choice_page_sequence = [FinalChoiceOnePage, FinalChoiceTwoPage]*Constants.num_choices + [PayoffInfoPage]
page_sequence = [TimerPage, InstructionTwoPage, PriorPage, TreatmentTypePage] + learning_page_sequence + choice_page_sequence