MASS/StaticAgent.py at main · TheJob21/MASS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
from Agent import Agent
import numpy as np
import random
from enum import Enum

class StaticType(Enum):
    Fat = 0
    Skinny = 1
    Pulsed = 2
    Rectangular = 3

class StaticAgent(Agent):

    def __init__(self, rng=None, currentAction=None, fftSize=1024, staticType=2):
        super().__init__(currentAction, fftSize)
        self.rng = rng
        self.staticType = staticType
        if self.staticType == 0:
            self.minBw = 150
            self.maxBw = 175
            self.txTimeMin = 100
            self.txTimeMax = 250
            self.deadTimeMin = 9000
            self.deadTimeMax = 10000
        elif self.staticType == 1:
            self.minBw = 15
            self.maxBw = 30
            self.txTimeMin = 225
            self.txTimeMax = 350
            self.deadTimeMin = 1900
            self.deadTimeMax = 2100
        elif self.staticType == 2:
            self.minBw = 25
            self.maxBw = 50
            self.txTimeMin = 20
            self.txTimeMax = 50
            self.deadTimeMin = 50
            self.deadTimeMax = 80
            self.intervalMin = 2000
            self.intervalMax = 5000
        elif self.staticType == 3:
            self.minBw = 60
            self.maxBw = 680
            self.txTimeMin = 45
            self.txTimeMax = 55
            self.deadTimeMin = 10
            self.deadTimeMax = 250
        self.takeRandomAction()


    def takeRandomAction(self):
        length = 0
        start = 0
        if self.rng == None:
            length = np.random.randint(self.minBw, self.maxBw + 1)
            start = np.random.randint(0, self.fftSize - length + 1)
            self.txTime = np.random.randint(self.txTimeMin, self.txTimeMax + 1)
            self.deadTime = np.random.randint(self.deadTimeMin, self.deadTimeMax + 1)
            if self.staticType == 2:
                self.interval = np.random.randint(self.intervalMin, self.intervalMax + 1)
        else:
            length = self.rng.integers(self.minBw, self.maxBw + 1)
            start = self.rng.integers(0, self.fftSize - length + 1)
            self.txTime = self.rng.integers(self.txTimeMin, self.txTimeMax + 1)
            self.deadTime = self.rng.integers(self.deadTimeMin, self.deadTimeMax + 1)
            if self.staticType == 2:
                self.interval = self.rng.integers(self.intervalMin, self.intervalMax + 1)

        stop = start + length

        self.actionToWobble = self.currentAction = (start, stop)

    def iterateCurrentAction(self, iteration):
        if iteration % (self.txTime+self.deadTime) > self.txTime:
            self.currentAction = None
            return
        if self.staticType == 2 and (iteration % (self.interval * 2)) > self.interval:
            self.currentAction = None
            return
        if self.staticType == 3 and (iteration % ((self.txTime+self.deadTime) * 2)) > self.txTime+self.deadTime:
            self.takeRandomAction()

        start, stop = self.actionToWobble

        randShift = 0
        if self.rng == None:
            randShift = random.randint(0, 20) # About 1-2 MHz modulation
        else:
            randShift = self.rng.integers(0, 20) # About 1-2 MHz modulation

        self.currentAction = (start+randShift, stop-randShift)