Re: Help with recursive destructive function

[Michael Heerdegen]

Eric Abrahamsen <address@hidden> writes:

> These are the sorts of things I can typically handle (sometimes even
> without bugs!), whereas Clement's (or your, or Stefan's) function
> isn't...

I tweaked it a bit and extended the idea for arrays.  Hope the comments
are helpful.  Uses seq.el and cl-lib.


#+begin_src emacs-lisp
;; -*- lexical-binding: t -*-

(defun deep-edit (edit-fun data)
  ;; DATA is the structure to process.
  ;;
  ;; EDIT-FUN is a function accepting one argument THING that returns
  ;; non-nil when THING is something to be replaced.  The non-nil
  ;; return value should be a function that when called with the THING as
  ;; argument returns the replacement for THING.
  ;; Example: (lambda (thing) (and (stringp thing) #'upcase)) as
  ;; EDIT-FUN would cause all strings being replaced with their upcased
  ;; version.
  (let ((stack `((identity ignore ,data))))
    (while stack
      (pcase-let* ((`(,getter ,setter ,cell) (pop stack))
                   (current                  (funcall getter cell))
                   (modify-fun               (funcall edit-fun current)))
        (cond
         ;; 1. When we should replace CURRENT, do it
         (modify-fun
          (funcall setter cell (funcall modify-fun current)))
         ;; FIXME: Do hash-tables fit here?

         ;; 2. Check whether we need to traverse CURRENT
         ((consp current)
          (cl-callf2 append
              `((car setcar ,current)
                (cdr setcdr ,current))
              stack))
         ((and (arrayp current) (not (stringp current)))
          (cl-callf2 append
              (let ((i -1))
                (seq-map
                 (lambda (_)
                   (let ((j (cl-incf i)))
                     `(,(lambda (x)   (aref x j))
                       ,(lambda (x v) (aset x j v))
                       ,current)))
                 current))
              stack)))))))

;; Example to try:
(let ((tree '("a" "b" "c"
              (2 ("d" . 3))
              (4 . "e")
              "f"
              (("g" . "h")
               (["i" "j"
                 ("k" "l")
                 nil
                 []])))))
  (deep-edit
   (lambda (thing) (and (stringp thing) #'upcase))
   tree)
  tree)
#+end_src


Michael.