def getState(): # made the order changes
state = [readscreen3.getLowerQlength() / 80,
readscreen3.getRightQlength() / 80,
readscreen3.getUpperQlength() / 80,
readscreen3.getLeftQlength() / 80,
traci.trafficlight.getPhase("0")]
# print (state)
return state
def getState(transition_time): # made the order changes
newState = []
for _ in range(transition_time):
traci.simulationStep()
state = [readscreen3.getLowerQlength() / 80,
readscreen3.getRightQlength() / 80,
readscreen3.getUpperQlength() / 80,
readscreen3.getLeftQlength() / 80
]
newState.insert(0, state)
# print (state)
newState = np.array(newState)
phaseState = getPhaseState(transition_time)
newState = np.dstack((newState, phaseState))
newState = np.expand_dims(newState, axis=0)
return newState