src/method-aggregate.lisp: Expose number of methods to custom combination.

This permits pre-allocating an array for collecting results, for
example.  Suggested by Tony Finch.

diff --git a/src/method-aggregate.lisp b/src/method-aggregate.lisp
index 9d0a6dde11df2e7dde2bb3a1e072ec11abfbca01..e374924e12ce9de6937ab61e862ad2333e134d4f 100644 (file)
--- a/src/method-aggregate.lisp
+++ b/src/method-aggregate.lisp
@@ -406,18 +406,21 @@ (defmethod aggregating-message-properties
     :before :fragment
     :first :fragment
     :each :fragment
     :before :fragment
     :first :fragment
     :each :fragment

-    :after :fragment))
+    :after :fragment
+    :count :id))

 
 (defmethod compute-aggregating-message-kernel
     ((message aggregating-message) (combination (eql :custom))
      codegen target methods arg-names
      &key (retvar "sod_ret") (valvar "sod_val")
 
 (defmethod compute-aggregating-message-kernel
     ((message aggregating-message) (combination (eql :custom))
      codegen target methods arg-names
      &key (retvar "sod_ret") (valvar "sod_val")

-         decls before each (first each) after)
+         decls before each (first each) after count)

   (let* ((type (c-type-subtype (sod-message-type message)))
         (not-void-p (not (eq type c-type-void))))
     (when not-void-p
       (ensure-var codegen retvar type)
       (ensure-var codegen valvar type))
   (let* ((type (c-type-subtype (sod-message-type message)))
         (not-void-p (not (eq type c-type-void))))
     (when not-void-p
       (ensure-var codegen retvar type)
       (ensure-var codegen valvar type))

+    (when count
+      (ensure-var codegen count c-type-int (length methods)))

     (when decls
       (emit-decl codegen decls))
     (labels ((maybe-emit (fragment)
     (when decls
       (emit-decl codegen decls))
     (labels ((maybe-emit (fragment)