summaryrefslogtreecommitdiff
path: root/src/GF/Conversion/SimpleToMCFG/Nondet.hs
blob: 83e5fec96c0b857259abbef631e38f9e6e3c7a4d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
----------------------------------------------------------------------
-- |
-- Maintainer  : PL
-- Stability   : (stable)
-- Portability : (portable)
--
-- > CVS $Date: 2005/04/12 10:49:44 $ 
-- > CVS $Author: peb $
-- > CVS $Revision: 1.2 $
--
-- Converting SimpleGFC grammars to MCFG grammars, nondeterministically.
-- Afterwards, the grammar has to be extended with coercion functions,
-- from the module 'GF.Conversion.SimpleToMCFG.Coercions'
--
-- the resulting grammars might be /very large/
--
-- the conversion is only equivalent if the GFC grammar has a context-free backbone.
-----------------------------------------------------------------------------


module GF.Conversion.SimpleToMCFG.Nondet 
    (convertGrammar) where

import GF.System.Tracing
import GF.Infra.Print

import Monad

import GF.Formalism.Utilities
import GF.Formalism.GCFG 
import GF.Formalism.MCFG 
import GF.Formalism.SimpleGFC 
import GF.Conversion.Types

import GF.Data.BacktrackM


------------------------------------------------------------
-- type declarations

type CnvMonad a = BacktrackM Env a

type Env    = (MCat, [MCat], LinRec, [SLinType])
type LinRec = [Lin SCat MLabel Token]


----------------------------------------------------------------------
-- main conversion function

convertGrammar :: SGrammar -> MGrammar 
convertGrammar rules = tracePrt "Nondet conversion: #MCFG rules" (prt . length) $
		       solutions conversion undefined
    where conversion = member rules >>= convertRule

convertRule :: SRule -> CnvMonad MRule
convertRule (Rule (Abs decl decls fun) (Cnc ctype ctypes (Just term)))
    = do let cat : args = map decl2cat (decl : decls)
	 writeState (initialMCat cat, map initialMCat args, [], ctypes)
	 rterm <- simplifyTerm term
	 reduceTerm ctype emptyPath rterm
	 (newCat, newArgs, linRec, _) <- readState
	 let newLinRec = map (instantiateArgs newArgs) linRec
	     catPaths : argsPaths = map (lintype2paths emptyPath) (ctype : ctypes)
	 return $ Rule (Abs newCat newArgs fun) (Cnc catPaths argsPaths newLinRec) 
convertRule _ = failure


----------------------------------------------------------------------
-- term simplification

simplifyTerm :: STerm -> CnvMonad STerm
simplifyTerm (term :! sel)      
    = do sterm <- simplifyTerm term
	 ssel <- simplifyTerm sel
	 case sterm of
	   Tbl table  -> do (pat, val) <- member table
			    pat =?= ssel
			    return val
	   _ -> do sel' <- expandTerm ssel
		   return (sterm +! sel')
simplifyTerm (con :^ terms)    = liftM (con :^) $ mapM simplifyTerm terms
simplifyTerm (Rec record)      = liftM Rec      $ mapM simplifyAssign record
simplifyTerm (term :. lbl)     = liftM (+. lbl) $      simplifyTerm term
simplifyTerm (Tbl table)       = liftM Tbl      $ mapM simplifyCase table
simplifyTerm (Variants terms)  = liftM Variants $ mapM simplifyTerm terms
simplifyTerm (term1 :++ term2) = liftM2 (:++) (simplifyTerm term1) (simplifyTerm term2)
simplifyTerm term              = return term
-- error constructors:
--   (I CIdent) - from resource
--   (LI Ident) - pattern variable
--   (EInt Integer) - integer

simplifyAssign :: (Label, STerm) -> CnvMonad (Label, STerm)
simplifyAssign (lbl, term) = liftM ((,) lbl) $ simplifyTerm term

simplifyCase :: (STerm, STerm) -> CnvMonad (STerm, STerm)
simplifyCase (pat, term) = liftM2 (,) (simplifyTerm pat) (simplifyTerm term)


------------------------------------------------------------
-- reducing simplified terms, collecting MCF rules

reduceTerm :: SLinType -> SPath -> STerm -> CnvMonad ()
reduceTerm ctype path (Variants terms) 
    = member terms >>= reduceTerm ctype path
reduceTerm (StrT)     path term = updateLin (path, term)
reduceTerm (ConT _ _) path term = do pat <- expandTerm term
				     updateHead (path, pat)
reduceTerm (RecT rtype) path term
    = sequence_ [ reduceTerm ctype (path ++. lbl) (term +. lbl) |
		  (lbl, ctype) <- rtype ]
reduceTerm (TblT ptype vtype) path table 
    = sequence_ [ reduceTerm vtype (path ++! pat) (table +! pat) |
		  pat <- enumeratePatterns ptype ]


------------------------------------------------------------
-- expanding a term to ground terms

expandTerm :: STerm -> CnvMonad STerm
expandTerm arg@(Arg nr _ path) 
    = do ctypes <- readArgCTypes
	 pat <- member $ enumeratePatterns $ lintypeFollowPath path $ ctypes !! nr
	 pat =?= arg
	 return pat
expandTerm (con :^ terms)   = liftM (con :^) $ mapM expandTerm terms
expandTerm (Rec record)     = liftM  Rec     $ mapM expandAssign record
expandTerm (Variants terms) = member terms >>= expandTerm  
expandTerm term = error $ "expandTerm: " ++ prt term

expandAssign :: (Label, STerm) -> CnvMonad (Label, STerm)
expandAssign (lbl, term) = liftM ((,) lbl) $ expandTerm term


------------------------------------------------------------
-- unification of patterns and selection terms

(=?=) :: STerm -> STerm -> CnvMonad ()
Wildcard      =?= _               = return ()
Rec precord   =?= arg@(Arg _ _ _) = sequence_ [ pat =?= (arg +. lbl) |
						(lbl, pat) <- precord ]
pat           =?= Arg nr _ path   = updateArg nr (path, pat)
(con :^ pats) =?= (con' :^ terms) = do guard (con==con' && length pats==length terms)
				       sequence_ $ zipWith (=?=) pats terms
Rec precord   =?= Rec record      = sequence_ [ maybe mzero (pat =?=) mterm |
						(lbl, pat) <- precord,
						let mterm = lookup lbl record ]
pat =?= term = error $ "(=?=): " ++ prt pat ++ " =?= " ++ prt term


------------------------------------------------------------
-- updating the MCF rule

readArgCTypes :: CnvMonad [SLinType]
readArgCTypes = do (_, _, _, env) <- readState
		   return env

updateArg :: Int -> Constraint -> CnvMonad ()
updateArg arg cn
    = do (head, args, lins, env) <- readState 
	 args' <- updateNth (addToMCat cn) arg args
	 writeState (head, args', lins, env)

updateHead :: Constraint -> CnvMonad ()
updateHead cn
    = do (head, args, lins, env) <- readState
	 head' <- addToMCat cn head
	 writeState (head', args, lins, env)

updateLin :: Constraint -> CnvMonad ()
updateLin (path, term) 
    = do let newLins = term2lins term
	 (head, args, lins, env) <- readState
	 let lins' = lins ++ map (Lin path) newLins
	 writeState (head, args, lins', env)

term2lins :: STerm -> [[Symbol (SCat, SPath, Int) Token]]
term2lins (Arg nr cat path) = return [Cat (cat, path, nr)]
term2lins (Token str)       = return [Tok str]
term2lins (t1 :++ t2)       = liftM2 (++) (term2lins t1) (term2lins t2)
term2lins (Empty)           = return []
term2lins (Variants terms)  = terms >>= term2lins
term2lins term = error $ "term2lins: " ++ show term

addToMCat :: Constraint -> MCat -> CnvMonad MCat
addToMCat cn (MCat cat cns) = liftM (MCat cat) $ addConstraint cn cns

addConstraint :: Constraint -> [Constraint] -> CnvMonad [Constraint]
addConstraint cn0 (cn : cns)
    | fst cn0 >  fst cn = liftM (cn:) (addConstraint cn0 cns)
    | fst cn0 == fst cn = guard (snd cn0 == snd cn) >>
			  return (cn : cns)
addConstraint cn0 cns   = return (cn0 : cns)


----------------------------------------------------------------------
-- utilities 

updateNth :: Monad m => (a -> m a) -> Int -> [a] -> m [a]
updateNth update 0 (a : as) = liftM (:as) (update a)
updateNth update n (a : as) = liftM (a:)  (updateNth update (n-1) as)