adamc@82: (* Copyright (c) 2008, Adam Chlipala adamc@82: * All rights reserved. adamc@82: * adamc@82: * Redistribution and use in source and binary forms, with or without adamc@82: * modification, are permitted provided that the following conditions are met: adamc@82: * adamc@82: * - Redistributions of source code must retain the above copyright notice, adamc@82: * this list of conditions and the following disclaimer. adamc@82: * - Redistributions in binary form must reproduce the above copyright notice, adamc@82: * this list of conditions and the following disclaimer in the documentation adamc@82: * and/or other materials provided with the distribution. adamc@82: * - The names of contributors may not be used to endorse or promote products adamc@82: * derived from this software without specific prior written permission. adamc@82: * adamc@82: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" adamc@82: * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE adamc@82: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE adamc@82: * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE adamc@82: * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR adamc@82: * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF adamc@82: * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS adamc@82: * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN adamc@82: * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) adamc@82: * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE adamc@82: * POSSIBILITY OF SUCH DAMAGE. adamc@82: *) adamc@82: adamc@82: structure Disjoint :> DISJOINT = struct adamc@82: adamc@82: open Elab adamc@82: open ElabOps adamc@82: adamc@207: datatype piece_fst = adamc@82: NameC of string adamc@82: | NameR of int adamc@82: | NameN of int adamc@88: | NameM of int * string list * string adamc@82: | RowR of int adamc@82: | RowN of int adamc@88: | RowM of int * string list * string adamc@88: adamc@207: type piece = piece_fst * int list adamc@207: adamc@88: fun p2s p = adamc@88: case p of adamc@88: NameC s => "NameC(" ^ s ^ ")" adamc@88: | NameR n => "NameR(" ^ Int.toString n ^ ")" adamc@88: | NameN n => "NameN(" ^ Int.toString n ^ ")" adamc@88: | NameM (n, _, s) => "NameR(" ^ Int.toString n ^ ", " ^ s ^ ")" adamc@88: | RowR n => "RowR(" ^ Int.toString n ^ ")" adamc@88: | RowN n => "RowN(" ^ Int.toString n ^ ")" adamc@88: | RowM (n, _, s) => "RowR(" ^ Int.toString n ^ ", " ^ s ^ ")" adamc@88: adamc@88: fun pp p = print (p2s p ^ "\n") adamc@88: adamc@88: structure PK = struct adamc@88: adamc@88: type ord_key = piece adamc@88: adamc@207: open Order adamc@88: adamc@207: fun compare' (p1, p2) = adamc@88: case (p1, p2) of adamc@88: (NameC s1, NameC s2) => String.compare (s1, s2) adamc@88: | (NameR n1, NameR n2) => Int.compare (n1, n2) adamc@88: | (NameN n1, NameN n2) => Int.compare (n1, n2) adamc@88: | (NameM (n1, ss1, s1), NameM (n2, ss2, s2)) => adamc@88: join (Int.compare (n1, n2), adamc@88: fn () => join (String.compare (s1, s2), fn () => adamc@88: joinL String.compare (ss1, ss2))) adamc@88: | (RowR n1, RowR n2) => Int.compare (n1, n2) adamc@88: | (RowN n1, RowN n2) => Int.compare (n1, n2) adamc@88: | (RowM (n1, ss1, s1), RowM (n2, ss2, s2)) => adamc@88: join (Int.compare (n1, n2), adamc@88: fn () => join (String.compare (s1, s2), fn () => adamc@88: joinL String.compare (ss1, ss2))) adamc@88: adamc@88: | (NameC _, _) => LESS adamc@88: | (_, NameC _) => GREATER adamc@88: adamc@88: | (NameR _, _) => LESS adamc@88: | (_, NameR _) => GREATER adamc@88: adamc@88: | (NameN _, _) => LESS adamc@88: | (_, NameN _) => GREATER adamc@88: adamc@88: | (NameM _, _) => LESS adamc@88: | (_, NameM _) => GREATER adamc@88: adamc@88: | (RowR _, _) => LESS adamc@88: | (_, RowR _) => GREATER adamc@88: adamc@88: | (RowN _, _) => LESS adamc@88: | (_, RowN _) => GREATER adamc@88: adamc@207: fun compare ((p1, ns1), (p2, ns2)) = adamc@207: join (compare' (p1, p2), adamc@207: fn () => joinL Int.compare (ns1, ns2)) adamc@207: adamc@88: end adamc@88: adamc@88: structure PS = BinarySetFn(PK) adamc@88: structure PM = BinaryMapFn(PK) adamc@88: adamc@88: type env = PS.set PM.map adamc@88: adamc@251: structure E = ElabEnv adamc@251: adamc@251: type goal = ErrorMsg.span * E.env * env * Elab.con * Elab.con adamc@90: adamc@88: val empty = PM.empty adamc@82: adamc@82: fun nameToRow (c, loc) = adamc@82: (CRecord ((KUnit, loc), [((c, loc), (CUnit, loc))]), loc) adamc@82: adamc@207: fun pieceToRow' (p, loc) = adamc@82: case p of adamc@82: NameC s => nameToRow (CName s, loc) adamc@82: | NameR n => nameToRow (CRel n, loc) adamc@82: | NameN n => nameToRow (CNamed n, loc) adamc@88: | NameM (n, xs, x) => nameToRow (CModProj (n, xs, x), loc) adamc@82: | RowR n => (CRel n, loc) adamc@88: | RowN n => (CNamed n, loc) adamc@88: | RowM (n, xs, x) => (CModProj (n, xs, x), loc) adamc@88: adamc@207: fun pieceToRow ((p, ns), loc) = adamc@207: foldl (fn (n, c) => (CProj (c, n), loc)) (pieceToRow' (p, loc)) ns adamc@207: adamc@88: datatype piece' = adamc@88: Piece of piece adamc@88: | Unknown of con adamc@82: adamc@207: fun pieceEnter' p = adamc@88: case p of adamc@88: NameR n => NameR (n + 1) adamc@88: | RowR n => RowR (n + 1) adamc@88: | _ => p adamc@82: adamc@207: fun pieceEnter (p, n) = (pieceEnter' p, n) adamc@207: adamc@88: fun enter denv = adamc@88: PM.foldli (fn (p, pset, denv') => adamc@88: PM.insert (denv', pieceEnter p, PS.map pieceEnter pset)) adamc@88: PM.empty denv adamc@82: adamc@335: val lowercase = CharVector.map Char.toLower adamc@335: adamc@82: fun prove1 denv (p1, p2) = adamc@82: case (p1, p2) of adamc@335: ((NameC s1, _), (NameC s2, _)) => lowercase s1 <> lowercase s2 adamc@88: | _ => adamc@88: case PM.find (denv, p1) of adamc@88: NONE => false adamc@88: | SOME pset => PS.member (pset, p2) adamc@82: adamc@90: fun decomposeRow (env, denv) c = adamc@82: let adamc@251: val loc = #2 c adamc@251: adamc@207: fun decomposeProj c = adamc@207: let adamc@207: val (c, gs) = hnormCon (env, denv) c adamc@207: in adamc@207: case #1 c of adamc@207: CProj (c, n) => adamc@207: let adamc@207: val (c', ns, gs') = decomposeProj c adamc@207: in adamc@207: (c', ns @ [n], gs @ gs') adamc@207: end adamc@207: | _ => (c, [], gs) adamc@207: end adamc@207: adamc@90: fun decomposeName (c, (acc, gs)) = adamc@90: let adamc@207: val (cAll as (c, _), ns, gs') = decomposeProj c adamc@90: adamc@90: val acc = case c of adamc@207: CName s => Piece (NameC s, ns) :: acc adamc@207: | CRel n => Piece (NameR n, ns) :: acc adamc@207: | CNamed n => Piece (NameN n, ns) :: acc adamc@207: | CModProj (m1, ms, x) => Piece (NameM (m1, ms, x), ns) :: acc adamc@90: | _ => Unknown cAll :: acc adamc@90: in adamc@90: (acc, gs' @ gs) adamc@90: end adamc@90: adamc@251: fun decomposeRow' (c, (acc, gs)) = adamc@90: let adamc@251: fun default () = adamc@251: let adamc@251: val (cAll as (c, _), ns, gs') = decomposeProj c adamc@251: val gs = gs' @ gs adamc@251: in adamc@251: case c of adamc@251: CRecord (_, xcs) => foldl (fn ((x, _), acc_gs) => decomposeName (x, acc_gs)) (acc, gs) xcs adamc@251: | CConcat (c1, c2) => decomposeRow' (c1, decomposeRow' (c2, (acc, gs))) adamc@251: | CRel n => (Piece (RowR n, ns) :: acc, gs) adamc@251: | CNamed n => (Piece (RowN n, ns) :: acc, gs) adamc@251: | CModProj (m1, ms, x) => (Piece (RowM (m1, ms, x), ns) :: acc, gs) adamc@251: | _ => (Unknown cAll :: acc, gs) adamc@251: end adamc@90: in adamc@326: (*Print.prefaces "decomposeRow'" [("c", ElabPrint.p_con env c), adamc@326: ("c'", ElabPrint.p_con env (#1 (hnormCon (env, denv) c)))];*) adamc@251: case #1 (#1 (hnormCon (env, denv) c)) of adamc@251: CApp ( adamc@251: (CApp ( adamc@251: (CApp ((CFold (dom, ran), _), f), _), adamc@251: i), _), adamc@251: r) => adamc@251: let adamc@251: val (env', nm) = E.pushCNamed env "nm" (KName, loc) NONE adamc@251: val (env', v) = E.pushCNamed env' "v" dom NONE adamc@251: val (env', st) = E.pushCNamed env' "st" ran NONE adamc@251: adamc@251: val (denv', gs') = assert env' denv ((CRecord (dom, [((CNamed nm, loc), adamc@251: (CUnit, loc))]), loc), adamc@251: (CNamed st, loc)) adamc@251: adamc@251: val c' = (CApp (f, (CNamed nm, loc)), loc) adamc@251: val c' = (CApp (c', (CNamed v, loc)), loc) adamc@251: val c' = (CApp (c', (CNamed st, loc)), loc) adamc@251: val (ps, gs'') = decomposeRow (env', denv') c' adamc@251: adamc@251: fun covered p = adamc@251: case p of adamc@251: Unknown _ => false adamc@251: | Piece p => adamc@251: case p of adamc@251: (NameN n, []) => n = nm adamc@251: | (RowN n, []) => n = st adamc@251: | _ => false adamc@251: adamc@251: val ps = List.filter (not o covered) ps adamc@251: in adamc@251: decomposeRow' (i, decomposeRow' (r, (ps @ acc, gs'' @ gs' @ gs))) adamc@251: end adamc@251: | _ => default () adamc@90: end adamc@90: in adamc@251: decomposeRow' (c, ([], [])) adamc@90: end adamc@90: adamc@90: and assert env denv (c1, c2) = adamc@90: let adamc@90: val (ps1, gs1) = decomposeRow (env, denv) c1 adamc@90: val (ps2, gs2) = decomposeRow (env, denv) c2 adamc@90: adamc@90: val unUnknown = List.mapPartial (fn Unknown _ => NONE | Piece p => SOME p) adamc@90: val ps1 = unUnknown ps1 adamc@90: val ps2 = unUnknown ps2 adamc@90: adamc@90: (*val () = print "APieces1:\n" adamc@90: val () = app pp ps1 adamc@90: val () = print "APieces2:\n" adamc@90: val () = app pp ps2*) adamc@90: adamc@90: fun assertPiece ps (p, denv) = adamc@90: let adamc@90: val pset = Option.getOpt (PM.find (denv, p), PS.empty) adamc@90: val ps = case p of adamc@207: (NameC _, _) => List.filter (fn (NameC _, _) => false | _ => true) ps adamc@90: | _ => ps adamc@90: val pset = PS.addList (pset, ps) adamc@90: in adamc@90: PM.insert (denv, p, pset) adamc@90: end adamc@90: adamc@90: val denv = foldl (assertPiece ps2) denv ps1 adamc@90: in adamc@90: (foldl (assertPiece ps1) denv ps2, gs1 @ gs2) adamc@90: end adamc@90: adamc@90: and prove env denv (c1, c2, loc) = adamc@90: let adamc@90: val (ps1, gs1) = decomposeRow (env, denv) c1 adamc@90: val (ps2, gs2) = decomposeRow (env, denv) c2 adamc@82: adamc@88: val hasUnknown = List.exists (fn Unknown _ => true | _ => false) adamc@88: val unUnknown = List.mapPartial (fn Unknown _ => NONE | Piece p => SOME p) adamc@82: in adamc@82: if hasUnknown ps1 orelse hasUnknown ps2 then adamc@90: [(loc, env, denv, c1, c2)] adamc@82: else adamc@88: let adamc@88: val ps1 = unUnknown ps1 adamc@88: val ps2 = unUnknown ps2 adamc@88: adamc@88: in adamc@88: (*print "Pieces1:\n"; adamc@88: app pp ps1; adamc@88: print "Pieces2:\n"; adamc@88: app pp ps2;*) adamc@88: adamc@88: foldl (fn (p1, rem) => adamc@88: foldl (fn (p2, rem) => adamc@88: if prove1 denv (p1, p2) then adamc@88: rem adamc@88: else adamc@90: (loc, env, denv, pieceToRow (p1, loc), pieceToRow (p2, loc)) :: rem) rem ps2) adamc@90: (gs1 @ gs2) ps1 adamc@88: end adamc@82: end adamc@82: adamc@90: and hnormCon (env, denv) c = adamc@90: let adamc@90: val cAll as (c, loc) = ElabOps.hnormCon env c adamc@90: adamc@90: fun doDisj (c1, c2, c) = adamc@90: let adamc@90: val (c, gs) = hnormCon (env, denv) c adamc@90: in adamc@90: (c, prove env denv (c1, c2, loc) @ gs) adamc@90: end adamc@90: in adamc@90: case c of adamc@345: CDisjoint (Instantiate, c1, c2, c) => doDisj (c1, c2, c) adamc@90: | _ => (cAll, []) adamc@90: end adamc@90: adamc@82: end