mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-28 14:12:52 -06:00
556 lines
24 KiB
Python
Executable file
556 lines
24 KiB
Python
Executable file
from direct.distributed import DistributedObjectAI
|
|
from direct.directnotify import DirectNotifyGlobal
|
|
from toontown.toonbase import ToontownGlobals
|
|
from pandac.PandaModules import *
|
|
from direct.fsm.FSM import FSM
|
|
from toontown.ai.ToonBarrier import *
|
|
from toontown.golf import GolfGlobals
|
|
import random
|
|
import math
|
|
|
|
class GolfHoleBase:
|
|
|
|
def __init__(self, canRender = 0):
|
|
self.canRender = canRender
|
|
self.recording = []
|
|
self.aVRecording = []
|
|
self.holePositions = []
|
|
self.grayCount = 0
|
|
self.skyContact = None
|
|
self.lastSkyContactPoint = None
|
|
self.doingRecording = 0
|
|
self.backAmount = 270
|
|
self.ballRocket = 0
|
|
self.inCount = 0
|
|
self.frame = 0
|
|
self.onSlick = 0
|
|
self.didHoleBreak = 0
|
|
return
|
|
|
|
def loadLevel(self):
|
|
tm = self.holeInfo['terrainModel']
|
|
self.terrainModel = loader.loadModel(tm)
|
|
td = self.holeInfo['physicsData']
|
|
if self.canRender:
|
|
self.terrainModel.reparentTo(render)
|
|
if self.canRender:
|
|
self.terrainModel.find('**/softSurface').setBin('ground', 0)
|
|
terrainData = self.terrainModel.find('**/softSurface')
|
|
grassData = terrainData.findAllMatches('**/grass*')
|
|
self.terrainData = []
|
|
for index in xrange(grassData.getNumPaths()):
|
|
someTerrainData = grassData[index]
|
|
terrainDataOde = OdeTriMeshData(someTerrainData)
|
|
self.meshDataList.append(terrainDataOde)
|
|
terrainGeomOde = OdeTriMeshGeom(self.space, terrainDataOde)
|
|
self.geomDataList.append(terrainGeomOde)
|
|
terrainGeomOde.setCollideBits(BitMask32(4026531840L))
|
|
terrainGeomOde.setCategoryBits(BitMask32(240))
|
|
self.space.setSurfaceType(terrainGeomOde, GolfGlobals.GRASS_SURFACE)
|
|
self.space.setCollideId(terrainGeomOde, 2)
|
|
|
|
slickData = terrainData.findAllMatches('**/slick*')
|
|
self.terrainData = []
|
|
for index in xrange(slickData.getNumPaths()):
|
|
someTerrainData = slickData[index]
|
|
terrainDataOde = OdeTriMeshData(someTerrainData)
|
|
self.meshDataList.append(terrainDataOde)
|
|
terrainGeomOde = OdeTriMeshGeom(self.space, terrainDataOde)
|
|
self.geomDataList.append(terrainGeomOde)
|
|
terrainGeomOde.setCollideBits(BitMask32(4026531840L))
|
|
terrainGeomOde.setCategoryBits(BitMask32(240))
|
|
self.space.setSurfaceType(terrainGeomOde, GolfGlobals.SLICK_SURFACE)
|
|
self.space.setCollideId(terrainGeomOde, GolfGlobals.SLICK_COLLIDE_ID)
|
|
|
|
cupData = terrainData.find('**/hole*')
|
|
cupData = OdeTriMeshData(cupData)
|
|
self.meshDataList.append(cupData)
|
|
cupGeom = OdeTriMeshGeom(self.space, cupData)
|
|
self.geomDataList.append(cupGeom)
|
|
cupGeom.setCollideBits(BitMask32(4026531840L))
|
|
cupGeom.setCategoryBits(BitMask32(240))
|
|
self.space.setSurfaceType(cupGeom, GolfGlobals.HOLE_SURFACE)
|
|
self.space.setCollideId(cupGeom, GolfGlobals.HOLE_CUP_COLLIDE_ID)
|
|
if self.canRender:
|
|
self.golfBarrier = self.terrainModel.find('**/collision1')
|
|
if not self.golfBarrier.isEmpty():
|
|
golfBarrierCollection = self.terrainModel.findAllMatches('**/collision?')
|
|
for i in xrange(golfBarrierCollection.getNumPaths()):
|
|
oneBarrier = golfBarrierCollection.getPath(i)
|
|
if oneBarrier != self.golfBarrier:
|
|
oneBarrier.wrtReparentTo(self.golfBarrier)
|
|
|
|
self.golfBarrier.hide()
|
|
else:
|
|
self.notify.warning('Could not find collision1 node ---------')
|
|
self.hardSurfaceNodePath = self.terrainModel.find('**/hardSurface')
|
|
if self.canRender:
|
|
self.terrainModel.find('**/hardSurface').setBin('ground', 0)
|
|
self.loadBlockers()
|
|
hardData = OdeTriMeshData(self.hardSurfaceNodePath)
|
|
self.meshDataList.append(hardData)
|
|
hardGeom = OdeTriMeshGeom(self.space, hardData)
|
|
self.geomDataList.append(hardGeom)
|
|
hardGeom.setCollideBits(BitMask32(4026531840L))
|
|
hardGeom.setCategoryBits(BitMask32(240))
|
|
self.space.setCollideId(hardGeom, 3)
|
|
hardSurface = self.space.getSurfaceType(hardGeom)
|
|
self.notify.debug('hardSurface = %s' % hardSurface)
|
|
if self.notify.getDebug():
|
|
self.notify.debug('self.hardGeom')
|
|
hardGeom.write()
|
|
self.notify.debug(' -')
|
|
self.holeBottomNodePath = self.terrainModel.find('**/holebottom0')
|
|
if self.holeBottomNodePath.isEmpty():
|
|
self.holeBottomPos = Vec3(*self.holeInfo['holePos'][0])
|
|
else:
|
|
self.holeBottomPos = self.holeBottomNodePath.getPos()
|
|
self.holePositions.append(self.holeBottomPos)
|
|
|
|
def isBallInHole(self, ball):
|
|
retval = False
|
|
for holePos in self.holePositions:
|
|
displacement = ball.getPosition() - holePos
|
|
length = displacement.length()
|
|
self.notify.debug('hole %s length=%s' % (holePos, length))
|
|
if length <= GolfGlobals.DistanceToBeInHole * 0.5:
|
|
retval = True
|
|
break
|
|
|
|
return retval
|
|
|
|
def createRays(self):
|
|
self.notify.debug('createRays')
|
|
body = OdeBody(self.world)
|
|
self.ballRay = OdeRayGeom(self.space, 50.0)
|
|
self.ballRay.setBody(body)
|
|
self.ballRay.setOffsetRotation(Mat3(1, 0, 0, 0, -1, 0, 0, 0, -1))
|
|
self.ballRay.setOffsetPosition(0, 0, 0.0)
|
|
self.ballRay.setCollideBits(BitMask32(16773375))
|
|
self.ballRay.setCategoryBits(BitMask32(4278190080L))
|
|
self.ballRayBody = body
|
|
self.space.setCollideId(self.ballRay, GolfGlobals.OOB_RAY_COLLIDE_ID)
|
|
self.rayList.append(self.ballRay)
|
|
self.rayList.append(self.ballRayBody)
|
|
self.skyRay = OdeRayGeom(self.space, 100.0)
|
|
self.skyRay.setCollideBits(BitMask32(240))
|
|
self.skyRay.setCategoryBits(BitMask32(0))
|
|
self.skyRay.setRotation(Mat3(1, 0, 0, 0, -1, 0, 0, 0, -1))
|
|
self.space.setCollideId(self.skyRay, GolfGlobals.SKY_RAY_COLLIDE_ID)
|
|
self.rayList.append(self.skyRay)
|
|
|
|
def delete(self):
|
|
self.ballRay = None
|
|
self.skyRay = None
|
|
self.recording = None
|
|
self.avRecording = None
|
|
self.llv = None
|
|
return
|
|
|
|
def initRecord(self):
|
|
del self.recording
|
|
self.recording = []
|
|
del self.aVRecording
|
|
self.aVRecording = []
|
|
self.skipFrame = 0.0
|
|
self.frame = 0
|
|
self.tXYMax = 1.0
|
|
self.tZMax = 1.0
|
|
self.tXYMin = 0.1
|
|
self.tZMin = 0.1
|
|
self.skyContact = 1
|
|
self.doingRecording = 1
|
|
self.ballRocket = 0
|
|
self.inCount = 0
|
|
self.ballInHoleFrame = 0
|
|
self.ballTouchedHoleFrame = 0
|
|
self.ballFirstTouchedHoleFrame = 0
|
|
self.ballLastTouchedGrass = 0
|
|
self.hasReset = 0
|
|
self.resetAt = 100000
|
|
self.greenIn = 0
|
|
for key in self.commonObjectDict:
|
|
self.commonObjectDict[key][2].enable()
|
|
|
|
def checkCommonObjectsNeedPass(self):
|
|
for index in self.commonObjectDict:
|
|
if self.commonObjectDict[index][1] in [4]:
|
|
return 1
|
|
|
|
return 0
|
|
|
|
def checkInRadius(self, ball):
|
|
smallestDist = None
|
|
for index in self.commonObjectDict:
|
|
if self.commonObjectDict[index][1] in [4]:
|
|
radius = self.commonObjectDict[index][8]
|
|
mover = self.commonObjectDict[index][2]
|
|
diffX = ball.getPosition()[0] - mover.getPosition()[0]
|
|
diffY = ball.getPosition()[1] - mover.getPosition()[1]
|
|
diffZ = ball.getPosition()[2] - mover.getPosition()[2]
|
|
dist = math.sqrt(diffX * diffX + diffY * diffY + diffZ * diffZ)
|
|
if dist < radius:
|
|
if not smallestDist or smallestDist[1] > dist:
|
|
smallestDist = [radius, dist]
|
|
self.notify.debug('Ball Pos %s\nMover Pos %s' % (ball.getPosition(), mover.getPosition()))
|
|
|
|
return smallestDist
|
|
|
|
def trackRecordBodyFlight(self, ball, cycleTime, power, startPos, dirX, dirY):
|
|
self.notify.debug('trackRecordBodyFlight')
|
|
self.ballInHoleFrame = 0
|
|
self.ballTouchedHoleFrame = 0
|
|
self.ballFirstTouchedHoleFrame = 0
|
|
self.ballLastTouchedGrass = 0
|
|
startTime = globalClock.getRealTime()
|
|
self.notify.debug('start position %s' % startPos)
|
|
self.swingTime = cycleTime
|
|
frameCount = 0
|
|
lift = 0
|
|
startTime = GolfGlobals.BALL_CONTACT_FRAME / 24
|
|
startFrame = int(startTime * self.FPS)
|
|
for frame in xrange(int(startFrame)):
|
|
self.simulate()
|
|
self.setTimeIntoCycle(self.swingTime + float(frameCount) * self.DTAStep)
|
|
frameCount += 1
|
|
|
|
forceMove = 1500
|
|
if power > 50:
|
|
lift = 0
|
|
self.didHoleBreak = 0
|
|
ball.setPosition(startPos)
|
|
ball.setLinearVel(0.0, 0.0, 0.0)
|
|
ball.setAngularVel(0.0, 0.0, 0.0)
|
|
ball.enable()
|
|
self.preStep()
|
|
self.simulate()
|
|
self.postStep()
|
|
ball.enable()
|
|
ball.addForce(Vec3(dirX * forceMove * power / 100.0, dirY * forceMove * power / 100.0, lift))
|
|
self.initRecord()
|
|
self.llv = None
|
|
self.lastSkyContactPoint = None
|
|
ran = 0
|
|
self.record(ball)
|
|
self.comObjNeedPass = self.checkCommonObjectsNeedPass()
|
|
self.notify.debug('self.comObjNeedPass %s' % self.comObjNeedPass)
|
|
firstDisabled = -1
|
|
reEnabled = 0
|
|
lastFrameEnabled = 0
|
|
checkFrames = self.FPS * (self.timingCycleLength + 1.0)
|
|
hasPrinted = 0
|
|
while ball.isEnabled() and len(self.recording) < 2100 or self.comObjNeedPass or len(self.recording) < 10:
|
|
ran = 1
|
|
if len(self.recording) > 2100 and not hasPrinted:
|
|
self.notify.debug('recording too long %s' % len(self.recording))
|
|
hasPrinted = 1
|
|
ball.disable()
|
|
self.preStep()
|
|
self.simulate()
|
|
self.setTimeIntoCycle(self.swingTime + float(frameCount) * self.DTAStep)
|
|
frameCount += 1
|
|
self.postStep()
|
|
self.record(ball)
|
|
if self.comObjNeedPass:
|
|
if firstDisabled == -1 and not ball.isEnabled():
|
|
firstDisabled = self.frame
|
|
self.notify.debug('firstDisabled %s' % firstDisabled)
|
|
check = self.checkInRadius(ball)
|
|
if check == None:
|
|
self.comObjNeedPass = 0
|
|
self.notify.debug('out radius')
|
|
else:
|
|
self.notify.debug('in radius %s dist %s' % (check[0], check[1]))
|
|
elif ball.isEnabled() and firstDisabled != -1 and not reEnabled:
|
|
reEnabled = self.frame
|
|
self.notify.debug('reEnabled %s' % reEnabled)
|
|
if reEnabled:
|
|
if self.frame > reEnabled + checkFrames:
|
|
self.comObjNeedPass = 0
|
|
self.notify.debug('renable limit passed')
|
|
elif self.frame > 2100 + checkFrames:
|
|
self.comObjNeedPass = 0
|
|
print 'recording limit passed comObj'
|
|
if ball.isEnabled():
|
|
lastFrameEnabled = self.frame
|
|
|
|
self.notify.debug('lastFrameEnabled %s' % lastFrameEnabled)
|
|
if lastFrameEnabled < 3:
|
|
lastFrameEnabled = 3
|
|
self.record(ball)
|
|
self.notify.debug('Frames %s' % self.frame)
|
|
midTime = globalClock.getRealTime()
|
|
self.recording = self.recording[:lastFrameEnabled]
|
|
self.aVRecording = self.aVRecording[:lastFrameEnabled]
|
|
self.frame = lastFrameEnabled
|
|
self.processRecording()
|
|
self.processAVRecording()
|
|
self.notify.debug('Recording End time %s cycle %s len %s avLen %s' % (self.timingSimTime,
|
|
self.getCycleTime(),
|
|
len(self.recording),
|
|
len(self.aVRecording)))
|
|
length = len(self.recording) - 1
|
|
x = self.recording[length][1]
|
|
y = self.recording[length][2]
|
|
z = self.recording[length][3]
|
|
endTime = globalClock.getRealTime()
|
|
diffTime = endTime - startTime
|
|
self.doingRecording = 0
|
|
fpsTime = self.frame / diffTime
|
|
self.notify.debug('Time Start %s Mid %s End %s Diff %s Fps %s frames %s' % (startTime,
|
|
midTime,
|
|
endTime,
|
|
diffTime,
|
|
fpsTime,
|
|
self.frame))
|
|
return Vec3(x, y, z)
|
|
|
|
def record(self, ball):
|
|
self.recording.append((self.frame,
|
|
ball.getPosition()[0],
|
|
ball.getPosition()[1],
|
|
ball.getPosition()[2]))
|
|
self.aVRecording.append((self.frame,
|
|
ball.getAngularVel()[0],
|
|
ball.getAngularVel()[1],
|
|
ball.getAngularVel()[2]))
|
|
if self.frame > 50 and not self.frame % 13:
|
|
curFrame = self.recording[self.frame]
|
|
pastFrame5 = self.recording[self.frame - 11]
|
|
pastFrame10 = self.recording[self.frame - 34]
|
|
currPosA = Vec3(curFrame[1], curFrame[2], curFrame[3])
|
|
past5PosA = Vec3(pastFrame5[1], pastFrame5[2], pastFrame5[3])
|
|
past10PosA = Vec3(pastFrame10[1], pastFrame10[2], pastFrame10[3])
|
|
displacement1 = currPosA - past5PosA
|
|
displacement2 = currPosA - past10PosA
|
|
if displacement1.lengthSquared() < 0.002 and displacement2.lengthSquared() < 0.002 and not self.grayCount and not self.onSlick:
|
|
ball.disable()
|
|
self.frame += 1
|
|
|
|
def preStep(self):
|
|
if hasattr(self, 'ballRay'):
|
|
bp = self.curGolfBall().getPosition()
|
|
self.ballRayBody.setPosition(bp[0], bp[1], bp[2])
|
|
self.skyRay.setPosition(bp[0], bp[1], 50.0)
|
|
|
|
def getOrderedContacts(self, entry):
|
|
c0 = self.space.getCollideId(entry.getGeom1())
|
|
c1 = self.space.getCollideId(entry.getGeom2())
|
|
if c0 > c1:
|
|
return (c1, c0)
|
|
else:
|
|
return (c0, c1)
|
|
|
|
def postStep(self):
|
|
if self.canRender:
|
|
self.translucentLastFrame = self.translucentCurFrame[:]
|
|
self.translucentCurFrame = []
|
|
self.onSlick = 0
|
|
rayCount = 0
|
|
skyRayHitPos = None
|
|
ballRayHitPos = None
|
|
bp = self.curGolfBall().getPosition()
|
|
for entry in self.colEntries:
|
|
c0, c1 = self.getOrderedContacts(entry)
|
|
x, y, z = entry.getContactPoint(0)
|
|
if c0 == GolfGlobals.OOB_RAY_COLLIDE_ID or c1 == GolfGlobals.OOB_RAY_COLLIDE_ID:
|
|
rayCount += 1
|
|
if self.canRender:
|
|
if self.currentGolfer:
|
|
self.ballShadowDict[self.currentGolfer].setPos(x, y, z + 0.1)
|
|
if c1 == GolfGlobals.GRASS_COLLIDE_ID or c1 == GolfGlobals.HARD_COLLIDE_ID:
|
|
if self.curGolfBall().getPosition()[2] < z + 0.2:
|
|
ballRayHitPos = Vec3(x, y, z)
|
|
if c0 == GolfGlobals.OOB_RAY_COLLIDE_ID and c1 == GolfGlobals.SLICK_COLLIDE_ID:
|
|
self.onSlick = 1
|
|
elif c0 == GolfGlobals.OOB_RAY_COLLIDE_ID and c1 == GolfGlobals.HARD_COLLIDE_ID:
|
|
self.onSlick = 1
|
|
if c0 == GolfGlobals.GRASS_COLLIDE_ID and c1 == GolfGlobals.SKY_RAY_COLLIDE_ID:
|
|
self.lastSkyContactPoint = (x, y, z)
|
|
if self.curGolfBall().getPosition()[2] < z + 0.2 and rayCount == 0:
|
|
if self.skyContact in [1, 2]:
|
|
skyRayHitPos = Vec3(x, y, z)
|
|
self.skyContact += 1
|
|
if self.doingRecording:
|
|
if c0 == GolfGlobals.OOB_RAY_COLLIDE_ID or c1 == GolfGlobals.OOB_RAY_COLLIDE_ID:
|
|
rayCount += 1
|
|
if c1 == GolfGlobals.GRASS_COLLIDE_ID:
|
|
self.greenIn = self.frame
|
|
self.llv = self.curGolfBall().getLinearVel()
|
|
elif GolfGlobals.BALL_COLLIDE_ID in [c0, c1] and GolfGlobals.HOLE_CUP_COLLIDE_ID in [c0, c1]:
|
|
self.ballTouchedHoleFrame = self.frame
|
|
ballUndersideZ = self.curGolfBall().getPosition()[2] - 0.05
|
|
if z < ballUndersideZ:
|
|
if not self.ballInHoleFrame:
|
|
self.ballInHoleFrame = self.frame
|
|
if self.ballFirstTouchedHoleFrame < self.ballLastTouchedGrass:
|
|
self.ballFirstTouchedHoleFrame = self.frame
|
|
if self.isBallInHole(self.curGolfBall()) and self.didHoleBreak == 0:
|
|
self.comObjNeedPass = 0
|
|
ballLV = self.curGolfBall().getLinearVel()
|
|
ballAV = self.curGolfBall().getAngularVel()
|
|
self.curGolfBall().setLinearVel(0.5 * ballLV[0], 0.5 * ballLV[1], 0.5 * ballLV[2])
|
|
self.curGolfBall().setAngularVel(0.5 * ballAV[0], 0.5 * ballAV[1], 0.5 * ballAV[2])
|
|
self.notify.debug('BALL IN THE HOLE!!! FOO!')
|
|
self.didHoleBreak = 1
|
|
return
|
|
elif GolfGlobals.BALL_COLLIDE_ID in [c0, c1] and GolfGlobals.GRASS_COLLIDE_ID in [c0, c1]:
|
|
if self.ballInHoleFrame:
|
|
self.ballInHoleFrame = 0
|
|
self.notify.debug('setting ballInHoleFrame=0')
|
|
self.ballLastTouchedGrass = self.frame
|
|
elif self.canRender:
|
|
if c0 == GolfGlobals.TOON_RAY_COLLIDE_ID or c1 == GolfGlobals.TOON_RAY_COLLIDE_ID:
|
|
self.toonRayCollisionCallback(x, y, z)
|
|
if GolfGlobals.CAMERA_RAY_COLLIDE_ID in [c0, c1] and GolfGlobals.WINDMILL_BASE_COLLIDE_ID in [c0, c1]:
|
|
self.translucentCurFrame.append(self.windmillFanNodePath)
|
|
self.translucentCurFrame.append(self.windmillBaseNodePath)
|
|
if GolfGlobals.BALL_COLLIDE_ID in [c0, c1] and GolfGlobals.GRASS_COLLIDE_ID not in [c0, c1]:
|
|
self.handleBallHitNonGrass(c0, c1)
|
|
|
|
if not self.curGolfBall().isEnabled():
|
|
return
|
|
if rayCount == 0:
|
|
self.notify.debug('out of bounds detected!')
|
|
self.grayCount += 1
|
|
self.outCommon = self.getCommonObjectData()
|
|
self.inCount = 0
|
|
if skyRayHitPos:
|
|
self.curGolfBall().setPosition(skyRayHitPos[0], skyRayHitPos[1], skyRayHitPos[2] + 0.27)
|
|
self.notify.debug('SKY RAY ADJUST?')
|
|
else:
|
|
if self.grayCount > 1:
|
|
self.notify.debug('Back in bounds')
|
|
self.grayCount = 0
|
|
self.inCount += 1
|
|
if ballRayHitPos:
|
|
self.curGolfBall().setPosition(ballRayHitPos[0], ballRayHitPos[1], ballRayHitPos[2] + 0.245)
|
|
ballRayHitPos = None
|
|
if self.doingRecording:
|
|
self.notify.debug('BALL RAY ADJUST!')
|
|
self.notify.debug('%s' % self.curGolfBall().getLinearVel())
|
|
if self.ballRocket > 0 and self.inCount > 1:
|
|
self.ballRocket -= 1
|
|
rocketVel = self.curGolfBall().getLinearVel()
|
|
self.curGolfBall().setLinearVel(2.0 * rocketVel[0], 2.0 * rocketVel[1], 2.0 * rocketVel[2])
|
|
self.notify.debug('ROCKET!!!!')
|
|
if self.grayCount > self.backAmount and self.doingRecording:
|
|
if self.greenIn > 2:
|
|
self.greenIn -= 2
|
|
if self.greenIn > self.resetAt:
|
|
self.greenIn = self.resetAt - 10
|
|
if self.greenIn < 0 or self.hasReset > 3:
|
|
self.greenIn = 0
|
|
self.hasReset += 1
|
|
self.notify.debug('BALL RESET frame %s greenIn %s resetAt %s' % (self.frame, self.greenIn, self.resetAt))
|
|
self.useCommonObjectData(self.outCommon)
|
|
self.curGolfBall().setPosition(self.recording[self.greenIn][1], self.recording[self.greenIn][2], self.recording[self.greenIn][3] + 0.27)
|
|
self.curGolfBall().setAngularVel(0, 0, 0)
|
|
if self.hasReset < 3 and self.llv:
|
|
self.ballRocket += 1
|
|
self.notify.debug(' BRAKE!!!!')
|
|
self.curGolfBall().setLinearVel(0.5 * self.llv[0], 0.5 * self.llv[1], 0.5 * self.llv[2])
|
|
else:
|
|
self.notify.debug('back disable %s' % self.frame)
|
|
if self.lastSkyContactPoint:
|
|
self.curGolfBall().setPosition(self.lastSkyContactPoint[0], self.lastSkyContactPoint[1], self.lastSkyContactPoint[2] + 0.27)
|
|
self.curGolfBall().setLinearVel(0, 0, 0)
|
|
self.curGolfBall().disable()
|
|
self.recording = self.recording[:self.greenIn]
|
|
self.aVRecording = self.aVRecording[:self.greenIn]
|
|
self.frame = self.greenIn
|
|
self.resetAt = self.greenIn
|
|
self.grayCount = 0
|
|
if self.ballFirstTouchedHoleFrame > self.frame:
|
|
self.notify.debug('reseting first touched hole, self.frame=%d self.ballFirstTouchedHoleFrame=%d' % (self.frame, self.ballFirstTouchedHoleFrame))
|
|
self.ballFirstTouchedHoleFrame = 0
|
|
if self.ballLastTouchedGrass > self.frame:
|
|
self.ballLastTouchedGrass = 0
|
|
return
|
|
|
|
def processRecording(self, errorMult = 1.0):
|
|
self.notify.debug('processRecording')
|
|
lastFrame = self.recording[len(self.recording) - 1][0]
|
|
countRemovals = 0
|
|
for frame in self.recording:
|
|
if frame[0] == 0 or frame[0] == lastFrame:
|
|
pass
|
|
else:
|
|
index = self.recording.index(frame)
|
|
prevFrame = self.recording[index - 1]
|
|
nextFrame = self.recording[index + 1]
|
|
if self.predict(frame, prevFrame, nextFrame, errorMult):
|
|
self.recording.remove(frame)
|
|
countRemovals += 1
|
|
|
|
if countRemovals > 5:
|
|
self.processRecording()
|
|
elif len(self.recording) > 120:
|
|
self.processRecording(errorMult * 1.25)
|
|
else:
|
|
for frame in self.recording:
|
|
pass
|
|
|
|
def processAVRecording(self, errorMult = 1.0, trials = 0):
|
|
self.notify.debug('processAVRecording')
|
|
lastFrame = self.recording[len(self.recording) - 1][0]
|
|
countRemovals = 0
|
|
countTrials = trials
|
|
for frame in self.aVRecording:
|
|
if frame[0] == 0 or frame[0] == lastFrame:
|
|
pass
|
|
else:
|
|
index = self.aVRecording.index(frame)
|
|
prevFrame = self.aVRecording[index - 1]
|
|
nextFrame = self.aVRecording[index + 1]
|
|
if self.predictAV(frame, prevFrame, nextFrame, errorMult):
|
|
self.aVRecording.remove(frame)
|
|
countRemovals += 1
|
|
else:
|
|
countTrials += 1
|
|
|
|
if countRemovals > 5:
|
|
self.processAVRecording(errorMult, countTrials)
|
|
elif len(self.aVRecording) > 80:
|
|
self.processAVRecording(errorMult * 1.25, countTrials)
|
|
else:
|
|
for frame in self.aVRecording:
|
|
pass
|
|
|
|
def predict(self, frame, sourceFrame, destFrame, errorMult = 1.0):
|
|
tXY = 0.05 * errorMult
|
|
tZ = 0.05 * errorMult
|
|
projLength = destFrame[0] - sourceFrame[0]
|
|
projPen = destFrame[0] - frame[0]
|
|
propSource = float(projPen) / float(projLength)
|
|
propDest = 1.0 - propSource
|
|
projX = sourceFrame[1] * propSource + destFrame[1] * propDest
|
|
projY = sourceFrame[2] * propSource + destFrame[2] * propDest
|
|
projZ = sourceFrame[3] * propSource + destFrame[3] * propDest
|
|
varX = abs(projX - frame[1])
|
|
varY = abs(projY - frame[2])
|
|
varZ = abs(projZ - frame[3])
|
|
if varX > tXY or varY > tXY or varZ > tZ:
|
|
return 0
|
|
else:
|
|
return 1
|
|
|
|
def predictAV(self, frame, sourceFrame, destFrame, errorMult = 1.0):
|
|
tXYZ = 1.5 * errorMult
|
|
projLength = destFrame[0] - sourceFrame[0]
|
|
projPen = destFrame[0] - frame[0]
|
|
propSource = float(projPen) / float(projLength)
|
|
propDest = 1.0 - propSource
|
|
projX = sourceFrame[1] * propSource + destFrame[1] * propDest
|
|
projY = sourceFrame[2] * propSource + destFrame[2] * propDest
|
|
projZ = sourceFrame[3] * propSource + destFrame[3] * propDest
|
|
varX = abs(projX - frame[1])
|
|
varY = abs(projY - frame[2])
|
|
varZ = abs(projZ - frame[3])
|
|
if varX > tXYZ or varY > tXYZ or varZ > tXYZ:
|
|
return 0
|
|
else:
|
|
return 1
|
|
|
|
def handleBallHitNonGrass(self, c0, c1):
|
|
pass
|