class Unit(Actor):
gravity = 30
def __init__(self, models = None, anims = None, sphereString = "**/CollisionSphere", game = None, xStart = 0, yStart = 0, zStart = 0, radius = 3):
Actor.__init__(self, models, anims)
self.game = game
self.health = 10
self.heightOffset = 3
#set up the position
self.setPos(xStart, yStart, zStart)
self.prevPosition = self.getPos()
#self.lastPosition = Point3()
self.vel = Vec3()
self.accel = Vec3(0, 0, -Unit.gravity)
#define the position that will be treated as the center of the map
self.wCenter = Point3(0, 0, 0)
#the radius of the sphere around this
self.radius = 3.5
#the base damage this unit deals upon collision
self.collisionAttackPower = 2.5
#set up Panda's collisions
#first the pusher
cSphere = CollisionSphere((0, 0, 1), 2)
cNode = CollisionNode("unit")
cNode.addSolid(cSphere)
cNode.setIntoCollideMask(BitMask32(PLAYER_ENEMY_OBJECTS))
cNode.setFromCollideMask(BitMask32(PLAYER_ENEMY_OBJECTS))
self.collisionNodePath = self.attachNewNode(cNode)
#self.collisionNodePath.show()
#set pattern for event sent on collision
# "%in" is substituted with the name of the into object, "%fn" is substituted with the name of the from object
#do the collision pusher
self.collisionPusher = CollisionHandlerPusher()
self.collisionPusher.addCollider(self.collisionNodePath, self)
self.collisionPusher.addInPattern("%fn-into-%in")
self.collisionPusher.addOutPattern("fn-out-%in")
game.cTrav.addCollider(self.collisionNodePath, self.collisionPusher)
#check for colllisions with the ground
self.groundRay = CollisionRay()
self.groundRay.setOrigin(0, 0, 4000)
self.groundRay.setDirection(0, 0, -1)
self.groundCol = CollisionNode('unitRay')
self.groundCol.addSolid(self.groundRay)
self.groundCol.setFromCollideMask(BitMask32(TERRAIN_RAY_MASK))
self.groundCol.setIntoCollideMask(BitMask32.allOff())
self.groundColNode = self.attachNewNode(self.groundCol)
self.groundHandler = CollisionHandlerQueue()
game.cTrav.addCollider(self.groundColNode, self.groundHandler)
#can be thought of as the inverse of the unit's mass
self.accelMultiplier = 45
self.friction = 1.7
self.disableFriction = False
self.nodePath = None
self.shootable = True
#finally set the python tag
self.setPythonTag("unit", self)
def applyForceFrom(self, magnitude, sourcePosition):
forceVector = self.getPos() - sourcePosition
forceVector.normalize()
forceVector *= magnitude
self.applyForce(forceVector)
def applyForce(self, forceVector):
self.accel += forceVector * self.accelMultiplier
def applyConstantVelocityFrom(self, magnitude, sourcePosition):
velVector = self.getPos() - sourcePosition
velVector.normalize()
velVector *= magnitude
self.applyConstantVelocity(velVector)
def applyConstantVelocity(self, velVector):
self.vel = velVector
def takeDamage(self, num):
self.health -= num
if self.health <= 0:
self.die()
def die(self):
self.game.actors[self.getName()] = None
self.delete()
def turn(self, magnitude):
pass
def update(self, time):
self.vel += self.accel * time
self.accel.set(0, 0, -Unit.gravity)
if not self.disableFriction:
self.vel -= self.vel * (self.friction * time)
self.setFluidPos(self.getPos() + self.vel * time)
self.setZ(max(-100, self.getZ()))
def collideWithUnit(self, other):
velDiff = self.vel - other.vel
if velDiff.lengthSquared() > 450:
Unit.takeDamage(self, other.collisionAttackPower)
Unit.takeDamage(other, self.collisionAttackPower)
self.vel *= 0.8
other.vel *= 0.8
def collideWithObstacle(self):
if self.vel.lengthSquared() > 500:
Unit.takeDamage(self, max(1, self.collisionAttackPower))
self.vel *= 0.5
def terrainCollisionCheck(self):
entries = []
length = self.groundHandler.getNumEntries()
for i in range(length):
entry = self.groundHandler.getEntry(i)
entries.append(entry)
entries.sort(lambda x, y: cmp(y.getSurfacePoint(render).getZ(), x.getSurfacePoint(render).getZ()))
if (len(entries) > 0):
for entry in entries:
if entry.getIntoNode().getName() == "craterCollisionPlane":
zVal = entry.getSurfacePoint(render).getZ()
if zVal >= MAX_HEIGHT:#apply a force toward the center
self.applyForce(self.wCenter - Point3((self.getX() * GROUND_REPULSION_MULTIPLIER), (self.getY() * GROUND_REPULSION_MULTIPLIER), 0))
'''
this is a little bit hackish, what is done is that a force in the x and y direction is created proportional to your distance from the origin. this will only
work effectively if the crater is xy centered in the environment
'''
else:
self.setZ(max(zVal, self.getZ()))
break
