X-Git-Url: https://git.jsancho.org/?p=guile-assimp.git;a=blobdiff_plain;f=src%2Fassimp.scm;h=f3d8893c7fe2265b23250ff012e2441d551f33e4;hp=4ecbcf4e3ff3f1824972d00ded1bbf64e793afd0;hb=97d868192018795a86bf88d6534c132303f6cd08;hpb=37ee792898f388533d48d02b9f732769faacaa07

diff --git a/src/assimp.scm b/src/assimp.scm
index 4ecbcf4..f3d8893 100644
--- a/src/assimp.scm
+++ b/src/assimp.scm
@@ -16,6 +16,7 @@
 
 
 (define-module (assimp assimp)
+  #:use-module (ice-9 iconv)
   #:use-module (rnrs bytevectors)
   #:use-module (system foreign))
 
@@ -26,24 +27,174 @@
 		      (dynamic-func "aiImportFile" libassimp)
 		      (list '* unsigned-int)))
 
-(define-wrapped-pointer-type scene
-  scene?
-  wrap-scene unwrap-scene
-  (lambda (s p)
-    (format p "#<scene ~x>"
-	    (pointer-address (unwrap-scene s)))))
+
+;;; Type Generation
+
+(define-syntax define-type
+  (lambda (x)
+    (define (mk-string . args)
+      (string-concatenate
+       (map (lambda (a)
+	      (if (string? a)
+		  a
+		  (symbol->string (syntax->datum a))))
+	    args)))
+    (define (mk-symbol . args)
+      (datum->syntax x
+        (string->symbol
+	 (apply mk-string args))))
+    (syntax-case x ()
+      ((_ name (field field-proc) ...)
+       (with-syntax ((type? (mk-symbol #'name "?"))
+		     (wrap-type (mk-symbol "wrap-" #'name))
+		     (unwrap-type (mk-symbol "unwrap-" #'name))
+		     (output-string (mk-string "#<" #'name " ~x>"))
+		     (type-contents (mk-symbol #'name "-contents")))
+         #'(begin
+	     (define-wrapped-pointer-type name
+	       type?
+	       wrap-type unwrap-type
+	       (lambda (x p)
+		 (format p output-string
+			 (pointer-address (unwrap-type x)))))
+	     (define (type-contents wrapped)
+	       (let ((unwrapped (unwrap-type wrapped)))
+		 (cond ((= (pointer-address unwrapped) 0)
+			'())
+		       (else
+			(filter
+			 (lambda (f)
+			   (not (null? (cdr f))))
+			 (list (cons 'field (field-proc unwrapped))
+			       ...))))))))))))
+
+
+(define (bv-uint-ref pointer index)
+  (bytevector-uint-ref
+   (pointer->bytevector pointer 4 index)
+   0
+   (native-endianness)
+   4))
+
+(define (get-aiString index)
+  (lambda (pointer)
+    (let* ((length (bv-uint-ref pointer index))
+	   (data (pointer->bytevector pointer length (+ index 4))))
+      (bytevector->string data (fluid-ref %default-port-encoding)))))
+
+(define* (get-pointer index #:optional wrap-proc)
+  (lambda (pointer)
+    (let ((p (bv-uint-ref pointer index)))
+      (cond ((= p 0) '())
+	    (else
+	     (let ((p2 (make-pointer p)))
+	       (list
+		(cond (wrap-proc (wrap-proc p2))
+		      (else p2)))))))))
+
+(define (get-array num-index root-index type)
+  (lambda (pointer)
+    (let ((num (bv-uint-ref pointer num-index))
+	  (rootp (make-pointer (bv-uint-ref pointer root-index))))
+      (cond ((> num 0)
+	     (array->list
+	      (pointer->bytevector rootp num 0 type)))
+	    (else
+	     '())))))
+
+(define* (get-structs-array num-index root-index struct-size #:optional wrap-proc)
+  (lambda (pointer)
+    (let ((num (bv-uint-ref pointer num-index))
+	  (rootp (bv-uint-ref pointer root-index)))
+      (let loop ((i (- num 1)))
+	(cond ((< i 0)
+	       '())
+	      (else
+	       (let* ((p (make-pointer (+ rootp (* i struct-size))))
+		      (wp (if wrap-proc (wrap-proc p) p)))
+		 (cons wp (loop (- i 1))))))))))
+
+(define* (get-pointer-of-pointers num-index root-index #:optional wrap-proc)
+  (lambda (pointer)
+    (let* ((num (bv-uint-ref pointer num-index))
+	   (rootp (make-pointer (bv-uint-ref pointer root-index))))
+      (let loop ((i 0))
+	(cond ((= i num)
+	       '())
+	      (else
+	       (let* ((p (make-pointer (bv-uint-ref rootp (* 4 i))))
+		      (wp (if wrap-proc (wrap-proc p) p)))
+		 (cons wp (loop (+ i 1))))))))))
+
+;;; Scenes
+
+(define-type scene
+  (flags (lambda (p) (bv-uint-ref p 0)))
+  (root-node (lambda (p) (wrap-node (make-pointer (bv-uint-ref p 4)))))
+  (meshes (get-pointer-of-pointers 8 12 wrap-mesh))
+  (materials (get-pointer-of-pointers 16 20 wrap-material))
+  (animations (get-pointer-of-pointers 24 28))
+  (textures (get-pointer-of-pointers 32 36))
+  (lights (get-pointer-of-pointers 40 44))
+  (cameras (get-pointer-of-pointers 48 52)))
 
 (define (load-scene filename flags)
   (wrap-scene
    (aiImportFile (string->pointer filename)
 		 flags)))
 
-(bytevector-uint-ref (pointer->bytevector numptob (sizeof long))
-                     0 (native-endianness)
-                     (sizeof long))
+(export load-scene
+	unwrap-scene
+	scene?
+	scene-contents)
+
+
+;;; Nodes
+
+(define-type node
+  (name (get-aiString 0))
+  (transformation (lambda (p) (array->list (pointer->bytevector p 16 1028 'f32))))
+  (parent (get-pointer 1092 wrap-node))
+  (children (get-pointer-of-pointers 1096 1100 wrap-node))
+  (meshes (get-array 1104 1108 'u32)))
+
+(export node?
+	node-contents)
+
+
+;;; Meshes
+
+(define-type mesh
+  (num-primitive-types (lambda (p) (bv-uint-ref p 0)))
+  (vertices (get-pointer-of-pointers 4 12))
+  (faces (get-structs-array 8 124 8 wrap-face))
+  (normals (lambda (p) (bv-uint-ref p 16)))
+  (tangents (lambda (p) (bv-uint-ref p 20)))
+  (bitangents (lambda (p) (bv-uint-ref p 24)))
+  (colors (lambda (p) (bv-uint-ref p 28))) ;AI_MAX_NUMBER_OF_COLOR_SETS
+  (texture-coords (lambda (p) (bv-uint-ref p 60))) ;AI_MAX_NUMBER_OF_TEXTURECOORDS
+  (num-uv-components (lambda (p) (bv-uint-ref p 92))) ;AI_MAX_NUMBER_OF_TEXTURECOORDS
+  (bones (get-pointer-of-pointers 128 132))
+  (material-index (lambda (p) (bv-uint-ref p 136))))
+
+(export mesh?
+	mesh-contents)
+
+
+;;; Materials
+
+(define-type material
+  (properties (get-pointer-of-pointers 4 0))
+  (allocated (lambda (p) (bv-uint-ref p 8))))
+
+(export material?
+	material-contents)
+
+
+;;; Faces
 
-(define (scene-contents scene)
-  (bytevector-uint-ref (pointer->bytevector (unwrap-scene scene) 16)
-		       0 (native-endiannes) 4))
+(define-type face
+  (indices (get-array 0 4 'u32)))
 
-(export load-scene)
+(export face?
+	face-contents)