trick/trick_source/codegen/Interface_Code_Gen/ClassTemplateVisitor.cpp
2015-02-26 09:02:31 -06:00

85 lines
2.3 KiB
C++

#include <iostream>
#include <sstream>
#include "llvm/Support/CommandLine.h"
#include "clang/Basic/SourceManager.h"
#include "clang/Basic/FileManager.h"
#include "clang/AST/RecordLayout.h"
#include "clang/AST/ASTContext.h"
#include "ClassTemplateVisitor.hh"
#include "ClassVisitor.hh"
#include "EnumVisitor.hh"
#include "EnumValues.hh"
#include "PrintAttributes.hh"
#include "Utilities.hh"
extern llvm::cl::opt< int > debug_level ;
ClassTemplateVisitor::ClassTemplateVisitor(
clang::CompilerInstance & in_ci ,
CommentSaver & in_cs ,
HeaderSearchDirs & in_hsd,
PrintAttributes & in_pa ) :
ci(in_ci) ,
cs(in_cs) ,
hsd(in_hsd),
pa(in_pa)
{}
/* We want to process enumerations out of template definitions */
bool ClassTemplateVisitor::TraverseDecl(clang::Decl *d) {
if (!d)
return true;
if (d->isImplicit())
return true;
if ( debug_level >=2 ) {
std::cout << "ClassTemplateVisitor TraverseDecl " << d->getDeclKindName() << std::endl ;
}
switch ( d->getKind() ) {
case clang::Decl::CXXRecord : {
TraverseCXXRecordDecl(static_cast<clang::CXXRecordDecl *>(d)) ;
}
break ;
case clang::Decl::Enum : {
clang::EnumDecl * ed = static_cast<clang::EnumDecl *>(d) ;
/* This is an embedded enumeration within a class template. Only process it if is public */
if ( ed->getAccess() == clang::AS_public ) {
if ( debug_level >=2 ) {
ed->dump() ; std::cout << std::endl ;
}
EnumVisitor evis(ci, hsd) ;
evis.TraverseDecl(ed) ;
//if ( evis.get_enum_data() != NULL ) {
evis.get_enum_data()->setHasDefinition(false) ;
pa.printEnum(evis.get_enum_data()) ;
//}
}
}
break ;
default :
break ;
}
return true ;
}
bool ClassTemplateVisitor::VisitDecl(clang::Decl *d) {
if ( debug_level >=2 ) {
std::cout << "\nClassTemplateVisitor Decl = " << d->getDeclKindName() << std::endl ;
}
return true ;
}
bool ClassTemplateVisitor::VisitType(clang::Type *t) {
if ( debug_level >=2 ) {
std::cout << "\nClassTemplateVisitor Type = " << t->getTypeClassName() << std::endl ;
}
return true;
}