RandomStartAgent.py

from CognitiveAgent import CognitiveAgent
import numpy as np

class FixedStartAgent(CognitiveAgent):
    def __init__(self, currentAction=None, fftSize=1024, cpiLen=256, rng=None):
        super().__init__(currentAction, fftSize, cpiLen)
        if currentAction==None:
            self.takeRandomAction(rng=rng)
        
    def takeRandomAction(self, rng=None, min_true=30, max_true=102):
        if max_true > self.fftSize:
            raise ValueError("max_true cannot exceed fftSize")
        start = 0
        length = 0
        if rng == None:
            length = np.random.randint(min_true, max_true + 1)
            start = np.random.randint(0, self.fftSize - length + 1)
        else:
            length = rng.integers(min_true, max_true + 1)
            start = rng.integers(0, self.fftSize - length + 1)
        stop = start + length

        self.currentAction = (start, stop)