Annotation of gforth/regexp.fs, revision 1.27

1.1       pazsan      1: \ Regexp compile
                      2: 
1.13      anton       3: \ Copyright (C) 2005,2006,2007,2008 Free Software Foundation, Inc.
1.3       anton       4: 
                      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
1.10      anton       9: \ as published by the Free Software Foundation, either version 3
1.3       anton      10: \ 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
1.10      anton      18: \ along with this program. If not, see http://www.gnu.org/licenses/.
1.3       anton      19: 
1.1       pazsan     20: \ 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: 
1.2       pazsan     29: \ special control structure
1.1       pazsan     30: 
1.8       pazsan     31: : FORK ( compilation -- orig ; run-time f -- ) \ gforth
                     32:     \G AHEAD-like control structure: calls the code after JOIN.
1.1       pazsan     33:     POSTPONE call >mark ; immediate restrict
1.8       pazsan     34: : JOIN ( orig -- ) \ gforth
                     35:     \G THEN-like control structure for FORK
                     36:     postpone THEN ; immediate restrict
1.1       pazsan     37: 
                     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
1.8       pazsan     45: : 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 ;
1.1       pazsan     57: : or! ( n addr -- )  dup @ rot or swap ! ;
                     58: : and! ( n addr -- )  dup @ rot and swap ! ;
1.8       pazsan     59: : +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 ;
1.1       pazsan     67: 
                     68: : char? ( addr class -- addr' flag )
                     69:     >r count r> + c@ ;
                     70: 
                     71: \ Charclass tests
                     72: 
1.8       pazsan     73: : 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
1.1       pazsan     79: 
                     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: 
1.8       pazsan     86: : \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
1.1       pazsan    103:     ]] count [[  char ]] Literal <> ?LEAVE [[ ;  immediate
1.17      pazsan    104: : -` ( "char" -- ) \ regexp-pattern
                    105:     \G check for particular char
                    106:     ]] count [[  char ]] Literal = ?LEAVE [[ ;  immediate
1.1       pazsan    107: 
                    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 ;
1.23      pazsan    116: : DONE, ( -- )  loops @ IF  loops> ]] DONE [[ ELSE ." no done left!" cr THEN ;
1.1       pazsan    117: 
                    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 + ;
1.26      pazsan    129: Variable greed-counts  9 cells allot \ no more than 9 nested greedy loops
                    130: : greed' ( -- addr )  greed-counts dup @ + ;
1.1       pazsan    131: 
                    132: \ start end
                    133: 
                    134: 0 Value end$
                    135: 0 Value start$
                    136: : !end ( addr u -- addr )  over + to end$ dup to start$ ;
1.12      pazsan    137: : end-rex? ( addr -- addr flag ) dup end$ u< ;
                    138: : start-rex? ( addr -- addr flag ) dup start$ u> ;
1.1       pazsan    139: : ?end ( addr -- addr ) ]] dup end$ u> ?LEAVE [[ ; immediate
1.20      pazsan    140: : rest$ ( addr -- addr addr u ) dup end$ over - ;
1.1       pazsan    141: 
                    142: \ start and end
                    143: 
1.8       pazsan    144: : \^ ( addr -- addr ) \ regexp-pattern
                    145:     \G check for string start
1.12      pazsan    146:     ]] start-rex? ?LEAVE [[ ; immediate
1.8       pazsan    147: : \$ ( addr -- addr ) \ regexp-pattern
                    148:     \G check for string end
1.12      pazsan    149:     ]] end-rex? ?LEAVE [[ ; immediate
1.1       pazsan    150: 
1.17      pazsan    151: \ A word for string comparison
                    152: 
1.22      pazsan    153: : (str=?) ( addr1 addr u -- addr2 )
1.20      pazsan    154:     dup >r 2>r rest$ r@ umin 2r> compare IF rdrop true ELSE r> + false THEN ;
1.22      pazsan    155: : str=? ( addr1 addr u -- addr2 ) ]] (str=?) ?LEAVE [[ ; immediate
                    156: : ,=" ( addr u -- ) tuck dup ]] rest$ Literal umin SLiteral compare ?LEAVE Literal + [[ ;
1.17      pazsan    157: : =" ( <string>" -- ) \ regexp-pattern
                    158:     \G check for string
                    159:     '" parse ,=" ; immediate
                    160: 
1.1       pazsan    161: \ regexp block
                    162: 
                    163: \ FORK/JOIN are like AHEAD THEN, but producing a call on AHEAD
                    164: \ instead of a jump.
                    165: 
1.8       pazsan    166: : (( ( addr u -- ) \ regexp-pattern
                    167:     \G start regexp block
1.26      pazsan    168:     vars off varsmax off loops off greed-counts off
1.1       pazsan    169:     ]] FORK  AHEAD BUT JOIN !end [[ BEGIN, ; immediate
1.8       pazsan    170: : )) ( -- addr f ) \ regexp-pattern
                    171:     \G end regexp block
1.19      pazsan    172:     ]] ?end drop true ;S [[
                    173:     DONE, ]] drop false ;S THEN [[ ; immediate
1.1       pazsan    174: 
                    175: \ greedy loops
                    176: 
                    177: \ Idea: scan as many characters as possible, try the rest of the pattern
                    178: \ and then back off one pattern at a time
                    179: 
                    180: : drops ( n -- ) 1+ cells sp@ + sp! ;
                    181: 
1.8       pazsan    182: : {** ( addr -- addr addr ) \ regexp-pattern
                    183:     \G greedy zero-or-more pattern
1.27    ! pazsan    184:     ]] false >r BEGIN  dup  FORK  BUT  WHILE  r> 1+ >r  REPEAT [[
        !           185:     ]] r>  AHEAD  BUT  JOIN [[
        !           186:     BEGIN, ; immediate
1.8       pazsan    187: ' {** Alias {++ ( addr -- addr addr ) \ regexp-pattern
                    188:     \G greedy one-or-more pattern
                    189:     immediate
                    190: : **} ( sys -- ) \ regexp-pattern
                    191:     \G end of greedy zero-or-more pattern
1.27    ! pazsan    192:     ]] dup end$ u<=  ;S [[ DONE, ]] false ;S  THEN [[
        !           193:     ]] nip 1+ false  U+DO  FORK BUT [[
        !           194:     ]] IF  I' I - 1- drops true UNLOOP ;S  THEN  LOOP [[
        !           195:     ]] dup LEAVE JOIN [[ ; immediate
1.8       pazsan    196: : ++} ( sys -- ) \ regexp-pattern
                    197:     \G end of greedy zero-or-more pattern
1.27    ! pazsan    198:     ]] dup end$ u<=  ;S [[ DONE, ]] false ;S  THEN [[
        !           199:     ]] nip false  U+DO  FORK BUT [[
        !           200:     ]] IF  I' I - drops true UNLOOP ;S  THEN  LOOP [[
        !           201:     ]] drop dup LEAVE JOIN [[ ; immediate
1.1       pazsan    202: 
                    203: \ non-greedy loops
                    204: 
                    205: \ Idea: Try to match rest of the regexp, and if that fails, try match
                    206: \ first expr and then try again rest of regexp.
                    207: 
1.8       pazsan    208: : {+ ( addr -- addr addr ) \ regexp-pattern
                    209:     \G non-greedy one-or-more pattern
1.1       pazsan    210:     ]] BEGIN  [[ BEGIN, ; immediate
1.8       pazsan    211: : {* ( addr -- addr addr ) \ regexp-pattern
                    212:     \G non-greedy zero-or-more pattern
1.19      pazsan    213:     ]] {+ dup FORK BUT  IF  drop true  ;S THEN [[ ; immediate
1.8       pazsan    214: : *} ( addr addr' -- addr' ) \ regexp-pattern
                    215:     \G end of non-greedy zero-or-more pattern
1.1       pazsan    216:     ]] dup end$ u>  UNTIL [[
1.19      pazsan    217:     DONE, ]] drop false  ;S  JOIN [[ ; immediate
1.8       pazsan    218: : +} ( addr addr' -- addr' ) \ regexp-pattern
                    219:     \G end of non-greedy one-or-more pattern
1.19      pazsan    220:     ]] dup FORK BUT  IF  drop true  ;S [[
1.23      pazsan    221:     DONE, ]] drop dup  LEAVE [[ BEGIN, ]] THEN *} [[ ; immediate
1.1       pazsan    222: 
1.8       pazsan    223: : // ( -- ) \ regexp-pattern
                    224:     \G search for string
                    225:     ]] {* 1+ *} [[ ; immediate
1.1       pazsan    226: 
                    227: \ alternatives
                    228: 
                    229: \ idea: try to match one alternative and then the rest of regexp.
                    230: \ if that fails, jump back to second alternative
                    231: 
1.24      pazsan    232: : THENs ( sys -- )  BEGIN  dup  WHILE  ]] THEN [[  REPEAT  drop ;
1.1       pazsan    233: 
1.8       pazsan    234: : {{ ( addr -- addr addr ) \ regexp-pattern
                    235:     \G Start of alternatives
1.24      pazsan    236:     0 ]] dup dup FORK  IF  2drop true ;S  BUT  JOIN [[ vars @ ; immediate
1.8       pazsan    237: : || ( addr addr -- addr addr ) \ regexp-pattern
                    238:     \G separator between alternatives
1.24      pazsan    239:     vars @ varsmax @ max varsmax !  vars !
                    240:     ]] AHEAD  BUT  THEN  drop [[
                    241:     ]] dup dup FORK  IF  2drop true ;S  BUT  JOIN [[ vars @ ; immediate
1.21      pazsan    242: : }} ( addr addr -- addr ) \ regexp-pattern
1.8       pazsan    243:     \G end of alternatives
1.24      pazsan    244:     vars @ varsmax @ max vars !  drop
1.25      pazsan    245:     ]] AHEAD  BUT  THEN  drop LEAVE [[  THENs ; immediate
1.1       pazsan    246: 
                    247: \ match variables
                    248: 
1.8       pazsan    249: : \( ( addr -- addr ) \ regexp-pattern
                    250:     \G start of matching variable; variables are referred as \\1--9
                    251:     ]] dup [[
1.1       pazsan    252:     >var ]] ALiteral ! [[ ; immediate
1.8       pazsan    253: : \) ( addr -- addr ) \ regexp-pattern
                    254:     \G end of matching variable
                    255:     ]] dup [[
1.1       pazsan    256:     var> ]] ALiteral ! [[ ; immediate
1.8       pazsan    257: : \0 ( -- addr u ) \ regexp-pattern
                    258:     \G the whole string
                    259:     start$ end$ over - ;
1.1       pazsan    260: : \: ( i -- )
                    261:     Create 2* 1+ cells vars + ,
                    262:   DOES> ( -- addr u ) @ 2@ tuck - ;
                    263: : \:s ( n -- ) 0 ?DO  I \:  LOOP ;
                    264: 9 \:s \1 \2 \3 \4 \5 \6 \7 \8 \9
1.6       pazsan    265: 
                    266: \ replacements, needs string.fs
                    267: 
                    268: require string.fs
                    269: 
                    270: 0 Value >>ptr
                    271: 0 Value <<ptr
                    272: Variable >>string
1.17      pazsan    273: : s>>  ( addr -- addr ) \ regexp-replace
1.8       pazsan    274:     \G Start replace pattern region
                    275:     dup to >>ptr ;
                    276: : << ( run-addr addr u -- run-addr ) \ regexp-replace
                    277:     \G Replace string from start of replace pattern region with
                    278:     \G @var{addr} @var{u}
1.6       pazsan    279:     <<ptr >>ptr over - >>string $+!
                    280:     >>string $+! dup to <<ptr ;
1.8       pazsan    281: : <<" ( "string<">" -- ) \ regexp-replace
                    282:     \G Replace string from start of replace pattern region with
                    283:     \G @var{string}
                    284:     '" parse postpone SLiteral postpone << ; immediate
1.6       pazsan    285: : >>string@ ( -- addr u )
1.16      pazsan    286:     >>string $@ ;
                    287: : >>string0 ( addr u -- addr u )  s" " >>string $!
                    288:     0 to >>ptr  over to <<ptr ;
1.6       pazsan    289: : >>next ( -- addr u ) <<ptr end$ over - ;
1.14      pazsan    290: : >>rest ( -- ) >>next >>string $+! ;
                    291: : s// ( addr u -- ptr )
1.8       pazsan    292:     \G start search/replace loop
1.17      pazsan    293:     ]] >>string0 (( // s>> [[ ; immediate
                    294: : >> ( addr -- addr )
                    295:     ]] <<ptr >>ptr u> ?LEAVE ?end [[ ; immediate
                    296: : //s ( ptr -- )
                    297:     \G search end
                    298:     ]] )) drop >>rest >>string@ [[ ; immediate
1.15      pazsan    299: : //o ( ptr addr u -- addr' u' )
1.14      pazsan    300:     \G end search/replace single loop
1.17      pazsan    301:     ]] << //s [[ ; immediate
1.14      pazsan    302: : //g ( ptr addr u -- addr' u' )
                    303:     \G end search/replace all loop
1.17      pazsan    304:     ]] << LEAVE //s [[ ; immediate

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>