//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()); } } }