rlm@73
|
1 #+title: First attempt at a creature!
|
rlm@73
|
2 #+author: Robert McIntyre
|
rlm@73
|
3 #+email: rlm@mit.edu
|
rlm@73
|
4 #+description:
|
rlm@73
|
5 #+keywords: simulation, jMonkeyEngine3, clojure
|
rlm@73
|
6 #+SETUPFILE: ../../aurellem/org/setup.org
|
rlm@73
|
7 #+INCLUDE: ../../aurellem/org/level-0.org
|
rlm@73
|
8
|
rlm@99
|
9 * objectives
|
rlm@103
|
10 - [X] get an overall bitmap-like image for touch
|
rlm@103
|
11 - [X] write code to visuliaze this bitmap
|
rlm@99
|
12 - [ ] directly change the UV-pixels to show touch sensor activation
|
rlm@99
|
13 - [ ] write an explination for why b&w bitmaps for senses is appropiate
|
rlm@99
|
14 - [ ] clean up touch code and write visulazation test
|
rlm@99
|
15 - [ ] do the same for eyes
|
rlm@99
|
16
|
rlm@73
|
17 * Intro
|
rlm@73
|
18 So far, I've made the following senses --
|
rlm@73
|
19 - Vision
|
rlm@73
|
20 - Hearing
|
rlm@73
|
21 - Touch
|
rlm@73
|
22 - Proprioception
|
rlm@73
|
23
|
rlm@73
|
24 And one effector:
|
rlm@73
|
25 - Movement
|
rlm@73
|
26
|
rlm@73
|
27 However, the code so far has only enabled these senses, but has not
|
rlm@73
|
28 actually implemented them. For example, there is still a lot of work
|
rlm@73
|
29 to be done for vision. I need to be able to create an /eyeball/ in
|
rlm@73
|
30 simulation that can be moved around and see the world from different
|
rlm@73
|
31 angles. I also need to determine weather to use log-polar or cartesian
|
rlm@73
|
32 for the visual input, and I need to determine how/wether to
|
rlm@73
|
33 disceritise the visual input.
|
rlm@73
|
34
|
rlm@73
|
35 I also want to be able to visualize both the sensors and the
|
rlm@104
|
36 effectors in pretty pictures. This semi-retarted creature will be my
|
rlm@73
|
37 first attempt at bringing everything together.
|
rlm@73
|
38
|
rlm@73
|
39 * The creature's body
|
rlm@73
|
40
|
rlm@73
|
41 Still going to do an eve-like body in blender, but due to problems
|
rlm@104
|
42 importing the joints, etc into jMonkeyEngine3, I'm going to do all
|
rlm@73
|
43 the connecting here in clojure code, using the names of the individual
|
rlm@73
|
44 components and trial and error. Later, I'll maybe make some sort of
|
rlm@73
|
45 creature-building modifications to blender that support whatever
|
rlm@73
|
46 discreitized senses I'm going to make.
|
rlm@73
|
47
|
rlm@73
|
48 #+name: body-1
|
rlm@73
|
49 #+begin_src clojure
|
rlm@73
|
50 (ns cortex.silly
|
rlm@73
|
51 "let's play!"
|
rlm@73
|
52 {:author "Robert McIntyre"})
|
rlm@73
|
53
|
rlm@73
|
54 ;; TODO remove this!
|
rlm@73
|
55 (require 'cortex.import)
|
rlm@73
|
56 (cortex.import/mega-import-jme3)
|
rlm@73
|
57 (use '(cortex world util body hearing touch vision))
|
rlm@73
|
58
|
rlm@73
|
59 (rlm.rlm-commands/help)
|
rlm@99
|
60 (import java.awt.image.BufferedImage)
|
rlm@99
|
61 (import javax.swing.JPanel)
|
rlm@99
|
62 (import javax.swing.SwingUtilities)
|
rlm@99
|
63 (import java.awt.Dimension)
|
rlm@99
|
64 (import javax.swing.JFrame)
|
rlm@99
|
65 (import java.awt.Dimension)
|
rlm@106
|
66 (import com.aurellem.capture.RatchetTimer)
|
rlm@99
|
67 (declare joint-create)
|
rlm@73
|
68
|
rlm@99
|
69 (defn view-image
|
rlm@99
|
70 "Initailizes a JPanel on which you may draw a BufferedImage.
|
rlm@99
|
71 Returns a function that accepts a BufferedImage and draws it to the
|
rlm@99
|
72 JPanel."
|
rlm@99
|
73 []
|
rlm@99
|
74 (let [image
|
rlm@99
|
75 (atom
|
rlm@99
|
76 (BufferedImage. 1 1 BufferedImage/TYPE_4BYTE_ABGR))
|
rlm@99
|
77 panel
|
rlm@99
|
78 (proxy [JPanel] []
|
rlm@99
|
79 (paint
|
rlm@99
|
80 [graphics]
|
rlm@99
|
81 (proxy-super paintComponent graphics)
|
rlm@99
|
82 (.drawImage graphics @image 0 0 nil)))
|
rlm@99
|
83 frame (JFrame. "Display Image")]
|
rlm@99
|
84 (SwingUtilities/invokeLater
|
rlm@99
|
85 (fn []
|
rlm@99
|
86 (doto frame
|
rlm@99
|
87 (-> (.getContentPane) (.add panel))
|
rlm@99
|
88 (.pack)
|
rlm@99
|
89 (.setLocationRelativeTo nil)
|
rlm@99
|
90 (.setResizable true)
|
rlm@99
|
91 (.setVisible true))))
|
rlm@99
|
92 (fn [#^BufferedImage i]
|
rlm@99
|
93 (reset! image i)
|
rlm@99
|
94 (.setSize frame (+ 8 (.getWidth i)) (+ 28 (.getHeight i)))
|
rlm@99
|
95 (.repaint panel 0 0 (.getWidth i) (.getHeight i)))))
|
rlm@99
|
96
|
rlm@100
|
97 (defn points->image
|
rlm@100
|
98 "Take a sparse collection of points and visuliaze it as a
|
rlm@100
|
99 BufferedImage."
|
rlm@102
|
100
|
rlm@102
|
101 ;; TODO maybe parallelize this since it's easy
|
rlm@102
|
102
|
rlm@100
|
103 [points]
|
rlm@106
|
104 (if (empty? points)
|
rlm@106
|
105 (BufferedImage. 1 1 BufferedImage/TYPE_BYTE_BINARY)
|
rlm@106
|
106 (let [xs (vec (map first points))
|
rlm@106
|
107 ys (vec (map second points))
|
rlm@106
|
108 x0 (apply min xs)
|
rlm@106
|
109 y0 (apply min ys)
|
rlm@106
|
110 width (- (apply max xs) x0)
|
rlm@106
|
111 height (- (apply max ys) y0)
|
rlm@106
|
112 image (BufferedImage. (inc width) (inc height)
|
rlm@106
|
113 BufferedImage/TYPE_BYTE_BINARY)]
|
rlm@106
|
114 (dorun
|
rlm@106
|
115 (for [index (range (count points))]
|
rlm@106
|
116 (.setRGB image (- (xs index) x0) (- (ys index) y0) -1)))
|
rlm@106
|
117
|
rlm@106
|
118 image)))
|
rlm@100
|
119
|
rlm@101
|
120 (defn test-data
|
rlm@101
|
121 []
|
rlm@101
|
122 (vec
|
rlm@102
|
123 (for [a (range 0 1000 2)
|
rlm@102
|
124 b (range 0 1000 2)]
|
rlm@101
|
125 (vector a b))
|
rlm@101
|
126 ))
|
rlm@99
|
127
|
rlm@101
|
128 (defn average [coll]
|
rlm@101
|
129 (/ (reduce + coll) (count coll)))
|
rlm@101
|
130
|
rlm@101
|
131 (defn collapse-1d
|
rlm@101
|
132 "One dimensional analogue of collapse"
|
rlm@101
|
133 [center line]
|
rlm@101
|
134 (let [length (count line)
|
rlm@101
|
135 num-above (count (filter (partial < center) line))
|
rlm@101
|
136 num-below (- length num-above)]
|
rlm@101
|
137 (range (- center num-below)
|
rlm@101
|
138 (+ center num-above))
|
rlm@101
|
139 ))
|
rlm@99
|
140
|
rlm@99
|
141 (defn collapse
|
rlm@99
|
142 "Take a set of pairs of integers and collapse them into a
|
rlm@99
|
143 contigous bitmap."
|
rlm@99
|
144 [points]
|
rlm@101
|
145 (let
|
rlm@101
|
146 [num-points (count points)
|
rlm@101
|
147 center (vector
|
rlm@102
|
148 (int (average (map first points)))
|
rlm@102
|
149 (int (average (map first points))))
|
rlm@101
|
150 flattened
|
rlm@101
|
151 (reduce
|
rlm@101
|
152 concat
|
rlm@101
|
153 (map
|
rlm@101
|
154 (fn [column]
|
rlm@101
|
155 (map vector
|
rlm@101
|
156 (map first column)
|
rlm@101
|
157 (collapse-1d (second center)
|
rlm@101
|
158 (map second column))))
|
rlm@101
|
159 (partition-by first (sort-by first points))))
|
rlm@101
|
160 squeezed
|
rlm@101
|
161 (reduce
|
rlm@101
|
162 concat
|
rlm@101
|
163 (map
|
rlm@101
|
164 (fn [row]
|
rlm@101
|
165 (map vector
|
rlm@101
|
166 (collapse-1d (first center)
|
rlm@101
|
167 (map first row))
|
rlm@101
|
168 (map second row)))
|
rlm@102
|
169 (partition-by second (sort-by second flattened))))
|
rlm@102
|
170 relocate
|
rlm@102
|
171 (let [min-x (apply min (map first squeezed))
|
rlm@102
|
172 min-y (apply min (map second squeezed))]
|
rlm@102
|
173 (map (fn [[x y]]
|
rlm@102
|
174 [(- x min-x)
|
rlm@102
|
175 (- y min-y)])
|
rlm@104
|
176 squeezed))]
|
rlm@102
|
177 relocate
|
rlm@101
|
178 ))
|
rlm@83
|
179
|
rlm@83
|
180 (defn load-bullet []
|
rlm@84
|
181 (let [sim (world (Node.) {} no-op no-op)]
|
rlm@102
|
182 (doto sim
|
rlm@102
|
183 (.enqueue
|
rlm@102
|
184 (fn []
|
rlm@102
|
185 (.stop sim)))
|
rlm@102
|
186 (.start))))
|
rlm@83
|
187
|
rlm@73
|
188 (defn load-blender-model
|
rlm@73
|
189 "Load a .blend file using an asset folder relative path."
|
rlm@73
|
190 [^String model]
|
rlm@73
|
191 (.loadModel
|
rlm@73
|
192 (doto (asset-manager)
|
rlm@73
|
193 (.registerLoader BlenderModelLoader (into-array String ["blend"])))
|
rlm@73
|
194 model))
|
rlm@73
|
195
|
rlm@74
|
196 (defn meta-data [blender-node key]
|
rlm@74
|
197 (if-let [data (.getUserData blender-node "properties")]
|
rlm@74
|
198 (.findValue data key)
|
rlm@74
|
199 nil))
|
rlm@73
|
200
|
rlm@78
|
201 (defn blender-to-jme
|
rlm@78
|
202 "Convert from Blender coordinates to JME coordinates"
|
rlm@78
|
203 [#^Vector3f in]
|
rlm@78
|
204 (Vector3f. (.getX in)
|
rlm@78
|
205 (.getZ in)
|
rlm@78
|
206 (- (.getY in))))
|
rlm@74
|
207
|
rlm@79
|
208 (defn jme-to-blender
|
rlm@79
|
209 "Convert from JME coordinates to Blender coordinates"
|
rlm@79
|
210 [#^Vector3f in]
|
rlm@79
|
211 (Vector3f. (.getX in)
|
rlm@79
|
212 (- (.getZ in))
|
rlm@79
|
213 (.getY in)))
|
rlm@79
|
214
|
rlm@78
|
215 (defn joint-targets
|
rlm@78
|
216 "Return the two closest two objects to the joint object, ordered
|
rlm@78
|
217 from bottom to top according to the joint's rotation."
|
rlm@78
|
218 [#^Node parts #^Node joint]
|
rlm@78
|
219 ;;(println (meta-data joint "joint"))
|
rlm@78
|
220 (.getWorldRotation joint)
|
rlm@78
|
221 (loop [radius (float 0.01)]
|
rlm@78
|
222 (let [results (CollisionResults.)]
|
rlm@78
|
223 (.collideWith
|
rlm@78
|
224 parts
|
rlm@78
|
225 (BoundingBox. (.getWorldTranslation joint)
|
rlm@78
|
226 radius radius radius)
|
rlm@78
|
227 results)
|
rlm@78
|
228 (let [targets
|
rlm@78
|
229 (distinct
|
rlm@78
|
230 (map #(.getGeometry %) results))]
|
rlm@78
|
231 (if (>= (count targets) 2)
|
rlm@78
|
232 (sort-by
|
rlm@79
|
233 #(let [v
|
rlm@79
|
234 (jme-to-blender
|
rlm@79
|
235 (.mult
|
rlm@79
|
236 (.inverse (.getWorldRotation joint))
|
rlm@79
|
237 (.subtract (.getWorldTranslation %)
|
rlm@79
|
238 (.getWorldTranslation joint))))]
|
rlm@79
|
239 (println-repl (.getName %) ":" v)
|
rlm@79
|
240 (.dot (Vector3f. 1 1 1)
|
rlm@79
|
241 v))
|
rlm@78
|
242 (take 2 targets))
|
rlm@78
|
243 (recur (float (* radius 2))))))))
|
rlm@74
|
244
|
rlm@87
|
245 (defn world-to-local
|
rlm@87
|
246 "Convert the world coordinates into coordinates relative to the
|
rlm@87
|
247 object (i.e. local coordinates), taking into account the rotation
|
rlm@87
|
248 of object."
|
rlm@87
|
249 [#^Spatial object world-coordinate]
|
rlm@87
|
250 (let [out (Vector3f.)]
|
rlm@88
|
251 (.worldToLocal object world-coordinate out) out))
|
rlm@87
|
252
|
rlm@96
|
253 (defn local-to-world
|
rlm@96
|
254 "Convert the local coordinates into coordinates into world relative
|
rlm@96
|
255 coordinates"
|
rlm@96
|
256 [#^Spatial object local-coordinate]
|
rlm@96
|
257 (let [world-coordinate (Vector3f.)]
|
rlm@96
|
258 (.localToWorld object local-coordinate world-coordinate)
|
rlm@96
|
259 world-coordinate))
|
rlm@96
|
260
|
rlm@96
|
261
|
rlm@87
|
262 (defmulti joint-dispatch
|
rlm@87
|
263 "Translate blender pseudo-joints into real JME joints."
|
rlm@88
|
264 (fn [constraints & _]
|
rlm@87
|
265 (:type constraints)))
|
rlm@87
|
266
|
rlm@87
|
267 (defmethod joint-dispatch :point
|
rlm@87
|
268 [constraints control-a control-b pivot-a pivot-b rotation]
|
rlm@87
|
269 (println-repl "creating POINT2POINT joint")
|
rlm@87
|
270 (Point2PointJoint.
|
rlm@87
|
271 control-a
|
rlm@87
|
272 control-b
|
rlm@87
|
273 pivot-a
|
rlm@87
|
274 pivot-b))
|
rlm@87
|
275
|
rlm@87
|
276 (defmethod joint-dispatch :hinge
|
rlm@87
|
277 [constraints control-a control-b pivot-a pivot-b rotation]
|
rlm@87
|
278 (println-repl "creating HINGE joint")
|
rlm@87
|
279 (let [axis
|
rlm@87
|
280 (if-let
|
rlm@87
|
281 [axis (:axis constraints)]
|
rlm@87
|
282 axis
|
rlm@87
|
283 Vector3f/UNIT_X)
|
rlm@87
|
284 [limit-1 limit-2] (:limit constraints)
|
rlm@87
|
285 hinge-axis
|
rlm@87
|
286 (.mult
|
rlm@87
|
287 rotation
|
rlm@87
|
288 (blender-to-jme axis))]
|
rlm@87
|
289 (doto
|
rlm@87
|
290 (HingeJoint.
|
rlm@87
|
291 control-a
|
rlm@87
|
292 control-b
|
rlm@87
|
293 pivot-a
|
rlm@87
|
294 pivot-b
|
rlm@87
|
295 hinge-axis
|
rlm@87
|
296 hinge-axis)
|
rlm@87
|
297 (.setLimit limit-1 limit-2))))
|
rlm@87
|
298
|
rlm@87
|
299 (defmethod joint-dispatch :cone
|
rlm@87
|
300 [constraints control-a control-b pivot-a pivot-b rotation]
|
rlm@87
|
301 (let [limit-xz (:limit-xz constraints)
|
rlm@87
|
302 limit-xy (:limit-xy constraints)
|
rlm@87
|
303 twist (:twist constraints)]
|
rlm@87
|
304
|
rlm@87
|
305 (println-repl "creating CONE joint")
|
rlm@87
|
306 (println-repl rotation)
|
rlm@87
|
307 (println-repl
|
rlm@87
|
308 "UNIT_X --> " (.mult rotation (Vector3f. 1 0 0)))
|
rlm@87
|
309 (println-repl
|
rlm@87
|
310 "UNIT_Y --> " (.mult rotation (Vector3f. 0 1 0)))
|
rlm@87
|
311 (println-repl
|
rlm@87
|
312 "UNIT_Z --> " (.mult rotation (Vector3f. 0 0 1)))
|
rlm@87
|
313 (doto
|
rlm@87
|
314 (ConeJoint.
|
rlm@87
|
315 control-a
|
rlm@87
|
316 control-b
|
rlm@87
|
317 pivot-a
|
rlm@87
|
318 pivot-b
|
rlm@87
|
319 rotation
|
rlm@87
|
320 rotation)
|
rlm@87
|
321 (.setLimit (float limit-xz)
|
rlm@87
|
322 (float limit-xy)
|
rlm@87
|
323 (float twist)))))
|
rlm@87
|
324
|
rlm@88
|
325 (defn connect
|
rlm@87
|
326 "here are some examples:
|
rlm@87
|
327 {:type :point}
|
rlm@87
|
328 {:type :hinge :limit [0 (/ Math/PI 2)] :axis (Vector3f. 0 1 0)}
|
rlm@87
|
329 (:axis defaults to (Vector3f. 1 0 0) if not provided for hinge joints)
|
rlm@87
|
330
|
rlm@89
|
331 {:type :cone :limit-xz 0]
|
rlm@89
|
332 :limit-xy 0]
|
rlm@89
|
333 :twist 0]} (use XZY rotation mode in blender!)"
|
rlm@87
|
334 [#^Node obj-a #^Node obj-b #^Node joint]
|
rlm@87
|
335 (let [control-a (.getControl obj-a RigidBodyControl)
|
rlm@87
|
336 control-b (.getControl obj-b RigidBodyControl)
|
rlm@87
|
337 joint-center (.getWorldTranslation joint)
|
rlm@87
|
338 joint-rotation (.toRotationMatrix (.getWorldRotation joint))
|
rlm@87
|
339 pivot-a (world-to-local obj-a joint-center)
|
rlm@87
|
340 pivot-b (world-to-local obj-b joint-center)]
|
rlm@89
|
341
|
rlm@87
|
342 (if-let [constraints
|
rlm@87
|
343 (map-vals
|
rlm@87
|
344 eval
|
rlm@87
|
345 (read-string
|
rlm@87
|
346 (meta-data joint "joint")))]
|
rlm@89
|
347 ;; A side-effect of creating a joint registers
|
rlm@89
|
348 ;; it with both physics objects which in turn
|
rlm@89
|
349 ;; will register the joint with the physics system
|
rlm@89
|
350 ;; when the simulation is started.
|
rlm@87
|
351 (do
|
rlm@87
|
352 (println-repl "creating joint between"
|
rlm@87
|
353 (.getName obj-a) "and" (.getName obj-b))
|
rlm@87
|
354 (joint-dispatch constraints
|
rlm@87
|
355 control-a control-b
|
rlm@87
|
356 pivot-a pivot-b
|
rlm@87
|
357 joint-rotation))
|
rlm@87
|
358 (println-repl "could not find joint meta-data!"))))
|
rlm@87
|
359
|
rlm@78
|
360 (defn assemble-creature [#^Node pieces joints]
|
rlm@78
|
361 (dorun
|
rlm@78
|
362 (map
|
rlm@78
|
363 (fn [geom]
|
rlm@78
|
364 (let [physics-control
|
rlm@78
|
365 (RigidBodyControl.
|
rlm@78
|
366 (HullCollisionShape.
|
rlm@78
|
367 (.getMesh geom))
|
rlm@78
|
368 (if-let [mass (meta-data geom "mass")]
|
rlm@78
|
369 (do
|
rlm@78
|
370 (println-repl
|
rlm@78
|
371 "setting" (.getName geom) "mass to" (float mass))
|
rlm@78
|
372 (float mass))
|
rlm@78
|
373 (float 1)))]
|
rlm@78
|
374
|
rlm@78
|
375 (.addControl geom physics-control)))
|
rlm@78
|
376 (filter #(isa? (class %) Geometry )
|
rlm@78
|
377 (node-seq pieces))))
|
rlm@78
|
378 (dorun
|
rlm@78
|
379 (map
|
rlm@78
|
380 (fn [joint]
|
rlm@78
|
381 (let [[obj-a obj-b]
|
rlm@78
|
382 (joint-targets pieces joint)]
|
rlm@88
|
383 (connect obj-a obj-b joint)))
|
rlm@78
|
384 joints))
|
rlm@78
|
385 pieces)
|
rlm@74
|
386
|
rlm@78
|
387 (defn blender-creature [blender-path]
|
rlm@78
|
388 (let [model (load-blender-model blender-path)
|
rlm@78
|
389 joints
|
rlm@78
|
390 (if-let [joint-node (.getChild model "joints")]
|
rlm@78
|
391 (seq (.getChildren joint-node))
|
rlm@78
|
392 (do (println-repl "could not find joints node")
|
rlm@78
|
393 []))]
|
rlm@78
|
394 (assemble-creature model joints)))
|
rlm@74
|
395
|
rlm@78
|
396 (def hand "Models/creature1/one.blend")
|
rlm@74
|
397
|
rlm@78
|
398 (def worm "Models/creature1/try-again.blend")
|
rlm@78
|
399
|
rlm@90
|
400 (def touch "Models/creature1/touch.blend")
|
rlm@90
|
401
|
rlm@90
|
402 (defn worm-model [] (load-blender-model worm))
|
rlm@90
|
403
|
rlm@80
|
404 (defn x-ray [#^ColorRGBA color]
|
rlm@80
|
405 (doto (Material. (asset-manager)
|
rlm@80
|
406 "Common/MatDefs/Misc/Unshaded.j3md")
|
rlm@80
|
407 (.setColor "Color" color)
|
rlm@80
|
408 (-> (.getAdditionalRenderState)
|
rlm@80
|
409 (.setDepthTest false))))
|
rlm@80
|
410
|
rlm@106
|
411
|
rlm@90
|
412
|
rlm@91
|
413 (defn colorful []
|
rlm@91
|
414 (.getChild (worm-model) "worm-21"))
|
rlm@90
|
415
|
rlm@90
|
416 (import jme3tools.converters.ImageToAwt)
|
rlm@90
|
417
|
rlm@90
|
418 (import ij.ImagePlus)
|
rlm@90
|
419
|
rlm@90
|
420 (defn triangle-indices
|
rlm@90
|
421 "Get the triangle vertex indices of a given triangle from a given
|
rlm@90
|
422 mesh."
|
rlm@90
|
423 [#^Mesh mesh triangle-index]
|
rlm@90
|
424 (let [indices (int-array 3)]
|
rlm@90
|
425 (.getTriangle mesh triangle-index indices)
|
rlm@90
|
426 (vec indices)))
|
rlm@90
|
427
|
rlm@90
|
428 (defn uv-coord
|
rlm@90
|
429 "Get the uv-coordinates of the vertex named by vertex-index"
|
rlm@90
|
430 [#^Mesh mesh vertex-index]
|
rlm@90
|
431 (let [UV-buffer
|
rlm@90
|
432 (.getData
|
rlm@90
|
433 (.getBuffer
|
rlm@90
|
434 mesh
|
rlm@90
|
435 VertexBuffer$Type/TexCoord))]
|
rlm@91
|
436 (Vector2f.
|
rlm@91
|
437 (.get UV-buffer (* vertex-index 2))
|
rlm@91
|
438 (.get UV-buffer (+ 1 (* vertex-index 2))))))
|
rlm@90
|
439
|
rlm@93
|
440 (defn tri-uv-coord
|
rlm@93
|
441 "Get the uv-cooridnates of the triangle's verticies."
|
rlm@93
|
442 [#^Mesh mesh #^Triangle triangle]
|
rlm@93
|
443 (map (partial uv-coord mesh)
|
rlm@93
|
444 (triangle-indices mesh (.getIndex triangle))))
|
rlm@93
|
445
|
rlm@91
|
446 (defn touch-receptor-image
|
rlm@97
|
447 "Return the touch-sensor distribution image in ImagePlus format, or
|
rlm@97
|
448 nil if it does not exist."
|
rlm@91
|
449 [#^Geometry obj]
|
rlm@97
|
450 (let [mat (.getMaterial obj)]
|
rlm@97
|
451 (if-let [texture-param
|
rlm@97
|
452 (.getTextureParam
|
rlm@97
|
453 mat
|
rlm@97
|
454 MaterialHelper/TEXTURE_TYPE_DIFFUSE)]
|
rlm@97
|
455 (let
|
rlm@97
|
456 [texture
|
rlm@97
|
457 (.getTextureValue texture-param)
|
rlm@97
|
458 im (.getImage texture)]
|
rlm@97
|
459 (ImagePlus.
|
rlm@97
|
460 "UV-map"
|
rlm@97
|
461 (ImageToAwt/convert im false false 0))))))
|
rlm@91
|
462
|
rlm@91
|
463 (import ij.process.ImageProcessor)
|
rlm@91
|
464 (import java.awt.image.BufferedImage)
|
rlm@91
|
465
|
rlm@91
|
466 (defprotocol Frame
|
rlm@91
|
467 (frame [this]))
|
rlm@91
|
468
|
rlm@91
|
469 (extend-type BufferedImage
|
rlm@91
|
470 Frame
|
rlm@91
|
471 (frame [image]
|
rlm@91
|
472 (merge
|
rlm@91
|
473 (apply
|
rlm@91
|
474 hash-map
|
rlm@91
|
475 (interleave
|
rlm@91
|
476 (doall (for [x (range (.getWidth image)) y (range (.getHeight image))]
|
rlm@91
|
477 (vector x y)))
|
rlm@91
|
478 (doall (for [x (range (.getWidth image)) y (range (.getHeight image))]
|
rlm@91
|
479 (let [data (.getRGB image x y)]
|
rlm@91
|
480 (hash-map :r (bit-shift-right (bit-and 0xff0000 data) 16)
|
rlm@91
|
481 :g (bit-shift-right (bit-and 0x00ff00 data) 8)
|
rlm@91
|
482 :b (bit-and 0x0000ff data)))))))
|
rlm@91
|
483 {:width (.getWidth image) :height (.getHeight image)})))
|
rlm@91
|
484
|
rlm@91
|
485
|
rlm@91
|
486 (extend-type ImagePlus
|
rlm@91
|
487 Frame
|
rlm@91
|
488 (frame [image+]
|
rlm@91
|
489 (frame (.getBufferedImage image+))))
|
rlm@91
|
490
|
rlm@94
|
491
|
rlm@92
|
492 (def white -1)
|
rlm@94
|
493
|
rlm@91
|
494 (defn filter-pixels
|
rlm@91
|
495 "List the coordinates of all pixels matching pred."
|
rlm@92
|
496 {:author "Dylan Holmes"}
|
rlm@91
|
497 [pred #^ImageProcessor ip]
|
rlm@91
|
498 (let
|
rlm@91
|
499 [width (.getWidth ip)
|
rlm@91
|
500 height (.getHeight ip)]
|
rlm@91
|
501 ((fn accumulate [x y matches]
|
rlm@91
|
502 (cond
|
rlm@91
|
503 (>= y height) matches
|
rlm@91
|
504 (>= x width) (recur 0 (inc y) matches)
|
rlm@91
|
505 (pred (.getPixel ip x y))
|
rlm@91
|
506 (recur (inc x) y (conj matches (Vector2f. x y)))
|
rlm@91
|
507 :else (recur (inc x) y matches)))
|
rlm@91
|
508 0 0 [])))
|
rlm@91
|
509
|
rlm@91
|
510 (defn white-coordinates
|
rlm@91
|
511 "List the coordinates of all the white pixels in an image."
|
rlm@91
|
512 [#^ImageProcessor ip]
|
rlm@92
|
513 (filter-pixels #(= % white) ip))
|
rlm@91
|
514
|
rlm@102
|
515 (defn same-side?
|
rlm@102
|
516 "Given the points p1 and p2 and the reference point ref, is point p
|
rlm@102
|
517 on the same side of the line that goes through p1 and p2 as ref is?"
|
rlm@102
|
518 [p1 p2 ref p]
|
rlm@91
|
519 (<=
|
rlm@91
|
520 0
|
rlm@91
|
521 (.dot
|
rlm@91
|
522 (.cross (.subtract p2 p1) (.subtract p p1))
|
rlm@91
|
523 (.cross (.subtract p2 p1) (.subtract ref p1)))))
|
rlm@91
|
524
|
rlm@94
|
525 (defn triangle->matrix4f
|
rlm@94
|
526 "Converts the triangle into a 4x4 matrix of vertices: The first
|
rlm@94
|
527 three columns contain the vertices of the triangle; the last
|
rlm@94
|
528 contains the unit normal of the triangle. The bottom row is filled
|
rlm@94
|
529 with 1s."
|
rlm@94
|
530 [#^Triangle t]
|
rlm@94
|
531 (let [mat (Matrix4f.)
|
rlm@94
|
532 [vert-1 vert-2 vert-3]
|
rlm@94
|
533 ((comp vec map) #(.get t %) (range 3))
|
rlm@94
|
534 unit-normal (do (.calculateNormal t)(.getNormal t))
|
rlm@94
|
535 vertices [vert-1 vert-2 vert-3 unit-normal]]
|
rlm@94
|
536 (dorun
|
rlm@94
|
537 (for [row (range 4) col (range 3)]
|
rlm@94
|
538 (do
|
rlm@94
|
539 (.set mat col row (.get (vertices row)col))
|
rlm@94
|
540 (.set mat 3 row 1))))
|
rlm@94
|
541 mat))
|
rlm@94
|
542
|
rlm@94
|
543 (defn triangle-transformation
|
rlm@94
|
544 "Returns the affine transformation that converts each vertex in the
|
rlm@94
|
545 first triangle into the corresponding vertex in the second
|
rlm@94
|
546 triangle."
|
rlm@94
|
547 [#^Triangle tri-1 #^Triangle tri-2]
|
rlm@94
|
548 (.mult
|
rlm@94
|
549 (triangle->matrix4f tri-2)
|
rlm@94
|
550 (.invert (triangle->matrix4f tri-1))))
|
rlm@94
|
551
|
rlm@94
|
552 (def death (Triangle.
|
rlm@94
|
553 (Vector3f. 1 1 1)
|
rlm@94
|
554 (Vector3f. 1 2 3)
|
rlm@94
|
555 (Vector3f. 5 6 7)))
|
rlm@94
|
556
|
rlm@94
|
557 (def death-2 (Triangle.
|
rlm@94
|
558 (Vector3f. 2 2 2)
|
rlm@94
|
559 (Vector3f. 1 1 1)
|
rlm@94
|
560 (Vector3f. 0 1 0)))
|
rlm@94
|
561
|
rlm@94
|
562 (defn vector2f->vector3f [v]
|
rlm@94
|
563 (Vector3f. (.getX v) (.getY v) 0))
|
rlm@94
|
564
|
rlm@94
|
565 (extend-type Triangle
|
rlm@94
|
566 Textual
|
rlm@94
|
567 (text [t]
|
rlm@94
|
568 (println "Triangle: " \newline (.get1 t) \newline
|
rlm@94
|
569 (.get2 t) \newline (.get3 t))))
|
rlm@94
|
570
|
rlm@94
|
571 (defn map-triangle [f #^Triangle tri]
|
rlm@94
|
572 (Triangle.
|
rlm@94
|
573 (f 0 (.get1 tri))
|
rlm@94
|
574 (f 1 (.get2 tri))
|
rlm@94
|
575 (f 2 (.get3 tri))))
|
rlm@94
|
576
|
rlm@94
|
577 (defn triangle-seq [#^Triangle tri]
|
rlm@94
|
578 [(.get1 tri) (.get2 tri) (.get3 tri)])
|
rlm@94
|
579
|
rlm@94
|
580 (defn vector3f-seq [#^Vector3f v]
|
rlm@94
|
581 [(.getX v) (.getY v) (.getZ v)])
|
rlm@94
|
582
|
rlm@91
|
583 (defn inside-triangle?
|
rlm@94
|
584 "Is the point inside the triangle? Now what do we do?
|
rlm@94
|
585 You might want to hold on there"
|
rlm@95
|
586 {:author "Dylan Holmes"}
|
rlm@94
|
587 [tri p]
|
rlm@94
|
588 (let [[vert-1 vert-2 vert-3] (triangle-seq tri)]
|
rlm@94
|
589 (and
|
rlm@94
|
590 (same-side? vert-1 vert-2 vert-3 p)
|
rlm@94
|
591 (same-side? vert-2 vert-3 vert-1 p)
|
rlm@94
|
592 (same-side? vert-3 vert-1 vert-2 p))))
|
rlm@91
|
593
|
rlm@94
|
594 (defn uv-triangle
|
rlm@94
|
595 "Convert the mesh triangle into the cooresponding triangle in
|
rlm@94
|
596 UV-space. Z-component of these triangles is always zero."
|
rlm@94
|
597 [#^Mesh mesh #^Triangle tri]
|
rlm@94
|
598 (apply #(Triangle. %1 %2 %3)
|
rlm@94
|
599 (map vector2f->vector3f
|
rlm@94
|
600 (tri-uv-coord mesh tri))))
|
rlm@91
|
601
|
rlm@94
|
602 (defn pixel-triangle
|
rlm@102
|
603 "Convert the mesh triangle into the corresponding triangle in
|
rlm@94
|
604 UV-pixel-space. Z compenent will be zero."
|
rlm@94
|
605 [#^Mesh mesh #^Triangle tri width height]
|
rlm@94
|
606 (map-triangle (fn [_ v]
|
rlm@94
|
607 (Vector3f. (* width (.getX v))
|
rlm@94
|
608 (* height (.getY v))
|
rlm@94
|
609 0))
|
rlm@94
|
610 (uv-triangle mesh tri)))
|
rlm@93
|
611
|
rlm@96
|
612 (def rasterize pixel-triangle)
|
rlm@96
|
613
|
rlm@96
|
614
|
rlm@94
|
615 (defn triangle-bounds
|
rlm@94
|
616 "Dimensions of the bounding square of the triangle in the form
|
rlm@94
|
617 [x y width height].
|
rlm@94
|
618 Assumes that the triangle lies in the XY plane."
|
rlm@94
|
619 [#^Triangle tri]
|
rlm@94
|
620 (let [verts (map vector3f-seq (triangle-seq tri))
|
rlm@94
|
621 x (apply min (map first verts))
|
rlm@94
|
622 y (apply min (map second verts))]
|
rlm@94
|
623 [x y
|
rlm@94
|
624 (- (apply max (map first verts)) x)
|
rlm@94
|
625 (- (apply max (map second verts)) y)
|
rlm@94
|
626 ]))
|
rlm@93
|
627
|
rlm@93
|
628
|
rlm@106
|
629 (defn sensors-in-triangle
|
rlm@106
|
630 "find the locations of the sensors within a triangle"
|
rlm@106
|
631 [image tri]
|
rlm@106
|
632 )
|
rlm@106
|
633
|
rlm@106
|
634
|
rlm@97
|
635 (defn locate-feelers
|
rlm@94
|
636 "Search the geometry's tactile UV image for touch sensors, returning
|
rlm@94
|
637 their positions in geometry-relative coordinates."
|
rlm@94
|
638 [#^Geometry geo]
|
rlm@97
|
639 (if-let [image (touch-receptor-image geo)]
|
rlm@97
|
640 (let [mesh (.getMesh geo)
|
rlm@97
|
641 tris (triangles geo)
|
rlm@102
|
642
|
rlm@97
|
643 width (.getWidth image)
|
rlm@97
|
644 height (.getHeight image)
|
rlm@97
|
645
|
rlm@97
|
646 ;; for each triangle
|
rlm@97
|
647 sensor-coords
|
rlm@97
|
648 (fn [tri]
|
rlm@97
|
649 ;; translate triangle to uv-pixel-space
|
rlm@97
|
650 (let [uv-tri
|
rlm@97
|
651 (pixel-triangle mesh tri width height)
|
rlm@97
|
652 bounds (vec (triangle-bounds uv-tri))]
|
rlm@97
|
653
|
rlm@97
|
654 ;; get that part of the picture
|
rlm@97
|
655
|
rlm@97
|
656 (apply #(.setRoi image %1 %2 %3 %4) bounds)
|
rlm@97
|
657 (let [cutout (.crop (.getProcessor image))
|
rlm@97
|
658 ;; extract white pixels inside triangle
|
rlm@97
|
659 cutout-tri
|
rlm@97
|
660 (map-triangle
|
rlm@97
|
661 (fn [_ v]
|
rlm@97
|
662 (.subtract
|
rlm@97
|
663 v
|
rlm@97
|
664 (Vector3f. (bounds 0) (bounds 1) (float 0))))
|
rlm@97
|
665 uv-tri)
|
rlm@97
|
666 whites (filter (partial inside-triangle? cutout-tri)
|
rlm@97
|
667 (map vector2f->vector3f
|
rlm@97
|
668 (white-coordinates cutout)))
|
rlm@97
|
669 ;; translate pixel coordinates to world-space
|
rlm@97
|
670 transform (triangle-transformation cutout-tri tri)]
|
rlm@97
|
671 (map #(.mult transform %) whites))))]
|
rlm@97
|
672 (vec (map sensor-coords tris)))
|
rlm@97
|
673 (repeat (count (triangles geo)) [])))
|
rlm@102
|
674
|
rlm@102
|
675 (use 'clojure.contrib.def)
|
rlm@102
|
676
|
rlm@102
|
677 (defn-memo touch-topology [#^Gemoetry geo]
|
rlm@106
|
678 (if-let [image (touch-receptor-image geo)]
|
rlm@106
|
679 (let [feeler-coords
|
rlm@106
|
680 (map
|
rlm@106
|
681 #(vector (int (.getX %)) (int (.getY %)))
|
rlm@106
|
682 (white-coordinates
|
rlm@106
|
683 (.getProcessor image)))]
|
rlm@106
|
684 (vec (collapse feeler-coords)))
|
rlm@106
|
685 []))
|
rlm@102
|
686
|
rlm@97
|
687 (defn enable-touch [#^Geometry geo]
|
rlm@97
|
688 (let [feeler-coords (locate-feelers geo)
|
rlm@96
|
689 tris (triangles geo)
|
rlm@97
|
690 limit 0.1]
|
rlm@97
|
691 (fn [node]
|
rlm@97
|
692 (let [sensor-origins
|
rlm@97
|
693 (map
|
rlm@97
|
694 #(map (partial local-to-world geo) %)
|
rlm@97
|
695 feeler-coords)
|
rlm@97
|
696 triangle-normals
|
rlm@97
|
697 (map (partial get-ray-direction geo)
|
rlm@97
|
698 tris)
|
rlm@97
|
699 rays
|
rlm@97
|
700 (flatten
|
rlm@97
|
701 (map (fn [origins norm]
|
rlm@97
|
702 (map #(doto (Ray. % norm)
|
rlm@97
|
703 (.setLimit limit)) origins))
|
rlm@97
|
704 sensor-origins triangle-normals))]
|
rlm@102
|
705 (vector
|
rlm@102
|
706 (touch-topology geo)
|
rlm@102
|
707 (vec
|
rlm@102
|
708 (for [ray rays]
|
rlm@102
|
709 (do
|
rlm@102
|
710 (let [results (CollisionResults.)]
|
rlm@102
|
711 (.collideWith node ray results)
|
rlm@102
|
712 (let [touch-objects
|
rlm@102
|
713 (set
|
rlm@102
|
714 (filter #(not (= geo %))
|
rlm@102
|
715 (map #(.getGeometry %) results)))]
|
rlm@102
|
716 (if (> (count touch-objects) 0)
|
rlm@102
|
717 1 0)))))))))))
|
rlm@94
|
718
|
rlm@97
|
719 (defn touch [#^Node pieces]
|
rlm@102
|
720 (map enable-touch
|
rlm@102
|
721 (filter #(isa? (class %) Geometry)
|
rlm@102
|
722 (node-seq pieces))))
|
rlm@102
|
723
|
rlm@103
|
724 (defn debug-window
|
rlm@103
|
725 "creates function that offers a debug view of sensor data"
|
rlm@103
|
726 []
|
rlm@103
|
727 (let [vi (view-image)]
|
rlm@103
|
728 (fn
|
rlm@103
|
729 [[coords sensor-data]]
|
rlm@103
|
730 (let [image (points->image coords)]
|
rlm@103
|
731 (dorun
|
rlm@103
|
732 (for [i (range (count coords))]
|
rlm@103
|
733 (.setRGB image ((coords i) 0) ((coords i) 1)
|
rlm@103
|
734 ({0 -16777216
|
rlm@103
|
735 1 -1} (sensor-data i)))))
|
rlm@103
|
736 (vi image)))))
|
rlm@103
|
737
|
rlm@83
|
738
|
rlm@106
|
739 ;;(defn test-touch [world creature]
|
rlm@83
|
740
|
rlm@78
|
741
|
rlm@106
|
742 (defn test-creature [thing]
|
rlm@106
|
743 (let [x-axis
|
rlm@106
|
744 (box 1 0.01 0.01 :physical? false :color ColorRGBA/Red)
|
rlm@106
|
745 y-axis
|
rlm@106
|
746 (box 0.01 1 0.01 :physical? false :color ColorRGBA/Green)
|
rlm@106
|
747 z-axis
|
rlm@106
|
748 (box 0.01 0.01 1 :physical? false :color ColorRGBA/Blue)
|
rlm@106
|
749 creature (blender-creature thing)
|
rlm@106
|
750 touch-nerves (touch creature)
|
rlm@106
|
751 touch-debug-windows (map (fn [_] (debug-window)) touch-nerves)
|
rlm@106
|
752 ]
|
rlm@106
|
753 (world
|
rlm@106
|
754 (nodify [creature
|
rlm@106
|
755 (box 10 2 10 :position (Vector3f. 0 -9 0)
|
rlm@106
|
756 :color ColorRGBA/Gray :mass 0)
|
rlm@106
|
757 x-axis y-axis z-axis
|
rlm@106
|
758 ])
|
rlm@106
|
759 standard-debug-controls
|
rlm@106
|
760 (fn [world]
|
rlm@106
|
761 (light-up-everything world)
|
rlm@106
|
762 (enable-debug world)
|
rlm@106
|
763 ;;(com.aurellem.capture.Capture/captureVideo
|
rlm@106
|
764 ;; world (file-str "/home/r/proj/ai-videos/hand"))
|
rlm@106
|
765 (.setTimer world (RatchetTimer. 60))
|
rlm@106
|
766 ;;(set-gravity world (Vector3f. 0 0 0))
|
rlm@106
|
767 )
|
rlm@106
|
768 (fn [world tpf]
|
rlm@106
|
769 (dorun
|
rlm@106
|
770 (map #(%1 (%2 (.getRootNode world))) touch-debug-windows touch-nerves))
|
rlm@106
|
771 )
|
rlm@106
|
772 ;;(let [timer (atom 0)]
|
rlm@106
|
773 ;; (fn [_ _]
|
rlm@106
|
774 ;; (swap! timer inc)
|
rlm@106
|
775 ;; (if (= (rem @timer 60) 0)
|
rlm@106
|
776 ;; (println-repl (float (/ @timer 60))))))
|
rlm@106
|
777 )))
|
rlm@83
|
778
|
rlm@87
|
779 #+end_src
|
rlm@83
|
780
|
rlm@87
|
781 #+results: body-1
|
rlm@104
|
782 : #'cortex.silly/tactile-coords
|
rlm@78
|
783
|
rlm@78
|
784
|
rlm@78
|
785 * COMMENT purgatory
|
rlm@78
|
786 #+begin_src clojure
|
rlm@77
|
787 (defn bullet-trans []
|
rlm@77
|
788 (let [obj-a (sphere 0.5 :color ColorRGBA/Red
|
rlm@77
|
789 :position (Vector3f. -10 5 0))
|
rlm@77
|
790 obj-b (sphere 0.5 :color ColorRGBA/Blue
|
rlm@77
|
791 :position (Vector3f. -10 -5 0)
|
rlm@77
|
792 :mass 0)
|
rlm@77
|
793 control-a (.getControl obj-a RigidBodyControl)
|
rlm@77
|
794 control-b (.getControl obj-b RigidBodyControl)
|
rlm@77
|
795 swivel
|
rlm@77
|
796 (.toRotationMatrix
|
rlm@77
|
797 (doto (Quaternion.)
|
rlm@77
|
798 (.fromAngleAxis (/ Math/PI 2)
|
rlm@77
|
799 Vector3f/UNIT_X)))]
|
rlm@77
|
800 (doto
|
rlm@77
|
801 (ConeJoint.
|
rlm@77
|
802 control-a control-b
|
rlm@77
|
803 (Vector3f. 0 5 0)
|
rlm@77
|
804 (Vector3f. 0 -5 0)
|
rlm@77
|
805 swivel swivel)
|
rlm@77
|
806 (.setLimit (* 0.6 (/ Math/PI 4))
|
rlm@77
|
807 (/ Math/PI 4)
|
rlm@77
|
808 (* Math/PI 0.8)))
|
rlm@77
|
809 (world (nodify
|
rlm@77
|
810 [obj-a obj-b])
|
rlm@77
|
811 standard-debug-controls
|
rlm@77
|
812 enable-debug
|
rlm@77
|
813 no-op)))
|
rlm@74
|
814
|
rlm@74
|
815
|
rlm@77
|
816 (defn bullet-trans* []
|
rlm@77
|
817 (let [obj-a (box 1.5 0.5 0.5 :color ColorRGBA/Red
|
rlm@77
|
818 :position (Vector3f. 5 0 0)
|
rlm@77
|
819 :mass 90)
|
rlm@77
|
820 obj-b (sphere 0.5 :color ColorRGBA/Blue
|
rlm@77
|
821 :position (Vector3f. -5 0 0)
|
rlm@77
|
822 :mass 0)
|
rlm@77
|
823 control-a (.getControl obj-a RigidBodyControl)
|
rlm@77
|
824 control-b (.getControl obj-b RigidBodyControl)
|
rlm@77
|
825 move-up? (atom nil)
|
rlm@77
|
826 move-down? (atom nil)
|
rlm@77
|
827 move-left? (atom nil)
|
rlm@77
|
828 move-right? (atom nil)
|
rlm@77
|
829 roll-left? (atom nil)
|
rlm@77
|
830 roll-right? (atom nil)
|
rlm@77
|
831 force 100
|
rlm@77
|
832 swivel
|
rlm@77
|
833 (.toRotationMatrix
|
rlm@77
|
834 (doto (Quaternion.)
|
rlm@77
|
835 (.fromAngleAxis (/ Math/PI 2)
|
rlm@77
|
836 Vector3f/UNIT_X)))
|
rlm@77
|
837 x-move
|
rlm@77
|
838 (doto (Matrix3f.)
|
rlm@77
|
839 (.fromStartEndVectors Vector3f/UNIT_X
|
rlm@77
|
840 (.normalize (Vector3f. 1 1 0))))
|
rlm@77
|
841
|
rlm@77
|
842 timer (atom 0)]
|
rlm@77
|
843 (doto
|
rlm@77
|
844 (ConeJoint.
|
rlm@77
|
845 control-a control-b
|
rlm@77
|
846 (Vector3f. -8 0 0)
|
rlm@77
|
847 (Vector3f. 2 0 0)
|
rlm@77
|
848 ;;swivel swivel
|
rlm@77
|
849 ;;Matrix3f/IDENTITY Matrix3f/IDENTITY
|
rlm@77
|
850 x-move Matrix3f/IDENTITY
|
rlm@77
|
851 )
|
rlm@77
|
852 (.setCollisionBetweenLinkedBodys false)
|
rlm@77
|
853 (.setLimit (* 1 (/ Math/PI 4)) ;; twist
|
rlm@77
|
854 (* 1 (/ Math/PI 4)) ;; swing span in X-Y plane
|
rlm@77
|
855 (* 0 (/ Math/PI 4)))) ;; swing span in Y-Z plane
|
rlm@77
|
856 (world (nodify
|
rlm@77
|
857 [obj-a obj-b])
|
rlm@77
|
858 (merge standard-debug-controls
|
rlm@77
|
859 {"key-r" (fn [_ pressed?] (reset! move-up? pressed?))
|
rlm@77
|
860 "key-t" (fn [_ pressed?] (reset! move-down? pressed?))
|
rlm@77
|
861 "key-f" (fn [_ pressed?] (reset! move-left? pressed?))
|
rlm@77
|
862 "key-g" (fn [_ pressed?] (reset! move-right? pressed?))
|
rlm@77
|
863 "key-v" (fn [_ pressed?] (reset! roll-left? pressed?))
|
rlm@77
|
864 "key-b" (fn [_ pressed?] (reset! roll-right? pressed?))})
|
rlm@77
|
865
|
rlm@77
|
866 (fn [world]
|
rlm@77
|
867 (enable-debug world)
|
rlm@77
|
868 (set-gravity world Vector3f/ZERO)
|
rlm@77
|
869 )
|
rlm@77
|
870
|
rlm@77
|
871 (fn [world _]
|
rlm@77
|
872
|
rlm@77
|
873 (if @move-up?
|
rlm@77
|
874 (.applyForce control-a
|
rlm@77
|
875 (Vector3f. force 0 0)
|
rlm@77
|
876 (Vector3f. 0 0 0)))
|
rlm@77
|
877 (if @move-down?
|
rlm@77
|
878 (.applyForce control-a
|
rlm@77
|
879 (Vector3f. (- force) 0 0)
|
rlm@77
|
880 (Vector3f. 0 0 0)))
|
rlm@77
|
881 (if @move-left?
|
rlm@77
|
882 (.applyForce control-a
|
rlm@77
|
883 (Vector3f. 0 force 0)
|
rlm@77
|
884 (Vector3f. 0 0 0)))
|
rlm@77
|
885 (if @move-right?
|
rlm@77
|
886 (.applyForce control-a
|
rlm@77
|
887 (Vector3f. 0 (- force) 0)
|
rlm@77
|
888 (Vector3f. 0 0 0)))
|
rlm@77
|
889
|
rlm@77
|
890 (if @roll-left?
|
rlm@77
|
891 (.applyForce control-a
|
rlm@77
|
892 (Vector3f. 0 0 force)
|
rlm@77
|
893 (Vector3f. 0 0 0)))
|
rlm@77
|
894 (if @roll-right?
|
rlm@77
|
895 (.applyForce control-a
|
rlm@77
|
896 (Vector3f. 0 0 (- force))
|
rlm@77
|
897 (Vector3f. 0 0 0)))
|
rlm@77
|
898
|
rlm@77
|
899 (if (zero? (rem (swap! timer inc) 100))
|
rlm@77
|
900 (.attachChild
|
rlm@77
|
901 (.getRootNode world)
|
rlm@77
|
902 (sphere 0.05 :color ColorRGBA/Yellow
|
rlm@77
|
903 :physical? false :position
|
rlm@77
|
904 (.getWorldTranslation obj-a)))))
|
rlm@77
|
905 )
|
rlm@77
|
906 ))
|
rlm@77
|
907
|
rlm@94
|
908 (defn transform-trianglesdsd
|
rlm@94
|
909 "Transform that converts each vertex in the first triangle
|
rlm@94
|
910 into the corresponding vertex in the second triangle."
|
rlm@94
|
911 [#^Triangle tri-1 #^Triangle tri-2]
|
rlm@94
|
912 (let [in [(.get1 tri-1)
|
rlm@94
|
913 (.get2 tri-1)
|
rlm@94
|
914 (.get3 tri-1)]
|
rlm@94
|
915 out [(.get1 tri-2)
|
rlm@94
|
916 (.get2 tri-2)
|
rlm@94
|
917 (.get3 tri-2)]]
|
rlm@94
|
918 (let [translate (doto (Matrix4f.) (.setTranslation (.negate (in 0))))
|
rlm@94
|
919 in* [(.mult translate (in 0))
|
rlm@94
|
920 (.mult translate (in 1))
|
rlm@94
|
921 (.mult translate (in 2))]
|
rlm@94
|
922 final-translation
|
rlm@94
|
923 (doto (Matrix4f.)
|
rlm@94
|
924 (.setTranslation (out 1)))
|
rlm@94
|
925
|
rlm@94
|
926 rotate-1
|
rlm@94
|
927 (doto (Matrix3f.)
|
rlm@94
|
928 (.fromStartEndVectors
|
rlm@94
|
929 (.normalize
|
rlm@94
|
930 (.subtract
|
rlm@94
|
931 (in* 1) (in* 0)))
|
rlm@94
|
932 (.normalize
|
rlm@94
|
933 (.subtract
|
rlm@94
|
934 (out 1) (out 0)))))
|
rlm@94
|
935 in** [(.mult rotate-1 (in* 0))
|
rlm@94
|
936 (.mult rotate-1 (in* 1))
|
rlm@94
|
937 (.mult rotate-1 (in* 2))]
|
rlm@94
|
938 scale-factor-1
|
rlm@94
|
939 (.mult
|
rlm@94
|
940 (.normalize
|
rlm@94
|
941 (.subtract
|
rlm@94
|
942 (out 1)
|
rlm@94
|
943 (out 0)))
|
rlm@94
|
944 (/ (.length
|
rlm@94
|
945 (.subtract (out 1)
|
rlm@94
|
946 (out 0)))
|
rlm@94
|
947 (.length
|
rlm@94
|
948 (.subtract (in** 1)
|
rlm@94
|
949 (in** 0)))))
|
rlm@94
|
950 scale-1 (doto (Matrix4f.) (.setScale scale-factor-1))
|
rlm@94
|
951 in*** [(.mult scale-1 (in** 0))
|
rlm@94
|
952 (.mult scale-1 (in** 1))
|
rlm@94
|
953 (.mult scale-1 (in** 2))]
|
rlm@94
|
954
|
rlm@94
|
955
|
rlm@94
|
956
|
rlm@94
|
957
|
rlm@94
|
958
|
rlm@94
|
959 ]
|
rlm@94
|
960
|
rlm@94
|
961 (dorun (map println in))
|
rlm@94
|
962 (println)
|
rlm@94
|
963 (dorun (map println in*))
|
rlm@94
|
964 (println)
|
rlm@94
|
965 (dorun (map println in**))
|
rlm@94
|
966 (println)
|
rlm@94
|
967 (dorun (map println in***))
|
rlm@94
|
968 (println)
|
rlm@94
|
969
|
rlm@99
|
970 ))))
|
rlm@94
|
971
|
rlm@94
|
972
|
rlm@106
|
973 (defn world-setup [joint]
|
rlm@106
|
974 (let [joint-position (Vector3f. 0 0 0)
|
rlm@106
|
975 joint-rotation
|
rlm@106
|
976 (.toRotationMatrix
|
rlm@106
|
977 (.mult
|
rlm@106
|
978 (doto (Quaternion.)
|
rlm@106
|
979 (.fromAngleAxis
|
rlm@106
|
980 (* 1 (/ Math/PI 4))
|
rlm@106
|
981 (Vector3f. -1 0 0)))
|
rlm@106
|
982 (doto (Quaternion.)
|
rlm@106
|
983 (.fromAngleAxis
|
rlm@106
|
984 (* 1 (/ Math/PI 2))
|
rlm@106
|
985 (Vector3f. 0 0 1)))))
|
rlm@106
|
986 top-position (.mult joint-rotation (Vector3f. 8 0 0))
|
rlm@106
|
987
|
rlm@106
|
988 origin (doto
|
rlm@106
|
989 (sphere 0.1 :physical? false :color ColorRGBA/Cyan
|
rlm@106
|
990 :position top-position))
|
rlm@106
|
991 top (doto
|
rlm@106
|
992 (sphere 0.1 :physical? false :color ColorRGBA/Yellow
|
rlm@106
|
993 :position top-position)
|
rlm@106
|
994
|
rlm@106
|
995 (.addControl
|
rlm@106
|
996 (RigidBodyControl.
|
rlm@106
|
997 (CapsuleCollisionShape. 0.5 1.5 1) (float 20))))
|
rlm@106
|
998 bottom (doto
|
rlm@106
|
999 (sphere 0.1 :physical? false :color ColorRGBA/DarkGray
|
rlm@106
|
1000 :position (Vector3f. 0 0 0))
|
rlm@106
|
1001 (.addControl
|
rlm@106
|
1002 (RigidBodyControl.
|
rlm@106
|
1003 (CapsuleCollisionShape. 0.5 1.5 1) (float 0))))
|
rlm@106
|
1004 table (box 10 2 10 :position (Vector3f. 0 -20 0)
|
rlm@106
|
1005 :color ColorRGBA/Gray :mass 0)
|
rlm@106
|
1006 a (.getControl top RigidBodyControl)
|
rlm@106
|
1007 b (.getControl bottom RigidBodyControl)]
|
rlm@106
|
1008
|
rlm@106
|
1009 (cond
|
rlm@106
|
1010 (= joint :cone)
|
rlm@106
|
1011
|
rlm@106
|
1012 (doto (ConeJoint.
|
rlm@106
|
1013 a b
|
rlm@106
|
1014 (world-to-local top joint-position)
|
rlm@106
|
1015 (world-to-local bottom joint-position)
|
rlm@106
|
1016 joint-rotation
|
rlm@106
|
1017 joint-rotation
|
rlm@106
|
1018 )
|
rlm@106
|
1019
|
rlm@106
|
1020
|
rlm@106
|
1021 (.setLimit (* (/ 10) Math/PI)
|
rlm@106
|
1022 (* (/ 4) Math/PI)
|
rlm@106
|
1023 0)))
|
rlm@106
|
1024 [origin top bottom table]))
|
rlm@106
|
1025
|
rlm@106
|
1026 (defn test-joint [joint]
|
rlm@106
|
1027 (let [[origin top bottom floor] (world-setup joint)
|
rlm@106
|
1028 control (.getControl top RigidBodyControl)
|
rlm@106
|
1029 move-up? (atom false)
|
rlm@106
|
1030 move-down? (atom false)
|
rlm@106
|
1031 move-left? (atom false)
|
rlm@106
|
1032 move-right? (atom false)
|
rlm@106
|
1033 roll-left? (atom false)
|
rlm@106
|
1034 roll-right? (atom false)
|
rlm@106
|
1035 timer (atom 0)]
|
rlm@106
|
1036
|
rlm@106
|
1037 (world
|
rlm@106
|
1038 (nodify [top bottom floor origin])
|
rlm@106
|
1039 (merge standard-debug-controls
|
rlm@106
|
1040 {"key-r" (fn [_ pressed?] (reset! move-up? pressed?))
|
rlm@106
|
1041 "key-t" (fn [_ pressed?] (reset! move-down? pressed?))
|
rlm@106
|
1042 "key-f" (fn [_ pressed?] (reset! move-left? pressed?))
|
rlm@106
|
1043 "key-g" (fn [_ pressed?] (reset! move-right? pressed?))
|
rlm@106
|
1044 "key-v" (fn [_ pressed?] (reset! roll-left? pressed?))
|
rlm@106
|
1045 "key-b" (fn [_ pressed?] (reset! roll-right? pressed?))})
|
rlm@106
|
1046
|
rlm@106
|
1047 (fn [world]
|
rlm@106
|
1048 (light-up-everything world)
|
rlm@106
|
1049 (enable-debug world)
|
rlm@106
|
1050 (set-gravity world (Vector3f. 0 0 0))
|
rlm@106
|
1051 )
|
rlm@106
|
1052
|
rlm@106
|
1053 (fn [world _]
|
rlm@106
|
1054 (if (zero? (rem (swap! timer inc) 100))
|
rlm@106
|
1055 (do
|
rlm@106
|
1056 ;; (println-repl @timer)
|
rlm@106
|
1057 (.attachChild (.getRootNode world)
|
rlm@106
|
1058 (sphere 0.05 :color ColorRGBA/Yellow
|
rlm@106
|
1059 :position (.getWorldTranslation top)
|
rlm@106
|
1060 :physical? false))
|
rlm@106
|
1061 (.attachChild (.getRootNode world)
|
rlm@106
|
1062 (sphere 0.05 :color ColorRGBA/LightGray
|
rlm@106
|
1063 :position (.getWorldTranslation bottom)
|
rlm@106
|
1064 :physical? false))))
|
rlm@106
|
1065
|
rlm@106
|
1066 (if @move-up?
|
rlm@106
|
1067 (.applyTorque control
|
rlm@106
|
1068 (.mult (.getPhysicsRotation control)
|
rlm@106
|
1069 (Vector3f. 0 0 10))))
|
rlm@106
|
1070 (if @move-down?
|
rlm@106
|
1071 (.applyTorque control
|
rlm@106
|
1072 (.mult (.getPhysicsRotation control)
|
rlm@106
|
1073 (Vector3f. 0 0 -10))))
|
rlm@106
|
1074 (if @move-left?
|
rlm@106
|
1075 (.applyTorque control
|
rlm@106
|
1076 (.mult (.getPhysicsRotation control)
|
rlm@106
|
1077 (Vector3f. 0 10 0))))
|
rlm@106
|
1078 (if @move-right?
|
rlm@106
|
1079 (.applyTorque control
|
rlm@106
|
1080 (.mult (.getPhysicsRotation control)
|
rlm@106
|
1081 (Vector3f. 0 -10 0))))
|
rlm@106
|
1082 (if @roll-left?
|
rlm@106
|
1083 (.applyTorque control
|
rlm@106
|
1084 (.mult (.getPhysicsRotation control)
|
rlm@106
|
1085 (Vector3f. -1 0 0))))
|
rlm@106
|
1086 (if @roll-right?
|
rlm@106
|
1087 (.applyTorque control
|
rlm@106
|
1088 (.mult (.getPhysicsRotation control)
|
rlm@106
|
1089 (Vector3f. 1 0 0))))))))
|
rlm@106
|
1090
|
rlm@99
|
1091
|
rlm@99
|
1092
|
rlm@99
|
1093 #+end_src
|
rlm@99
|
1094
|
rlm@99
|
1095
|
rlm@99
|
1096 * COMMENT generate source
|
rlm@99
|
1097 #+begin_src clojure :tangle ../src/cortex/silly.clj
|
rlm@99
|
1098 <<body-1>>
|
rlm@99
|
1099 #+end_src
|
rlm@99
|
1100
|
rlm@99
|
1101
|
rlm@94
|
1102
|
rlm@94
|
1103
|