From 349c01fff273ffcaf5d46873121a067eb82f7997 Mon Sep 17 00:00:00 2001
From: Kaz Kylheku <kaz@kylheku.com>
Date: Fri, 30 Jun 2023 21:42:48 -0700
Subject: Callable integers become assignable places.

* lib.c (dwim_set): Handle seq argument being an integer
or range.

* tests/012/callable.tl: A few tests.

* txr.1: Documented.
---
 lib.c | 96 ++++++++++++++++++++++++++++++++++++++++++++++---------------------
 1 file changed, 66 insertions(+), 30 deletions(-)

(limited to 'lib.c')

diff --git a/lib.c b/lib.c
index 086041a0..44f5005f 100644
--- a/lib.c
+++ b/lib.c
@@ -13134,42 +13134,78 @@ val replace(val seq, val items, val from, val to)
 val dwim_set(val place_p, val seq, varg vargs)
 {
   val self = lit("index/range assignment");
+  type_t st = type(seq);
 
-  switch (type(seq)) {
-  case COBJ:
-    if (type(seq) == COBJ) {
-      if (seq->co.cls == hash_cls) {
-        args_normalize_least(vargs, 3);
-
-        switch (vargs->fill) {
-        case 2:
-          (void) sethash(seq, vargs->arg[0], vargs->arg[1]);
-          break;
-        case 3:
-          if (vargs->list)
-            goto excargs;
-          (void) sethash(seq, vargs->arg[0], vargs->arg[2]);
-          break;
-        default:
-          goto fewargs;
+  switch (st) {
+  case NUM:
+  case BGNUM:
+  case RNG:
+    {
+      args_normalize_least(vargs, 3);
+      switch (vargs->fill) {
+      case 2:
+        {
+          val arg = vargs->arg[0];
+          val newval = vargs->arg[1];
+          switch (type(arg)) {
+          case NUM:
+          case BGNUM:
+          case RNG:
+            goto notplace;
+          default:
+            if (st == RNG) {
+              range_bind (x, y, seq);
+              if (!place_p && listp(arg))
+                goto notplace;
+              return replace(arg, newval, x, y);
+            } else {
+              (void) refset(arg, seq, newval);
+              return seq;
+            }
+          }
         }
+      case 1:
+      case 0:
+        goto fewargs;
+      default:
+        goto excargs;
+      }
+    }
+  case COBJ:
+    if (seq->co.cls == hash_cls) {
+      args_normalize_least(vargs, 3);
 
-        return seq;
+      switch (vargs->fill) {
+      case 2:
+        (void) sethash(seq, vargs->arg[0], vargs->arg[1]);
+        break;
+      case 3:
+        if (vargs->list)
+          goto excargs;
+        (void) sethash(seq, vargs->arg[0], vargs->arg[2]);
+        break;
+      case 1:
+      case 0:
+        goto fewargs;
+      default:
+        goto excargs;
       }
-      if (obj_struct_p(seq)) {
-        {
-          val lambda_set_meth = get_special_slot(seq, lambda_set_m);
-          if (lambda_set_meth) {
-            (void) funcall(method_args(seq, lambda_set_s, vargs));
-            return seq;
-          }
+
+      return seq;
+    }
+    if (obj_struct_p(seq)) {
+      {
+        val lambda_set_meth = get_special_slot(seq, lambda_set_m);
+        if (lambda_set_meth) {
+          (void) funcall(method_args(seq, lambda_set_s, vargs));
+          return seq;
         }
-        if (get_special_slot(seq, car_m))
-          goto list;
-        type_mismatch(lit("~a: object ~s lacks "
-                          "~s or ~s method"),
-                      self, seq, lambda_set_s, car_s, nao);
       }
+      if (get_special_slot(seq, car_m))
+        goto list;
+      type_mismatch(lit("~a: object ~s lacks "
+                        "~s or ~s method"),
+                    self, seq, lambda_set_s, car_s, nao);
     }
     /* fallthrough */
   default:
-- 
cgit v1.2.3