new version

.gitignore

@@ -5,4 +5,4 @@
 *.o
 *.so
 
-test/bin
+/test/bin/**

haxelib.json

@@ -9,5 +9,8 @@
   "classPath": "src/",
   "contributors": [
     "Aurel300"
-  ]
+  ],
+  "dependencies": {
+    "ammer-core": ""
+  }
 }

src/ammer/Lib.hx

@@ -0,0 +1,24 @@
+// ammer-bake: ammer Lib true
+package ammer;
+
+class Lib {
+  // struct methods
+  public static macro function allocStruct<T>(cls:Class<T>, ?initVals:{}):T;
+  public static macro function freeStruct<T>(e:T):Void;
+  public static macro function nullPtrStruct<T>(cls:Class<T>):T;
+
+  // box methods
+  public static macro function allocBox<T>(cls:Class<T>, ?initVal:T):ammer.ffi.Box<T>;
+  public static macro function nullPtrBox<T>(cls:Class<T>):ammer.ffi.Box<T>;
+
+  // array methods
+  public static macro function allocArray<T>(cls:Class<T>, size:Int, ?initVal:T):ammer.ffi.Array<T>;
+  public static macro function nullPtrArray<T>(cls:Class<T>):ammer.ffi.Array<T>;
+
+  public static macro function vecToArrayCopy<T>(vec:haxe.ds.Vector<T>):ammer.ffi.Array<T>;
+  public static macro function vecToArrayRef<T>(vec:haxe.ds.Vector<T>):ammer.ffi.ArrayRef<T>;
+  public static macro function vecToArrayRefForce<T>(vec:haxe.ds.Vector<T>):ammer.ffi.ArrayRef<T>;
+
+  // Haxe ref methods
+  public static macro function createHaxeRef<T>(cls:Class<T>, e:T):ammer.ffi.Haxe<T>;
+}

src/ammer/Lib.macro.baked.hx

@@ -0,0 +1,22 @@
+// ammer-bake: ammer Lib.macro macro
+package ammer;
+
+import haxe.macro.Context;
+import haxe.macro.Context.currentPos;
+import haxe.macro.Context.fatalError as fail;
+import haxe.macro.Context.resolveType;
+import haxe.macro.Expr;
+import haxe.macro.Type;
+import haxe.macro.TypeTools;
+import ammer.internal.*;
+import ammer.internal.v1.AmmerBaked.mergedInfo as info;
+
+using Lambda;
+
+class Lib {
+  static function withPos<T>(pos:Position, f:()->T):T {
+    return f();
+  }
+// ammer-include: internal/Utils.hx lib-baked
+// ammer-include: Lib.macro.hx lib-baked
+}

src/ammer/Lib.macro.hx

@@ -0,0 +1,251 @@
+package ammer;
+
+import haxe.macro.Context;
+import haxe.macro.Context.fatalError as fail;
+import haxe.macro.Expr;
+import haxe.macro.TypeTools;
+import ammer.internal.*;
+import ammer.internal.Ammer.mergedInfo as info;
+import ammer.internal.Utils.access;
+import ammer.internal.Utils.accessTp;
+import ammer.internal.Utils.complexTypeExpr;
+import ammer.internal.Utils.expectTypePath;
+import ammer.internal.Utils.isNull;
+import ammer.internal.Utils.triggerTyping;
+import ammer.internal.Utils.typeId;
+import ammer.internal.Utils.typeId2;
+
+using Lambda;
+
+class Lib {
+  static function withPos<T>(pos:Position, f:()->T):T {
+    return Reporting.withPosition(pos, f);
+  }
+
+  // These methods are inserted into `Lib.macro.hx` when baking a library.
+  // This avoids code duplication/synchronisation issues. Importantly, the code
+  // is just string-pasted, so it is important that the `import`s that are
+  // in `Lib.macro.baked.hx` are sufficient for the code to work.
+
+// ammer-fragment-begin: lib-baked
+  public static function allocStruct(cls:Expr, ?initVals:Expr):Expr {
+    var ct = complexTypeExpr(cls);
+    var clsType = withPos(cls.pos, () -> triggerTyping(ct));
+    clsType != null || throw fail("invalid type in allocStruct call", cls.pos);
+    var struct = info.structs[typeId(clsType)];
+    struct != null || throw fail("not a struct type in allocStruct call", cls.pos);
+    struct.gen.alloc != null || throw fail("struct type was not marked with @:ammer.alloc", cls.pos);
+    var alloc = struct.gen.alloc;
+    if (isNull(initVals)) {
+      return macro @:privateAccess $p{access(clsType)}.$alloc();
+    }
+    var assigns = (switch (initVals) {
+      case {expr: EObjectDecl(fields)}:
+        [ for (field in fields) macro @:pos(field.expr.pos) $p{["_allocated", field.field]} = $e{field.expr} ];
+      case _: throw fail("expected initial values (e.g. {a: 1, b: 2, ...}) as second argument of allocStruct call", initVals.pos);
+    });
+    return macro {
+      var _allocated = @:privateAccess $p{access(clsType)}.$alloc();
+      $b{assigns};
+      _allocated;
+    };
+  }
+
+  public static function freeStruct(e:Expr):Expr {
+    var typed = withPos(e.pos, () -> Context.typeExpr(e));
+    var clsType = (switch (typed.t) {
+      case TInst(t, []): t.get();
+      case _: throw fail("invalid type in freeStruct call", e.pos);
+    });
+    var struct = info.structs[typeId(clsType)];
+    struct != null || throw fail("not a struct type in freeStruct call", e.pos);
+    struct.gen.free != null || throw fail("struct type was not marked with @:ammer.alloc", e.pos);
+    var free = struct.gen.free;
+    return macro @:privateAccess ${Context.storeTypedExpr(typed)}.$free();
+  }
+
+  public static function nullPtrStruct(cls:Expr):Expr {
+    var ct = complexTypeExpr(cls);
+    var clsType = withPos(cls.pos, () -> triggerTyping(ct));
+    clsType != null || throw fail("invalid type in nullPtrStruct call", cls.pos);
+    var typeId = typeId(clsType);
+    var struct = info.structs[typeId];
+    //var opaque = info.opaques[typeId];
+    //(struct != null || opaque != null) || throw fail("not a struct type or opaque type in nullPtrStruct call", cls.pos);
+    struct != null || throw fail("not a struct type in nullPtrStruct call", cls.pos);
+    struct.gen.nullPtr != null || throw fail("struct type was not marked with @:ammer.alloc", cls.pos);
+    var nullPtr = struct.gen.nullPtr;
+    return macro @:privateAccess $p{access(clsType)}.$nullPtr();
+  }
+
+  public static function allocBox(cls:Expr, ?initVal:Expr):Expr {
+    var elCt = complexTypeExpr(cls);
+    var elType = withPos(cls.pos, () -> Context.resolveType(elCt, cls.pos));
+    #if ammer
+    // if baked, ammer.ffi.Box does not exist, only its monomorphisations
+    withPos(cls.pos, () -> Context.resolveType((macro : ammer.ffi.Box<$elCt>), cls.pos));
+    #end
+    var box = info.boxes.byElementTypeId[typeId2(elType)];
+    box != null || throw fail("not a known box type in allocBox call", cls.pos);
+    var tp = expectTypePath(box.boxCt);
+    if (isNull(initVal)) {
+      return macro @:privateAccess new $tp($e{box.alloc});
+    }
+    return macro {
+      var _allocated = @:privateAccess new $tp($e{box.alloc});
+      _allocated.set($initVal);
+      _allocated;
+    };
+  }
+
+  public static function nullPtrBox(cls:Expr):Expr {
+    var elCt = complexTypeExpr(cls);
+    var elType = withPos(cls.pos, () -> Context.resolveType(elCt, cls.pos));
+    #if ammer
+    // if baked, ammer.ffi.Box does not exist, only its monomorphisations
+    withPos(cls.pos, () -> Context.resolveType((macro : ammer.ffi.Box<$elCt>), cls.pos));
+    #end
+    var box = info.boxes.byElementTypeId[typeId2(elType)];
+    box != null || throw fail("not a known box type in nullPtrBox call", cls.pos);
+    var tp = expectTypePath(box.boxCt);
+    return macro @:privateAccess $p{accessTp(tp)}.nullPtr();
+  }
+
+  public static function allocArray(cls:Expr, size:Expr, ?initVal:Expr):Expr {
+    var elCt = complexTypeExpr(cls);
+    var elType = withPos(cls.pos, () -> Context.resolveType(elCt, cls.pos));
+    #if ammer
+    // if baked, ammer.ffi.Array does not exist, only its monomorphisations
+    withPos(cls.pos, () -> Context.resolveType((macro : ammer.ffi.Array<$elCt>), cls.pos));
+    #end
+    var array = info.arrays.byElementTypeId[typeId2(elType)];
+    array != null || throw fail("not a known array type in allocArray call", cls.pos);
+    var tp = expectTypePath(array.arrayCt);
+    if (isNull(initVal)) {
+      return macro {
+        var _size = $size;
+        @:privateAccess new $tp($e{array.alloc});
+      }
+    }
+    return macro {
+      var _size = $size;
+      var _val = $initVal;
+      var _allocated = @:privateAccess new $tp($e{array.alloc});
+      for (i in 0..._size) {
+        _allocated.set(i, _val);
+      }
+      _allocated;
+    };
+  }
+
+  public static function nullPtrArray(cls:Expr):Expr {
+    var elCt = complexTypeExpr(cls);
+    var elType = withPos(cls.pos, () -> Context.resolveType(elCt, cls.pos));
+    #if ammer
+    // if baked, ammer.ffi.Array does not exist, only its monomorphisations
+    withPos(cls.pos, () -> Context.resolveType((macro : ammer.ffi.Array<$elCt>), cls.pos));
+    #end
+    var array = info.arrays.byElementTypeId[typeId2(elType)];
+    array != null || throw fail("not a known array type in allocArray call", cls.pos);
+    var tp = expectTypePath(array.arrayCt);
+    return macro @:privateAccess $p{accessTp(tp)}.nullPtr();
+  }
+
+  public static function vecToArrayCopy(vec:Expr):Expr {
+    var typed = withPos(vec.pos, () -> Context.typeExpr(vec));
+    var elType = (switch (typed.t) {
+      case TInst(typeId(_.get()) => "haxe.ds.Vector.Vector", [el]): el;
+      case TAbstract(typeId(_.get()) => "haxe.ds.Vector.Vector", [el]): el;
+      case _: throw fail("argument should be a haxe.ds.Vector", vec.pos);
+    });
+    var elCt = TypeTools.toComplexType(elType);
+    var stored = Context.storeTypedExpr(typed);
+    #if ammer
+    // if baked, ammer.ffi.Array does not exist, only its monomorphisations
+    withPos(vec.pos, () -> Context.resolveType((macro : ammer.ffi.Array<$elCt>), vec.pos));
+    #end
+    var array = info.arrays.byElementTypeId[typeId2(elType)];
+    array != null || throw fail("not a known array type in vecToArrayCopy call", vec.pos);
+    var tp = expectTypePath(array.arrayCt);
+    return macro {
+      var _vec = $vec;
+      @:privateAccess new $tp($e{array.fromHaxeCopy});
+    };
+  }
+
+  public static function vecToArrayRef(vec:Expr):Expr {
+    var typed = withPos(vec.pos, () -> Context.typeExpr(vec));
+    var elType = (switch (typed.t) {
+      case TInst(typeId(_.get()) => "haxe.ds.Vector.Vector", [el]): el;
+      case TAbstract(typeId(_.get()) => "haxe.ds.Vector.Vector", [el]): el;
+      case _: throw fail("argument should be a haxe.ds.Vector", vec.pos);
+    });
+    var elCt = TypeTools.toComplexType(elType);
+    var stored = Context.storeTypedExpr(typed);
+    #if ammer
+    // if baked, ammer.ffi.Array does not exist, only its monomorphisations
+    withPos(vec.pos, () -> Context.resolveType((macro : ammer.ffi.Array<$elCt>), vec.pos));
+    #end
+    var array = info.arrays.byElementTypeId[typeId2(elType)];
+    array != null || throw fail("not a known array type in vecToArrayRef call", vec.pos);
+    var tp = expectTypePath(array.arrayRefCt);
+    if (array.fromHaxeRef != null) {
+      // if references are supported, create an array ref
+      return macro {
+        var _vec = $vec;
+        @:privateAccess new $tp($e{array.fromHaxeRef}, _vec);
+      };
+    }
+    // if not, create a fake ref with the same API
+    return macro {
+      var _vec = $vec;
+      @:privateAccess new $tp($e{array.fromHaxeCopy}, _vec);
+    };
+  }
+
+  public static function vecToArrayRefForce(vec:Expr):Expr {
+    var typed = withPos(vec.pos, () -> Context.typeExpr(vec));
+    var elType = (switch (typed.t) {
+      case TInst(typeId(_.get()) => "haxe.ds.Vector.Vector", [el]): el;
+      case TAbstract(typeId(_.get()) => "haxe.ds.Vector.Vector", [el]): el;
+      case _: throw fail("argument should be a haxe.ds.Vector", vec.pos);
+    });
+    var elCt = TypeTools.toComplexType(elType);
+    var stored = Context.storeTypedExpr(typed);
+    #if ammer
+    // if baked, ammer.ffi.Array does not exist, only its monomorphisations
+    withPos(vec.pos, () -> Context.resolveType((macro : ammer.ffi.Array<$elCt>), vec.pos));
+    #end
+    var array = info.arrays.byElementTypeId[typeId2(elType)];
+    array != null || throw fail("not a known array type in vecToArrayRefForce call", vec.pos);
+    var tp = expectTypePath(array.arrayRefCt);
+    array.fromHaxeRef != null || throw fail("platform does not support non-copy references to Vector", vec.pos);
+    return macro {
+      var _vec = $vec;
+      @:privateAccess new $tp($e{array.fromHaxeRef}, _vec);
+    };
+  }
+
+  public static function createHaxeRef(cls:Expr, e:Expr):Expr {
+    var elCt = complexTypeExpr(cls);
+    var elType = withPos(cls.pos, () -> Context.resolveType(elCt, cls.pos));
+    #if ammer
+    // if baked, ammer.ffi.Haxe does not exist, only its monomorphisations
+    withPos(cls.pos, () -> Context.resolveType((macro : ammer.ffi.Haxe<$elCt>), cls.pos));
+    #end
+    var elId = typeId2(elType);
+    if (info.haxeRefs.byElementTypeId.exists(elId)) {
+      var haxeRef = info.haxeRefs.byElementTypeId[elId];
+      return macro {
+        var _hxval = $e;
+        $e{haxeRef.create};
+      };
+    }
+    info.haxeRefs.byElementTypeId.exists(".Any.Any") || throw 0;
+    return macro {
+      var _hxval = $e;
+      new ammer.internal.LibTypes.HaxeAnyRef<$elCt>($e{info.haxeRefs.byElementTypeId[".Any.Any"].create});
+    };
+  }
+// ammer-fragment-end: lib-baked
+}

src/ammer/Syntax.hx

@@ -0,0 +1,4 @@
+package ammer;
+
+@:autoBuild(ammer.Syntax.build())
+interface Syntax {}