#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon May 24 10:55:00 2021

@author: elybenve
"""
import random
import numpy as np
A_1 = [[7, 1],[4, 2],[1, 3]]
A_2 = [[10, 1],[7, 2], [4, 3], [1, 4]]
A_3 = [[7, 1],[5, 2],[3, 3], [1, 4]]
A_4 = [[9, 1],[7, 2],[5, 3], [3, 4], [1, 5]]
A_5 = [[5, 1], [4, 2], [3, 3], [2, 4], [1, 5]]
A_6 = [[6, 1], [5, 2], [4, 3], [3, 4], [2, 5], [1, 6]]
A_7 = [[7, 1], [6, 2], [5,3], [4,4], [3, 5], [2, 6], [1, 7]]
A_8 = [[4, 1], [3, 3], [2 ,5], [1, 7]]
A_9 = [[3, 1], [2, 4], [1, 7]]
A_10 = [[5, 1], [4, 3], [3, 5], [2, 7], [1, 9]]
A_11 = [[4, 1], [3, 4], [2, 7], [1,10]]

A = [A_1, A_2, A_3, A_4, A_5, A_6, A_7, A_8, A_9, A_10, A_11]

tasks = np.arange(0,len(A))
num_rounds = len(tasks)

#FUNCTION 

def creating_session(subsession: Subsession):
    if subsession.round_number == 1:
        for p in subsession.get_players():
            round_numbers = list(range(1, Constants.num_rounds + 1))
            random.shuffle(round_numbers)
            p.participant.vars['task_rounds'] = dict(zip(Constants.tasks, round_numbers))
            

round_numbers = list(range(1, num_rounds + 1))   
random.shuffle(round_numbers)
pvtr['task_rounds'] = dict(zip(tasks, round_numbers))