#-*- encoding: utf_8 -*-
from pandac.PandaModules import *
loadPrcFileData("editor-startup", "show-frame-rate-meter #t")
loadPrcFileData("", "win-size 1024 768")
from direct.directbase import DirectStart

from random import random      
import time

from direct.gui.OnscreenText import OnscreenText    
from direct.gui.DirectGui import *

class LoaderFrame():
       """ Класс для отображения загрузки
       """
       def __init__(self):
           bk_text = "..."
           self.frame = DirectFrame(frameSize=(-1*base.getAspectRatio(), 1*base.getAspectRatio(), -1, 1))
           self.textObject = OnscreenText(text = bk_text, pos = (0, -0.02),    
                      scale = 0.07,fg=(0,0,1,1),    
                      align=TextNode.ACenter,    
                      mayChange=1)
           self.bar = DirectWaitBar(text = "", value = 0, pos = (0,.4,.4),    
                      range = 100, frameColor = (1,1,1,1))
           self.textObject.reparentTo(self.bar)

       def progress(self, arg):
           self.bar['value'] = arg
           text = 'Progress is: %i' % arg
           self.textObject.setText(text)
           base.graphicsEngine.renderFrame()
              
       def hide(self):
           self.textObject.hide()
           self.frame.hide()
           self.bar.hide()
          
#---------------------- Основной тест ---------------------
start = time.time()
models = []
for i in xrange(3):
       models.append(loader.loadModel("box"))
       tex = loader.loadTexture('%i.png' % (i+1))    
       tex.setMagfilter(Texture.FTLinearMipmapLinear)    
       tex.setMinfilter(Texture.FTLinearMipmapLinear)    
       models[i].setTexture(tex)

hf = PNMImage()
hf.read('hf.png')
hfx = hf.getReadXSize()
hfy = hf.getReadYSize()
bx, by = 5, 5 # кол-во RBC X на Y
bs = 40       # сторона одного RBC в кубиках )
lf = LoaderFrame()
lf.progress(0)
for rx in xrange(bx):
       for ry in xrange(by):
           #print '%i/%i RBC of %i/%i' % (rx, ry, bx - 1, by - 1)
           lf.progress(int((rx * by + ry + 1) * (100/(bx*by))))
           rbc = RigidBodyCombiner('rbc_%i_%i' % (rx, ry))
           rbcnp = NodePath(rbc)      
           rbcnp.reparentTo(render)      
           for x in xrange(bs):
               for y in xrange(bs):
                   px = x + rx*bs
                   py = y + ry*bs
                   pz = int(hf.getGray(int((hfx/((bx+1)*bs))*px),    
                      int((hfy/((by+1)*bs))*py)) * 10)
                   if pz < 2:
                       f = models[2].copyTo(rbcnp)
                   elif pz < 9:
                       f = models[1].copyTo(rbcnp)
                   else:
                       f = models[0].copyTo(rbcnp)
                   f.setPos(px, py, pz)
           rbcnp.flattenLight()
           rbc.collect()      
           f.detachNode()

lf.hide()

dlight = DirectionalLight('dlight')     
dlight.setColor(VBase4(0.8, 0.8, 0.5, 1))     
dlnp = render.attachNewNode(dlight)     
dlnp.setHpr(0, -60, 0)     
render.setLight(dlnp)     

colour = (0.5,0.8,0.8)
linfog = Fog("A linear-mode Fog node")
linfog.setColor(*colour)
linfog.setLinearRange(0,320)
linfog.setLinearFallback(0,500,550)
fn = base.camera.attachNewNode(linfog)
fn.setY(250)
render.setFog(linfog)
base.setBackgroundColor(*colour)

render.analyze()

print 'Load at %.2f sec' % (time.time() - start)

run()