from otree.api import *
from sqlalchemy import true
from numpy import random

c = Currency

doc = """
TREATMENTS HAVE BEEN MODIFIED:
1 - LINEAR
2 - SUSTAINABILITY CONVEX
3 - QUALITY CONVEX
"""


class Constants(BaseConstants):
    name_in_url = 'Infographics'
    players_per_group = None
    num_rounds = 1
    iRandomTreatment = 5 # Number for the treatment variable such that it randomizes between all conditions from Exp1 (which go from 1 to iRandomTreatment-1)
    Bonus = '5p (0.05 pounds)'
    # Image path
    sImagePath          = 'global/figures/'
    imgLeaf_symbol      = sImagePath+'one_leaf.png'
    imgStar_symbol      = sImagePath+'one_star.png'
    sPathQ_l            = sImagePath+'Infographic_graphs/qual_lin2.png'
    sPathQ_cv           = sImagePath+'Infographic_graphs/qual_concave2.png'
    sPathQ_cx           = sImagePath+'Infographic_graphs/qual_convex2.png'
    sPathS_l            = sImagePath+'Infographic_graphs/sus_linear2.png'
    sPathS_cv           = sImagePath+'Infographic_graphs/sus_concave2.png'
    sPathS_cx           = sImagePath+'Infographic_graphs/sus_convex2.png' 
    # Variables for Infographics
    Q1l,Q1h = 60, 70
    Q2l,Q2h = 70, 80
    Q3l,Q3h = 80, 90
    S1l, S1h = 0, 10
    S2l_l, S2h_l = 10, 20
    S2l_cv, S2h_cv = 15, 25
    S2l_cx, S2h_cx = 5, 15
    S3l, S3h  = 20, 30 
    BQ1,BQ2,BQ3 = 60,75,90
    BS1,BS2,BS3 = 0,15,30
    Scx,Qcx = 2.5,62.5
    Currency = 'Pounds'
    lAttrQ = dict(
        attr = 'Quality',
        attr_lower = 'quality',
        conversion = '10 points = 0.5 pound',
        min = '60',
        max = '90',
        explain = 'is worth 3 pounds and the best one is worth 4.5',
        symbolName = 'stars',
        symbolPath = imgStar_symbol,
        extra = '',
        symbol1Path = sImagePath+'star_1.png',
        symbol2Path = sImagePath+'star_2.png',
        symbol3Path = sImagePath+'star_3.png',
        graphPath   = sPathQ_l,
    )
    lAttrS = dict(
        attr = 'Sustainability',
        attr_lower = 'sustainability',
        conversion = '10 points = 1 tree planted',
        min = '0',
        max = '30',
        explain = 'will plant 0 trees and the best will plant 3 trees',
        symbolName = 'leaves',
        symbolPath = imgLeaf_symbol,
        extra = '<li> The total amount of points donated to the area you selected will be rounded up, so no point will be lost. For example, 102 points will mean 11 trees planted (instead of 10). </li>',
        symbol1Path = sImagePath+'leaf_1.png',
        symbol2Path = sImagePath+'leaf_2.png',
        symbol3Path = sImagePath+'leaf_3.png',
        graphPath   = '',
   )


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    sAttrOrder          = models.StringField()
    dRTbelief           = models.FloatField(blank=true)
    dRTinfographics     = models.FloatField(blank=true)
    iTreatment          = models.IntegerField()
    # Beliefs
    B01 = models.IntegerField()
    B02 = models.IntegerField()
    B03 = models.IntegerField()
    B11 = models.IntegerField()
    B12 = models.IntegerField()
    B13 = models.IntegerField()



# FUNCTIONS 
def creating_session(subsession):
    ## SETUP FOR PARTICIPANT
    for player in subsession.get_players():
        p, session = player.participant, subsession.session
        iTreatment = session.config['iTreatment']
        if (iTreatment!=Constants.iRandomTreatment):
            player.iTreatment = p.iTreatment = iTreatment 
        else:
            # player.iTreatment =  p.iTreatment = iTreatment = random.randint(1,Constants.iRandomTreatment) 
            # For the follow up we omitted the non information treatment
            player.iTreatment =  p.iTreatment = iTreatment = random.randint(1,Constants.iRandomTreatment-1)
        print('Treatment for participant: {}'.format(p.iTreatment))
        # Add path to graph to treatment dictionary
        lAttrS = Constants.lAttrS
        lAttrQ = Constants.lAttrQ
        if (iTreatment==1):
            lAttrS['graphPath'] = Constants.sPathS_l
        elif (iTreatment==2):
            lAttrS['graphPath'] = Constants.sPathS_cx
        elif (iTreatment==3):
            # lAttrS['graphPath'] = Constants.sPathS_cx
            lAttrQ['graphPath'] = Constants.sPathQ_cx
        #sAttrOrder = p.sAttrOrder # (Change when ready!!)
        sAttrOrder = random.choice(['Quality','Sustainability']) # Delete when ready
        player.sAttrOrder = p.sAttrOrder  = sAttrOrder # remove last equal when ready
        if sAttrOrder == 'Quality':
            p.lAttr  = [lAttrQ, lAttrS]
        else:
            p.lAttr  = [lAttrS, lAttrQ]



# PAGES
class Information(Page):

    @staticmethod
    def vars_for_template(player: Player):
        return dict(
            lAttr = player.participant.lAttr,
            AddInfo = dict(
                exists  = False,
                content = 'InfoMid/InfoValues.html',
            )
        )


class Belief(Page):
    # Load template
    template_name = 'InfoMid/Belief.html'

    form_model = 'player'
    form_fields = [
        'dRTbelief', 'B01','B02','B03','B11','B12','B13'
    ]


    @staticmethod
    def vars_for_template(player: Player):
        return dict(
            lAttr = player.participant.lAttr,
            AddInfo = dict(
                exists  = True,
                content = 'InfoMid/InfoValues.html',
            ),
            IntroBelief = 'Now we want to know how much do you think each rating is worth in terms of points',
            BonusText = '<u>For each correct value, you will get '+str(Constants.Bonus)+' as an additional bonus.</u>',
        )    

    @staticmethod
    def js_vars(player: Player):
        if (player.sAttrOrder=="Quality"):
            iMin1 = Constants.Q1l
            iMin2 = Constants.S1l
            iMax1 = Constants.Q3h
            iMax2 = Constants.S3h
        else:
            iMin2 = Constants.Q1l
            iMin1 = Constants.S1l
            iMax2 = Constants.Q3h
            iMax1 = Constants.S3h

        return dict(
            iMin1 = iMin1,
            iMin2 = iMin2,
            iMax1 = iMax1,
            iMax2 = iMax2,
            lInputs = ['dRTbelief','B01','B02','B03','B11','B12','B13'],
            RTvar = 'dRTbelief',
        )

    @staticmethod
    def before_next_page(player: Player, timeout_happened):
        p = player.participant
        iTreatment = p.iTreatment
        # Check order of attributes
        if (player.sAttrOrder=='Quality'):
            lB = [player.B01,player.B02,player.B03,player.B11,player.B12,player.B13]
        else:
            lB = [player.B11,player.B12,player.B13,player.B01,player.B02,player.B03]
        # Check if treatment is non-info, then use linear
        if (iTreatment==4):
            iTreatment = 2 
        # Assign 2-leaves value depending on treatment
        Q2,S2 = Constants.BQ2, Constants.BS2
        if (iTreatment==2):
            S2 = Constants.Scx
        elif (iTreatment==3): 
            Q2 = Constants.Qcx
        # lVal = [
        #     0.5*(Constants.Q1l+Constants.Q1h),
        #     0.5*(Constants.Q2l+Constants.Q2h),
        #     0.5*(Constants.Q3l+Constants.Q3h),
        #     0.5*(Constants.S1l+Constants.S1h),
        #     S2,
        #     0.5*(Constants.S3l+Constants.S3h)
        # ]
        lVal = [Constants.BQ1,Q2,Constants.BQ3,Constants.BS1,S2,Constants.BS2]
        iCorrect = 0
        for i in range(len(lVal)):
            iCorrect += (lVal[i]==lB[i]) 
        p.iCorrectBeliefs = iCorrect 
        print('Participant got {} correct beliefs'.format(str(iCorrect)))
        

class Infographics(Page):

    form_model = 'player'
    form_fields = [
        'dRTinfographics', 
    ]
    @staticmethod
    def vars_for_template(player: Player):
        return dict(
            lAttr = player.participant.lAttr,
            AddInfo = dict(
                exists  = True,
                content = 'InfoMid/InfoValues.html',
            )
        )    

    @staticmethod
    def js_vars(player: Player):
        if (player.sAttrOrder=="Quality"):
            sA1 = '3'
            sA2 = '1'
        else:
            sA1 = '1'
            sA2 = '3'

        return dict(
            sA1= sA1, 
            sA2 = sA2,
            lInputs = ['dRTinfographics'],
        )

    @staticmethod
    def is_displayed(player: Player):
        return (player.iTreatment != int(Constants.iRandomTreatment-1))

class First(Page):
    pass
page_sequence = [First,Information, Belief, Infographics]