def getHistory0b()

in multiple_futures_prediction/dataset_ngsim.py [0:0]

• 16 lines of code
• 6 McCabe index (conditional complexity)

  def getHistory0b(self, vehId:int, t:int, dsId:int ) -> np.ndarray :
    """Get track history trajectory. VehId and dsId are zero-based.
    No normalizations are performed.
    """
    if vehId < 0:
      return np.empty([0,2])
    else:
      if vehId >= self.T.shape[1]:
        return np.empty([0,2])                
      vehTrack = self.T[dsId][vehId].transpose()
      if vehTrack.size==0 or np.argwhere(vehTrack[:, 0] == t).size==0:
         return np.empty([0,2])
      else:
        stpt = np.maximum(0, np.argwhere(vehTrack[:, 0] == t).item() - self.t_h)
        enpt = np.argwhere(vehTrack[:, 0] == t).item() + 1
        hist = vehTrack[stpt:enpt:self.d_s,1:3]            
      if len(hist) < self.t_h//self.d_s + 1:
        return np.empty([0,2])
      return hist