From 822c93e562009625b3ff47f3c0309d6ba813bfa7 Mon Sep 17 00:00:00 2001
From: Marshall Lochbaum <mwlochbaum@gmail.com>
Date: Sun, 14 Feb 2021 22:09:48 -0500
Subject: =?UTF-8?q?Support=20length=20value=20at=20the=20end=20of=20a=20ra?=
 =?UTF-8?q?nk-1=20component=20of=20=F0=9D=95=A8=20in=20Group?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/r.bqn | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

(limited to 'src')

diff --git a/src/r.bqn b/src/r.bqn
index 7ace52e6..702de192 100644
--- a/src/r.bqn
+++ b/src/r.bqn
@@ -534,7 +534,8 @@ Replicate ← (0<=∘⊣)◶{
 _group←{
   "⊔: Grouping argument must consist of integers" ! ∧´Int⌜𝕩
   "⊔: Grouping argument values cannot be less than ¯1" ! ∧´¯1≤𝕩
-  d←(l←GroupLen𝕩)GroupOrd𝕩
+  GL←GroupLen⋄𝕩↩𝕨(-˜⟜≠{GL↩(𝕨⊑𝕩)GL⊢⋄𝕨↑𝕩}⊢)⍟(0⊘⊣)𝕩
+  d←(l←GL𝕩)GroupOrd𝕩
   i←0⋄(𝔽{𝕩⋄(i↩i+1)⊢i⊑d}⌜∘↕)⌜l
 }
 GroupInds←{
@@ -549,12 +550,14 @@ GroupGen←{
   "𝕨⊔𝕩: 𝕩 must be an array" ! IsArray 𝕩
   𝕨↩Pair∘ToArray⍟(2>≡)𝕨
   "𝕨⊔𝕩: Compound 𝕨 must be a list" ! 1==𝕨
-  n←+´=⌜𝕨
+  n←+´r←=⌜𝕨
   "𝕨⊔𝕩: Total rank of 𝕨 must be at most rank of 𝕩" ! n≤=𝕩
-  "𝕨⊔𝕩: Lengths of components of 𝕨 must be compatible with 𝕩" ! ∧´(Join≢⌜𝕨)=n↑≢𝕩
+  ld←(Join≢⌜𝕨)-n↑≢𝕩
+  "𝕨⊔𝕩: Lengths of 𝕨 must equal to 𝕩, or one more only in a rank-1 component" ! ∧´(0⊸≤∧≤⟜(r/1=r))ld
+  dr←r⌊(0»+`r)⊏ld∾⟨0⟩
   l←≠⌜𝕨↩⥊⌜𝕨 ⋄ LS←∾⟜(n Cell 𝕩) Reshape 𝕩˙
   S←⊏⟜(LS⟨×´l⟩)
-  (LS 0⌜𝕨) Fill (1≠≠)◶(S _group 0⊸⊑)‿(S⌜ ·+⌜⌜´ (⌽×`1»⌽l) × ⊢_group⌜) 𝕨
+  (LS 0⌜𝕨) Fill dr (1≠≠∘⊢)◶⟨S _group○(0⊸⊑), S⌜ ·+⌜⌜´ (⌽×`1»⌽l) × ⊢_group¨⟩ 𝕨
 }
 
 Pick1←{
-- 
cgit v1.2.3