Add RefC Buffer support

Author: madman-bob

libs/base/Data/Buffer.idr

@@ -16,6 +16,7 @@ export
 data Buffer : Type where [external]
 
 %foreign "scheme:blodwen-buffer-size"
+         "C:idris2_getBufferSize, libidris2_support, idris_buffer.h"
          "node:lambda:b => BigInt(b.length)"
 prim__bufferSize : Buffer -> Int
 
@@ -24,6 +25,7 @@ rawSize : HasIO io => Buffer -> io Int
 rawSize buf = pure (prim__bufferSize buf)
 
 %foreign "scheme:blodwen-new-buffer"
+         "C:idris2_newBuffer, libidris2_support, idris_buffer.h"
          "node:lambda:s=>Buffer.alloc(Number(s))"
 prim__newBuffer : Int -> PrimIO Buffer
 
@@ -36,16 +38,22 @@ newBuffer size
 --             then pure Nothing
 --             else pure $ Just $ MkBuffer buf size 0
 
--- might be needed if we do this in C...
+%foreign "scheme:blodwen-buffer-free"
+         "C:idris2_freeBuffer, libidris2_support, idris_buffer.h"
+         "node:lambda:buf=>undefined"
+prim__freeBuffer : Buffer -> PrimIO ()
+
 export
 freeBuffer : HasIO io => Buffer -> io ()
-freeBuffer buf = pure ()
+freeBuffer buf = primIO (prim__freeBuffer buf)
 
 %foreign "scheme:blodwen-buffer-setbyte"
+         "C:idris2_setBufferByte, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset,value)=>buf.writeUInt8(Number(value), Number(offset))"
 prim__setByte : Buffer -> Int -> Int -> PrimIO ()
 
 %foreign "scheme:blodwen-buffer-setbyte"
+         "C:idris2_setBufferByte, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset,value)=>buf.writeUInt8(Number(value), Number(offset))"
 prim__setBits8 : Buffer -> Int -> Bits8 -> PrimIO ()
 
@@ -61,10 +69,12 @@ setBits8 buf loc val
     = primIO (prim__setBits8 buf loc val)
 
 %foreign "scheme:blodwen-buffer-getbyte"
+         "C:idris2_getBufferByte, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset)=>BigInt(buf.readUInt8(Number(offset)))"
 prim__getByte : Buffer -> Int -> PrimIO Int
 
 %foreign "scheme:blodwen-buffer-getbyte"
+         "C:idris2_getBufferByte, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset)=>BigInt(buf.readUInt8(Number(offset)))"
 prim__getBits8 : Buffer -> Int -> PrimIO Bits8
 
@@ -149,6 +159,7 @@ getInt32 buf loc
     = primIO (prim__getInt32 buf loc)
 
 %foreign "scheme:blodwen-buffer-setint"
+         "C:idris2_setBufferInt, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset,value)=>buf.writeInt64(Number(value), Number(offset))"
 prim__setInt : Buffer -> Int -> Int -> PrimIO ()
 
@@ -158,6 +169,7 @@ setInt buf loc val
     = primIO (prim__setInt buf loc val)
 
 %foreign "scheme:blodwen-buffer-getint"
+         "C:idris2_getBufferInt, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset)=>BigInt(buf.readInt64(Number(offset)))"
 prim__getInt : Buffer -> Int -> PrimIO Int
 
@@ -167,6 +179,7 @@ getInt buf loc
     = primIO (prim__getInt buf loc)
 
 %foreign "scheme:blodwen-buffer-setdouble"
+         "C:idris2_setBufferDouble, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset,value)=>buf.writeDoubleLE(value, Number(offset))"
 prim__setDouble : Buffer -> Int -> Double -> PrimIO ()
 
@@ -176,6 +189,7 @@ setDouble buf loc val
     = primIO (prim__setDouble buf loc val)
 
 %foreign "scheme:blodwen-buffer-getdouble"
+         "C:idris2_getBufferDouble, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset)=>buf.readDoubleLE(Number(offset))"
 prim__getDouble : Buffer -> Int -> PrimIO Double
 
@@ -190,6 +204,7 @@ export
 stringByteLength : String -> Int
 
 %foreign "scheme:blodwen-buffer-setstring"
+         "C:idris2_setBufferString, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset,value)=>buf.write(value, Number(offset),buf.length - Number(offset), 'utf-8')"
 prim__setString : Buffer -> Int -> String -> PrimIO ()
 
@@ -199,6 +214,7 @@ setString buf loc val
     = primIO (prim__setString buf loc val)
 
 %foreign "scheme:blodwen-buffer-getstring"
+         "C:idris2_getBufferString, libidris2_support, idris_buffer.h"
          "node:lambda:(buf,offset,len)=>buf.slice(Number(offset), Number(offset+len)).toString('utf-8')"
 prim__getString : Buffer -> Int -> Int -> PrimIO String
 
@@ -221,6 +237,7 @@ bufferData buf
 
 
 %foreign "scheme:blodwen-buffer-copydata"
+         "C:idris2_copyBuffer, libidris2_support, idris_buffer.h"
          "node:lambda:(b1,o1,length,b2,o2)=>b1.copy(b2,Number(o2), Number(o1), Number(o1+length))"
 prim__copyData : Buffer -> Int -> Int -> Buffer -> Int -> PrimIO ()
 

support/chez/support.ss

@@ -186,6 +186,9 @@
 (define (blodwen-buffer-size buf)
   (bytevector-length buf))
 
+(define (blodwen-buffer-free buf)
+  (void))  ; Rely on built-in memory management
+
 (define (blodwen-buffer-setbyte buf loc val)
   (bytevector-u8-set! buf loc val))
 

support/racket/support.rkt

@@ -179,6 +179,9 @@
 (define (blodwen-buffer-size buf)
   (bytevector-length buf))
 
+(define (blodwen-buffer-free buf)
+  (void))  ; Rely on built-in memory management
+
 (define (blodwen-buffer-setbyte buf loc val)
   (bytevector-u8-set! buf loc val))
 

support/refc/memoryManagement.c

@@ -131,6 +131,14 @@ Value_GCPointer *makeGCPointer(void *ptr_Raw, Value_Closure *onCollectFct)
     return p;
 }
 
+Value_Buffer *makeBuffer(void *buf)
+{
+    Value_Buffer *b = (Value_Buffer *)newValue();
+    b->header.tag = BUFFER_TAG;
+    b->buffer = buf;
+    return b;
+}
+
 Value_Array *makeArray(int length)
 {
     Value_Array *a = (Value_Array *)newValue();

support/refc/memoryManagement.h

@@ -23,6 +23,7 @@ Value_String *makeString(char *);
 
 Value_Pointer *makePointer(void *);
 Value_GCPointer *makeGCPointer(void *ptr_Raw, Value_Closure *onCollectFct);
+Value_Buffer *makeBuffer(void *buf);
 Value_Array *makeArray(int length);
 Value_World *makeWorld(void);
 

tests/Main.idr

@@ -224,6 +224,7 @@ refcTests : TestPool
 refcTests = MkTestPool "Reference counting C backend" [C]
     [ "refc001" , "refc002"
     , "strings", "doubles"
+    , "buffer"
     ]
 
 racketTests : TestPool

tests/refc/buffer/TestBuffer.idr

@@ -0,0 +1,35 @@
+module TestBuffer
+
+import Data.Buffer
+
+put : Show a => IO a -> IO ()
+put = (>>= putStrLn . show)
+
+main : IO ()
+main = do
+    Just buf <- newBuffer 23
+        | Nothing => pure ()
+
+    setByte buf 0 1
+    setBits8 buf 1 2
+    setDouble buf 2 (sqrt 2)
+
+    let helloWorld = "Hello, world"
+
+    Just helloWorldBuf <- newBuffer (stringByteLength helloWorld)
+        | Nothing => pure ()
+
+    setString helloWorldBuf 0 "Hello, world"
+    copyData helloWorldBuf 0 12 buf 10
+
+    put $ rawSize buf
+
+    put $ getByte buf 0
+    put $ getBits8 buf 1
+    put $ getDouble buf 2
+    put $ getString buf 10 12
+
+    put $ bufferData buf
+
+    freeBuffer helloWorldBuf
+    freeBuffer buf

tests/refc/buffer/expected

@@ -0,0 +1,6 @@
+23
+1
+2
+1.414214
+"Hello, world"
+[1, 2, 205, 59, 127, 102, 158, 160, 246, 63, 72, 101, 108, 108, 111, 44, 32, 119, 111, 114, 108, 100, 0]

tests/refc/buffer/run

@@ -0,0 +1,4 @@
+$1 --no-banner --no-color --console-width 0 --cg refc -o refc_buffer TestBuffer.idr > /dev/null
+./build/exec/refc_buffer
+
+rm -rf build