lib: intro the c module
authorAlexis Laferrière <alexis.laf@xymus.net>
Tue, 1 Jul 2014 19:17:52 +0000 (15:17 -0400)
committerAlexis Laferrière <alexis.laf@xymus.net>
Mon, 7 Jul 2014 19:02:13 +0000 (15:02 -0400)
Signed-off-by: Alexis Laferrière <alexis.laf@xymus.net>

lib/c.nit [new file with mode: 0644]

diff --git a/lib/c.nit b/lib/c.nit
new file mode 100644 (file)
index 0000000..c0bd650
--- /dev/null
+++ b/lib/c.nit
@@ -0,0 +1,95 @@
+# This file is part of NIT (http://www.nitlanguage.org).
+#
+# Copyright 2014 Alexis Laferrière <alexis.laf@xymus.net>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Utilities and performant structure for the FFI with C
+module c
+
+# A thin wrapper around a `NativeCArray` adding length information
+abstract class CArray[E]
+       super AbstractArrayRead[E]
+
+       private init(length: Int) do self._length = length
+end
+
+# A native C array, as in a pointer to the first element of the array
+extern class NativeCArray `{ void * `}
+       type E: nullable Object
+       type SELF: NativeCArray
+
+       fun [](index: E): E is abstract
+       fun []=(index: E, val: E) is abstract
+
+       # Return pointer to the address to the second element of this array
+       #
+       # This is the standard `+` operator on pointers in C
+       fun +(offset: Int): SELF is abstract
+end
+
+# Wrapper around an array of `int` in C (`int*`) with length and destroy state
+class CIntArray
+       super CArray[Int]
+
+       var native_array: NativeCIntArray
+       init(size: Int)
+       do
+               native_array = new NativeCIntArray(size)
+               super size
+       end
+
+       redef fun [](index)
+       do
+               assert not destroyed
+               assert index >= 0 and index < length
+               return native_array[index]
+       end
+
+       fun []=(index: Int, val: Int)
+       do
+               assert not destroyed
+               assert index >= 0 and index < length
+               native_array[index] = val
+       end
+
+       var destroyed = false
+       fun destroy
+       do
+               if destroyed then return
+
+               native_array.free
+               destroyed = true
+       end
+end
+
+# An array of `int` in C (`int*`)
+extern class NativeCIntArray `{ int* `}
+       super NativeCArray
+       redef type E: Int
+       redef type SELF: NativeCIntArray
+
+       new(size: Int) `{ return calloc(size, sizeof(int)); `}
+       redef fun [](index) `{ return recv[index]; `}
+       redef fun []=(index, val) `{ recv[index] = val; `}
+
+       redef fun +(offset) `{ return recv + offset; `}
+end
+
+redef class NativeString
+       super NativeCArray
+       redef type E: Char
+       redef type SELF: NativeString
+
+       redef fun +(offset) `{ return recv + offset; `}
+end