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
|
using Microsoft.Boogie;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
namespace Microsoft.Dafny.Triggers {
class QuantifierSplitter : BottomUpVisitor {
private static BinaryExpr.Opcode FlipOpcode(BinaryExpr.Opcode opCode) {
if (opCode == BinaryExpr.Opcode.And) {
return BinaryExpr.Opcode.Or;
} else if (opCode == BinaryExpr.Opcode.Or) {
return BinaryExpr.Opcode.And;
} else {
throw new ArgumentException();
}
}
// NOTE: If we wanted to split quantifiers as far as possible, it would be
// enough to put the formulas in DNF (for foralls) or CNF (for exists).
// Unfortunately, this would cause ill-behaved quantifiers to produce
// exponentially many smaller quantifiers.
internal static IEnumerable<Expression> SplitExpr(Expression expr, BinaryExpr.Opcode separator) {
expr = expr.Resolved;
var unary = expr as UnaryOpExpr;
var binary = expr as BinaryExpr;
if (unary != null && unary.Op == UnaryOpExpr.Opcode.Not) {
foreach (var e in SplitExpr(unary.E, FlipOpcode(separator))) { yield return new UnaryOpExpr(unary.tok, UnaryOpExpr.Opcode.Not, e); }
} else if (binary != null && binary.Op == separator) {
foreach (var e in SplitExpr(binary.E0, separator)) { yield return e; }
foreach (var e in SplitExpr(binary.E1, separator)) { yield return e; }
} else if (binary != null && binary.Op == BinaryExpr.Opcode.Imp && separator == BinaryExpr.Opcode.Or) {
foreach (var e in SplitExpr(new UnaryOpExpr(unary.tok, UnaryOpExpr.Opcode.Not, binary.E0), separator)) { yield return e; }
foreach (var e in SplitExpr(binary.E1, separator)) { yield return e; }
}
}
internal static IEnumerable<Expression> SplitAndStich(BinaryExpr pair, BinaryExpr.Opcode separator) {
foreach (var e1 in SplitExpr(pair.E1, separator)) {
yield return new BinaryExpr(pair.tok, pair.Op, pair.E0, e1);
}
}
internal static IEnumerable<Expression> SplitQuantifier(QuantifierExpr quantifier) {
var body = quantifier.LogicalBody();
var binary = body as BinaryExpr;
if (quantifier is ForallExpr) {
IEnumerable<Expression> stream;
if (binary != null && (binary.Op == BinaryExpr.Opcode.Imp || binary.Op == BinaryExpr.Opcode.Or)) {
stream = SplitAndStich(binary, BinaryExpr.Opcode.And);
} else {
stream = SplitExpr(body, BinaryExpr.Opcode.And);
}
foreach (var e in stream) {
yield return new ForallExpr(quantifier.tok, quantifier.BoundVars, quantifier.Range, quantifier.Term, quantifier.Attributes);
}
} else if (quantifier is ExistsExpr) {
IEnumerable<Expression> stream;
if (binary != null && binary.Op == BinaryExpr.Opcode.And) {
stream = SplitAndStich(binary, BinaryExpr.Opcode.Or);
} else {
stream = SplitExpr(body, BinaryExpr.Opcode.Or);
}
foreach (var e in stream) {
yield return new ExistsExpr(quantifier.tok, quantifier.BoundVars, quantifier.Range, quantifier.Term, quantifier.Attributes);
}
} else {
yield return quantifier;
}
}
protected override void VisitOneExpr(Expression expr) { //FIXME: This doesn't save the rewritten quantifier
var quantifier = expr as QuantifierExpr;
if (quantifier != null) {
var rew = SplitQuantifier(quantifier);
//Console.WriteLine("!!! {0} => {1}", Printer.ExprToString(expr), rew.MapConcat(Printer.ExprToString, " ||| "));
}
}
}
}
|