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@478: fun rp2s (p, ns) = String.concatWith " " (p2s p :: map Int.toString ns) adamc@478: 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@478: fun p_env x = adamc@478: (print "\nDENV:\n"; adamc@478: PM.appi (fn (p1, ps) => adamc@478: PS.app (fn p2 => adamc@478: print (rp2s p1 ^ " ~ " ^ rp2s p2 ^ "\n")) ps) x) adamc@478: 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@628: fun decomposeRow env c = adamc@82: let adamc@251: val loc = #2 c adamc@251: adamc@207: fun decomposeProj c = adamc@207: let adamc@628: val c = hnormCon env c adamc@207: in adamc@207: case #1 c of adamc@207: CProj (c, n) => adamc@207: let adamc@628: val (c', ns) = decomposeProj c adamc@207: in adamc@628: (c', ns @ [n]) adamc@207: end adamc@628: | _ => (c, []) adamc@207: end adamc@207: adamc@628: fun decomposeName (c, acc) = adamc@90: let adamc@628: val (cAll as (c, _), ns) = decomposeProj c adamc@90: in adamc@628: case c of adamc@628: CName s => Piece (NameC s, ns) :: acc adamc@628: | CRel n => Piece (NameR n, ns) :: acc adamc@628: | CNamed n => Piece (NameN n, ns) :: acc adamc@628: | CModProj (m1, ms, x) => Piece (NameM (m1, ms, x), ns) :: acc adamc@628: | _ => Unknown cAll :: acc adamc@90: end adamc@90: adamc@628: fun decomposeRow' (c, acc) = adamc@90: let adamc@251: fun default () = adamc@251: let adamc@628: val (cAll as (c, _), ns) = decomposeProj c adamc@251: in adamc@251: case c of adamc@628: CRecord (_, xcs) => foldl (fn ((x, _), acc) => decomposeName (x, acc)) acc xcs adamc@628: | CConcat (c1, c2) => decomposeRow' (c1, decomposeRow' (c2, acc)) adamc@628: | CRel n => Piece (RowR n, ns) :: acc adamc@628: | CNamed n => Piece (RowN n, ns) :: acc adamc@628: | CModProj (m1, ms, x) => Piece (RowM (m1, ms, x), ns) :: acc adamc@628: | _ => Unknown cAll :: acc adamc@251: end adamc@90: in adamc@628: case #1 (hnormCon env c) of adamc@251: CApp ( adamc@621: (CApp ((CMap _, _), _), _), adamc@628: r) => decomposeRow' (r, acc) adamc@251: | _ => default () adamc@90: end adamc@90: in adamc@628: decomposeRow' (c, []) adamc@90: end adamc@90: adamc@90: and assert env denv (c1, c2) = adamc@90: let adamc@628: val ps1 = decomposeRow env c1 adamc@628: val ps2 = decomposeRow env 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@628: foldl (assertPiece ps1) denv ps2 adamc@90: end adamc@90: adamc@90: and prove env denv (c1, c2, loc) = adamc@90: let adamc@628: val ps1 = decomposeRow env c1 adamc@628: val ps2 = decomposeRow env 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: 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@628: [] ps1 adamc@88: end adamc@82: end adamc@82: adamc@82: end