Mercurial > cortex
view org/util.org @ 76:5e75b616ca60
going to work on constraints for the joints in the hand.
author | Robert McIntyre <rlm@mit.edu> |
---|---|
date | Wed, 28 Dec 2011 23:12:32 -0700 |
parents | fb810a2c50c2 |
children | 77b506ac64f3 |
line wrap: on
line source
1 #+title: Clojure Utilities for jMonkeyEngine32 #+author: Robert McIntyre3 #+email: rlm@mit.edu4 #+description:5 #+keywords: JME3, clojure, import, utilities6 #+SETUPFILE: ../../aurellem/org/setup.org7 #+INCLUDE: ../../aurellem/org/level-0.org9 [TABLE-OF-CONTENTS]11 These are a collection of functions to make programming jMonkeyEngine12 in clojure easier.14 * Imports16 #+name: import17 #+begin_src clojure :results silent18 (ns cortex.import19 (:require swank.util.class-browse))21 (defn permissive-import22 [classname]23 (eval `(try (import '~classname)24 (catch java.lang.Exception e#25 (println "couldn't import " '~classname))))26 classname)28 (defn jme-class? [classname]29 (and30 (.startsWith classname "com.jme3.")31 ;; Don't import the Lwjgl stuff since it can throw exceptions32 ;; upon being loaded.33 (not (re-matches #".*Lwjgl.*" classname))))35 (defn jme-classes36 "returns a list of all jme3 classes"37 []38 (filter39 jme-class?40 (map :name41 swank.util.class-browse/available-classes)))43 (defn mega-import-jme344 "Import ALL the jme classes. For REPL use."45 []46 (doall47 (map (comp permissive-import symbol) (jme-classes))))48 #+end_src50 jMonkeyEngine3 has a plethora of classes which can be overwhelming to51 manage. This code uses reflection to import all of them. Once I'm52 happy with the general structure of a namespace I can deal with53 importing only the classes it actually needs.55 The =mega-import-jme3= is quite usefull for debugging purposes since56 it allows completion for almost all of JME's classes from the REPL.58 Out of curiousity, let's see just how many classes =mega-import-jme3=59 imports:61 #+begin_src clojure :exports both :results output62 (println (clojure.core/count (cortex.import/jme-classes)) "classes")63 #+end_src65 #+results:66 : 955 classes69 * Utilities71 The utilities here come in three main groups:72 - Changing settings in a running =Application=73 - Creating objects74 - Debug Actions75 - Visualizing objects77 *** Changing Settings79 #+name: util80 #+begin_src clojure81 (ns cortex.util82 "Utility functions for making jMonkeyEngine3 easier to program from83 clojure."84 {:author "Robert McIntyre"}85 (:use cortex.world)86 (:use clojure.contrib.def)87 (:import com.jme3.math.Vector3f)88 (:import com.jme3.math.Quaternion)89 (:import com.jme3.asset.TextureKey)90 (:import com.jme3.bullet.control.RigidBodyControl)91 (:import com.jme3.bullet.collision.shapes.GImpactCollisionShape)92 (:import com.jme3.scene.shape.Box)93 (:import com.jme3.scene.Node)94 (:import com.jme3.scene.shape.Sphere)95 (:import com.jme3.light.AmbientLight)96 (:import com.jme3.light.DirectionalLight)97 (:import com.jme3.math.ColorRGBA)98 (:import com.jme3.bullet.BulletAppState)99 (:import com.jme3.material.Material)100 (:import com.jme3.scene.Geometry)101 (:import (java.util.logging Level Logger)))105 (defvar println-repl106 (bound-fn [& args] (apply println args))107 "println called from the LWJGL thread will not go to the REPL, but108 instead to whatever terminal started the JVM process. This function109 will always output to the REPL")111 (defn position-camera112 "Change the position of the in-world camera."113 ([world position direction up]114 (doto (.getCamera world)115 (.setLocation )116 (.lookAt direction up)))117 ([world position direction]118 (position-camera119 world position direction Vector3f/UNIT_Y)))121 (defn enable-debug122 "Turn on debug wireframes for every object in this simulation."123 [world]124 (.enableDebug125 (.getPhysicsSpace126 (.getState127 (.getStateManager world)128 BulletAppState))129 (asset-manager)))131 (defn no-logging132 "Disable all of jMonkeyEngine's logging."133 []134 (.setLevel (Logger/getLogger "com.jme3") Level/OFF))136 (defn set-accuracy137 "Change the accuracy at which the World's Physics is calculated."138 [world new-accuracy]139 (let [physics-manager140 (.getState141 (.getStateManager world) BulletAppState)]142 (.setAccuracy143 (.getPhysicsSpace physics-manager)144 (float new-accuracy))))147 (defn set-gravity148 "In order to change the gravity of a scene, it is not only necessary149 to set the gravity variable, but to \"tap\" every physics object in150 the scene to reactivate physics calculations."151 [world gravity]152 (traverse153 (fn [geom]154 (if-let155 ;; only set gravity for physical objects.156 [control (.getControl geom RigidBodyControl)]157 (do158 (.setGravity control gravity)159 ;; tappsies!160 (.applyImpulse control Vector3f/ZERO Vector3f/ZERO))))161 (.getRootNode world)))163 (defn add-element164 "Add the Spatial to the world's environment"165 ([world element node]166 (.addAll167 (.getPhysicsSpace168 (.getState169 (.getStateManager world)170 BulletAppState))171 element)172 (.attachChild node element))173 ([world element]174 (add-element world element (.getRootNode world))))176 (defn apply-map177 "Like apply, but works for maps and functions that expect an178 implicit map and nothing else as in (fn [& {}]).179 ------- Example -------180 (defn demo [& {:keys [www] :or {www \"oh yeah\"} :as env}]181 (println www))182 (apply-map demo {:www \"hello!\"})183 -->\"hello\""184 [fn m]185 (apply fn (reduce #(into %1 %2) [] m)))187 #+end_src189 #+results: util190 : #'cortex.util/apply-map193 *** Creating Basic Shapes195 #+name: shapes196 #+begin_src clojure :results silent197 (in-ns 'cortex.util)199 (defrecord shape-description200 [name201 color202 mass203 friction204 texture205 material206 position207 rotation208 shape209 physical?210 GImpact?211 ])213 (defvar base-shape214 (shape-description.215 "default-shape"216 false217 ;;ColorRGBA/Blue218 1.0 ;; mass219 1.0 ;; friction220 ;; texture221 "Textures/Terrain/BrickWall/BrickWall.jpg"222 ;; material223 "Common/MatDefs/Misc/Unshaded.j3md"224 Vector3f/ZERO225 Quaternion/IDENTITY226 (Box. Vector3f/ZERO 0.5 0.5 0.5)227 true228 false)229 "Basic settings for shapes.")231 (defn make-shape232 [#^shape-description d]233 (let [asset-manager (asset-manager)234 mat (Material. asset-manager (:material d))235 geom (Geometry. (:name d) (:shape d))]236 (if (:texture d)237 (let [key (TextureKey. (:texture d))]238 ;;(.setGenerateMips key true)239 ;;(.setTexture mat "ColorMap" (.loadTexture asset-manager key))240 ))241 (if (:color d) (.setColor mat "Color" (:color d)))242 (.setMaterial geom mat)243 (if-let [rotation (:rotation d)] (.rotate geom rotation))244 (.setLocalTranslation geom (:position d))245 (if (:physical? d)246 (let [physics-control247 (if (:GImpact d)248 ;; Create an accurate mesh collision shape if desired.249 (RigidBodyControl.250 (doto (GImpactCollisionShape.251 (.getMesh geom))252 (.createJmeMesh)253 ;;(.setMargin 0)254 )255 (float (:mass d)))256 ;; otherwise use jme3's default257 (RigidBodyControl. (float (:mass d))))]258 (.addControl geom physics-control)259 ;;(.setSleepingThresholds physics-control (float 0) (float 0))260 (.setFriction physics-control (:friction d))))261 geom))263 (defn box264 ([l w h & {:as options}]265 (let [options (merge base-shape options)]266 (make-shape (assoc options267 :shape (Box. l w h)))))268 ([] (box 0.5 0.5 0.5)))270 (defn sphere271 ([r & {:as options}]272 (let [options (merge base-shape options)]273 (make-shape (assoc options274 :shape (Sphere. 32 32 (float r))))))275 ([] (sphere 0.5)))277 (defn green-x-ray278 "A usefull material for debuging -- it can be seen no matter what279 object occuldes it."280 []281 (doto (Material. (asset-manager)282 "Common/MatDefs/Misc/Unshaded.j3md")283 (.setColor "Color" ColorRGBA/Green)284 (-> (.getAdditionalRenderState)285 (.setDepthTest false))))287 (defn node-seq288 "Take a node and return a seq of all its children289 recursively. There will be no nodes left in the resulting290 structure"291 [#^Node node]292 (tree-seq #(isa? (class %) Node) #(.getChildren %) node))294 (defn nodify295 "Take a sequence of things that can be attached to a node and return296 a node with all of them attached"297 ([name children]298 (let [node (Node. name)]299 (dorun (map #(.attachChild node %) children))300 node))301 ([children] (nodify "" children)))304 #+end_src307 *** Debug Actions308 #+name: debug-actions309 #+begin_src clojure :results silent310 (in-ns 'cortex.util)312 (defn basic-light-setup313 "returns a sequence of lights appropiate for fully lighting a scene"314 []315 (conj316 (doall317 (map318 (fn [direction]319 (doto (DirectionalLight.)320 (.setDirection direction)321 (.setColor ColorRGBA/White)))322 [;; six faces of a cube323 Vector3f/UNIT_X324 Vector3f/UNIT_Y325 Vector3f/UNIT_Z326 (.mult Vector3f/UNIT_X (float -1))327 (.mult Vector3f/UNIT_Y (float -1))328 (.mult Vector3f/UNIT_Z (float -1))]))329 (doto (AmbientLight.)330 (.setColor ColorRGBA/White))))332 (defn light-up-everything333 "Add lights to a world appropiate for quickly seeing everything334 in the scene. Adds six DirectionalLights facing in orthogonal335 directions, and one AmbientLight to provide overall lighting336 coverage."337 [world]338 (dorun339 (map340 #(.addLight (.getRootNode world) %)341 (basic-light-setup))))343 (defn fire-cannon-ball344 "Creates a function that fires a cannon-ball from the current game's345 camera. The cannon-ball will be attached to the node if provided, or346 to the game's RootNode if no node is provided."347 ([node]348 (fn [game value]349 (if (not value)350 (let [camera (.getCamera game)351 cannon-ball352 (sphere 0.7353 :material "Common/MatDefs/Misc/Unshaded.j3md"354 :texture "Textures/PokeCopper.jpg"355 :position356 (.add (.getLocation camera)357 (.mult (.getDirection camera) (float 1)))358 :mass 3)] ;200 0.05359 (.setLinearVelocity360 (.getControl cannon-ball RigidBodyControl)361 (.mult (.getDirection camera) (float 50))) ;50362 (add-element game cannon-ball (if node node (.getRootNode game)))))))363 ([]364 (fire-cannon-ball false)))366 (def standard-debug-controls367 {"key-space" (fire-cannon-ball)})372 #+end_src375 *** Viewing Objects377 #+name: world-view378 #+begin_src clojure :results silent379 (in-ns 'cortex.util)381 (defprotocol Viewable382 (view [something]))384 (extend-type com.jme3.scene.Geometry385 Viewable386 (view [geo]387 (view (doto (Node.)(.attachChild geo)))))389 (extend-type com.jme3.scene.Node390 Viewable391 (view392 [node]393 (.start394 (world395 node396 {}397 (fn [world]398 (enable-debug world)399 (set-gravity world Vector3f/ZERO)400 (light-up-everything world))401 no-op))))403 (extend-type com.jme3.math.ColorRGBA404 Viewable405 (view406 [color]407 (view (doto (Node.)408 (.attachChild (box 1 1 1 :color color))))))410 (defprotocol Textual411 (text [something]412 "Display a detailed textual analysis of the given object."))414 (extend-type com.jme3.scene.Node415 Textual416 (text [node]417 (println "Total Vertexes: " (.getVertexCount node))418 (println "Total Triangles: " (.getTriangleCount node))419 (println "Controls :")420 (dorun (map #(text (.getControl node %)) (range (.getNumControls node))))421 (println "Has " (.getQuantity node) " Children:")422 (doall (map text (.getChildren node)))))424 (extend-type com.jme3.animation.AnimControl425 Textual426 (text [control]427 (let [animations (.getAnimationNames control)]428 (println "Animation Control with " (count animations) " animation(s):")429 (dorun (map println animations)))))431 (extend-type com.jme3.animation.SkeletonControl432 Textual433 (text [control]434 (println "Skeleton Control with the following skeleton:")435 (println (.getSkeleton control))))437 (extend-type com.jme3.bullet.control.KinematicRagdollControl438 Textual439 (text [control]440 (println "Ragdoll Control")))442 (extend-type com.jme3.scene.Geometry443 Textual444 (text [control]445 (println "...geo...")))446 #+end_src448 Here I make the =Viewable= protocol and extend it to JME's types. Now449 JME3's =hello-world= can be written as easily as:451 #+begin_src clojure :results silent452 (cortex.util/view (cortex.util/box))453 #+end_src456 * COMMENT code generation457 #+begin_src clojure :tangle ../src/cortex/import.clj458 <<import>>459 #+end_src462 #+begin_src clojure :tangle ../src/cortex/util.clj :noweb yes463 <<util>>464 <<shapes>>465 <<debug-actions>>466 <<world-view>>467 #+end_src