import csv
import math

from otree.api import *

doc = """
Social Value Orientation (Murphy et al)
"""


class Choice(ExtraModel):
    """only used as a helper for read_csv"""

    round_number = models.IntegerField()
    to_self = models.IntegerField()
    to_other = models.IntegerField()


def group_rows():
    from collections import defaultdict

    d = defaultdict(list)
    for row in read_csv(__name__ + '/g_SVO.csv', Choice):
        round_number = row['round_number']
        d[round_number].append(row)
    return d


class C(BaseConstants):
    NAME_IN_URL = 'g_svo'
    PLAYERS_PER_GROUP = None
    NUM_ROUNDS = 15  # change to 15 for the full task
    ROWS = group_rows()


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    g_svo_choice = models.IntegerField(choices=list(range(9)))
    angle = models.FloatField()
    category = models.StringField()


def assign_category(angle):
    if angle > 57.15:
        return 'Altruistic'
    if angle > 22.45:
        return 'Prosocial'
    if angle > -12.04:
        return 'Individualistic'
    return 'Competitive'


class Decide(Page):
    form_model = 'player'
    form_fields = ['g_svo_choice']

    @staticmethod
    def js_vars(player: Player):
        return dict(rows=C.ROWS[player.round_number])

    @staticmethod
    def before_next_page(player: Player, timeout_happened):
        participant = player.participant
        # remember to add 'progress' to PARTICIPANT_FIELDS.
        participant.progress += 1


class Results(Page):
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == C.NUM_ROUNDS

    @staticmethod
    def vars_for_template(player: Player):
        participant = player.participant

        to_self_total = 0
        to_other_total = 0
        for p in player.in_all_rounds():
            row = C.ROWS[p.round_number][p.g_svo_choice]
            to_self_total += row['to_self']
            to_other_total += row['to_other']

        avg_self = to_self_total / C.NUM_ROUNDS
        avg_other = to_other_total / C.NUM_ROUNDS
        radians = math.atan((avg_other - 50) / (avg_self - 50))
        participant.svo_angle = round(math.degrees(radians), 2)
        participant.svo_category = assign_category(participant.svo_angle)

    @staticmethod
    def before_next_page(player: Player, timeout_happened):
        participant = player.participant
        # remember to add 'progress' to PARTICIPANT_FIELDS.
        participant.progress += 1

page_sequence = [Decide, Results]