chiark / gitweb /
46270c47d6509301dfac9d8e1ecd592c586843db
[sod] / src / utilities.lisp
1 ;;; -*-lisp-*-
2 ;;;
3 ;;; Various handy utilities
4 ;;;
5 ;;; (c) 2009 Straylight/Edgeware
6 ;;;
7
8 ;;;----- Licensing notice ---------------------------------------------------
9 ;;;
10 ;;; This file is part of the Sensible Object Design, an object system for C.
11 ;;;
12 ;;; SOD is free software; you can redistribute it and/or modify
13 ;;; it under the terms of the GNU General Public License as published by
14 ;;; the Free Software Foundation; either version 2 of the License, or
15 ;;; (at your option) any later version.
16 ;;;
17 ;;; SOD is distributed in the hope that it will be useful,
18 ;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
19 ;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20 ;;; GNU General Public License for more details.
21 ;;;
22 ;;; You should have received a copy of the GNU General Public License
23 ;;; along with SOD; if not, write to the Free Software Foundation,
24 ;;; Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
25
26 (cl:defpackage #:sod-utilities
27   (:use #:common-lisp
28
29         ;; MOP from somewhere.
30         #+sbcl #:sb-mop
31         #+(or cmu clisp) #:mop
32         #+ecl #:clos))
33
34 (cl:in-package #:sod-utilities)
35
36 ;;;--------------------------------------------------------------------------
37 ;;; Macro hacks.
38
39 (export 'with-gensyms)
40 (defmacro with-gensyms ((&rest binds) &body body)
41   "Evaluate BODY with variables bound to fresh symbols.
42
43    The BINDS are a list of entries (VAR [NAME]), and a singleton list can be
44    replaced by just a symbol; each VAR is bound to a fresh symbol generated
45    by (gensym NAME), where NAME defaults to the symbol-name of VAR."
46   `(let (,@(mapcar (lambda (bind)
47                      (multiple-value-bind (var name)
48                          (if (atom bind)
49                              (values bind (concatenate 'string
50                                            (symbol-name bind) "-"))
51                              (destructuring-bind
52                                  (var &optional
53                                       (name (concatenate 'string
54                                              (symbol-name var) "-")))
55                                  bind
56                                (values var name)))
57                        `(,var (gensym ,name))))
58                    binds))
59      ,@body))
60
61 (eval-when (:compile-toplevel :load-toplevel :execute)
62   (defun strip-quote (form)
63     "If FORM looks like (quote FOO) for self-evaluating FOO, return FOO.
64
65    If FORM is a symbol whose constant value is `nil' then return `nil'.
66    Otherwise return FORM unchanged.  This makes it easier to inspect constant
67    things.  This is a utility for `once-only'."
68
69     (cond ((and (consp form)
70                 (eq (car form) 'quote)
71                 (cdr form)
72                 (null (cddr form)))
73            (let ((body (cadr form)))
74              (if (or (not (or (consp body) (symbolp body)))
75                      (member body '(t nil))
76                      (keywordp body))
77                  body
78                  form)))
79           ((and (symbolp form) (boundp form) (null (symbol-value form)))
80            nil)
81           (t
82            form))))
83
84 (export 'once-only)
85 (defmacro once-only (binds &body body)
86   "Macro helper for preventing repeated evaluation.
87
88    The syntax is actually hairier than shown:
89
90         once-only ( [[ :environment ENV ]] { VAR | (VAR [VALUE-FORM]) }* )
91           { FORM }*
92
93    So, the BINDS are a list of entries (VAR [VALUE-FORM]); a singleton list
94    can be replaced by just a symbol VAR, and the VALUE-FORM defaults to VAR.
95    But before them you can have keyword arguments.  Only one is defined so
96    far.  See below for the crazy things that does.
97
98    The result of evaluating a ONCE-ONLY form is a form with the structure
99
100         (let ((#:GS1 VALUE-FORM1)
101               ...
102               (#:GSn VALUE-FORMn))
103           STUFF)
104
105    where STUFF is the value of the BODY forms, as an implicit progn, in an
106    environment with the VARs bound to the corresponding gensyms.
107
108    As additional magic, if any of the VALUE-FORMs is actually constant (as
109    determined by inspection, and aided by `constantp' if an :environment is
110    supplied, then no gensym is constructed for it, and the VAR is bound
111    directly to the constant form.  Moreover, if the constant form looks like
112    (quote FOO) for a self-evaluating FOO then the outer layer of quoting is
113    stripped away."
114
115   ;; We need an extra layer of gensyms in our expansion: we'll want the
116   ;; expansion to examine the various VALUE-FORMs to find out whether they're
117   ;; constant without evaluating them repeatedly.  This also helps with
118   ;; another problem: we explicitly encourage the rebinding of a VAR
119   ;; (probably a macro argument) to a gensym which will be bound to the value
120   ;; of the form previously held in VAR itself -- so the gensym and value
121   ;; form must exist at the same time and we need two distinct variables.
122
123   (with-gensyms ((envvar "ENV-") lets sym (bodyfunc "BODY-"))
124     (let ((env nil))
125
126       ;; First things first: let's pick up the keywords.
127       (loop
128         (unless (and binds (keywordp (car binds)))
129           (return))
130         (ecase (pop binds)
131           (:environment (setf env (pop binds)))))
132
133       ;; Now we'll investigate the bindings.  Turn each one into a list (VAR
134       ;; VALUE-FORM TEMP) where TEMP is an appropriate gensym -- see the note
135       ;; above.
136       (let ((canon (mapcar (lambda (bind)
137                              (multiple-value-bind (var form)
138                                  (if (atom bind)
139                                      (values bind bind)
140                                      (destructuring-bind
141                                          (var &optional (form var)) bind
142                                        (values var form)))
143                                (list var form
144                                      (gensym (format nil "T-~A-"
145                                                      (symbol-name var))))))
146                            binds)))
147
148         `(let* (,@(and env `((,envvar ,env)))
149                 (,lets nil)
150                 ,@(mapcar (lambda (bind)
151                             (destructuring-bind (var form temp) bind
152                               (declare (ignore var))
153                               `(,temp ,form)))
154                           canon)
155                 ,@(mapcar (lambda (bind)
156                             (destructuring-bind (var form temp) bind
157                               (declare (ignore form))
158                               `(,var
159                                 (cond ((constantp ,temp
160                                                   ,@(and env `(,envvar)))
161                                        (strip-quote ,temp))
162                                       ((symbolp ,temp)
163                                        ,temp)
164                                       (t
165                                        (let ((,sym (gensym
166                                                     ,(concatenate 'string
167                                                       (symbol-name var)
168                                                       "-"))))
169                                          (push (list ,sym ,temp) ,lets)
170                                          ,sym))))))
171                           canon))
172            (flet ((,bodyfunc () ,@body))
173              (if ,lets
174                  `(let (,@(nreverse ,lets)) ,(,bodyfunc))
175                  (,bodyfunc))))))))
176
177 (export 'parse-body)
178 (defun parse-body (body &key (docp t) (declp t))
179   "Parse the BODY into a docstring, declarations and the body forms.
180
181    These are returned as three lists, so that they can be spliced into a
182    macro expansion easily.  The declarations are consolidated into a single
183    `declare' form.  If DOCP is nil then a docstring is not permitted; if
184    DECLP is nil, then declarations are not permitted."
185   (let ((decls nil)
186         (doc nil))
187     (loop
188       (cond ((null body) (return))
189             ((and declp (consp (car body)) (eq (caar body) 'declare))
190              (setf decls (append decls (cdr (pop body)))))
191             ((and docp (stringp (car body)) (not doc) (cdr body))
192              (setf doc (pop body)))
193             (t (return))))
194     (values (and doc (list doc))
195             (and decls (list (cons 'declare decls)))
196             body)))
197
198 ;;;--------------------------------------------------------------------------
199 ;;; Locatives.
200
201 (export '(loc locp))
202 (defstruct (loc (:predicate locp) (:constructor make-loc (reader writer)))
203   "Locative data type.  See `locf' and `ref'."
204   (reader nil :type function)
205   (writer nil :type function))
206
207 (export 'locf)
208 (defmacro locf (place &environment env)
209   "Slightly cheesy locatives.
210
211    (locf PLACE) returns an object which, using the `ref' function, can be
212    used to read or set the value of PLACE.  It's cheesy because it uses
213    closures rather than actually taking the address of something.  Also,
214    unlike Zetalisp, we don't overload `car' to do our dirty work."
215   (multiple-value-bind
216       (valtmps valforms newtmps setform getform)
217       (get-setf-expansion place env)
218     `(let* (,@(mapcar #'list valtmps valforms))
219        (make-loc (lambda () ,getform)
220                  (lambda (,@newtmps) ,setform)))))
221
222 (export 'ref)
223 (declaim (inline ref (setf ref)))
224 (defun ref (loc)
225   "Fetch the value referred to by a locative."
226   (funcall (loc-reader loc)))
227 (defun (setf ref) (new loc)
228   "Store a new value in the place referred to by a locative."
229   (funcall (loc-writer loc) new))
230
231 (export 'with-locatives)
232 (defmacro with-locatives (locs &body body)
233   "Evaluate BODY with implicit locatives.
234
235    LOCS is a list of items of the form (SYM [LOC-EXPR]), where SYM is a
236    symbol and LOC-EXPR evaluates to a locative.  If LOC-EXPR is omitted, it
237    defaults to SYM.  As an abbreviation for a common case, LOCS may be a
238    symbol instead of a list.
239
240    The BODY is evaluated in an environment where each SYM is a symbol macro
241    which expands to (ref LOC-EXPR) -- or, in fact, something similar which
242    doesn't break if LOC-EXPR has side-effects.  Thus, references, including
243    `setf' forms, fetch or modify the thing referred to by the LOC-EXPR.
244    Useful for covering over where something uses a locative."
245   (setf locs (mapcar (lambda (item)
246                        (cond ((atom item) (list item item))
247                              ((null (cdr item)) (list (car item) (car item)))
248                              (t item)))
249                      (if (listp locs) locs (list locs))))
250   (let ((tt (mapcar (lambda (l) (declare (ignore l)) (gensym)) locs))
251         (ll (mapcar #'cadr locs))
252         (ss (mapcar #'car locs)))
253     `(let (,@(mapcar (lambda (tmp loc) `(,tmp ,loc)) tt ll))
254        (symbol-macrolet (,@(mapcar (lambda (sym tmp)
255                                      `(,sym (ref ,tmp))) ss tt))
256          ,@body))))
257
258 ;;;--------------------------------------------------------------------------
259 ;;; Anaphorics.
260
261 (export 'it)
262
263 (export 'aif)
264 (defmacro aif (cond cons &optional (alt nil altp))
265   "If COND is not nil, evaluate CONS with `it' bound to the value of COND.
266
267    Otherwise, if given, evaluate ALT; `it' isn't bound in ALT."
268   (once-only (cond)
269     `(if ,cond (let ((it ,cond)) ,cons) ,@(and altp `(,alt)))))
270
271 (export 'awhen)
272 (defmacro awhen (cond &body body)
273   "If COND, evaluate BODY as a progn with `it' bound to the value of COND."
274   `(let ((it ,cond)) (when it ,@body)))
275
276 (export 'aand)
277 (defmacro aand (&rest forms)
278   "Like `and', but anaphoric.
279
280    Each FORM except the first is evaluated with `it' bound to the value of
281    the previous one.  If there are no forms, then the result it `t'; if there
282    is exactly one, then wrapping it in `aand' is pointless."
283   (labels ((doit (first rest)
284              (if (null rest)
285                  first
286                  `(let ((it ,first))
287                     (if it ,(doit (car rest) (cdr rest)) nil)))))
288     (if (null forms)
289         't
290         (doit (car forms) (cdr forms)))))
291
292 (export 'acond)
293 (defmacro acond (&body clauses &environment env)
294   "Like COND, but with `it' bound to the value of the condition.
295
296    Each of the CLAUSES has the form (CONDITION FORM*); if a CONDITION is
297    non-nil then evaluate the FORMs with `it' bound to the non-nil value, and
298    return the value of the last FORM; if there are no FORMs, then return `it'
299    itself.  If the CONDITION is nil then continue with the next clause; if
300    all clauses evaluate to nil then the result is nil."
301   (labels ((walk (clauses)
302              (if (null clauses)
303                  `nil
304                  (once-only (:environment env (cond (caar clauses)))
305                    (if (and (constantp cond)
306                             (if (and (consp cond) (eq (car cond) 'quote))
307                                 (cadr cond) cond))
308                        (if (cdar clauses)
309                            `(let ((it ,cond))
310                               (declare (ignorable it))
311                               ,@(cdar clauses))
312                            cond)
313                        `(if ,cond
314                             ,(if (cdar clauses)
315                                  `(let ((it ,cond))
316                                     (declare (ignorable it))
317                                     ,@(cdar clauses))
318                                  cond)
319                             ,(walk (cdr clauses))))))))
320     (walk clauses)))
321
322 (export '(acase aecase atypecase aetypecase))
323 (defmacro acase (value &body clauses)
324   `(let ((it ,value)) (case it ,@clauses)))
325 (defmacro aecase (value &body clauses)
326   `(let ((it ,value)) (ecase it ,@clauses)))
327 (defmacro atypecase (value &body clauses)
328   `(let ((it ,value)) (typecase it ,@clauses)))
329 (defmacro aetypecase (value &body clauses)
330   `(let ((it ,value)) (etypecase it ,@clauses)))
331
332 (export 'asetf)
333 (defmacro asetf (&rest places-and-values &environment env)
334   "Anaphoric update of places.
335
336    The PLACES-AND-VALUES are alternating PLACEs and VALUEs.  Each VALUE is
337    evaluated with IT bound to the current value stored in the corresponding
338    PLACE."
339   `(progn ,@(loop for (place value) on places-and-values by #'cddr
340                   collect (multiple-value-bind
341                               (temps inits newtemps setform getform)
342                               (get-setf-expansion place env)
343                             `(let* (,@(mapcar #'list temps inits)
344                                     (it ,getform))
345                                (multiple-value-bind ,newtemps ,value
346                                  ,setform))))))
347
348 ;;;--------------------------------------------------------------------------
349 ;;; MOP hacks (not terribly demanding).
350
351 (export 'instance-initargs)
352 (defgeneric instance-initargs (instance)
353   (:documentation
354    "Return a plausble list of initargs for INSTANCE.
355
356    The idea is that you can make a copy of INSTANCE by invoking
357
358         (apply #'make-instance (class-of INSTANCE)
359                (instance-initargs INSTANCE))
360
361    The default implementation works by inspecting the slot definitions and
362    extracting suitable initargs, so this will only succeed if enough slots
363    actually have initargs specified that `initialize-instance' can fill in
364    the rest correctly.
365
366    The list returned is freshly consed, and you can destroy it if you like.")
367   (:method ((instance standard-object))
368     (mapcan (lambda (slot)
369               (aif (slot-definition-initargs slot)
370                    (list (car it)
371                          (slot-value instance (slot-definition-name slot)))
372                    nil))
373             (class-slots (class-of instance)))))
374
375 (export '(copy-instance copy-instance-using-class))
376 (defgeneric copy-instance-using-class (class instance &rest initargs)
377   (:documentation
378    "Metaobject protocol hook for `copy-instance'.")
379   (:method ((class standard-class) instance &rest initargs)
380     (let ((copy (allocate-instance class)))
381       (dolist (slot (class-slots class))
382         (let ((name (slot-definition-name slot)))
383           (when (slot-boundp instance name)
384             (setf (slot-value copy name) (slot-value instance name)))))
385       (apply #'shared-initialize copy nil initargs))))
386 (defun copy-instance (object &rest initargs)
387   "Construct and return a copy of OBJECT.
388
389    The new object has the same class as OBJECT, and the same slot values
390    except where overridden by INITARGS."
391   (apply #'copy-instance-using-class (class-of object) object initargs))
392
393 (export '(generic-function-methods method-specializers
394           eql-specializer eql-specializer-object))
395
396 ;;;--------------------------------------------------------------------------
397 ;;; List utilities.
398
399 (export 'make-list-builder)
400 (defun make-list-builder (&optional initial)
401   "Return a simple list builder."
402
403   ;; The `builder' is just a cons cell whose cdr will be the list that's
404   ;; wanted.  Effectively, then, we have a list that's one item longer than
405   ;; we actually want.  The car of this extra initial cons cell is always the
406   ;; last cons in the list -- which is now well defined because there's
407   ;; always at least one.
408
409   (let ((builder (cons nil initial)))
410     (setf (car builder) (last builder))
411     builder))
412
413 (export 'lbuild-add)
414 (defun lbuild-add (builder item)
415   "Add an ITEM to the end of a list BUILDER."
416   (let ((new (cons item nil)))
417     (setf (cdar builder) new
418           (car builder) new))
419   builder)
420
421 (export 'lbuild-add-list)
422 (defun lbuild-add-list (builder list)
423   "Add a LIST to the end of a list BUILDER.  The LIST will be clobbered."
424   (when list
425     (setf (cdar builder) list
426           (car builder) (last list)))
427   builder)
428
429 (export 'lbuild-list)
430 (defun lbuild-list (builder)
431   "Return the constructed list."
432   (cdr builder))
433
434 (export 'mappend)
435 (defun mappend (function list &rest more-lists)
436   "Like a nondestructive `mapcan'.
437
438    Map FUNCTION over the the corresponding elements of LIST and MORE-LISTS,
439    and return the result of appending all of the resulting lists."
440   (reduce #'append (apply #'mapcar function list more-lists) :from-end t))
441
442 (export '(inconsistent-merge-error merge-error-candidates))
443 (define-condition inconsistent-merge-error (error)
444   ((candidates :initarg :candidates
445                :reader merge-error-candidates))
446   (:documentation
447    "Reports an inconsistency in the arguments passed to `merge-lists'.")
448   (:report (lambda (condition stream)
449              (format stream "Merge inconsistency: failed to decide between ~
450                              ~{~#[~;~A~;~A and ~A~:;~
451                                   ~@{~A, ~#[~;and ~A~]~}~]~}"
452                      (merge-error-candidates condition)))))
453
454 (export 'merge-lists)
455 (defun merge-lists (lists &key pick (test #'eql) (present #'identity))
456   "Return a merge of the given LISTS.
457
458    The resulting list contains the items of the given LISTS, with duplicates
459    removed.  The order of the resulting list is consistent with the orders of
460    the input LISTS in the sense that if A precedes B in some input list then
461    A will also precede B in the output list.  If the lists aren't consistent
462    (e.g., some list contains A followed by B, and another contains B followed
463    by A) then an error of type `inconsistent-merge-error' is signalled.  The
464    offending items are filtered for presentation through the PRESENT function
465    before being attached to the condition, so as to produce a more useful
466    diagnostic message.
467
468    Item equality is determined by TEST.
469
470    If there is an ambiguity at any point -- i.e., a choice between two or
471    more possible next items to emit -- then PICK is called to arbitrate.
472    PICK is called with two arguments: the list of candidate next items, and
473    the current output list.  It should return one of the candidate items.
474    The order of the candidates in the list given to the PICK function
475    reflects their order in the input LISTS: item A will precede item B in the
476    candidates list if and only if an occurrence of A appears in an earlier
477    input list than any occurrence of item B.  (This completely determines the
478    order of the candidates: it is not possible that two candidates appear in
479    the same input list would resolve the ambiguity between them.)  If PICK is
480    omitted then the item chosen is the one appearing in the earliest of the
481    input lists: i.e., effectively, the default PICK function is
482
483         (lambda (candidates output-so-far)
484           (declare (ignore output-so-far))
485           (car candidates))
486
487    The primary use of this function is in computing class precedence lists.
488    By building the input lists and selecting the PICK function appropriately,
489    a variety of different CPL algorithms can be implemented."
490
491   (do ((lb (make-list-builder)))
492       ((null lists) (lbuild-list lb))
493
494     ;; The candidate items are the ones at the front of the input lists.
495     ;; Gather them up, removing duplicates.  If a candidate is somewhere in
496     ;; one of the other lists other than at the front then we reject it.  If
497     ;; we've just rejected everything, then we can make no more progress and
498     ;; the input lists were inconsistent.
499     (let* ((candidates (delete-duplicates (mapcar #'car lists)
500                                           :test test :from-end t))
501            (leasts (remove-if (lambda (item)
502                                 (some (lambda (list)
503                                         (member item (cdr list) :test test))
504                                       lists))
505                               candidates))
506            (winner (cond ((null leasts)
507                           (error 'inconsistent-merge-error
508                                  :candidates (mapcar present candidates)))
509                          ((null (cdr leasts))
510                           (car leasts))
511                          (pick
512                           (funcall pick leasts (lbuild-list lb)))
513                          (t (car leasts)))))
514
515       ;; Check that the PICK function isn't conning us.
516       (assert (member winner leasts :test test))
517
518       ;; Update the output list and remove the winning item from the input
519       ;; lists.  We know that it must be at the front of each input list
520       ;; containing it.  At this point, we discard input lists entirely when
521       ;; they run out of entries.  The loop ends when there are no more input
522       ;; lists left, i.e., when we've munched all of the input items.
523       (lbuild-add lb winner)
524       (setf lists (delete nil (mapcar (lambda (list)
525                                         (if (funcall test winner (car list))
526                                             (cdr list)
527                                             list))
528                                       lists))))))
529
530 (export 'categorize)
531 (defmacro categorize ((itemvar items &key bind) categories &body body)
532   "Categorize ITEMS into lists and invoke BODY.
533
534    The ITEMVAR is a symbol; as the macro iterates over the ITEMS, ITEMVAR
535    will contain the current item.  The BIND argument is a list of LET*-like
536    clauses.  The CATEGORIES are a list of clauses of the form (SYMBOL
537    PREDICATE).
538
539    The behaviour of the macro is as follows.  ITEMVAR is assigned (not
540    bound), in turn, each item in the list ITEMS.  The PREDICATEs in the
541    CATEGORIES list are evaluated in turn, in an environment containing
542    ITEMVAR and the BINDings, until one of them evaluates to a non-nil value.
543    At this point, the item is assigned to the category named by the
544    corresponding SYMBOL.  If none of the PREDICATEs returns non-nil then an
545    error is signalled; a PREDICATE consisting only of T will (of course)
546    match anything; it is detected specially so as to avoid compiler warnings.
547
548    Once all of the ITEMS have been categorized in this fashion, the BODY is
549    evaluated as an implicit PROGN.  For each SYMBOL naming a category, a
550    variable named after that symbol will be bound in the BODY's environment
551    to a list of the items in that category, in the same order in which they
552    were found in the list ITEMS.  The final values of the macro are the final
553    values of the BODY."
554
555   (let* ((cat-names (mapcar #'car categories))
556          (cat-match-forms (mapcar #'cadr categories))
557          (cat-vars (mapcar (lambda (name) (gensym (concatenate 'string
558                                                    (symbol-name name) "-")))
559                            cat-names))
560          (items-var (gensym "ITEMS-")))
561     `(let ((,items-var ,items)
562            ,@(mapcar (lambda (cat-var) (list cat-var nil)) cat-vars))
563        (dolist (,itemvar ,items-var)
564          (let* ,bind
565            (cond ,@(mapcar (lambda (cat-match-form cat-var)
566                              `(,cat-match-form
567                                (push ,itemvar ,cat-var)))
568                            cat-match-forms cat-vars)
569                  ,@(and (not (member t cat-match-forms))
570                         `((t (error "Failed to categorize ~A" ,itemvar)))))))
571        (let ,(mapcar (lambda (name var)
572                        `(,name (nreverse ,var)))
573                      cat-names cat-vars)
574          ,@body))))
575
576 ;;;--------------------------------------------------------------------------
577 ;;; Strings and characters.
578
579 (export 'frob-identifier)
580 (defun frob-identifier (string &key (swap-case t) (swap-hyphen t))
581   "Twiddles the case of STRING.
582
583    If all the letters in STRING are uppercase, and SWAP-CASE is true, then
584    switch them to lowercase; if they're all lowercase then switch them to
585    uppercase.  If there's a mix then leave them all alone.  At the same time,
586    if there are underscores but no hyphens, and SWAP-HYPHEN is true, then
587    switch them to hyphens, if there are hyphens and no underscores, switch
588    them underscores, and if there are both then leave them alone.
589
590    This is an invertible transformation, which turns vaguely plausible Lisp
591    names into vaguely plausible C names and vice versa.  Lisp names with
592    `funny characters' like stars and percent signs won't be any use, of
593    course."
594
595   ;; Work out what kind of a job we've got to do.  Gather flags: bit 0 means
596   ;; there are upper-case letters; bit 1 means there are lower-case letters;
597   ;; bit 2 means there are hyphens; bit 3 means there are underscores.
598   ;;
599   ;; Consequently, (logxor flags (ash flags 1)) is interesting: bit 1 is set
600   ;; if we have to frob case; bit 3 is set if we have to swap hyphens and
601   ;; underscores.  So use this to select functions which do bits of the
602   ;; mapping, and then compose them together.
603   (let* ((flags (reduce (lambda (state ch)
604                           (logior state
605                                   (cond ((upper-case-p ch) 1)
606                                         ((lower-case-p ch) 2)
607                                         ((char= ch #\-) 4)
608                                         ((char= ch #\_) 8)
609                                         (t 0))))
610                         string
611                         :initial-value 0))
612          (mask (logxor flags (ash flags 1)))
613          (letter (cond ((or (not swap-case) (not (logbitp 1 mask)))
614                         (constantly nil))
615                        ((logbitp 0 flags)
616                         (lambda (ch)
617                           (and (alpha-char-p ch) (char-downcase ch))))
618                        (t
619                         (lambda (ch)
620                           (and (alpha-char-p ch) (char-upcase ch))))))
621          (uscore-hyphen (cond ((or (not (logbitp 3 mask)) (not swap-hyphen))
622                                (constantly nil))
623                               ((logbitp 2 flags)
624                                (lambda (ch) (and (char= ch #\-) #\_)))
625                               (t
626                                (lambda (ch) (and (char= ch #\_) #\-))))))
627
628     (if (logbitp 3 (logior mask (ash mask 2)))
629         (map 'string (lambda (ch)
630                        (or (funcall letter ch)
631                            (funcall uscore-hyphen ch)
632                            ch))
633              string)
634         string)))
635
636 (export 'whitespace-char-p)
637 (declaim (inline whitespace-char-p))
638 (defun whitespace-char-p (char)
639   "Returns whether CHAR is a whitespace character.
640
641    Whitespaceness is determined relative to the compile-time readtable, which
642    is probably good enough for most purposes."
643   (case char
644     (#.(loop for i below char-code-limit
645              for ch = (code-char i)
646              unless (with-input-from-string (in (string ch))
647                       (peek-char t in nil))
648              collect ch) t)
649     (t nil)))
650
651 (export 'update-position)
652 (declaim (inline update-position))
653 (defun update-position (char line column)
654   "Updates LINE and COLUMN appropriately for having read the character CHAR.
655
656    Returns the new LINE and COLUMN numbers."
657   (case char
658     ((#\newline #\vt #\page)
659      (values (1+ line) 0))
660     ((#\tab)
661      (values line (logandc2 (+ column 8) 7)))
662     (t
663      (values line (1+ column)))))
664
665 (export 'backtrack-position)
666 (declaim (inline backtrack-position))
667 (defun backtrack-position (char line column)
668   "Updates LINE and COLUMN appropriately for having unread CHAR.
669
670    Well, actually an approximation for it; it will likely be wrong if the
671    last character was a tab.  But when the character is read again, it will
672    be correct."
673
674   ;; This isn't perfect: if the character doesn't actually match what was
675   ;; really read then it might not actually be possible: for example, if we
676   ;; push back a newline while in the middle of a line, or a tab while not at
677   ;; a tab stop.  In that case, we'll just lose, but hopefully not too badly.
678   (case char
679
680     ;; In the absence of better ideas, I'll set the column number to zero.
681     ;; This is almost certainly wrong, but with a little luck nobody will ask
682     ;; and it'll be all right soon.
683     ((#\newline #\vt #\page) (values (1- line) 0))
684
685     ;; Winding back a single space is sufficient.  If the position is
686     ;; currently on a tab stop then it'll advance back here next time.  If
687     ;; not, we're going to lose anyway because the previous character
688     ;; certainly couldn't have been a tab.
689     (#\tab (values line (1- column)))
690
691     ;; Anything else: just decrement the column and cross fingers.
692     (t (values line (1- column)))))
693
694 ;;;--------------------------------------------------------------------------
695 ;;; Functions.
696
697 (export 'compose)
698 (defun compose (function &rest more-functions)
699   "Composition of functions.  Functions are applied left-to-right.
700
701    This is the reverse order of the usual mathematical notation, but I find
702    it easier to read.  It's also slightly easier to work with in programs.
703    That is, (compose F1 F2 ... Fn) is what a category theorist might write as
704    F1 ; F2 ; ... ; Fn, rather than F1 o F2 o ... o Fn."
705
706   (labels ((compose1 (func-a func-b)
707              (lambda (&rest args)
708                (multiple-value-call func-b (apply func-a args)))))
709     (reduce #'compose1 more-functions :initial-value function)))
710
711 ;;;--------------------------------------------------------------------------
712 ;;; Variables.
713
714 (export 'defvar-unbound)
715 (defmacro defvar-unbound (var doc)
716   "Make VAR a special variable with documentation DOC, but leave it unbound."
717   `(eval-when (:compile-toplevel :load-toplevel :execute)
718      (defvar ,var)
719      (setf (documentation ',var 'variable) ',doc)
720      ',var))
721
722 ;;;--------------------------------------------------------------------------
723 ;;; Symbols.
724
725 (export 'symbolicate)
726 (defun symbolicate (&rest symbols)
727   "Return a symbol named after the concatenation of the names of the SYMBOLS.
728
729    The symbol is interned in the current `*package*'.  Trad."
730   (intern (apply #'concatenate 'string (mapcar #'symbol-name symbols))))
731
732 ;;;--------------------------------------------------------------------------
733 ;;; Object printing.
734
735 (export 'maybe-print-unreadable-object)
736 (defmacro maybe-print-unreadable-object
737     ((object stream &rest args) &body body)
738   "Print helper for usually-unreadable objects.
739
740    If `*print-escape*' is set then print OBJECT unreadably using BODY.
741    Otherwise just print using BODY."
742   (with-gensyms (print)
743     `(flet ((,print () ,@body))
744        (if *print-escape*
745            (print-unreadable-object (,object ,stream ,@args)
746              (,print))
747            (,print)))))
748
749 (export 'print-ugly-stuff)
750 (defun print-ugly-stuff (stream func)
751   "Print not-pretty things to the stream underlying STREAM.
752
753    The Lisp pretty-printing machinery, notably `pprint-logical-block', may
754    interpose additional streams between its body and the original target
755    stream.  This makes it difficult to make use of the underlying stream's
756    special features, whatever they might be."
757
758   ;; This is unpleasant.  Hacky hacky.
759   #.(or #+sbcl '(if (typep stream 'sb-pretty:pretty-stream)
760                   (let ((target (sb-pretty::pretty-stream-target stream)))
761                     (pprint-newline :mandatory stream)
762                     (funcall func target))
763                   (funcall func stream))
764         #+cmu '(if (typep stream 'pp:pretty-stream)
765                   (let ((target (pp::pretty-stream-target stream)))
766                     (pprint-newline :mandatory stream)
767                     (funcall func target))
768                   (funcall func stream))
769         '(funcall func stream)))
770
771 ;;;--------------------------------------------------------------------------
772 ;;; Iteration macros.
773
774 (export 'dosequence)
775 (defmacro dosequence ((var seq &key (start 0) (end nil) indexvar)
776                       &body body
777                       &environment env)
778   "Macro for iterating over general sequences.
779
780    Iterates over a (sub)sequence SEQ, delimited by START and END (which are
781    evaluated).  For each item of SEQ, BODY is invoked with VAR bound to the
782    item, and INDEXVAR (if requested) bound to the item's index.  (Note that
783    this is different from most iteration constructs in Common Lisp, which
784    work by mutating the variable.)
785
786    The loop is surrounded by an anonymous BLOCK and the loop body forms an
787    implicit TAGBODY, as is usual.  There is no result-form, however."
788
789   (once-only (:environment env seq start end)
790     (with-gensyms ((ivar "INDEX-") (endvar "END-") (bodyfunc "BODY-"))
791       (multiple-value-bind (docs decls body) (parse-body body :docp nil)
792         (declare (ignore docs))
793
794         (flet ((loopguts (indexp listp endvar)
795                  ;; Build a DO-loop to do what we want.
796                  (let* ((do-vars nil)
797                         (end-condition (if endvar
798                                            `(>= ,ivar ,endvar)
799                                            `(endp ,seq)))
800                         (item (if listp
801                                   `(car ,seq)
802                                   `(aref ,seq ,ivar)))
803                         (body-call `(,bodyfunc ,item)))
804                    (when listp
805                      (push `(,seq (nthcdr ,start ,seq) (cdr ,seq))
806                            do-vars))
807                    (when indexp
808                      (push `(,ivar ,start (1+ ,ivar)) do-vars))
809                    (when indexvar
810                      (setf body-call (append body-call (list ivar))))
811                    `(do ,do-vars (,end-condition) ,body-call))))
812
813           `(block nil
814              (flet ((,bodyfunc (,var ,@(and indexvar `(,indexvar)))
815                       ,@decls
816                       (tagbody ,@body)))
817                (etypecase ,seq
818                  (vector
819                   (let ((,endvar (or ,end (length ,seq))))
820                     ,(loopguts t nil endvar)))
821                  (list
822                   (if ,end
823                       ,(loopguts t t end)
824                       ,(loopguts indexvar t nil)))))))))))
825
826 ;;;--------------------------------------------------------------------------
827 ;;; Structure accessor hacks.
828
829 (export 'define-access-wrapper)
830 (defmacro define-access-wrapper (from to &key read-only)
831   "Make (FROM THING) work like (TO THING).
832
833    If not READ-ONLY, then also make (setf (FROM THING) VALUE) work like
834    (setf (TO THING) VALUE).
835
836    This is mostly useful for structure slot accessors where the slot has to
837    be given an unpleasant name to avoid it being an external symbol."
838   `(progn
839      (declaim (inline ,from ,@(and (not read-only) `((setf ,from)))))
840      (defun ,from (object)
841        (,to object))
842      ,@(and (not read-only)
843             `((defun (setf ,from) (value object)
844                 (setf (,to object) value))))))
845
846 ;;;--------------------------------------------------------------------------
847 ;;; Condition and error utilities.
848
849 (export 'designated-condition)
850 (defun designated-condition (default-type datum arguments
851                              &key allow-pointless-arguments)
852   "Return the condition designated by DATUM and ARGUMENTS.
853
854    DATUM and ARGUMENTS together are a `condition designator' of (some
855    supertype of) DEFAULT-TYPE; return the condition so designated."
856   (typecase datum
857     (condition
858      (unless (or allow-pointless-arguments (null arguments))
859        (error "Argument list provided with specific condition"))
860      datum)
861     (symbol
862      (apply #'make-condition datum arguments))
863     ((or string function)
864      (make-condition default-type
865                      :format-control datum
866                      :format-arguments arguments))
867     (t
868      (error "Unexpected condition designator datum ~S" datum))))
869
870 ;;;--------------------------------------------------------------------------
871 ;;; CLOS hacking.
872
873 (export 'default-slot)
874 (defmacro default-slot ((instance slot &optional (slot-names t))
875                           &body value
876                           &environment env)
877   "If INSTANCE's slot named SLOT is unbound, set it to VALUE.
878
879    Only set SLOT if it's listed in SLOT-NAMES, or SLOT-NAMES is `t' (i.e., we
880    obey the `shared-initialize' protocol).  SLOT-NAMES defaults to `t', so
881    you can use it in `initialize-instance' or similar without ill effects.
882    Both INSTANCE and SLOT are evaluated; VALUE is an implicit progn and only
883    evaluated if it's needed."
884
885   (once-only (:environment env instance slot slot-names)
886     `(when ,(if (eq slot-names t)
887                   `(not (slot-boundp ,instance ,slot))
888                   `(and (not (slot-boundp ,instance ,slot))
889                         (or (eq ,slot-names t)
890                             (member ,slot ,slot-names))))
891        (setf (slot-value ,instance ,slot)
892              (progn ,@value)))))
893
894 (export 'define-on-demand-slot)
895 (defmacro define-on-demand-slot (class slot (instance) &body body)
896   "Defines a slot which computes its initial value on demand.
897
898    Sets up the named SLOT of CLASS to establish its value as the implicit
899    progn BODY, by defining an appropriate method on `slot-unbound'."
900   (multiple-value-bind (docs decls body) (parse-body body)
901     (with-gensyms (classvar slotvar)
902       `(defmethod slot-unbound
903            (,classvar (,instance ,class) (,slotvar (eql ',slot)))
904          ,@docs ,@decls
905          (declare (ignore ,classvar))
906          (setf (slot-value ,instance ',slot) (block ,slot ,@body))))))
907
908 ;;;----- That's all, folks --------------------------------------------------