//not used
public static void prettyPrint(string fileName)
{
DTE dte = (DTE)ExtractMethodPackage.GetGlobalService(typeof(DTE));
TextWriter x = new StringWriter();
Printer p = new Printer(x);
dte.ActiveDocument.Save();
p.PrintProgram(HelpFunctions.GetProgram(fileName), false);
var resultText = x.ToString();
//get length of doc
TextDocument doc = (TextDocument)(dte.ActiveDocument.Object("TextDocument"));
var pp = doc.StartPoint.CreateEditPoint();
string s = pp.GetText(doc.EndPoint);
ITextEdit edit = HelpFunctions.GetWpfView().TextBuffer.CreateEdit();
edit.Delete(0, s.Length);
edit.Insert(0, resultText);
edit.Apply();
dte.ActiveDocument.Save();
}
private void button_Click(object sender, RoutedEventArgs e)
{
this.Hide();
DVariableComparer comparer = new DVariableComparer();
List <Statement> afterSelected;
List <Statement> selectedStatements = (List <Statement>)HelpFunctions.GetSelectedStatements();
if (selectedStatements.Count == 0)
{
return;
}
//start and end of all relevant stmts and not only the selected ones.
var start = selectedStatements[0].Tok.pos;
if (selectedStatements[0] is UpdateStmt)
{
if (((UpdateStmt)selectedStatements[0]).Lhss.Count > 0)
{
start = ((UpdateStmt)selectedStatements[0]).Lhss[0].tok.pos;
}
else if (((UpdateStmt)selectedStatements[0]).Rhss.Count > 0)
{
if (((UpdateStmt)selectedStatements[0]).Rhss[0] is ExprRhs)
{
if (((ExprRhs)((UpdateStmt)selectedStatements[0]).Rhss[0]).Expr is ApplySuffix)
{
start -= ((NameSegment)((ApplySuffix)((ExprRhs)((UpdateStmt)selectedStatements[0]).Rhss[0]).Expr).Lhs).Name.Length;
}
}
}
}
var end = selectedStatements[selectedStatements.Count - 1].EndTok.pos;
HelpFunctions.GetSelectedStatements(out afterSelected).ToList();
List <DVariable> tempIns = new List <DVariable>();
List <DVariable> tempOuts = new List <DVariable>();
List <DVariable> tempToDeclare = new List <DVariable>();
HelpFunctions.GetVariables(selectedStatements, afterSelected, out tempIns, out tempOuts, out tempToDeclare);
tempOuts = tempOuts.Union(tempToDeclare).ToList();
List <DVariable> intersect = tempIns.Intersect(tempOuts, comparer).ToList <DVariable>();
Dictionary <string, string> renameDic = new Dictionary <string, string>();
foreach (var x in intersect)
{
renameDic.Add(x.name, x.name + HelpFunctions.getNextIndex(x.name, tempIns, tempOuts, tempToDeclare));
}
List <Formal> ins = new List <Formal>();
List <Formal> outs = new List <Formal>();
foreach (var x in tempIns)
{
DVariable temp = new DVariable(x.name, x.type);
if (renameDic.ContainsKey(temp.name))
{
temp.name = renameDic[temp.name];
}
ins.Add(temp.ToFormal());
}
foreach (var x in tempOuts)
{
outs.Add(x.ToFormal());
}
var requires = HelpFunctions.getPreAsserts(selectedStatements);
var ensures = HelpFunctions.getPostAsserts(selectedStatements);
selectedStatements = HelpFunctions.removeReqAndEnsFromStmts(selectedStatements);
selectedStatements = HelpFunctions.removeVarDeclFromStmts(selectedStatements);
var method = HelpFunctions.GetCurrentMethod();
List <MaybeFreeExpression> req = new List <MaybeFreeExpression>();
foreach (var x in requires)
{
//HelpFunctions.renameB(x.Expr, renameDic);
req.Add(new MaybeFreeExpression(HelpFunctions.renameB(x.Expr, renameDic)));
}
List <MaybeFreeExpression> ens = new List <MaybeFreeExpression>();
foreach (var x in ensures)
{
ens.Add(new MaybeFreeExpression(x.Expr));
}
var newMethod = new Method(null, this.textBox.Text, false, false, new List <TypeParameter>(), ins, outs, req, new Specification <FrameExpression>(null, null), ens, new Specification <Microsoft.Dafny.Expression>(null, null), null, null, null);
List <Microsoft.Dafny.Expression> Lhs = new List <Microsoft.Dafny.Expression>();
List <AssignmentRhs> Rhs = new List <AssignmentRhs>();
foreach (var x in intersect)
{
Lhs.Add(new NameSegment(null, x.name, new List <Type>()));
Rhs.Add(new ExprRhs(new NameSegment(null, renameDic[x.name], new List <Type>())));
}
if (Lhs.Count > 0)
{
UpdateStmt correctingNames = new UpdateStmt(null, null, Lhs, Rhs);
selectedStatements.Insert(0, correctingNames);
}
newMethod.Body = new BlockStmt(null, null, selectedStatements);
// HelpFunctions.renameBody(newMethod.Body, new Dictionary<string, string>());///debug purposes only!!!!
string signature = "aklt 5ra";
string body = "ya m3lm";
try
{
signature = Printer.MethodSignatureToString(newMethod);
body = Printer.StatementToString(newMethod.Body);
}
catch (System.Exception ee)
{
}
// Place the new method implementation in the code.
int position = HelpFunctions.GetCurrentMethod().BodyEndTok.pos + 1;
ITextEdit edit = HelpFunctions.GetWpfView().TextBuffer.CreateEdit();
edit.Insert(position, "\r\n\r\n" + signature + "\r\n" + body);
string methodcall = HelpFunctions.generateMethodCall(newMethod, tempIns, tempOuts);
//DTE dte = (DTE)ExtractMethodPackage.GetGlobalService(typeof(DTE));
//var selection = (EnvDTE.TextSelection)dte.ActiveDocument.Selection;
edit.Delete(start, end - start + 1);
edit.Insert(start, HelpFunctions.toDeclareString(tempToDeclare) + methodcall);
edit.Apply();
//selection.Text = methodcall;
}
public static void inline(UpdateStmt stmt, string fileName, Method m /*the called method*/)
{
DTE dte = (DTE)ExtractMethodPackage.GetGlobalService(typeof(DTE));
if (!(((ExprRhs)stmt.Rhss[0]).Expr is ApplySuffix))
{
return;
}
//var arg = ((ApplySuffix)((ExprRhs)stmt.Rhss[0]).Expr).Args;
//Microsoft.Dafny.Type[] InTypes = getTypes(arg);
//var rets = stmt.Lhss;
//Microsoft.Dafny.Type[] retTypes = getTypes(rets);
//Method m = HelpFunctions.FindMethod(fileName, ((NameSegment)((ApplySuffix)((ExprRhs)stmt.Rhss[0]).Expr).Lhs).Name, InTypes, retTypes);
//Method m = HelpFunctions.FindMethod(fileName, "MinAndMax");
Dictionary <string, string> rename = new Dictionary <string, string>();
List <Microsoft.Dafny.Expression> currArgs = ((ApplySuffix)((ExprRhs)stmt.Rhss[0]).Expr).Args;
List <Microsoft.Dafny.Expression> currRet = stmt.Lhss;
for (int i = 0; i < m.Ins.Count && i < currArgs.Count; i++)
{
if (currArgs[i] is Microsoft.Dafny.Expression)
{
rename.Add(m.Ins[i].Name, Printer.ExprToString(currArgs[i]));
}
else
{
throw new NotImplementedException();
}
}
for (int i = 0; i < m.Outs.Count && i < currRet.Count; i++)
{
rename.Add(m.Outs[i].Name, Printer.ExprToString(currRet[i]));
}
List <MaybeFreeExpression> Ens = new List <MaybeFreeExpression>(m.Ens);
List <MaybeFreeExpression> Req = new List <MaybeFreeExpression>(m.Req);
//foreach (var x in Ens)
//{
// HelpFunctions.renameBody(x, rename);
// //ens.Add(new AssertStmt());
//}
StringBuilder res = new StringBuilder();
int indentLength = (stmt.Tok.col - ((NameSegment)((ApplySuffix)((ExprRhs)stmt.Rhss[0]).Expr).Lhs).Name.Length) - 1;
if (stmt.Lhss.Count > 0)
{
indentLength = stmt.Lhss[0].tok.col - 1;
}
string indent = new string('\t', indentLength);
foreach (var x in Req)
{
//HelpFunctions.renameBody(x, rename);
res.Append(indent + "assert " + Printer.ExprToString(HelpFunctions.renameB(x.E, rename)) + ";\r\n");
}
//HelpFunctions.renameBody(m.Body, rename);
m.Body = (BlockStmt)HelpFunctions.renameB(m.Body, rename);
foreach (var x in m.Body.Body)
{
res.Append(indent + Printer.StatementToString(x, indentLength * 4) + "\r\n");
}
foreach (var x in Ens)
{
res.Append(indent + "assert " + Printer.ExprToString(HelpFunctions.renameB(x.E, rename)) + ";\r\n");
}
string result = res.ToString();
if (Ens.Count > 0)
{
result = result.Remove(result.Length - 2, 2);
}
//ITextEdit edit = HelpFunctions.GetWpfView().TextBuffer.CreateEdit();
int from = stmt.Tok.pos - m.Name.Length;
if (stmt.Lhss.Count > 0)
{
from = stmt.Lhss[0].tok.pos;
}
edit.Delete(from, stmt.EndTok.pos - from + 1);
if (result.Length > 1)
{
edit.Insert(from, result.Substring(indentLength));
}
//edit.Apply();
//((EnvDTE.TextSelection)dte.ActiveDocument.Selection).Text = "";
////((EnvDTE.TextSelection)dte.ActiveDocument.Selection).Delete();
//if (result.Length > 1) ((EnvDTE.TextSelection)dte.ActiveDocument.Selection).Insert(result.Substring(1));
}
public static void traverse(Statement statement, Method m)
{
if (statement is AssignStmt)
{
statement = statement as AssignStmt;
traverse(((AssignStmt)statement).Lhs, m);
}
else if (statement is UpdateStmt)
{
DTE dte = (DTE)ExtractMethodPackage.GetGlobalService(typeof(DTE));
var fileName = dte.ActiveDocument.FullName;
UpdateStmt curr = (UpdateStmt)statement;
if (!(((ExprRhs)curr.Rhss[0]).Expr is ApplySuffix))
{
return;
}
var arg = ((ApplySuffix)((ExprRhs)curr.Rhss[0]).Expr).Args;
Microsoft.Dafny.Type[] InTypes = getTypes(arg);
var rets = curr.Lhss;
Microsoft.Dafny.Type[] retTypes = getTypes(rets);
Method temp = HelpFunctions.FindMethod(fileName, ((NameSegment)((ApplySuffix)((ExprRhs)curr.Rhss[0]).Expr).Lhs).Name, InTypes, retTypes);
//check name of called func first;
if (MethodsAreEqual(m, temp))
{
inline((UpdateStmt)statement, fileName, m);
}
}
else if (statement is WhileStmt)
{
traverse(((WhileStmt)statement).Body, m);
traverse(((WhileStmt)statement).Guard, m);
}
else if (statement is IfStmt)
{
traverse(((IfStmt)statement).Guard, m);
traverse(((IfStmt)statement).Thn, m);
traverse(((IfStmt)statement).Els, m);
}
else if (statement is BlockStmt)
{
foreach (var st in ((BlockStmt)statement).SubStatements)
{
traverse(st, m);
}
}
else if (statement is AssertStmt)
{
traverse(((AssertStmt)statement).Proof, m);
}
else if (statement is AlternativeStmt)
{
}
else if (statement is VarDeclStmt)
{
if (((VarDeclStmt)statement).Update != null)
{
traverse(((VarDeclStmt)statement).Update, m);
}
foreach (var localvar in ((VarDeclStmt)statement).Locals)
{
//traverse(localvar);
}
}
else if (statement is CallStmt)
{
foreach (var expr in ((CallStmt)statement).Lhs)
{
traverse(expr, m);
}
foreach (var expr in ((CallStmt)statement).Args)
{
traverse(expr, m);
}
}
else if (statement is PrintStmt)
{
foreach (var expr in ((PrintStmt)statement).Args)
{
traverse(expr, m);
}
}
else if (statement is AssignSuchThatStmt)
{
foreach (var expr in ((AssignSuchThatStmt)statement).Lhss)
{
traverse(expr, m);
}
traverse(((AssignSuchThatStmt)statement).Expr, m);
}
else if (statement is CalcStmt)
{
}
else if (statement is MatchStmt)
{
}
else
{
if (statement != null)
{
throw new NotImplementedException(statement.ToString());
}
}
}
