+
+////////////////////////////////////////////////////////////////////////////////////////
+//
+// This file is part of Boost Statechart Viewer.
+//
+// Boost Statechart Viewer is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// Boost Statechart Viewer is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with Boost Statechart Viewer. If not, see <http://www.gnu.org/licenses/>.
+//
+////////////////////////////////////////////////////////////////////////////////////////
+
//standard header files
#include <iostream>
-#include <string>
-#include <fstream>
-#include <list>
//LLVM Header files
#include "llvm/Support/raw_ostream.h"
#include "clang/Driver/Compilation.h"
//my own header files
-#include "stringoper.h"
+#include "iooper.h"
using namespace clang;
using namespace clang::driver;
class MyDiagnosticClient : public TextDiagnosticPrinter // My diagnostic Client
{
+ int nwarnings;
+ int nnotes;
+ int nignored;
+ int nerrors;
public:
- MyDiagnosticClient(llvm::raw_ostream &os, const DiagnosticOptions &diags, bool OwnsOutputStream = false):TextDiagnosticPrinter(os, diags, OwnsOutputStream = false){}
+ MyDiagnosticClient(llvm::raw_ostream &os, const DiagnosticOptions &diags, bool OwnsOutputStream = false):TextDiagnosticPrinter(os, diags, OwnsOutputStream = false)
+ {
+ nwarnings=0;
+ nnotes=0;
+ nignored=0;
+ nerrors = 0;
+ }
virtual void HandleDiagnostic(Diagnostic::Level DiagLevel, const DiagnosticInfo &Info)
{
TextDiagnosticPrinter::HandleDiagnostic(DiagLevel, Info); // print diagnostic information
- if(DiagLevel > 2) // if error/fatal error stop the program
- {
- exit(1);
- }
+ switch (DiagLevel)
+ {
+ case 0 : nignored+=1; break;
+ case 1 : nnotes+=1; break;
+ case 2 : nwarnings+=1; break;
+ default : nerrors+=1;
+ print_stats();
+ exit(1);
+ }
+ }
+
+ void print_stats()
+ {
+ cout<<"\n--Diagnostic Info--\n";
+ cout<<"Number of ignored: "<<nignored<<"\n";
+ cout<<"Number of notes: "<<nnotes<<"\n";
+ cout<<"Number of warnings: "<<nwarnings<<"\n";
+ cout<<"Number of errors and fatal errors: "<<nerrors<<"\n";
+ }
+
+ int getNbrOfWarnings()
+ {
+ return nwarnings;
+ }
+
+ int getNbrOfNotes()
+ {
+ return nnotes;
+ }
+
+ int getNbrOfIgnored()
+ {
+ return nignored;
}
};
class FindStates : public ASTConsumer
{
list<string> transitions;
-
list<string> cReactions;
list<string> events;
+ list<string> states;
string name_of_machine;
string name_of_start;
- StringDecl sd;
- int nbrStates;
+ FullSourceLoc *fsloc;
+
public:
- list<string> states;
+
+ list<string> getStates()
+ {
+ return states;
+ }
+
+ list<string> getTransitions()
+ {
+ return transitions;
+ }
+
+ list<string> getEvents()
+ {
+ return events;
+ }
+
+ string getStateMachine()
+ {
+ return name_of_machine;
+ }
+
+ string getNameOfFirstState()
+ {
+ return name_of_start;
+ }
virtual void Initialize(ASTContext &ctx)//run after the AST is constructed before the consumer starts to work
{
+ fsloc = new FullSourceLoc(* new SourceLocation(), ctx.getSourceManager());
name_of_start = "";
name_of_machine = "";
- nbrStates = 0;
}
virtual void HandleTopLevelDecl(DeclGroupRef DGR)// traverse all top level declarations
{
SourceLocation loc;
- std::string line, output;
+ string line, output, event;
llvm::raw_string_ostream x(output);
for (DeclGroupRef::iterator i = DGR.begin(), e = DGR.end(); i != e; ++i)
{
loc = decl->getLocation();
if(loc.isValid())
{
- //cout<<decl->getKind()<<"ss\n";
- if(const CXXMethodDecl *cMDecl = dyn_cast<CXXMethodDecl>(decl))
- {
- //decl->dump();
+ if(decl->getKind()==35)
+ {
+ method_decl(decl);
}
if (const TagDecl *tagDecl = dyn_cast<TagDecl>(decl))
{
}
}
+ output = "";
}
}
void recursive_visit(const DeclContext *declCont) //recursively visit all decls hidden inside namespaces
{
- std::string line, output;
+ string line, output, event;
+ llvm::raw_string_ostream x(output);
SourceLocation loc;
for (DeclContext::decl_iterator i = declCont->decls_begin(), e = declCont->decls_end(); i != e; ++i)
{
loc = decl->getLocation();
if(loc.isValid())
{
- if(const CXXMethodDecl *cMDecl = dyn_cast<CXXMethodDecl>(decl))
+ if(decl->getKind()==35)
{
- /*TODO
- get Base class, test name and chcek if it is a state
-
- */
+ method_decl(decl);
}
else if (const TagDecl *tagDecl = dyn_cast<TagDecl>(decl))
{
recursive_visit(declCont);
}
}
+ output = "";
}
}
void struct_class(const Decl *decl) // works with struct or class decl
{
- string output, line, ret, trans;
+ string output, line, ret, trans, event;
llvm::raw_string_ostream x(output);
decl->print(x);
- line = sd.get_line_of_code(x.str());
+ line = get_line_of_code(x.str());
output = "";
- int pos, num;
- const TagDecl *tagDecl = dyn_cast<TagDecl>(decl);
- const NamedDecl *namedDecl = dyn_cast<NamedDecl>(decl);
- if(sd.is_derived(line))
+ int pos;
+ const NamedDecl *namedDecl = dyn_cast<NamedDecl>(decl);
+ if(is_derived(line))
{
const CXXRecordDecl *cRecDecl = dyn_cast<CXXRecordDecl>(decl);
- if(sd.find_events(cRecDecl, line))
+ if(find_events(cRecDecl, line))
{
events.push_back(namedDecl->getNameAsString());
cout<<"New event: "<<namedDecl->getNameAsString()<<"\n";
}
else if(name_of_machine == "")
{
- ret = sd.find_name_of_machine(cRecDecl, line);
+ ret = find_name_of_machine(cRecDecl, line);
if(!ret.empty())
{
pos = ret.find(",");
}
else
{
- ret = sd.find_states(cRecDecl, line);
+ ret = find_states(cRecDecl, line);
if(!ret.empty())
- {
- const DeclContext *declCont = tagDecl->castToDeclContext(tagDecl);
- //states.push_back(namedDecl->getNameAsString());
- std::cout << "New state: " << namedDecl->getNameAsString() << "\n";
- states.push_back(ret);
- ret = sd.find_transitions(namedDecl->getNameAsString(), declCont);
- if(!ret.empty())
- {
- num = sd.count(ret,';')+1;
- for(int i = 0;i<num;i++)
- {
- pos = ret.find(";");
- if(pos == 0)
- {
- ret = ret.substr(1);
- pos = ret.find(";");
- if(pos==-1) cReactions.push_back(ret);
- else cReactions.push_back(ret.substr(0,pos));
- num-=1;
- }
- else
- {
- if(pos==-1) transitions.push_back(ret);
- else transitions.push_back(ret.substr(0,pos));
- }
- //cout<<ret<<"\n";
- if(i!=num-1) ret = ret.substr(pos+1);
- }
- }
+ {
+ cout << "New state: " << namedDecl->getNameAsString() << "\n";
+ states.push_back(ret);
+ methods_in_class(decl,namedDecl->getNameAsString());
}
}
}
}
-
- void save_to_file(std::string output) // save all to the output file
+
+ void methods_in_class(const Decl *decl, const string state)
{
- nbrStates = states.size();
- std::string state, str, context, ctx;
- int pos1, pos2, cnt, subs;
- std::ofstream filestr(output.c_str());
- //std::cout<<output<<"\n";
- filestr<<"digraph "<< name_of_machine<< " {\n";
- context = name_of_machine;
- for(list<std::string>::iterator i = states.begin();i!=states.end();i++) // write all states in the context of the automaton
+ string output, line, ret, trans, event;
+ llvm::raw_string_ostream x(output);
+ int pos, num;
+ const TagDecl *tagDecl = dyn_cast<TagDecl>(decl);
+ const DeclContext *declCont = tagDecl->castToDeclContext(tagDecl);
+ //states.push_back(namedDecl->getNameAsString());
+
+ output="";
+ for (DeclContext::decl_iterator i = declCont->decls_begin(), e = declCont->decls_end(); i != e; ++i)
{
- state = *i;
- cnt = sd.count(state,',');
- if(cnt==1)
+ if (i->getKind()==26)
{
- pos1 = state.find(",");
- ctx = sd.cut_namespaces(state.substr(pos1+1));
- //std::cout<<name_of_machine.length();
- if(ctx.compare(0,context.length(),context)==0)
+ i->print(x);
+ output = x.str();
+ line = clean_spaces(cut_type(output));
+ ret = find_transitions(state,line);
+ if(!ret.empty())
{
- filestr<<sd.cut_namespaces(state.substr(0,pos1))<<";\n";
- states.erase(i);
- i--;
- }
- }
- if(cnt==2)
- {
- pos1 = state.find(",");
- pos2 = state.rfind(",");
- ctx = sd.cut_namespaces(state.substr(pos1+1,pos2-pos1-1));
- //std::cout<<ctx<<" "<<context<<"\n";
- if(ctx.compare(0,context.length(),context)==0)
- {
- filestr<<sd.cut_namespaces(state.substr(0,pos1))<<";\n";
+ num = count(ret,';')+1;
+ for(int i = 0;i<num;i++)
+ {
+ pos = ret.find(";");
+ if(pos == 0)
+ {
+ ret = ret.substr(1);
+ pos = ret.find(";");
+ if(pos==-1) cReactions.push_back(ret);
+ else cReactions.push_back(ret.substr(0,pos));
+ num-=1;
+ }
+ else
+ {
+ if(pos==-1) transitions.push_back(ret);
+ else transitions.push_back(ret.substr(0,pos));
+ }
+ //cout<<ret<<"\n";
+ if(i!=num-1) ret = ret.substr(pos+1);
+ }
+ output="";
}
}
+ if(i->getKind()==35) method_decl(decl);
}
- filestr<<name_of_start<<" [peripheries=2] ;\n";
- subs = 0;
- while(!states.empty()) // substates ?
+ }
+
+ void method_decl(const Decl *decl)
+ {
+ string output, line, event;
+ llvm::raw_string_ostream x(output);
+ if(decl->hasBody())
{
- state = states.front();
- filestr<<"subgraph cluster"<<subs<<" {\n";
- pos1 = state.find(",");
- pos2 = state.rfind(",");
- context = sd.cut_namespaces(state.substr(0,pos1));
- filestr<<"label=\""<<context<<"\";\n";
- filestr<<sd.cut_namespaces(state.substr(pos2+1))<<" [peripheries=2] ;\n";
- states.pop_front();
- //std::cout<<states.size();
- for(list<string>::iterator i = states.begin();i!=states.end();i++)
+ decl->print(x);
+ line = get_return(x.str());
+ if(test_model(line,"result"))
{
- state = *i;
- cnt = sd.count(state,',');
- //std::cout<<state<<" \n";
- if(cnt==1)
+ const FunctionDecl *fDecl = dyn_cast<FunctionDecl>(decl);
+ const ParmVarDecl *pvd = fDecl->getParamDecl(0);
+ QualType qt = pvd->getOriginalType();
+ event = qt.getAsString();
+ if(event[event.length()-1]=='&') event = event.substr(0,event.length()-2);
+ event = event.substr(event.rfind(" ")+1);
+ line = dyn_cast<NamedDecl>(decl)->getQualifiedNameAsString();
+ line = cut_namespaces(line.substr(0,line.rfind("::")));
+ line.append(",");
+ line.append(event);
+ find_return_stmt(decl->getBody(),line);
+ for(list<string>::iterator i = cReactions.begin();i!=cReactions.end();i++)
{
- pos1 = state.find(",");
- ctx = sd.cut_namespaces(state.substr(pos1+1));
-
- //std::cout<<ctx<<" "<<context<<"\n";
- if(ctx.compare(0,context.length(),context)==0)
+ event = *i;
+ if(line.compare(event)==0)
{
- filestr<<sd.cut_namespaces(state.substr(0,pos1))<<";\n";
- states.erase(i);
- i--;
- }
- }
- if(cnt==2)
- {
- pos1 = state.find(",");
- pos2 = state.rfind(",");
- ctx = sd.cut_namespaces(state.substr(pos1+1,pos2-pos1-1));
- if(ctx.compare(0,context.length(),context)==0)
- {
- filestr<<sd.cut_namespaces(state.substr(0,pos1))<<";\n";
- //std::cout<<ctx<<"\n";
+ cReactions.erase(i);
+ break;
}
}
}
- filestr<<"}\n";
- subs+=1;
- }
- for(list<string>::iterator i = transitions.begin();i!=transitions.end();i++) // write all transitions
+ }
+ }
+
+ void find_return_stmt(Stmt *statemt,string event)
+ {
+ if(statemt->getStmtClass() == 99) test_stmt(dyn_cast<CaseStmt>(statemt)->getSubStmt(), event);
+ else
{
- state = *i;
- pos1 = state.find(",");
- filestr<<sd.cut_namespaces(state.substr(0,pos1))<<"->";
- pos2 = state.rfind(",");
- filestr<<sd.cut_namespaces(state.substr(pos2+1));
- filestr<<"[label=\""<<sd.cut_namespaces(state.substr(pos1+1,pos2-pos1-1))<<"\"];\n";
- }
- filestr<<"}";
- filestr.close();
+ for (Stmt::child_range range = statemt->children(); range; ++range)
+ {
+ test_stmt(*range, event);
+ }
+ }
}
- void print_stats() // print statistics
+
+ void test_stmt(Stmt *stmt, string event)
{
- cout<<"\n"<<"Statistics: \n";
- cout<<"Number of states: "<<nbrStates<<"\n";
- cout<<"Number of events: "<<events.size()<<"\n";
- cout<<"Number of transitions: "<<transitions.size()<<"\n";
- return;
+ const SourceManager &sman = fsloc->getManager();
+ int type;
+ string line, param;
+ type = stmt->getStmtClass();
+ switch(type)
+ {
+ case 8 : find_return_stmt(dyn_cast<DoStmt>(stmt)->getBody(), event); // do
+ break;
+ case 86 : find_return_stmt(dyn_cast<ForStmt>(stmt)->getBody(), event); // for
+ break;
+ case 88 : find_return_stmt(dyn_cast<IfStmt>(stmt)->getThen(), event); //if then
+ find_return_stmt(dyn_cast<IfStmt>(stmt)->getElse(), event); //if else
+ break;
+ case 90 : find_return_stmt(dyn_cast<LabelStmt>(stmt)->getSubStmt(), event); //label
+ break;
+ case 98 : line = sman.getCharacterData(dyn_cast<ReturnStmt>(stmt)->getReturnLoc());
+ line = get_line_of_code(line).substr(6);
+ line = line.substr(0,line.find("("));
+ if(test_model(line,"transit"))
+ {
+ param = get_params(line);
+ transitions.push_back(event.append(",").append(param));
+ }
+ break;
+ case 99 : find_return_stmt(stmt, event);
+ break;
+ case 101 : find_return_stmt(dyn_cast<SwitchStmt>(stmt)->getBody(), event); // switch
+ break;
+ case 102 : find_return_stmt(dyn_cast<WhileStmt>(stmt)->getBody(), event); // while
+ break;
+ }
}
};
{
string inputFilename = "";
string outputFilename = "graph.dot"; // initialize output Filename
- MyDiagnosticClient *mdc = new MyDiagnosticClient(llvm::errs(), * new DiagnosticOptions());
+ DiagnosticOptions dopts;
+ dopts.ShowColors=1;
+ MyDiagnosticClient *mdc = new MyDiagnosticClient(llvm::errs(), dopts);
llvm::IntrusiveRefCntPtr<DiagnosticIDs> dis(new DiagnosticIDs());
Diagnostic diag(dis,mdc);
FileManager fm( * new FileSystemOptions());
SourceManager sm (diag, fm);
HeaderSearch *headers = new HeaderSearch(fm);
- Driver TheDriver(LLVM_PREFIX "/bin", llvm::sys::getHostTriple(), "", false, false, diag);
+ Driver TheDriver(LLVM_BINDIR, llvm::sys::getHostTriple(), "", false, false, diag);
TheDriver.setCheckInputsExist(true);
TheDriver.CCCIsCXX = 1;
+ TheDriver.ResourceDir = LLVM_PREFIX "/lib/clang/" CLANG_VERSION_STRING;
+
CompilerInvocation compInv;
llvm::SmallVector<const char *, 16> Args(argv, argv + argc);
llvm::OwningPtr<Compilation> C(TheDriver.BuildCompilation(Args.size(),
diag);
HeaderSearchOptions hsopts = compInv.getHeaderSearchOpts();
- hsopts.ResourceDir = LLVM_PREFIX "/lib/clang/" CLANG_VERSION_STRING;
LangOptions lang = compInv.getLangOpts();
CompilerInvocation::setLangDefaults(lang, IK_ObjCXX);
TargetInfo *ti = TargetInfo::CreateTargetInfo(diag, compInv.getTargetOpts());
- ApplyHeaderSearchOptions(*headers, hsopts, lang, ti->getTriple());
FrontendOptions f = compInv.getFrontendOpts();
inputFilename = f.Inputs[0].second;
mdc->BeginSourceFile(lang, &pp);//start using diagnostic
ParseAST(pp, &c, ctx, false, false);
mdc->EndSourceFile(); //end using diagnostic
- if(c.states.size()>0) c.save_to_file(outputFilename);
- else cout<<"No state machine was found\n";
- c.print_stats();
+ IO_operations *io = new IO_operations(outputFilename, c.getStateMachine(), c.getNameOfFirstState(), c.getTransitions(), c.getStates(), c.getEvents());
+ io->save_to_file();
+ io->print_stats();
+ mdc->print_stats();
return 0;
}