# -*- coding: utf-8 -*-
"""
Octree Collisions

by fabius astelix @2010-02-06

Level: INTERMEDIATE

Now we introduce a new term: octree. This is mainly a way to reshape your mesh structure to be more efficient and have higher frame rates in your games. There is a deep David Rose's speech on this subject in the forum that should clear you all that, therefore I'll suggest you to read it here: http://www.panda3d.org/phpbb2/viewtopic.php?p=34405#34405
Here I'll just show you how to prepare your models and scene to use this important technique. Note that this technique will be effective just for collision meshes of an high polygon count - in low count meshes you won't notice appreciable performance improvements.

NOTE If you won't find here some line of code explained, probably you missed it in the previous steps - if you don't find there as well though, or still isn't clear for you, browse at http://www.panda3d.org/phpbb2/viewtopic.php?t=7918 and post your issue to the thread.
"""
from direct.showbase.DirectObject import DirectObject
from pandac.PandaModules import ActorNode, CollisionHandlerEvent, CollisionHandlerGravity, CollisionHandlerPusher, CollisionNode, CollisionSphere, CollisionTraverser, BitMask32, CollisionRay, NodePath
from direct.interval.IntervalGlobal import *

from pandac.PandaModules import loadPrcFileData
loadPrcFileData("", """sync-video 0
"""
)
import direct.directbase.DirectStart
#** snippet support routines - not concerning the tutorial part
import snipstuff

#=========================================================================
# Scenographic stuff
#=========================================================================

base.cam.setPos(40, -150, 35)

splash=snipstuff.splashCard()
snipstuff.info.append("Collision Octree")
snipstuff.info.append("A snippet to see performance improvement using octree splitting\nfor collision meshes.")
snipstuff.info.append("Change collision mesh and compare the frame rate difference\n\n1=single collision mesh\n2=octreefied collision mesh")
snipstuff.info_message("check out the frame rate meter above...")

snipstuff.info_show()

#=========================================================================
# Main
"""
Here we settle up an (apparently) simple scene with a wide terrain surface, where to compare 2 different floor colliders applied on it: the first is a whole single mesh with approx. 16.000 polygons and the other is the same former mesh but after being octreefied, using the octree script incorporated in the blender chicken exporter - have to know that the resulting model is not composed by a whole mesh anymore but rather by dozens of small mesh chunks that will relief the panda collision system for an extent you'll going to see by yourself with this snippet. The real deal with this tutorial is not much how to settle the collisions - explained in detail in former steps - but rather how to export in blender the mesh and how will change here the usual floor collision settings, because of the octree script. Check infos in oc3models.blend and down here to know what I mean.
"""
#=========================================================================

#** Collision system ignition
base.cTrav=CollisionTraverser()

#** The floor collision handler - all stuff explained in step3
avatarFloorHandler = CollisionHandlerGravity()
avatarFloorHandler.setGravity(9.81+25)
avatarFloorHandler.setMaxVelocity(100)

#** Collision masks
FLOOR_MASK=BitMask32.bit(1)
MASK_OFF=BitMask32.allOff()

#** Our steering avatar
avatarNP=base.render.attachNewNode(ActorNode('smileyNP'))
avatarNP.reparentTo(base.render)
avatar = loader.loadModel('smiley')
avatar.reparentTo(avatarNP)
avatar.setPos(0,0,1)
avatar.setCollideMask(MASK_OFF)
avatarNP.setPos(0,0,15)

#** Stick and set the ray collider to the avatar
raygeometry = CollisionRay(0, 0, 2, 0, 0, -1)
avatarRay = avatarNP.attachNewNode(CollisionNode('avatarRay'))
avatarRay.node().addSolid(raygeometry)
# let's mask our floor FROM collider
avatarRay.node().setFromCollideMask(FLOOR_MASK)
avatarRay.node().setIntoCollideMask(MASK_OFF)

#** This is the terrain map - differently than what we used to in previous steps, the collision solids will be loaded separately. See this below.
terrain = loader.loadModel("oc3ground")
terrain.reparentTo(render)
terrain.setCollideMask(MASK_OFF)
terrain.setScale(10)

#** Let's load ands mask our collision surfaces - the task is repeated 2 times: one for a single-mesh collider and the other for the octreefied one and the two colliders data is conveniently stored in the dictionary 'floorcolliderdata', for a later usage to show the performance difference between the two setups. The important thing to know here is that when your collision model mesh will be octreefied, the octree script will change a bit the mesh naming, changing indeed the root name, that we use here to find its geometry and mask it to be recognized by the floor collision handler - in a octreefied mesh the root name will always be (at the date I'm writing and until the script will changes) "octree-root" - just be advised.
# the colliders storage
floorcolliderdata={}
# this block of lines is to grab the data for the single-mesh collider:
model = loader.loadModel('groundc')
model.reparentTo(terrain)
collider = model.find("**/normal-root")
collider.node().setIntoCollideMask(MASK_OFF)
floorcolliderdata['normal_floor']=collider
# and this for the octreefied one. Afterwards the
model = loader.loadModel('groundco3')
model.reparentTo(terrain)
collider = model.find("**/octree-root")
collider.node().setIntoCollideMask(MASK_OFF)
floorcolliderdata['oc3fied_floor']=collider

#** as said in step1 we tells to our collision handlers who take part to the respective tasks: for the floor the avatar ray and the avatar nodepath...
avatarFloorHandler.addCollider(avatarRay, avatarNP)
# and then, as usual, we inform the main collision traverser that are the colliders and the relative handlers coupled with
base.cTrav.addCollider(avatarRay, avatarFloorHandler)

#** These two functions are here to switch from normal to octreefied floor colliders at runtime, just to visually see the difference, especially between the frame rate perfortmances.
def oc3_floor_collider():
  floorcolliderdata['oc3fied_floor'].node().setIntoCollideMask(FLOOR_MASK)
  floorcolliderdata['normal_floor'].node().setIntoCollideMask(MASK_OFF)
  snipstuff.infotext['message'].setText("Using ocreefied collision mesh")
#
def normal_floor_collider():
  floorcolliderdata['normal_floor'].node().setIntoCollideMask(FLOOR_MASK)
  floorcolliderdata['oc3fied_floor'].node().setIntoCollideMask(MASK_OFF)
  snipstuff.infotext['message'].setText("Using single mesh collider")
normal_floor_collider()

snipstuff.DO.accept('1', normal_floor_collider)
snipstuff.DO.accept('2', oc3_floor_collider)

#** Make the smiley roll on the the floor...
def avatarwalk(dt, vel):
  if vel[0]: avatar.setR(avatar.getR()+(40*vel[0]))
  if vel[1]: avatar.setP(avatar.getP()+(-38*vel[1]))
#... and jump
def avatarjump(dt):
  if avatarFloorHandler.isOnGround():
    avatarFloorHandler.addVelocity(20)

#** Now we're ready to activate the avatar steering and go
steering=snipstuff.avatar_steer(avatarNP, avatarwalk, avatarjump, fwspeed=12.)
steering.start()

splash.destroy()
run()