def testRumbleStrip(self):
cW = self.traffic.mustang.maxX
pos = self.getRoadPosition()
if (pos+cW*0.6 > 0 and pos+cW*0.3 < 0) or (pos-cW*0.6 < 0 and pos-cW*0.3 > 0):
#print 'rumble'
if helpers.conditionTime() - self.rumblerTime > 0.2:
rumbler = pyglet.resource.media('bump.wav')
rumbler.play()
self.rumblerTime = helpers.conditionTime()
def testRumbleStrip(self):
cW = self.traffic.mustang.maxX
pos = self.getRoadPosition()
if (pos + cW * 0.6 > 0
and pos + cW * 0.3 < 0) or (pos - cW * 0.6 < 0
and pos - cW * 0.3 > 0):
#print 'rumble'
if helpers.conditionTime() - self.rumblerTime > 0.2:
rumbler = pyglet.resource.media('bump.wav')
rumbler.play()
self.rumblerTime = helpers.conditionTime()
def checkBlinkerUse(self, cardir):
curtime = helpers.currentTime()
if curtime - self.lastBlinkerTime > globals.MAX_TRANSITION_LAG_TIME:
if globals.debug:
print 'No blinker was used during a lane transition'
if doPractice and block < 2:
self.blinkerPracWarning = 1.5
globals.bonusCounter = max(
0.0, globals.bonusCounter - globals.BLINKER_PENALTY)
globals.db['lanetransition'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'blinkerused', '0', 'blinkdir',
'?', 'block', globals.flow.block, 'cardir', cardir,
'congruent', '0', 'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
else:
if (cardir == 'L' and self.lastBlinkerDir
== 'L') or (cardir == 'R' and self.lastBlinkerDir == 'R'):
globals.db['lanetransition'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'blinkerused', '1',
'blinkdir', self.lastBlinkerDir, 'block',
globals.flow.block, 'cardir', cardir, 'congruent', '1',
'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
if globals.debug:
print 'Correct use of blinker during lane transition'
else:
globals.db['lanetransition'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'blinkerused', '1',
'blinkdir', self.lastBlinkerDir, 'block',
globals.flow.block, 'cardir', cardir, 'congruent', '0',
'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
if globals.debug:
print 'Incorrect use of blinker during lane transition'
globals.bonusCounter -= BLINKER_PENALTY
def reportNearbyStaticCars(self, car):
iMin = bisect.bisect(self.tStaticCars, -car.xPos - 25.0)
iMax = bisect.bisect(self.tStaticCars, -car.xPos + 25.0)
speed = (self.staticSpeed * 3600) * 0.001
for i in xrange(iMin, iMax):
s = self.staticCars[i]
globals.db['slowcars'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'id', s.id, 'type', 'slow',
'speed', r3(speed), 'xposition', r3(s.x), 'distance', r3(s.x-(-car.xPos)), 'block', globals.flow.block,
'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
def reportNearbyFastCars(self, car):
iMin = bisect.bisect(self.tFastCars, -car.xPos - 25.0) # get plants close to the car
iMax = bisect.bisect(self.tFastCars, -car.xPos + 25.0)
for i in xrange(iMin, iMax):
f = self.fastCars[i]
speed = (f.speed * 3600) * 0.001
globals.db['fastcars'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'id', f.id, 'type', 'fast',
'speed', r3(speed), 'xposition', r3(f.x), 'distance', r3(f.x-(-car.xPos)), 'block', globals.flow.block,
'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
def checkBlinkerUse(self, cardir):
curtime = helpers.currentTime()
if curtime - self.lastBlinkerTime > globals.MAX_TRANSITION_LAG_TIME:
if globals.debug: print 'No blinker was used during a lane transition'
if doPractice and block < 2:
self.blinkerPracWarning = 1.5
globals.bonusCounter = max(0.0, globals.bonusCounter - globals.BLINKER_PENALTY)
globals.db['lanetransition'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'blinkerused', '0', 'blinkdir', '?', 'block', globals.flow.block,
'cardir', cardir, 'congruent', '0', 'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
else:
if (cardir == 'L' and self.lastBlinkerDir == 'L') or (cardir == 'R' and self.lastBlinkerDir == 'R'):
globals.db['lanetransition'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'blinkerused', '1', 'blinkdir', self.lastBlinkerDir, 'block', globals.flow.block,
'cardir', cardir, 'congruent', '1', 'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
if globals.debug: print 'Correct use of blinker during lane transition'
else:
globals.db['lanetransition'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'blinkerused', '1', 'blinkdir', self.lastBlinkerDir, 'block', globals.flow.block,
'cardir', cardir, 'congruent', '0', 'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
if globals.debug: print 'Incorrect use of blinker during lane transition'
globals.bonusCounter -= BLINKER_PENALTY
def reportNearbyStaticCars(self, car):
iMin = bisect.bisect(self.tStaticCars, -car.xPos - 25.0)
iMax = bisect.bisect(self.tStaticCars, -car.xPos + 25.0)
speed = (self.staticSpeed * 3600) * 0.001
for i in xrange(iMin, iMax):
s = self.staticCars[i]
globals.db['slowcars'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'id', s.id, 'type', 'slow',
'speed',
r3(speed), 'xposition',
r3(s.x), 'distance',
r3(s.x - (-car.xPos)), 'block', globals.flow.block, 'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
def update(dt):
global window, renderer
global car, traffic
#global hasWHeel, joystick
if globals.flow.getScreenType() == 'screen':
if globals.flow.isTextScreenDone(renderer):
globals.flow.endState(car, traffic, renderer, window)
elif globals.flow.getScreenType() == 'drive':
if helpers.conditionTime() <= globals.flow.blockDuration:
traffic.updateStaticCars(car, dt)
traffic.updateFastCar(car, dt)
car.updateDrivingInput(dt)
car.updatePosition(dt)
globals.flow.secondary.update(dt)
globals.flow.secondary.checkInput()
else:
globals.flow.endState(car, traffic, renderer, window)
def reportNearbyFastCars(self, car):
iMin = bisect.bisect(self.tFastCars,
-car.xPos - 25.0) # get plants close to the car
iMax = bisect.bisect(self.tFastCars, -car.xPos + 25.0)
for i in xrange(iMin, iMax):
f = self.fastCars[i]
speed = (f.speed * 3600) * 0.001
globals.db['fastcars'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'id', f.id, 'type', 'fast',
'speed',
r3(speed), 'xposition',
r3(f.x), 'distance',
r3(f.x - (-car.xPos)), 'block', globals.flow.block, 'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
def update(dt):
global window, renderer
global car, traffic
#global hasWHeel, joystick
if globals.flow.getScreenType() == 'screen':
if globals.flow.isTextScreenDone(renderer):
globals.flow.endState(car, traffic, renderer, window)
elif globals.flow.getScreenType() == 'drive':
if helpers.conditionTime() <= globals.flow.blockDuration:
traffic.updateStaticCars(car, dt)
traffic.updateFastCar(car, dt)
car.updateDrivingInput(dt)
car.updatePosition(dt)
globals.flow.secondary.update(dt)
globals.flow.secondary.checkInput()
else:
globals.flow.endState(car, traffic, renderer, window)
def setBlinker(self, direction):
global db
self.activeBlinker = direction
d = 'L'
if (direction == globals.RIGHT_BLINKER):
d = 'R'
self.blinkerCooldown = 1.0
self.lastBlinkerDir = d
self.lastBlinkerTime = helpers.currentTime()
blinkSound = pyglet.resource.media('blinker.wav')
blinkSound.play()
globals.db['blinker'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'direction', d, 'condtime',
r3(helpers.conditionTime()), 'block', block, 'time',
r3(helpers.currentTime())
], True)
def updatePosition(self, dt):
if self.autoPilot:
x = self.xPos
xnew = x - globals.kphToMps(90.0) * dt
(xl, zl, heading, hx,
hz) = self.road.getLanePosition(xnew, globals.LEFT_LANE)
self.xPos = xnew
self.zPos = zl
self.xHeading = hx
self.zHeading = hz
else:
self.prevRoadPos = self.getRoadPosition() #self.zPos
roadPos = self.prevRoadPos
self.updateAcceleration()
self.updateBreaking()
self.collisionCooldown = max(0.0, self.collisionCooldown - dt)
self.blinkerCooldown = max(0.0, self.blinkerCooldown - dt)
self.blinkerPracWarning = max(0.0, self.blinkerPracWarning - dt)
self.traffic.calcPassingStates(self)
# Hit a fast moving car (left lane)
(idx, collisionId,
collisionZone) = self.traffic.foundFastCollisions(self)
if collisionId >= 0:
if self.collisionCooldown < 0.001:
self.collisionCooldown = 1.0
self.traffic.removeFastCar(idx)
self.incrementDamage()
globals.db['collision'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'zone', collisionZone,
'carType', 'fast', 'block', block, 'carNum',
collisionId, 'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
if globals.debug:
print 'Collision detected with fast car [' + str(
idx) + ', zone: ' + str(collisionZone) + ']'
# hit a slow moving car (right lane)
(idx, collisionId,
collisionZone) = self.traffic.foundSlowCollisions(self)
if collisionId >= 0:
if collisionZone == FRONT:
mul = 0.9
self.xVelocity = min(self.xVelocity * mul,
self.traffic.staticSpeed * mul)
self.zVelocity = min(self.zVelocity * mul,
self.traffic.staticSpeed * mul)
else: # BACK
bumpspeed = self.traffic.staticSpeed
ratio = bumpspeed / self.currentSpeed
self.xVelocity *= ratio * 1.2
self.zVelocity *= ratio * 1.2
self.currentSpeed = self.lenVector(self.xVelocity,
self.zVelocity)
self.xPos -= self.xVelocity * dt
self.zPos -= self.zVelocity * dt
if self.collisionCooldown < 0.001:
#print 'boem! '+str(collisionZone)
self.collisionCooldown = 1.0
self.incrementDamage()
globals.db['collision'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'zone', collisionZone,
'carType', 'slow', 'block', block, 'carNum',
collisionId, 'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
if globals.debug:
print 'Collision detected with slow car [' + str(
idx) + ', zone: ' + str(collisionZone) + ']'
roadPos = self.getRoadPosition()
else:
dotProd = sum(
map(operator.mul, (self.xHeading, self.zHeading),
(self.xVelocity, self.zVelocity)))
#print dotProd
xLatVel = self.xHeading * dotProd
zLatVel = self.zHeading * dotProd
xLatFric = 0 #-xLatVel*self.latFricFactor
zLatFric = 0 #-zLatVel*self.latFricFactor
xBackFric = -self.xVelocity * self.backFricFactor
zBackFric = -self.zVelocity * self.backFricFactor
#print xLatFric
self.xVelocity += (xBackFric + xLatFric) * dt
self.zVelocity += (zBackFric + zLatFric) * dt
#print self.xVelocity
dm = self.currentSpeed * dt
self.currentSpeed = self.lenVector(self.xVelocity,
self.zVelocity)
if self.currentSpeed < globals.MAX_SPEED:
self.xVelocity += self.xAccel * dt
self.zVelocity += self.zAccel * dt
self.xVelocity *= self.breakPower
self.zVelocity *= self.breakPower
#self.xVelocity = max(0.0001*self.xVelocity, self.xVelocity - self.xBreak * dt)
#self.zVelocity = max(0.0001*self.zVelocity, self.zVelocity - self.zBreak * dt)
self.xPos -= self.xVelocity * dt
self.zPos -= self.zVelocity * dt
#self.xPos -= self.xHeading*dm
#self.zPos -= self.zHeading*dm
vHx = self.currentSpeed * self.xHeading
vHz = self.currentSpeed * self.zHeading
latFactor = 0.2
(x, z) = self.lerpVectors(vHx, vHz, self.xVelocity,
self.zVelocity, latFactor)
self.xVelocity = x
self.zVelocity = z
self.angleHeading = (atan2(self.xVelocity, self.zVelocity) -
atan2(1.0, 0.0)) * globals.RAD2DEG
roadPos = self.getRoadPosition()
if self.prevRoadPos <= 0.0 and roadPos > 0.0: # left lane transition
if globals.flow.drivingCondition == 'complex':
(cid, x) = self.traffic.findNextStaticCar(self)
dist = x - (-self.xPos)
self.laneCrossDist = dist
if globals.debug:
print 'Lane transition to left lane at ' + str(
-self.xPos
) + '. Distance to next static car (' + str(
cid) + '): ' + str(dist)
else:
if globals.debug:
print 'Lane transition to left lane at ' + str(
-self.xPos)
self.checkBlinkerUse('L')
elif self.prevRoadPos >= 0.0 and roadPos < 0.0: #right lane transition
if globals.debug:
print 'Lane transition to right lane at ' + str(-self.xPos)
self.checkBlinkerUse('R')
self.testRumbleStrip()
if self.isOffRoad():
self.offroadTime += dt
if self.offroadTime >= 1.0:
globals.bonusCounter = max(
0.0, globals.bonusCounter - globals.OFFROAD_PENALTY)
self.offroadTime = 0.0
# See if we're next to a car we're overtaking
self.inSafeZone = False
if self.traffic.recentPassedCar + 1 < len(self.traffic.staticCars):
try:
nextx = self.traffic.staticCars[self.traffic.recentPassedCar +
1].x
except IndexError:
nextx = -1
if nextx > 0.0:
#print 'next: '+str(nextx)+' car: '+str(self.xPos)
d = abs(nextx - -self.xPos)
#print 'distance to next car: '+str(d)
if (d < 6.0 and self.prevRoadPos > 0.1) or d < -1.0:
#print 'safe!'
self.inSafeZone = True
def calcPassingStates(self, car):
iMin = bisect.bisect(self.tStaticCars, -car.xPos - 20.0)
iMax = bisect.bisect(self.tStaticCars, -car.xPos + 20.0)
for i in xrange(iMin, iMax):
if self.waitingForLaneTransition: # don't generate new cars until the particpant is back on the right lane
if car.getRoadPosition() < 0.0 or helpers.conditionTime(
) - self.waitingForLaneTransTime > 10.0:
if car.getRoadPosition() < 0.0:
if globals.debug:
print 'Participant returned to right lane after overtaking a car'
else:
if globals.debug:
print 'Participant is sticking left too long, generating cars anyway'
self.waitingForLaneTransition = False
oldNews = [
j for j, v in enumerate(self.beingOvertaken)
if v > self.staticCars[i].id
]
if len(
oldNews
) == 0: # participant has not started overtaking another car yet
#rand = random.random()
#if rand < 1.0 - NO_OVERTAKES_CHANCE:
if self.recentPassedCar + 1 < len(self.staticCars):
try:
approxDist = self.staticCars[
self.recentPassedCar + 1].x - (-car.xPos)
except IndexError:
approxDist = -1
if globals.debug:
print 'Distance to next car: ' + str(
approxDist)
#critRand = random.random()
#secRand = random.random()
#if secRand <= 0.50:
ad = approxDist
if approxDist > -1:
#if random.random() >= 0.5:
for dynCar in self.dynamicCarTypes:
if random.random(
) <= dynCar.genChance and approxDist > dynCar.minRoom:
self.genDynamicCar(
dynCar.distanceFactor * approxDist
+ random.uniform(
-dynCar.jitterFactor * ad,
dynCar.jitterFactor * ad),
dynCar.generateDistance)
# if self.recentPassedCar < len(self.fastCars3):
# if self.fastCars3[self.recentPassedCar]:
# self.genDynamicCar(0.9*ad + random.uniform(-0.05*ad, 0.05*ad), random.uniform(40.0, 41.0))
#
# #if random.random() >= 0.5:
# if self.recentPassedCar < len(self.fastCars4) and approxDist > 75.0:
# if self.fastCars4[self.recentPassedCar]:
# self.genDynamicCar(0.7*ad + random.uniform(-0.05*ad, 0.05*ad), random.uniform(60.0, 61.0))
#
#
# secLoc = 0.5*approxDist + random.uniform(-0.05*approxDist, 0.05*approxDist)
# if self.recentPassedCar < len(self.fastCars1):
# if self.fastCars1[self.recentPassedCar]:
# if approxDist > 20.0:
# #print 'Non-critical fast car generated after overtaking ' + str(self.recentPassedCar)+' at '+str(secLoc)
# self.genDynamicCar(secLoc, random.uniform(71.0, 85.0))
#
#if random.random() > 0.001 and approxDist > 80.0:
#self.genDynamicCar(0.3*ad + random.uniform(-0.05*ad, 0.05*ad), random.uniform(100.0, 101.0))
#if random.random() >= 0.5 and overtakePoint * 3 < 0.3*ad:
# overtakePoint = car.getAvgOvertakePoint()
#
# if self.recentPassedCar < len(self.fastCars5) and approxDist > 75.0:
# if self.fastCars5[self.recentPassedCar]:
# self.genDynamicCar(overtakePoint*3, random.uniform(110.0, 111.0))
#
# if self.recentPassedCar < len(self.fastCars2):
# if self.fastCars2[self.recentPassedCar]:
# critLoc = overtakePoint
# self.genDynamicCar(critLoc, random.uniform(135.0, 145.0))
# if globals.debug: print 'Critical fast car generated after overtaking ' + str(self.recentPassedCar) + ', will overtake at distance '+str(critLoc)
# #else:
# #print 'Not generating any fast cars after overtaking ' + str(self.staticCars[i].id)
#
# # tailer
# #if random.random() >= 0.5:
# if self.recentPassedCar < len(self.fastCars6):
# if self.fastCars6[self.recentPassedCar]:
# #if 1:
# if self.genDynamicCar(0.0, random.uniform(155.0, 161.0)):
# if len(self.fastCars) > 0:
# self.fastCars[-1].kind = 'tailer'
#
#print 'car '+str(i)
oldState = self.staticCars[i].passState
(state, dist) = self.determinePassingState(car, self.staticCars[i])
if oldState < globals.PASS_COMPLETE:
self.staticCars[i].passState = state
if not self.staticCars[i].beingOvertaken:
if state == globals.PASS_ALONG_CAR and oldState == globals.PASS_BEHIND_CAR:
if globals.debug:
print 'Starting overtake [slow car ID: ' + str(
self.staticCars[i].id) + ']'
self.staticCars[i].beingOvertaken = True
self.beingOvertaken.append(self.staticCars[i].id)
globals.db['overtake'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'state', 'start',
'carNum', self.staticCars[i].id, 'carDist',
car.laneCrossDist, 'block', globals.flow.block,
'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
else:
if state == globals.PASS_INFRONT_CAR and dist > 2.0:
if globals.debug:
print 'Finished overtake [slow car ID: ' + str(
self.staticCars[i].id) + ']'
globals.db['overtake'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'state', 'complete',
'carNum', self.staticCars[i].id, 'carDist',
car.laneCrossDist, 'block', globals.flow.block,
'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
self.staticCars[i].passState = globals.PASS_COMPLETE
self.beingOvertaken.remove(self.staticCars[i].id)
self.beenOvertaken.append(self.staticCars[i].id)
self.recentPassedCar = i
car.saveOvertakePoint()
self.fastCars = []
self.tFastCars = []
self.waitingForLaneTransition = True
self.waitingForLaneTransTime = helpers.conditionTime()
elif state == globals.PASS_BEHIND_CAR and dist < -9.0:
self.staticCars[i].beingOvertaken = False
globals.db['overtake'].addData([
'pp', globals.participant, 'condition',
globals.flow.getCondition(), 'state', 'failed',
'carNum', self.staticCars[i].id, 'carDist',
car.laneCrossDist, 'block', globals.flow.block,
'condtime',
r3(helpers.conditionTime()), 'time',
r3(helpers.currentTime())
], True)
if globals.debug:
print 'Overtake incomplete [slow car ID: ' + str(
self.staticCars[i].id) + ']'
def updatePosition(self, dt):
if self.autoPilot:
x = self.xPos
xnew = x-globals.kphToMps(90.0)*dt
(xl, zl, heading, hx, hz) = self.road.getLanePosition(xnew, globals.LEFT_LANE)
self.xPos = xnew
self.zPos = zl
self.xHeading = hx
self.zHeading = hz
else:
self.prevRoadPos = self.getRoadPosition() #self.zPos
roadPos = self.prevRoadPos
self.updateAcceleration()
self.updateBreaking()
self.collisionCooldown = max(0.0, self.collisionCooldown - dt)
self.blinkerCooldown = max(0.0, self.blinkerCooldown - dt)
self.blinkerPracWarning = max(0.0, self.blinkerPracWarning - dt)
self.traffic.calcPassingStates(self)
# Hit a fast moving car (left lane)
(idx, collisionId, collisionZone) = self.traffic.foundFastCollisions(self)
if collisionId >= 0:
if self.collisionCooldown < 0.001:
self.collisionCooldown = 1.0
self.traffic.removeFastCar(idx)
self.incrementDamage()
globals.db['collision'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'zone', collisionZone, 'carType', 'fast', 'block', block,
'carNum', collisionId ,'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
if globals.debug: print 'Collision detected with fast car ['+str(idx)+', zone: '+str(collisionZone)+']'
# hit a slow moving car (right lane)
(idx, collisionId, collisionZone) = self.traffic.foundSlowCollisions(self)
if collisionId >= 0:
if collisionZone == FRONT:
mul = 0.9
self.xVelocity = min(self.xVelocity*mul, self.traffic.staticSpeed*mul)
self.zVelocity = min(self.zVelocity*mul, self.traffic.staticSpeed*mul)
else: # BACK
bumpspeed = self.traffic.staticSpeed
ratio = bumpspeed / self.currentSpeed
self.xVelocity *= ratio*1.2
self.zVelocity *= ratio*1.2
self.currentSpeed = self.lenVector(self.xVelocity, self.zVelocity)
self.xPos -= self.xVelocity*dt
self.zPos -= self.zVelocity*dt
if self.collisionCooldown < 0.001:
#print 'boem! '+str(collisionZone)
self.collisionCooldown = 1.0
self.incrementDamage()
globals.db['collision'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'zone', collisionZone, 'carType', 'slow', 'block', block,
'carNum', collisionId ,'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
if globals.debug: print 'Collision detected with slow car ['+str(idx)+', zone: '+str(collisionZone)+']'
roadPos = self.getRoadPosition()
else:
dotProd = sum(map( operator.mul, (self.xHeading, self.zHeading), (self.xVelocity, self.zVelocity)))
#print dotProd
xLatVel = self.xHeading * dotProd
zLatVel = self.zHeading * dotProd
xLatFric = 0#-xLatVel*self.latFricFactor
zLatFric = 0#-zLatVel*self.latFricFactor
xBackFric = -self.xVelocity*self.backFricFactor
zBackFric = -self.zVelocity*self.backFricFactor
#print xLatFric
self.xVelocity += (xBackFric + xLatFric) * dt
self.zVelocity += (zBackFric + zLatFric) * dt
#print self.xVelocity
dm = self.currentSpeed*dt
self.currentSpeed = self.lenVector(self.xVelocity, self.zVelocity)
if self.currentSpeed < globals.MAX_SPEED:
self.xVelocity += self.xAccel * dt
self.zVelocity += self.zAccel * dt
self.xVelocity *= self.breakPower
self.zVelocity *= self.breakPower
#self.xVelocity = max(0.0001*self.xVelocity, self.xVelocity - self.xBreak * dt)
#self.zVelocity = max(0.0001*self.zVelocity, self.zVelocity - self.zBreak * dt)
self.xPos -= self.xVelocity*dt
self.zPos -= self.zVelocity*dt
#self.xPos -= self.xHeading*dm
#self.zPos -= self.zHeading*dm
vHx = self.currentSpeed*self.xHeading
vHz = self.currentSpeed*self.zHeading
latFactor = 0.2
(x,z) = self.lerpVectors(vHx, vHz, self.xVelocity, self.zVelocity, latFactor)
self.xVelocity = x
self.zVelocity = z
self.angleHeading = (atan2(self.xVelocity, self.zVelocity) - atan2(1.0, 0.0))*globals.RAD2DEG
roadPos = self.getRoadPosition()
if self.prevRoadPos <= 0.0 and roadPos > 0.0: # left lane transition
if globals.flow.drivingCondition == 'complex':
(cid, x) = self.traffic.findNextStaticCar(self)
dist = x - (-self.xPos)
self.laneCrossDist = dist
if globals.debug: print 'Lane transition to left lane at ' + str(-self.xPos) + '. Distance to next static car ('+str(cid)+'): '+str(dist)
else:
if globals.debug: print 'Lane transition to left lane at ' + str(-self.xPos)
self.checkBlinkerUse('L')
elif self.prevRoadPos >= 0.0 and roadPos < 0.0: #right lane transition
if globals.debug: print 'Lane transition to right lane at ' + str(-self.xPos)
self.checkBlinkerUse('R')
self.testRumbleStrip()
if self.isOffRoad():
self.offroadTime += dt
if self.offroadTime >= 1.0:
globals.bonusCounter = max(0.0, globals.bonusCounter - globals.OFFROAD_PENALTY)
self.offroadTime = 0.0
# See if we're next to a car we're overtaking
self.inSafeZone = False
if self.traffic.recentPassedCar+1 < len(self.traffic.staticCars):
try:
nextx = self.traffic.staticCars[self.traffic.recentPassedCar+1].x
except IndexError:
nextx = -1
if nextx > 0.0:
#print 'next: '+str(nextx)+' car: '+str(self.xPos)
d = abs(nextx - -self.xPos)
#print 'distance to next car: '+str(d)
if (d < 6.0 and self.prevRoadPos > 0.1) or d < -1.0:
#print 'safe!'
self.inSafeZone = True
def measureData(self, car, traffic, dt):
if self.flowIdx < len(self.flow) and self.flow[self.flowIdx].type == 'drive':
# get all the car measurements
laneDeviation = car.getLaneDeviation()
roadPos = car.getRoadPosition()
wheelAngle = car.currentWheelAngle
speed = (car.currentSpeed * 3600) * 0.001
acceleration = car.accelerationInput
breaking = car.breakInput
xpos = car.xPos
globals.db['car'].addData(['pp', globals.participant, 'condition', self.getCondition(), 'deviation', r3(laneDeviation), 'yposition', r3(roadPos),
'speed', r3(speed), 'wheelangle', r3(wheelAngle), 'accel', r3(acceleration), 'break', r3(breaking),
'xposition', r3(xpos), 'block', self.block, 'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
traffic.reportNearbyStaticCars(car)
traffic.reportNearbyFastCars(car)
def calcPassingStates(self, car):
iMin = bisect.bisect(self.tStaticCars, -car.xPos - 20.0)
iMax = bisect.bisect(self.tStaticCars, -car.xPos + 20.0)
for i in xrange(iMin, iMax):
if self.waitingForLaneTransition: # don't generate new cars until the particpant is back on the right lane
if car.getRoadPosition() < 0.0 or helpers.conditionTime() - self.waitingForLaneTransTime > 10.0:
if car.getRoadPosition() < 0.0:
if globals.debug: print 'Participant returned to right lane after overtaking a car'
else:
if globals.debug: print 'Participant is sticking left too long, generating cars anyway'
self.waitingForLaneTransition = False
oldNews = [j for j,v in enumerate(self.beingOvertaken) if v > self.staticCars[i].id]
if len(oldNews) == 0: # participant has not started overtaking another car yet
#rand = random.random()
#if rand < 1.0 - NO_OVERTAKES_CHANCE:
if self.recentPassedCar+1 < len(self.staticCars):
try:
approxDist = self.staticCars[self.recentPassedCar+1].x - (-car.xPos)
except IndexError:
approxDist = -1
if globals.debug: print 'Distance to next car: '+str(approxDist)
#critRand = random.random()
#secRand = random.random()
#if secRand <= 0.50:
ad = approxDist
if approxDist > -1:
#if random.random() >= 0.5:
for dynCar in self.dynamicCarTypes:
if random.random() <= dynCar.genChance and approxDist > dynCar.minRoom:
self.genDynamicCar(dynCar.distanceFactor*approxDist + random.uniform(-dynCar.jitterFactor*ad, dynCar.jitterFactor*ad), dynCar.generateDistance)
# if self.recentPassedCar < len(self.fastCars3):
# if self.fastCars3[self.recentPassedCar]:
# self.genDynamicCar(0.9*ad + random.uniform(-0.05*ad, 0.05*ad), random.uniform(40.0, 41.0))
#
# #if random.random() >= 0.5:
# if self.recentPassedCar < len(self.fastCars4) and approxDist > 75.0:
# if self.fastCars4[self.recentPassedCar]:
# self.genDynamicCar(0.7*ad + random.uniform(-0.05*ad, 0.05*ad), random.uniform(60.0, 61.0))
#
#
# secLoc = 0.5*approxDist + random.uniform(-0.05*approxDist, 0.05*approxDist)
# if self.recentPassedCar < len(self.fastCars1):
# if self.fastCars1[self.recentPassedCar]:
# if approxDist > 20.0:
# #print 'Non-critical fast car generated after overtaking ' + str(self.recentPassedCar)+' at '+str(secLoc)
# self.genDynamicCar(secLoc, random.uniform(71.0, 85.0))
#
#if random.random() > 0.001 and approxDist > 80.0:
#self.genDynamicCar(0.3*ad + random.uniform(-0.05*ad, 0.05*ad), random.uniform(100.0, 101.0))
#if random.random() >= 0.5 and overtakePoint * 3 < 0.3*ad:
# overtakePoint = car.getAvgOvertakePoint()
#
# if self.recentPassedCar < len(self.fastCars5) and approxDist > 75.0:
# if self.fastCars5[self.recentPassedCar]:
# self.genDynamicCar(overtakePoint*3, random.uniform(110.0, 111.0))
#
# if self.recentPassedCar < len(self.fastCars2):
# if self.fastCars2[self.recentPassedCar]:
# critLoc = overtakePoint
# self.genDynamicCar(critLoc, random.uniform(135.0, 145.0))
# if globals.debug: print 'Critical fast car generated after overtaking ' + str(self.recentPassedCar) + ', will overtake at distance '+str(critLoc)
# #else:
# #print 'Not generating any fast cars after overtaking ' + str(self.staticCars[i].id)
#
# # tailer
# #if random.random() >= 0.5:
# if self.recentPassedCar < len(self.fastCars6):
# if self.fastCars6[self.recentPassedCar]:
# #if 1:
# if self.genDynamicCar(0.0, random.uniform(155.0, 161.0)):
# if len(self.fastCars) > 0:
# self.fastCars[-1].kind = 'tailer'
#
#print 'car '+str(i)
oldState = self.staticCars[i].passState
(state, dist) = self.determinePassingState(car,self.staticCars[i])
if oldState < globals.PASS_COMPLETE:
self.staticCars[i].passState = state
if not self.staticCars[i].beingOvertaken:
if state == globals.PASS_ALONG_CAR and oldState == globals.PASS_BEHIND_CAR:
if globals.debug: print 'Starting overtake [slow car ID: ' + str(self.staticCars[i].id) +']'
self.staticCars[i].beingOvertaken = True
self.beingOvertaken.append(self.staticCars[i].id)
globals.db['overtake'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'state', 'start',
'carNum', self.staticCars[i].id, 'carDist', car.laneCrossDist, 'block', globals.flow.block,
'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
else:
if state == globals.PASS_INFRONT_CAR and dist > 2.0:
if globals.debug: print 'Finished overtake [slow car ID: ' + str(self.staticCars[i].id) +']'
globals.db['overtake'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'state', 'complete',
'carNum', self.staticCars[i].id, 'carDist', car.laneCrossDist, 'block', globals.flow.block,
'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
self.staticCars[i].passState = globals.PASS_COMPLETE
self.beingOvertaken.remove(self.staticCars[i].id)
self.beenOvertaken.append(self.staticCars[i].id)
self.recentPassedCar = i
car.saveOvertakePoint()
self.fastCars = []
self.tFastCars = []
self.waitingForLaneTransition = True
self.waitingForLaneTransTime = helpers.conditionTime()
elif state == globals.PASS_BEHIND_CAR and dist < -9.0:
self.staticCars[i].beingOvertaken = False
globals.db['overtake'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'state', 'failed',
'carNum', self.staticCars[i].id, 'carDist', car.laneCrossDist, 'block', globals.flow.block,
'condtime', r3(helpers.conditionTime()), 'time', r3(helpers.currentTime())], True)
if globals.debug: print 'Overtake incomplete [slow car ID: ' + str(self.staticCars[i].id) +']'
def setBlinker(self, direction):
global db
self.activeBlinker = direction
d = 'L'
if (direction == globals.RIGHT_BLINKER):
d = 'R'
self.blinkerCooldown = 1.0
self.lastBlinkerDir = d
self.lastBlinkerTime = helpers.currentTime()
blinkSound = pyglet.resource.media('blinker.wav')
blinkSound.play()
globals.db['blinker'].addData(['pp', globals.participant, 'condition', globals.flow.getCondition(), 'direction', d, 'condtime', r3(helpers.conditionTime()), 'block', block,
'time', r3(helpers.currentTime())], True)