[gforth] / gforth / regexp.fs  

gforth: gforth/regexp.fs


1 : pazsan 1.31 \ Regexp compiler
2 : pazsan 1.1
3 : pazsan 1.31 \ Copyright (C) 2005,2006,2007,2008,2010 Free Software Foundation, Inc.
4 : anton 1.3
5 :     \ This file is part of Gforth.
6 :    
7 :     \ Gforth is free software; you can redistribute it and/or
8 :     \ modify it under the terms of the GNU General Public License
9 : anton 1.10 \ as published by the Free Software Foundation, either version 3
10 : anton 1.3 \ of the License, or (at your option) any later version.
11 :    
12 :     \ This program is distributed in the hope that it will be useful,
13 :     \ but WITHOUT ANY WARRANTY; without even the implied warranty of
14 :     \ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 :     \ GNU General Public License for more details.
16 :    
17 :     \ You should have received a copy of the GNU General Public License
18 : anton 1.10 \ along with this program. If not, see http://www.gnu.org/licenses/.
19 : anton 1.3
20 : pazsan 1.1 \ The idea of the parser is the following:
21 :     \ As long as there's a match, continue
22 :     \ On a mismatch, LEAVE.
23 :     \ Insert appropriate control structures on alternative branches
24 :     \ Keep the old pointer (backtracking) on the stack
25 :     \ I try to keep the syntax as close to a real regexp system as possible
26 :     \ All regexp stuff is compiled into one function as forward branching
27 :     \ state machine
28 :    
29 : pazsan 1.2 \ special control structure
30 : pazsan 1.1
31 : pazsan 1.8 : FORK ( compilation -- orig ; run-time f -- ) \ gforth
32 :     \G AHEAD-like control structure: calls the code after JOIN.
33 : pazsan 1.1 POSTPONE call >mark ; immediate restrict
34 : pazsan 1.8 : JOIN ( orig -- ) \ gforth
35 :     \G THEN-like control structure for FORK
36 :     postpone THEN ; immediate restrict
37 : pazsan 1.1
38 :     \ Charclasses
39 :    
40 :     : +bit ( addr n -- ) + 1 swap c! ;
41 :     : -bit ( addr n -- ) + 0 swap c! ;
42 :     : @+ ( addr -- n addr' ) dup @ swap cell+ ;
43 :    
44 :     0 Value cur-class
45 : pazsan 1.8 : charclass ( -- ) \ regexp-cg
46 :     \G Create a charclass
47 :     Create here dup to cur-class $100 dup allot erase ;
48 :     : +char ( char -- ) \ regexp-cg
49 :     \G add a char to the current charclass
50 :     cur-class swap +bit ;
51 :     : -char ( char -- ) \ regexp-cg
52 :     \G remove a char from the current charclass
53 :     cur-class swap -bit ;
54 :     : ..char ( start end -- ) \ regexp-cg
55 :     \G add a range of chars to the current charclass
56 :     1+ swap ?DO I +char LOOP ;
57 : pazsan 1.1 : or! ( n addr -- ) dup @ rot or swap ! ;
58 :     : and! ( n addr -- ) dup @ rot and swap ! ;
59 : pazsan 1.8 : +class ( class -- ) \ regexp-cg
60 :     \G union of charclass @var{class} and the current charclass
61 :     $100 0 ?DO @+ swap
62 :     cur-class I + or! cell +LOOP drop ;
63 :     : -class ( class -- ) \ regexp-cg
64 :     \G subtract the charclass @var{class} from the current charclass
65 :     $100 0 ?DO @+ swap invert
66 :     cur-class I + and! cell +LOOP drop ;
67 : pazsan 1.1
68 :     : char? ( addr class -- addr' flag )
69 :     >r count r> + c@ ;
70 :    
71 :     \ Charclass tests
72 :    
73 : pazsan 1.8 : c? ( addr class -- ) \ regexp-pattern
74 :     \G check @var{addr} for membership in charclass @var{class}
75 :     ]] char? 0= ?LEAVE [[ ; immediate
76 :     : -c? ( addr class -- ) \ regexp-pattern
77 :     \G check @var{addr} for not membership in charclass @var{class}
78 :     ]] char? ?LEAVE [[ ; immediate
79 : pazsan 1.1
80 :     charclass digit '0 '9 ..char
81 :     charclass blanks 0 bl ..char
82 :     \ bl +char #tab +char #cr +char #lf +char ctrl L +char
83 :     charclass letter 'a 'z ..char 'A 'Z ..char
84 :     charclass any 0 $FF ..char #lf -char
85 :    
86 : pazsan 1.8 : \d ( addr -- addr' ) \ regexp-pattern
87 :     \G check for digit
88 :     ]] digit c? [[ ; immediate
89 :     : \s ( addr -- addr' ) \ regexp-pattern
90 :     \G check for blanks
91 :     ]] blanks c? [[ ; immediate
92 :     : .? ( addr -- addr' ) \ regexp-pattern
93 :     \G check for any single charachter
94 :     ]] any c? [[ ; immediate
95 :     : -\d ( addr -- addr' ) \ regexp-pattern
96 :     \G check for not digit
97 :     ]] digit -c? [[ ; immediate
98 :     : -\s ( addr -- addr' ) \ regexp-pattern
99 :     \G check for not blank
100 :     ]] blanks -c? [[ ; immediate
101 :     : ` ( "char" -- ) \ regexp-pattern
102 :     \G check for particular char
103 : pazsan 1.1 ]] count [[ char ]] Literal <> ?LEAVE [[ ; immediate
104 : pazsan 1.17 : -` ( "char" -- ) \ regexp-pattern
105 :     \G check for particular char
106 :     ]] count [[ char ]] Literal = ?LEAVE [[ ; immediate
107 : pazsan 1.1
108 :     \ loop stack
109 :    
110 :     Variable loops $40 3 * cells allot
111 :     : 3@ ( addr -- a b c ) dup >r 2 cells + @ r> 2@ ;
112 :     : 3! ( a b c addr -- ) dup >r 2! r> 2 cells + ! ;
113 :     : loops> ( -- addr ) -3 loops +! loops @+ swap cells + 3@ ;
114 :     : >loops ( addr -- ) loops @+ swap cells + 3! 3 loops +! ;
115 :     : BEGIN, ( -- ) ]] BEGIN [[ >loops ;
116 : pazsan 1.23 : DONE, ( -- ) loops @ IF loops> ]] DONE [[ ELSE ." no done left!" cr THEN ;
117 : pazsan 1.1
118 :     \ variables
119 :    
120 :     Variable vars &18 cells allot
121 :     Variable varstack 9 cells allot
122 :     Variable varsmax
123 :     : >var ( -- addr ) vars @+ swap 2* cells +
124 :     vars @ varstack @+ swap cells + !
125 :     1 vars +! 1 varstack +! ;
126 :     : var> ( -- addr ) -1 varstack +!
127 :     varstack @+ swap cells + @
128 :     1+ 2* cells vars + ;
129 : pazsan 1.26 Variable greed-counts 9 cells allot \ no more than 9 nested greedy loops
130 :     : greed' ( -- addr ) greed-counts dup @ + ;
131 : pazsan 1.1
132 :     \ start end
133 :    
134 :     0 Value end$
135 : pazsan 1.28 0 Value last$
136 : pazsan 1.1 0 Value start$
137 :     : !end ( addr u -- addr ) over + to end$ dup to start$ ;
138 : pazsan 1.12 : end-rex? ( addr -- addr flag ) dup end$ u< ;
139 :     : start-rex? ( addr -- addr flag ) dup start$ u> ;
140 : pazsan 1.1 : ?end ( addr -- addr ) ]] dup end$ u> ?LEAVE [[ ; immediate
141 : pazsan 1.20 : rest$ ( addr -- addr addr u ) dup end$ over - ;
142 : pazsan 1.28 : >last ( addr -- flag ) dup to last$ end$ u<= ;
143 : pazsan 1.1
144 :     \ start and end
145 :    
146 : pazsan 1.8 : \^ ( addr -- addr ) \ regexp-pattern
147 :     \G check for string start
148 : pazsan 1.12 ]] start-rex? ?LEAVE [[ ; immediate
149 : pazsan 1.8 : \$ ( addr -- addr ) \ regexp-pattern
150 :     \G check for string end
151 : pazsan 1.12 ]] end-rex? ?LEAVE [[ ; immediate
152 : pazsan 1.1
153 : pazsan 1.17 \ A word for string comparison
154 :    
155 : pazsan 1.22 : (str=?) ( addr1 addr u -- addr2 )
156 : pazsan 1.20 dup >r 2>r rest$ r@ umin 2r> compare IF rdrop true ELSE r> + false THEN ;
157 : pazsan 1.22 : str=? ( addr1 addr u -- addr2 ) ]] (str=?) ?LEAVE [[ ; immediate
158 :     : ,=" ( addr u -- ) tuck dup ]] rest$ Literal umin SLiteral compare ?LEAVE Literal + [[ ;
159 : pazsan 1.17 : =" ( <string>" -- ) \ regexp-pattern
160 :     \G check for string
161 :     '" parse ,=" ; immediate
162 :    
163 : pazsan 1.1 \ regexp block
164 :    
165 :     \ FORK/JOIN are like AHEAD THEN, but producing a call on AHEAD
166 :     \ instead of a jump.
167 :    
168 : pazsan 1.8 : (( ( addr u -- ) \ regexp-pattern
169 :     \G start regexp block
170 : pazsan 1.26 vars off varsmax off loops off greed-counts off
171 : pazsan 1.1 ]] FORK AHEAD BUT JOIN !end [[ BEGIN, ; immediate
172 : pazsan 1.29 : )) ( -- flag ) \ regexp-pattern
173 : pazsan 1.8 \G end regexp block
174 : pazsan 1.28 ]] >last ;S [[
175 : pazsan 1.19 DONE, ]] drop false ;S THEN [[ ; immediate
176 : pazsan 1.1
177 :     \ greedy loops
178 :    
179 :     \ Idea: scan as many characters as possible, try the rest of the pattern
180 :     \ and then back off one pattern at a time
181 :    
182 :     : drops ( n -- ) 1+ cells sp@ + sp! ;
183 :    
184 : pazsan 1.8 : {** ( addr -- addr addr ) \ regexp-pattern
185 :     \G greedy zero-or-more pattern
186 : pazsan 1.30 ]] false >r BEGIN dup FORK BUT WHILE last$ r> 1+ >r REPEAT [[
187 : pazsan 1.27 ]] r> AHEAD BUT JOIN [[
188 :     BEGIN, ; immediate
189 : pazsan 1.8 ' {** Alias {++ ( addr -- addr addr ) \ regexp-pattern
190 :     \G greedy one-or-more pattern
191 :     immediate
192 :     : **} ( sys -- ) \ regexp-pattern
193 :     \G end of greedy zero-or-more pattern
194 : pazsan 1.30 ]] >last ;S [[ DONE, ]] drop false ;S THEN [[
195 :     ]] 1+ false U+DO FORK BUT [[
196 : pazsan 1.28 ]] IF I' I - 1- drops UNLOOP true ;S THEN LOOP [[
197 : pazsan 1.29 ]] false ;S JOIN [[ ; immediate
198 : pazsan 1.8 : ++} ( sys -- ) \ regexp-pattern
199 :     \G end of greedy zero-or-more pattern
200 : pazsan 1.30 ]] >last ;S [[ DONE, ]] drop false ;S THEN [[
201 :     ]] false U+DO FORK BUT [[
202 : pazsan 1.28 ]] IF I' I - drops UNLOOP true ;S THEN LOOP [[
203 : pazsan 1.29 ]] drop false ;S JOIN [[ ; immediate
204 : pazsan 1.1
205 :     \ non-greedy loops
206 :    
207 :     \ Idea: Try to match rest of the regexp, and if that fails, try match
208 :     \ first expr and then try again rest of regexp.
209 :    
210 : pazsan 1.8 : {+ ( addr -- addr addr ) \ regexp-pattern
211 :     \G non-greedy one-or-more pattern
212 : pazsan 1.1 ]] BEGIN [[ BEGIN, ; immediate
213 : pazsan 1.8 : {* ( addr -- addr addr ) \ regexp-pattern
214 :     \G non-greedy zero-or-more pattern
215 : pazsan 1.19 ]] {+ dup FORK BUT IF drop true ;S THEN [[ ; immediate
216 : pazsan 1.8 : *} ( addr addr' -- addr' ) \ regexp-pattern
217 :     \G end of non-greedy zero-or-more pattern
218 : pazsan 1.1 ]] dup end$ u> UNTIL [[
219 : pazsan 1.19 DONE, ]] drop false ;S JOIN [[ ; immediate
220 : pazsan 1.8 : +} ( addr addr' -- addr' ) \ regexp-pattern
221 :     \G end of non-greedy one-or-more pattern
222 : pazsan 1.19 ]] dup FORK BUT IF drop true ;S [[
223 : pazsan 1.28 DONE, ]] drop false ;S [[ BEGIN, ]] THEN *} [[ ; immediate
224 : pazsan 1.1
225 : pazsan 1.8 : // ( -- ) \ regexp-pattern
226 :     \G search for string
227 :     ]] {* 1+ *} [[ ; immediate
228 : pazsan 1.1
229 :     \ alternatives
230 :    
231 :     \ idea: try to match one alternative and then the rest of regexp.
232 :     \ if that fails, jump back to second alternative
233 :    
234 : pazsan 1.24 : THENs ( sys -- ) BEGIN dup WHILE ]] THEN [[ REPEAT drop ;
235 : pazsan 1.1
236 : pazsan 1.8 : {{ ( addr -- addr addr ) \ regexp-pattern
237 :     \G Start of alternatives
238 : pazsan 1.29 0 ]] dup FORK IF drop true ;S BUT JOIN [[ vars @ ; immediate
239 : pazsan 1.8 : || ( addr addr -- addr addr ) \ regexp-pattern
240 :     \G separator between alternatives
241 : pazsan 1.24 vars @ varsmax @ max varsmax ! vars !
242 : pazsan 1.29 ]] AHEAD BUT THEN [[
243 :     ]] dup FORK IF drop true ;S BUT JOIN [[ vars @ ; immediate
244 : pazsan 1.21 : }} ( addr addr -- addr ) \ regexp-pattern
245 : pazsan 1.8 \G end of alternatives
246 : pazsan 1.24 vars @ varsmax @ max vars ! drop
247 : pazsan 1.29 ]] AHEAD BUT THEN drop false ;S [[ THENs ; immediate
248 : pazsan 1.1
249 :     \ match variables
250 :    
251 : pazsan 1.8 : \( ( addr -- addr ) \ regexp-pattern
252 :     \G start of matching variable; variables are referred as \\1--9
253 :     ]] dup [[
254 : pazsan 1.1 >var ]] ALiteral ! [[ ; immediate
255 : pazsan 1.8 : \) ( addr -- addr ) \ regexp-pattern
256 :     \G end of matching variable
257 :     ]] dup [[
258 : pazsan 1.1 var> ]] ALiteral ! [[ ; immediate
259 : pazsan 1.8 : \0 ( -- addr u ) \ regexp-pattern
260 :     \G the whole string
261 :     start$ end$ over - ;
262 : pazsan 1.1 : \: ( i -- )
263 :     Create 2* 1+ cells vars + ,
264 :     DOES> ( -- addr u ) @ 2@ tuck - ;
265 :     : \:s ( n -- ) 0 ?DO I \: LOOP ;
266 :     9 \:s \1 \2 \3 \4 \5 \6 \7 \8 \9
267 : pazsan 1.6
268 :     \ replacements, needs string.fs
269 :    
270 :     require string.fs
271 :    
272 :     0 Value >>ptr
273 :     0 Value <<ptr
274 :     Variable >>string
275 : pazsan 1.17 : s>> ( addr -- addr ) \ regexp-replace
276 : pazsan 1.8 \G Start replace pattern region
277 :     dup to >>ptr ;
278 :     : << ( run-addr addr u -- run-addr ) \ regexp-replace
279 :     \G Replace string from start of replace pattern region with
280 :     \G @var{addr} @var{u}
281 : pazsan 1.6 <<ptr >>ptr over - >>string $+!
282 :     >>string $+! dup to <<ptr ;
283 : pazsan 1.8 : <<" ( "string<">" -- ) \ regexp-replace
284 :     \G Replace string from start of replace pattern region with
285 :     \G @var{string}
286 :     '" parse postpone SLiteral postpone << ; immediate
287 : pazsan 1.6 : >>string@ ( -- addr u )
288 : pazsan 1.16 >>string $@ ;
289 :     : >>string0 ( addr u -- addr u ) s" " >>string $!
290 :     0 to >>ptr over to <<ptr ;
291 : pazsan 1.6 : >>next ( -- addr u ) <<ptr end$ over - ;
292 : pazsan 1.14 : >>rest ( -- ) >>next >>string $+! ;
293 :     : s// ( addr u -- ptr )
294 : pazsan 1.8 \G start search/replace loop
295 : pazsan 1.17 ]] >>string0 (( // s>> [[ ; immediate
296 :     : >> ( addr -- addr )
297 :     ]] <<ptr >>ptr u> ?LEAVE ?end [[ ; immediate
298 :     : //s ( ptr -- )
299 :     \G search end
300 :     ]] )) drop >>rest >>string@ [[ ; immediate
301 : pazsan 1.15 : //o ( ptr addr u -- addr' u' )
302 : pazsan 1.14 \G end search/replace single loop
303 : pazsan 1.17 ]] << //s [[ ; immediate
304 : pazsan 1.14 : //g ( ptr addr u -- addr' u' )
305 :     \G end search/replace all loop
306 : pazsan 1.17 ]] << LEAVE //s [[ ; immediate

CVS Admin

Powered by ViewCVS 1.0-dev
(Powered by ViewCVS)

ViewCVS and CVS Help