import random
from ..BaseBehaviour import *
from .. import sentimentCorrelationNodeArray


class BassModel(BaseBehaviour):
    """
    Settings:
        innovation_prob
        
        imitation_prob
    """

    def __init__(self, environment=None, agent_id=0, state=()):
        super().__init__(environment=environment, agent_id=agent_id, state=state)
        self.innovation_prob = environment.environment_params['innovation_prob']
        self.imitation_prob = environment.environment_params['imitation_prob']
        sentimentCorrelationNodeArray[self.id][self.env.now] = 0

    def step(self, now):
        self.behaviour()
        super().step(now)

    def behaviour(self):
        # Outside effects
        if random.random() < self.innovation_prob:
            if self.state['id'] == 0:
                self.state['id'] = 1
                sentimentCorrelationNodeArray[self.id][self.env.now] = 1
            else:
                pass

            self.attrs['status'] = self.state['id']
            return

        # Imitation effects
        if self.state['id'] == 0:
            aware_neighbors = self.get_neighboring_agents(state_id=1)
            num_neighbors_aware = len(aware_neighbors)
            if random.random() < (self.imitation_prob*num_neighbors_aware):
                self.state['id'] = 1
                sentimentCorrelationNodeArray[self.id][self.env.now] = 1

            else:
                pass
            self.attrs['status'] = self.state['id']