]> gitweb.factorcode.org Git - factor.git/blobdiff - core/alien/alien.factor
core: trim using lists with lint.vocabs tool
[factor.git] / core / alien / alien.factor
index 10012ea3d0393091a4f53a832a72c1fea3eb9c9d..689053efd88bfadc9b07ea8722e15d533e3c08ab 100644 (file)
@@ -1,21 +1,56 @@
 ! Copyright (C) 2004, 2010 Slava Pestov.
 ! See http://factorcode.org/license.txt for BSD license.
-USING: accessors assocs kernel math namespaces sequences system
-kernel.private byte-arrays arrays init ;
+USING: accessors assocs byte-arrays byte-vectors continuations
+continuations.private kernel kernel.private math namespaces
+sequences ;
+USE: init ! required but does not reference words
 IN: alien
 
+BUILTIN: alien { underlying c-ptr read-only initial: f } expired ;
+BUILTIN: dll { path byte-array read-only initial: B{ } } ;
+
+PRIMITIVE: <callback> ( word return-rewind -- alien )
+PRIMITIVE: <displaced-alien> ( displacement c-ptr -- alien )
+PRIMITIVE: alien-address ( c-ptr -- addr )
+PRIMITIVE: free-callback ( alien -- )
+
+<PRIVATE
+PRIMITIVE: current-callback ( -- n )
+PRIVATE>
+
 PREDICATE: pinned-alien < alien underlying>> not ;
 
 UNION: pinned-c-ptr pinned-alien POSTPONE: f ;
 
-GENERIC: >c-ptr ( obj -- c-ptr )
+GENERIC: element-size ( seq -- n ) flushable
+
+M: byte-array element-size drop 1 ; inline
+
+M: byte-vector element-size drop 1 ; inline
+
+M: slice element-size seq>> element-size ; inline
+
+M: f element-size drop 1 ; inline
+
+GENERIC: byte-length ( obj -- n ) flushable
+
+M: object byte-length [ length ] [ element-size ] bi * ; inline
+
+GENERIC: >c-ptr ( obj -- c-ptr ) flushable
 
 M: c-ptr >c-ptr ; inline
 
+M: slice >c-ptr
+    [ [ from>> ] [ element-size ] bi * ] [ seq>> >c-ptr ] bi
+    <displaced-alien> ; inline
+
 SLOT: underlying
 
 M: object >c-ptr underlying>> ; inline
 
+: binary-object ( obj -- c-ptr n )
+    [ >c-ptr ] [ byte-length ] bi ; inline
+
 GENERIC: expired? ( c-ptr -- ? ) flushable
 
 M: alien expired? expired>> ;
@@ -33,7 +68,7 @@ M: alien equal?
         2dup [ expired? ] either? [
             [ expired? ] both?
         ] [
-            [ alien-address ] bi@ =
+            [ alien-address ] same?
         ] if
     ] [
         2drop f
@@ -42,25 +77,28 @@ M: alien equal?
 M: pinned-alien hashcode*
     nip dup expired>> [ drop 1234 ] [ alien-address ] if ;
 
-ERROR: alien-callback-error ;
+SINGLETONS: stdcall thiscall fastcall cdecl mingw ;
 
-: alien-callback ( return parameters abi quot -- alien )
-    alien-callback-error ;
+UNION: abi stdcall thiscall fastcall cdecl mingw ;
 
-ERROR: alien-indirect-error ;
+: callee-cleanup? ( abi -- ? )
+    { stdcall fastcall thiscall } member? ;
 
-: alien-indirect ( ... funcptr return parameters abi -- ... )
-    alien-indirect-error ;
+ERROR: callsite-not-compiled word ;
 
-ERROR: alien-invoke-error library symbol ;
+: alien-assembly ( args... return parameters abi quot -- return... )
+    \ alien-assembly callsite-not-compiled ;
 
-: alien-invoke ( ... return library function parameters -- ... )
-    2over alien-invoke-error ;
+: alien-callback ( return parameters abi quot -- alien )
+    \ alien-callback callsite-not-compiled ;
 
-ERROR: alien-assembly-error code ;
+: alien-indirect ( args... funcptr return parameters abi -- return... )
+    \ alien-indirect callsite-not-compiled ;
 
-: alien-assembly ( ... return library parameters abi quot -- ... )
-    dup alien-assembly-error ;
+: alien-invoke ( args... return library function parameters varargs? -- return... )
+    \ alien-invoke callsite-not-compiled ;
+
+<PRIVATE
 
 ! Callbacks are registered in a global hashtable. Note that they
 ! are also pinned in a special callback area, so clearing this
@@ -68,17 +106,34 @@ ERROR: alien-assembly-error code ;
 ! cleared on startup.
 SYMBOL: callbacks
 
-[ H{ } clone callbacks set-global ] "alien" add-startup-hook
+STARTUP-HOOK: [ H{ } clone callbacks set-global ]
 
-<PRIVATE
+! Used by compiler.codegen to wrap callback bodies
+: do-callback ( callback-quot wait-quot: ( callback -- ) -- )
+    t CONTEXT-OBJ-IN-CALLBACK-P set-context-object
+    init-namestack
+    init-catchstack
+    current-callback
+    [ 2drop call ] [ swap call( callback -- ) drop ] 3bi ; inline
 
+! A utility for defining global variables that are recompiled in
+! every session
 TUPLE: expiry-check object alien ;
 
 : recompute-value? ( check -- ? )
-    dup [ alien>> expired? ] [ drop t ] if ;
+    [ alien>> expired? ] [ t ] if* ;
+
+: delete-values ( value assoc -- )
+    [ rot drop = ] with assoc-reject! drop ;
 
 PRIVATE>
 
+: unregister-and-free-callback ( alien -- )
+    [ callbacks get delete-values ] [ free-callback ] bi ;
+
+: with-callback ( alien quot -- )
+    over [ unregister-and-free-callback ] curry finally ; inline
+
 : initialize-alien ( symbol quot -- )
     swap dup get-global dup recompute-value?
     [ drop [ call dup 31337 <alien> expiry-check boa ] dip set-global ]