1 /*
2  * Copyright (c) 2009-2010 jMonkeyEngine
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are
7  * met:
8  *
9  * * Redistributions of source code must retain the above copyright
10  *   notice, this list of conditions and the following disclaimer.
11  *
12  * * Redistributions in binary form must reproduce the above copyright
13  *   notice, this list of conditions and the following disclaimer in the
14  *   documentation and/or other materials provided with the distribution.
15  *
16  * * Neither the name of 'jMonkeyEngine' nor the names of its contributors
17  *   may be used to endorse or promote products derived from this software
18  *   without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
22  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
23  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
24  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
25  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
26  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
27  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
28  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
29  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
30  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31  */
32 
33 package jme3test.tools;
34 
35 import com.jme3.app.SimpleApplication;
36 import com.jme3.bounding.BoundingBox;
37 import com.jme3.light.DirectionalLight;
38 import com.jme3.material.Material;
39 import com.jme3.material.MaterialList;
40 import com.jme3.math.ColorRGBA;
41 import com.jme3.math.Vector3f;
42 import com.jme3.post.SceneProcessor;
43 import com.jme3.renderer.RenderManager;
44 import com.jme3.renderer.ViewPort;
45 import com.jme3.renderer.queue.RenderQueue;
46 import com.jme3.scene.Geometry;
47 import com.jme3.scene.Spatial;
48 import com.jme3.scene.debug.WireBox;
49 import com.jme3.scene.plugins.ogre.MeshLoader;
50 import com.jme3.scene.plugins.ogre.OgreMeshKey;
51 import com.jme3.texture.FrameBuffer;
52 import java.util.ArrayList;
53 import java.util.HashSet;
54 import java.util.Set;
55 import jme3tools.optimize.FastOctnode;
56 import jme3tools.optimize.Octree;
57 
58 
59 public class TestOctree extends SimpleApplication implements SceneProcessor {
60 
61     private Octree tree;
62     private FastOctnode fastRoot;
63     private Geometry[] globalGeoms;
64     private BoundingBox octBox;
65 
66     private Set<Geometry> renderSet = new HashSet<Geometry>(300);
67     private Material mat, mat2;
68     private WireBox box = new WireBox(1,1,1);
69 
main(String[] args)70     public static void main(String[] args){
71         TestOctree app = new TestOctree();
72         app.start();
73     }
74 
simpleInitApp()75     public void simpleInitApp() {
76 //        this.flyCam.setMoveSpeed(2000);
77 //        this.cam.setFrustumFar(10000);
78         MeshLoader.AUTO_INTERLEAVE = false;
79 
80 //        mat = new Material(assetManager, "Common/MatDefs/Misc/WireColor.j3md");
81 //        mat.setColor("Color", ColorRGBA.White);
82 
83 //        mat2 = new Material(assetManager, "Common/MatDefs/Misc/ShowNormals.j3md");
84 
85         assetManager.registerLocator("quake3level.zip", "com.jme3.asset.plugins.ZipLocator");
86         MaterialList matList = (MaterialList) assetManager.loadAsset("Scene.material");
87         OgreMeshKey key = new OgreMeshKey("main.meshxml", matList);
88         Spatial scene = assetManager.loadModel(key);
89 
90 //        Spatial scene = assetManager.loadModel("Models/Teapot/teapot.obj");
91 //        scene.scale(3);
92 
93         DirectionalLight dl = new DirectionalLight();
94         dl.setColor(ColorRGBA.White);
95         dl.setDirection(new Vector3f(-1, -1, -1).normalize());
96         rootNode.addLight(dl);
97 
98         DirectionalLight dl2 = new DirectionalLight();
99         dl2.setColor(ColorRGBA.White);
100         dl2.setDirection(new Vector3f(1, -1, 1).normalize());
101         rootNode.addLight(dl2);
102 
103         // generate octree
104 //        tree = new Octree(scene, 20000);
105         tree = new Octree(scene, 50);
106         tree.construct();
107 
108         ArrayList<Geometry> globalGeomList = new ArrayList<Geometry>();
109         tree.createFastOctnodes(globalGeomList);
110         tree.generateFastOctnodeLinks();
111 
112         for (Geometry geom : globalGeomList){
113             geom.addLight(dl);
114             geom.addLight(dl2);
115             geom.updateGeometricState();
116         }
117 
118         globalGeoms = globalGeomList.toArray(new Geometry[0]);
119         fastRoot = tree.getFastRoot();
120         octBox = tree.getBound();
121 
122         viewPort.addProcessor(this);
123     }
124 
initialize(RenderManager rm, ViewPort vp)125     public void initialize(RenderManager rm, ViewPort vp) {
126     }
127 
reshape(ViewPort vp, int w, int h)128     public void reshape(ViewPort vp, int w, int h) {
129     }
130 
isInitialized()131     public boolean isInitialized() {
132         return true;
133     }
134 
preFrame(float tpf)135     public void preFrame(float tpf) {
136     }
137 
postQueue(RenderQueue rq)138     public void postQueue(RenderQueue rq) {
139         renderSet.clear();
140         //tree.generateRenderSet(renderSet, cam);
141         fastRoot.generateRenderSet(globalGeoms, renderSet, cam, octBox, true);
142 //        System.out.println("Geoms: "+renderSet.size());
143         int tris = 0;
144 
145         for (Geometry geom : renderSet){
146             tris += geom.getTriangleCount();
147 //            geom.setMaterial(mat2);
148             rq.addToQueue(geom, geom.getQueueBucket());
149         }
150 
151 //        Matrix4f transform = new Matrix4f();
152 //        transform.setScale(0.2f, 0.2f, 0.2f);
153 //        System.out.println("Tris: "+tris);
154 
155 //        tree.renderBounds(rq, transform, box, mat);
156 
157 //        renderManager.flushQueue(viewPort);
158     }
159 
postFrame(FrameBuffer out)160     public void postFrame(FrameBuffer out) {
161     }
162 
cleanup()163     public void cleanup() {
164     }
165 }
166