improve janet cloning

polygolf-lang · Mar 11, 2024 · 6c0d222 · 6c0d222
1 parent d9d47d5
commit 6c0d222
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/src/languages/janet/emit.ts b/src/languages/janet/emit.ts
@@ -45,6 +45,11 @@ export class JanetEmitter extends VisitorEmitter {
     const prop = spine.pathFragment?.prop;
 
     switch (n.kind) {
+      case "Cast":
+        if (n.targetType === "array") {
+          return ["@[;", "$GLUE$", $.expr, "]"];
+        }
+        throw new EmitError(n, "unsuported cast target type");
       case "Block": {
         return prop === "consequent" || prop === "alternate"
           ? list("do", $.children.join())

diff --git a/src/languages/janet/index.ts b/src/languages/janet/index.ts
@@ -29,6 +29,7 @@ import {
   isInt,
   intToDecOpOrText,
   isForEachChar,
+  cast,
 } from "../../IR";
 import {
   golfLastPrint,
@@ -220,8 +221,9 @@ const janetLanguage: Language = {
           type.kind === "List" &&
           ["boolean", "integer", "text"].includes(type.member.kind)
         ) {
-          return func("array/slice", node, int(0), int(-1));
+          return cast(node, "array");
         }
+        return func("thaw", node);
       }),
     ),
     simplegolf(

diff --git a/src/programs/control-flow.test.md b/src/programs/control-flow.test.md
@@ -365,7 +365,7 @@ $b <- $a;
 _Janet_
 
 ```janet nogolf
-(var a @[1 2 3])(var b(array/slice a 0 -1))
+(var a @[1 2 3])(var b @[;a])
 ```
 
 _Javascript_
@@ -403,6 +403,12 @@ $a <- (list (list 1 2 3) (list 4 5 6) (list 7 8 9));
 $b <- $a;
 ```
 
+_Janet_
+
+```janet nogolf
+(var a @[@[1 2 3]@[4 5 6]@[7 8 9]])(var b(thaw a))
+```
+
 _Javascript_
 
 ```js nogolf