forked from qt-creator/qt-creator
New matchers
This commit is contained in:
Roberto Raggi
@@ -55,13 +55,13 @@ static const char copyrightHeader[] =
|
||||
"**************************************************************************/\n"
|
||||
;
|
||||
|
||||
class VisitCG: protected ASTVisitor
|
||||
class Accept0CG: protected ASTVisitor
|
||||
{
|
||||
QDir _cplusplusDir;
|
||||
QTextStream *out;
|
||||
|
||||
public:
|
||||
VisitCG(const QDir &cplusplusDir, Control *control)
|
||||
Accept0CG(const QDir &cplusplusDir, Control *control)
|
||||
: ASTVisitor(control), _cplusplusDir(cplusplusDir), out(0)
|
||||
{ }
|
||||
|
||||
@@ -194,13 +194,13 @@ protected:
|
||||
}
|
||||
};
|
||||
|
||||
class MatchCG: protected ASTVisitor
|
||||
class Match0CG: protected ASTVisitor
|
||||
{
|
||||
QDir _cplusplusDir;
|
||||
QTextStream *out;
|
||||
|
||||
public:
|
||||
MatchCG(const QDir &cplusplusDir, Control *control)
|
||||
Match0CG(const QDir &cplusplusDir, Control *control)
|
||||
: ASTVisitor(control), _cplusplusDir(cplusplusDir), out(0)
|
||||
{ }
|
||||
|
||||
@@ -245,49 +245,11 @@ protected:
|
||||
Overview oo;
|
||||
const QString className = oo(klass->name());
|
||||
|
||||
*out << " if (" << className << " *_other = pattern->as" << className.left(className.length() - 3) << "()) {" << endl;
|
||||
*out << " if (" << className << " *_other = pattern->as" << className.left(className.length() - 3) << "())" << endl;
|
||||
|
||||
*out << " if (! matcher->match(this, _other))" << endl
|
||||
<< " return false;" << endl;
|
||||
*out << " return matcher->match(this, _other);" << endl;
|
||||
|
||||
for (unsigned i = 0; i < klass->memberCount(); ++i) {
|
||||
Symbol *member = klass->memberAt(i);
|
||||
if (! member->name())
|
||||
continue;
|
||||
|
||||
Identifier *id = member->name()->identifier();
|
||||
|
||||
if (! id)
|
||||
continue;
|
||||
|
||||
const QByteArray memberName = QByteArray::fromRawData(id->chars(), id->size());
|
||||
if (member->type().isUnsigned() && memberName.endsWith("_token")) {
|
||||
// nothing to do here.
|
||||
|
||||
} else if (PointerType *ptrTy = member->type()->asPointerType()) {
|
||||
|
||||
if (NamedType *namedTy = ptrTy->elementType()->asNamedType()) {
|
||||
QByteArray typeName = namedTy->name()->identifier()->chars();
|
||||
|
||||
if (typeName.endsWith("AST")) {
|
||||
*out << " if (! match(" << memberName << ", _other->" << memberName << ", matcher))" << endl
|
||||
<< " return false;" << endl;
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (unsigned i = 0; i < klass->baseClassCount(); ++i) {
|
||||
const QByteArray baseClassName = klass->baseClassAt(i)->identifier()->chars();
|
||||
|
||||
if (ClassSpecifierAST *baseClassSpec = classMap.value(baseClassName, 0)) {
|
||||
visitMembers(baseClassSpec->symbol);
|
||||
}
|
||||
}
|
||||
|
||||
*out << " return true;" << endl;
|
||||
*out << " }" << endl;
|
||||
*out << endl;
|
||||
}
|
||||
|
||||
bool checkMethod(Symbol *accept0Method) const
|
||||
@@ -341,6 +303,149 @@ protected:
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
class MatcherCPPCG: protected ASTVisitor
|
||||
{
|
||||
QDir _cplusplusDir;
|
||||
QTextStream *out;
|
||||
|
||||
public:
|
||||
MatcherCPPCG(const QDir &cplusplusDir, Control *control)
|
||||
: ASTVisitor(control), _cplusplusDir(cplusplusDir), out(0)
|
||||
{ }
|
||||
|
||||
void operator()(AST *ast)
|
||||
{
|
||||
QFileInfo fileInfo(_cplusplusDir, QLatin1String("ASTMatcher.cpp"));
|
||||
|
||||
QFile file(fileInfo.absoluteFilePath());
|
||||
if (! file.open(QFile::WriteOnly))
|
||||
return;
|
||||
|
||||
QTextStream output(&file);
|
||||
out = &output;
|
||||
|
||||
*out << copyrightHeader <<
|
||||
"\n"
|
||||
"#include \"AST.h\"\n"
|
||||
"#include \"ASTMatcher.h\"\n"
|
||||
"\n"
|
||||
"using namespace CPlusPlus;\n" << endl;
|
||||
|
||||
accept(ast);
|
||||
}
|
||||
|
||||
protected:
|
||||
using ASTVisitor::visit;
|
||||
|
||||
QMap<QByteArray, ClassSpecifierAST *> classMap;
|
||||
|
||||
QByteArray id_cast(NameAST *name)
|
||||
{
|
||||
if (! name)
|
||||
return QByteArray();
|
||||
|
||||
Identifier *id = identifier(name->asSimpleName()->identifier_token);
|
||||
|
||||
return QByteArray::fromRawData(id->chars(), id->size());
|
||||
}
|
||||
|
||||
void visitMembers(Class *klass)
|
||||
{
|
||||
const QByteArray className = klass->name()->identifier()->chars();
|
||||
|
||||
for (unsigned i = 0; i < klass->memberCount(); ++i) {
|
||||
Symbol *member = klass->memberAt(i);
|
||||
if (! member->name())
|
||||
continue;
|
||||
|
||||
Identifier *id = member->name()->identifier();
|
||||
|
||||
if (! id)
|
||||
continue;
|
||||
|
||||
const QByteArray memberName = QByteArray::fromRawData(id->chars(), id->size());
|
||||
if (member->type().isUnsigned() && memberName.endsWith("_token")) {
|
||||
// nothing to do. The member is a token.
|
||||
|
||||
*out << " if (node->" << memberName << " != pattern->" << memberName << ")" << endl
|
||||
<< " return false;" << endl;
|
||||
|
||||
} else if (PointerType *ptrTy = member->type()->asPointerType()) {
|
||||
|
||||
if (NamedType *namedTy = ptrTy->elementType()->asNamedType()) {
|
||||
QByteArray typeName = namedTy->name()->identifier()->chars();
|
||||
|
||||
if (typeName.endsWith("AST")) {
|
||||
*out << " if (! AST::match(node->" << memberName << ", pattern->" << memberName << ", this))" << endl
|
||||
<< " return false;" << endl;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (unsigned i = 0; i < klass->baseClassCount(); ++i) {
|
||||
const QByteArray baseClassName = klass->baseClassAt(i)->identifier()->chars();
|
||||
|
||||
if (ClassSpecifierAST *baseClassSpec = classMap.value(baseClassName, 0)) {
|
||||
visitMembers(baseClassSpec->symbol);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
bool checkMethod(Symbol *accept0Method) const
|
||||
{
|
||||
Declaration *decl = accept0Method->asDeclaration();
|
||||
if (! decl)
|
||||
return false;
|
||||
|
||||
Function *funTy = decl->type()->asFunctionType();
|
||||
if (! funTy)
|
||||
return false;
|
||||
|
||||
else if (funTy->isPureVirtual())
|
||||
return false;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
virtual bool visit(ClassSpecifierAST *ast)
|
||||
{
|
||||
Class *klass = ast->symbol;
|
||||
const QByteArray className = id_cast(ast->name);
|
||||
|
||||
Identifier *match0_id = control()->findOrInsertIdentifier("match0");
|
||||
Symbol *match0Method = klass->members()->lookat(match0_id);
|
||||
for (; match0Method; match0Method = match0Method->next()) {
|
||||
if (match0Method->identifier() != match0_id)
|
||||
continue;
|
||||
|
||||
if (checkMethod(match0Method))
|
||||
break;
|
||||
}
|
||||
|
||||
if (! match0Method)
|
||||
return true;
|
||||
|
||||
classMap.insert(className, ast);
|
||||
|
||||
*out
|
||||
<< "bool ASTMatcher::match(" << className.constData() << " *node, " << className.constData() << " *pattern)" << endl
|
||||
<< "{" << endl;
|
||||
|
||||
visitMembers(klass);
|
||||
|
||||
*out
|
||||
<< " return true;" << endl
|
||||
<< "}" << endl
|
||||
<< endl;
|
||||
|
||||
return true;
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
|
||||
QTextCursor createCursor(TranslationUnit *unit, AST *ast, QTextDocument *document)
|
||||
{
|
||||
unsigned startLine, startColumn, endLine, endColumn;
|
||||
@@ -545,12 +650,15 @@ QStringList generateAST_H(const Snapshot &snapshot, const QDir &cplusplusDir)
|
||||
out << document.toPlainText();
|
||||
}
|
||||
|
||||
VisitCG cg(cplusplusDir, doc->control());
|
||||
Accept0CG cg(cplusplusDir, doc->control());
|
||||
cg(doc->translationUnit()->ast());
|
||||
|
||||
MatchCG cg2(cplusplusDir, doc->control());
|
||||
Match0CG cg2(cplusplusDir, doc->control());
|
||||
cg2(doc->translationUnit()->ast());
|
||||
|
||||
MatcherCPPCG cg3(cplusplusDir, doc->control());
|
||||
cg3(doc->translationUnit()->ast());
|
||||
|
||||
return astDerivedClasses;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user