def __init__(self, driver_id):
print("driver : %s created" % (driver_id))
self.driver_id = driver_id
self.driver_path = os.path.join(data_path, str(driver_id))
# print self.driver_path
trips = onlyfiles = [
f for f in listdir(self.driver_path)
if isfile(join(self.driver_path, f))
]
self.nptrips = []
self.nptripsdict = {}
counter = 0
for trip in trips:
if trip.split('.')[1] != 'npy':
# print "for trip %s"%trip
tripnum = int(trip.split('.')[0])
fn = join(self.driver_path, trip)
data = np.genfromtxt(fn,
dtype=float,
delimiter=',',
skip_header=1)
data = sigut.removeRotation(data)
data = self.augmentTripData(data, AUGMENT_TRIP_DATA)
self.nptripsdict[tripnum] = data
counter += 1
# if counter == 20: break
# print counter
# break
for tripnum in np.arange(1, 201):
self.nptrips.append(self.nptripsdict[tripnum])
self.aggregateCatData()
def __init__(self, driver_id):
print "driver : %s created"%(driver_id)
self.driver_id = driver_id
self.driver_path = os.path.join(data_path,str(driver_id))
# print self.driver_path
trips = onlyfiles = [ f for f in listdir(self.driver_path) if isfile(join(self.driver_path,f)) ]
self.nptrips = []
self.nptripsdict = {}
counter = 0
for trip in trips:
if trip.split('.')[1] != 'npy':
# print "for trip %s"%trip
tripnum = int(trip.split('.')[0])
fn = join(self.driver_path,trip)
data = np.genfromtxt(fn, dtype=float, delimiter=',', skip_header=1)
data = sigut.removeRotation(data)
data = self.augmentTripData(data, AUGMENT_TRIP_DATA)
self.nptripsdict[tripnum] = data
counter += 1
# if counter == 20: break
# print counter
# break
for tripnum in np.arange(1,201):
self.nptrips.append(self.nptripsdict[tripnum])
self.aggregateCatData()
def interpAroundId(self, idtest, trip, Nsteps, nmid, samples):
newzero = trip[idtest, :]
# import pdb
# pdb.set_trace()
ndist = trip[:, self.headers['cum_d']] - newzero[self.headers['cum_d']]
# print ndist[idtest]
res = np.zeros((Nsteps, 2), dtype=np.float)
res[:, 0] = np.interp(samples, ndist, trip[:, self.headers['x']])
res[:, 1] = np.interp(samples, ndist, trip[:, self.headers['y']])
# res[:,2] = np.interp(samples, ndist, trip[:,4])
res = sigut.removeRotation(res, removeBias=True)
shifted = -res[:, 1]
res = np.c_[res, shifted]
return res
def interpAroundId( self, idtest, trip, Nsteps, nmid, samples):
newzero = trip[idtest,:]
# import pdb
# pdb.set_trace()
ndist = trip[:, self.headers['cum_d']] - newzero[self.headers['cum_d']]
# print ndist[idtest]
res = np.zeros((Nsteps, 2), dtype=np.float)
res[:,0] = np.interp(samples, ndist, trip[:,self.headers['x']])
res[:,1] = np.interp(samples, ndist, trip[:,self.headers['y']])
# res[:,2] = np.interp(samples, ndist, trip[:,4])
res = sigut.removeRotation(res, removeBias=True)
shifted = -res[:,1]
res = np.c_[res, shifted]
return res
