basis/regexp/disambiguate/disambiguate.factor

   1 ! Copyright (C) 2009 Daniel Ehrenberg.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: kernel accessors regexp.classes math.bits assocs sequences
   4 arrays sets regexp.dfa math fry regexp.minimize ;
   5 IN: regexp.disambiguate
   6
   7 TUPLE: parts in out ;
   8
   9 : make-partition ( choices classes -- partition )
  10     zip [ first ] partition [ values ] bi@ parts boa ;
  11
  12 : powerset-partition ( classes -- partitions )
  13     [ length [ 2^ ] keep ] keep '[
  14         _ <bits> _ make-partition
  15     ] map rest ;
  16
  17 : partition>class ( parts -- class )
  18     [ out>> [ <not-class> ] map ]
  19     [ in>> <and-class> ] bi
  20     prefix <and-class> ;
  21
  22 : get-transitions ( partition state-transitions -- next-states )
  23     [ in>> ] dip '[ _ at ] map prune ;
  24
  25 : disambiguate ( dfa -- nfa )
  26     [
  27         [
  28             [ keys powerset-partition ] keep '[
  29                 [ partition>class ]
  30                 [ _ get-transitions ] bi
  31             ] H{ } map>assoc
  32             [ drop ] assoc-filter
  33         ] assoc-map
  34     ] change-transitions ;
  35
  36 : nfa>dfa ( nfa -- dfa )
  37     construct-dfa
  38     minimize disambiguate
  39     construct-dfa minimize ;