(define-module (assimp assimp)
+ #:use-module (assimp low-level material)
+ #:use-module (assimp low-level mesh)
+ #:use-module (assimp low-level scene)
+ #:use-module (ice-9 iconv)
#:use-module (rnrs bytevectors)
#:use-module (system foreign))
(string->symbol
(apply mk-string args))))
(syntax-case x ()
- ((_ name (field field-proc) ...)
+ ((_ name parser (field field-proc) ...)
(with-syntax ((type? (mk-symbol #'name "?"))
(wrap-type (mk-symbol "wrap-" #'name))
(unwrap-type (mk-symbol "unwrap-" #'name))
(list (cons 'field (field-proc unwrapped))
...))))))))))))
-
(define (bv-uint-ref pointer index)
(bytevector-uint-ref
(pointer->bytevector pointer 4 index)
(native-endianness)
4))
-(define* (get-pointer-of-pointers-procedure num-index root-index #:optional wrap-proc)
+(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))))
(wp (if wrap-proc (wrap-proc p) p)))
(cons wp (loop (+ i 1))))))))))
+(define-syntax define-conversion-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 (->)
+ ((_ parser -> name (field-name 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"))
+ (type-parse (mk-symbol #'name "-parse"))
+ ((field-reader ...) (map (lambda (f) (mk-symbol #'name "-" (car f))) #'((field-name field-proc) ...))))
+ #'(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-parse wrapped)
+ (let ((unwrapped (unwrap-type wrapped)))
+ (cond ((= (pointer-address unwrapped) 0)
+ '())
+ (else
+ (parser unwrapped)))))
+ (define (type-contents wrapped)
+ (let ((alist (type-parse wrapped)))
+ (list (cons 'field-name (field-proc alist))
+ ...)))
+ (define (field-reader wrapped)
+ (let ((alist (type-parse wrapped)))
+ (field-proc alist)))
+ ...))))))
+
+(define (field name)
+ (lambda (alist)
+ (assoc-ref alist name)))
+
+(define (get-element-address root-pointer offset)
+ (make-pointer (+ (pointer-address root-pointer) offset)))
+
+(define* (array size-tag root-tag #:key (element-size 4) (element-proc bv-uint-ref))
+ (lambda (alist)
+ (let ((size (assoc-ref alist size-tag))
+ (root (assoc-ref alist root-tag)))
+ (cond ((= (pointer-address root) 0)
+ '())
+ (else
+ (let loop ((i 0))
+ (cond ((= i size)
+ '())
+ (else
+ (cons (element-proc root (* element-size i))
+ (loop (+ i 1)))))))))))
+
+(define (wrap proc wrap-proc)
+ (define (make-wrap element)
+ (let ((pointer
+ (cond ((pointer? element)
+ (if (= (pointer-address element) 0)
+ #f
+ element))
+ ((= element 0)
+ #f)
+ (else
+ (make-pointer element)))))
+ (cond (pointer
+ (wrap-proc pointer))
+ (else
+ #f))))
+ (lambda (alist)
+ (let ((res (proc alist)))
+ (cond ((list? res)
+ (map make-wrap res))
+ (else
+ (make-wrap res))))))
+
+(define (sized-string string-tag)
+ (lambda (alist)
+ (let ((s (assoc-ref alist string-tag)))
+ (cond (s
+ (bytevector->string
+ (u8-list->bytevector (list-head (cadr s) (car s)))
+ (fluid-ref %default-port-encoding)))
+ (else
+ #f)))))
+
+
;;; 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-procedure 8 12 wrap-mesh))
- (materials (get-pointer-of-pointers-procedure 16 20))
- (animations (get-pointer-of-pointers-procedure 24 28))
- (textures (get-pointer-of-pointers-procedure 32 36))
- (lights (get-pointer-of-pointers-procedure 40 44))
- (cameras (get-pointer-of-pointers-procedure 48 52)))
-
-(define (load-scene filename flags)
- (wrap-scene
- (aiImportFile (string->pointer filename)
- flags)))
+(define-conversion-type parse-aiScene -> scene
+ (flags (field 'mFlags))
+ (root-node (wrap (field 'mRootNode) wrap-node))
+ (meshes (wrap (array 'mNumMeshes 'mMeshes) wrap-mesh))
+ (materials (wrap (array 'mNumMaterials 'mMaterials) wrap-material))
+ (animations (array 'mNumAnimations 'mAnimations))
+ (textures (array 'mNumTextures 'mTextures))
+ (lights (array 'mNumLights 'mLights))
+ (cameras (array 'mNumCameras 'mCameras)))
+
+ (define (load-scene filename flags)
+ (wrap-scene
+ (aiImportFile (string->pointer filename)
+ flags)))
(export load-scene
- unwrap-scene
scene?
- scene-contents)
+ scene-contents
+ scene-flags
+ scene-root-node
+ scene-meshes
+ scene-materials
+ scene-animations
+ scene-textures
+ scene-lights
+ scene-cameras)
;;; Nodes
-(define-type node
- (name (lambda (p) (bv-uint-ref p 0))) ;check, it's a struct
- (transformation (lambda (p) (bv-uint-ref p 1028))) ;check, it's a struct
- (parent (lambda (p) (bv-uint-ref p 1092)))
- (children (get-pointer-of-pointers-procedure 1096 1100))
- (meshes (get-pointer-of-pointers-procedure 1104 1108)))
+(define-conversion-type parse-aiNode -> node
+ (name (sized-string 'mName))
+ (transformation (field 'mTransformation))
+ (parent (wrap (field 'mParent) wrap-node))
+ (children (wrap (array 'mNumChildren 'mChildren) wrap-node))
+ (meshes (array 'mNumMeshes 'mMeshes)))
(export node?
- node-contents)
+ node-contents
+ node-name
+ node-transformation
+ node-parent
+ node-children
+ node-meshes)
;;; Meshes
-(define-type mesh
- (num-primitive-types (lambda (p) (bv-uint-ref p 0)))
- (vertices (get-pointer-of-pointers-procedure 4 12))
- (faces (get-pointer-of-pointers-procedure 8 124))
- (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-procedure 128 132))
- (material-index (lambda (p) (bv-uint-ref p 136))))
+(define-conversion-type parse-aiMesh -> mesh
+ (name (sized-string 'mName))
+ (primitive-types (field 'mPrimitiveTypes))
+ (vertices (array 'mNumVertices 'mVertices #:element-proc get-element-address))
+ (faces (wrap (array 'mNumFaces 'mFaces #:element-size 8 #:element-proc get-element-address) wrap-face))
+ (normals (array 'mNumVertices 'mNormals #:element-size 12 #:element-proc get-element-address))
+ (tangents (array 'mNumVertices 'mTangents #:element-size 12 #:element-proc get-element-address))
+ (bitangents (array 'mNumVertices 'mBitangents #:element-size 12 #:element-proc get-element-address))
+ (colors (field 'mColors))
+ (texture-coords (field 'mTextureCoords))
+ (num-uv-components (field 'mNumUVComponents))
+ (bones (array 'mNumBones 'mBones))
+ (material-index (field 'mMaterialIndex))
+)
(export mesh?
- mesh-contents)
+ mesh-contents
+ mesh-name
+ mesh-primitive-types
+ mesh-vertices
+ mesh-faces
+ mesh-normals
+ mesh-tangents
+ mesh-bitangents
+ mesh-colors
+ mesh-texture-coords
+ mesh-num-uv-components
+ mesh-bones
+ mesh-material-index)
+
+
+;;; Materials
+
+(define-conversion-type parse-aiMaterial -> material
+ (properties (array 'mNumProperties 'mProperties))
+ (num-allocated (field 'mNumAllocated)))
+
+(export material?
+ material-contents
+ material-properties
+ material-num-allocated)
+
+
+;;; Faces
+
+(define-conversion-type parse-aiFace -> face
+ (indices (array 'mNumIndices 'mIndices)))
+
+(export face?
+ face-contents
+ face-indices)