-! Copyright (C) 2004, 2009 Slava Pestov.
+! 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>> ;
2dup [ expired? ] either? [
[ expired? ] both?
] [
- [ alien-address ] bi@ =
+ [ alien-address ] same?
] if
] [
2drop f
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 ;
+
+: callee-cleanup? ( abi -- ? )
+ { stdcall fastcall thiscall } member? ;
+
+ERROR: callsite-not-compiled word ;
+
+: alien-assembly ( args... return parameters abi quot -- return... )
+ \ alien-assembly callsite-not-compiled ;
-ERROR: alien-indirect-error ;
+: alien-callback ( return parameters abi quot -- alien )
+ \ alien-callback callsite-not-compiled ;
-: alien-indirect ( ... funcptr return parameters abi -- )
- alien-indirect-error ;
+: alien-indirect ( args... funcptr return parameters abi -- return... )
+ \ alien-indirect callsite-not-compiled ;
-ERROR: alien-invoke-error library symbol ;
+: alien-invoke ( args... return library function parameters varargs? -- return... )
+ \ alien-invoke callsite-not-compiled ;
-: alien-invoke ( ... return library function parameters -- ... )
- 2over alien-invoke-error ;
+<PRIVATE
! Callbacks are registered in a global hashtable. Note that they
! are also pinned in a special callback area, so clearing this
! 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 ]