aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/c.bqn16
-rw-r--r--wc.bqn2
2 files changed, 10 insertions, 8 deletions
diff --git a/src/c.bqn b/src/c.bqn
index 594984ef..3e6718f3 100644
--- a/src/c.bqn
+++ b/src/c.bqn
@@ -51,26 +51,28 @@ Tokenize←{
str←1↓¨𝕩⊔˜1-˜(+`si←a>»qe)×≠`dm∧ab # Strings (indices /si)
# Extract words: identifiers and numbers
- t←CharCode (ind←/f)⊏𝕩⋄Fi←{ind/˜↩𝕨⋄𝕨/𝕩} # Track source code indices
+ is←/1»f⋄ie←/f⋄Fs←{is/˜↩𝕨⋄𝕨/𝕩} # Token start and end
+ t←CharCode ie⊏𝕩
w←»⊸<l←t M bD(⊣≍-˜)○⊑bW⋄us←t=¯1++´bA # Word chars l, start w
wk←na⌊∘÷˜(⊑bA)-˜w/t # Kind of word from first char
t-↩na×l∧t≥na+⊑bA # Case-insensitive
i←l>n←l∧(+`w)⊏0∾wk<0 # Identifier/Number masks
- num←ind ReadNums○((∨⟜«0∾n)/0⊸∾) t×l # Numbers
+ num←is ReadNums○((∨⟜«0∾n)/0⊸∾) t×l # Numbers
ir←(us/˜«⊸<i)(⊢+∧⟜(2⊸=))0⊸≤⊸/wk # Identifier role
id←(1-˜(i>us)×+`w>n)⊔t # Identifiers
# Deduplicate literals and identifiers; other cleanup
ki←((⍋⊏⟜n)⊸⊏/w)∾(ci∾/si)⊏+`»f # Indices in t
k←id‿num‿chr‿str⋄k(⊢>¯1»⌈`)⊸/¨˜↩j←⊐¨k # IDs j into uniques k
- t↩(w∨¬l∨t M bW)Fi(∾j++`vi»≠¨k)⌾(ki⊸⊏)t # Add IDs; remove words/whitespace
+ wf←¬l∨t M bW⋄ie/˜↩wf∨>⟜«l # Index management for...
+ t↩(w∨wf)Fs(∾j++`vi»≠¨k)⌾(ki⊸⊏)t # Add IDs; remove words/whitespace
t-↩t(M×-⟜⊑)bS # Separators are equivalent
p←≠`1¨sb←¯1↓1↓/1(∾≠∾˜)t=sep # Separator group boundaries (excludes leading and trailing)
sk←sb/˜p>∨⟜«(p+(sb-p)⊏t)∊3‿5+⊑bB # Keep the first of each group that's not just inside a bracket
- t Fi˜↩1¨⌾(sk⊸⊏)t≠sep # Remove the rest
+ t{ie/˜↩𝕨⋄𝕨Fs𝕩}˜↩1¨⌾(sk⊸⊏)t≠sep # Remove the rest
r←ir⌾((t M vi∾≠⊑k)⊸/)(vi⌊t)⊏charRole∾0 # Role
t+↩5×t M⟨⊑bI,5⟩ # Case-insensitive special names
- ⟨t,r,k,ind⟩
+ ⟨t,r,k,is,ie⟩
}
# 𝕩 is a list of tokens that contains the numeric literals, each
@@ -195,7 +197,7 @@ LEB←{
}
Compile←{
- ⟨tok,role,val,ind⟩←tx←Tokenize 𝕩
- ⟨bc,prim,blk,indb⟩←⟨role,⊑val,ind⟩ Parse tok
+ ⟨tok,role,val,inds,inde⟩←tx←Tokenize 𝕩
+ ⟨bc,prim,blk,indb⟩←⟨role,⊑val,inds⟩ Parse tok
⟨bc, ∾⟨prim⊏𝕨⟩∾1↓val, <˘⍉>blk, indb, tx⟩
}
diff --git a/wc.bqn b/wc.bqn
index 5a8fc398..eed73e99 100644
--- a/wc.bqn
+++ b/wc.bqn
@@ -47,7 +47,7 @@ fntab←⍉(0¨tab1)∾tab1≍tab2
f64←127-3
GenFn←{
- ⟨t,r,k,i⟩←Tokenize𝕩
+ ⟨t,r,k⟩←3↑Tokenize𝕩
nVar←≠⊑k⋄lits←∾1↓k
nLoc←nVar-𝕨⊢0
t⊏˜↩⍋+`-˝(2‿3+⊑bB)=⌜t