101 lines
2.8 KiB
C++
101 lines
2.8 KiB
C++
#include "./codefactory.h"
|
|
#include "./clangversionabstraction.h"
|
|
#include "./frontendaction.h"
|
|
|
|
#include "resources/config.h"
|
|
|
|
#include <clang/Basic/FileManager.h>
|
|
#include <clang/Frontend/FrontendActions.h>
|
|
#include <clang/Tooling/Tooling.h>
|
|
|
|
#include <memory>
|
|
|
|
using namespace std;
|
|
|
|
namespace ReflectiveRapidJSON {
|
|
|
|
struct CodeFactory::ToolInvocation {
|
|
ToolInvocation(CodeFactory &factory);
|
|
|
|
clang::FileManager fileManager;
|
|
clang::tooling::ToolInvocation invocation;
|
|
};
|
|
|
|
CodeFactory::ToolInvocation::ToolInvocation(CodeFactory &factory)
|
|
: fileManager({ "." })
|
|
, invocation(factory.makeClangArgs(), maybe_unique(new FrontendAction(factory)), &fileManager)
|
|
{
|
|
fileManager.Retain();
|
|
}
|
|
|
|
/*!
|
|
* \brief Constructs a new instance.
|
|
* \remarks The specified arguments are not copied and must remain valid for the live-time of the code factory.
|
|
*/
|
|
CodeFactory::CodeFactory(std::string_view applicationPath, const std::vector<const char *> &sourceFiles,
|
|
const std::vector<std::string_view> &clangOptions, std::ostream &os)
|
|
: m_applicationPath(applicationPath)
|
|
, m_sourceFiles(sourceFiles)
|
|
, m_clangOptions(clangOptions)
|
|
, m_os(os)
|
|
, m_compilerInstance(nullptr)
|
|
, m_errorResilient(true)
|
|
{
|
|
}
|
|
|
|
CodeFactory::~CodeFactory()
|
|
{
|
|
}
|
|
|
|
/*!
|
|
* \brief Constructs arguments for the Clang tool invocation.
|
|
*/
|
|
std::vector<string> CodeFactory::makeClangArgs() const
|
|
{
|
|
static const initializer_list<std::string_view> flags
|
|
= { m_applicationPath, "-x", "c++", "-Wno-pragma-once-outside-header", "-std=c++14", "-fsyntax-only", "-D" PROJECT_VARNAME_UPPER };
|
|
vector<string> clangArgs;
|
|
clangArgs.reserve(flags.size() + m_clangOptions.size() + m_sourceFiles.size());
|
|
clangArgs.insert(clangArgs.end(), flags.begin(), flags.end());
|
|
clangArgs.insert(clangArgs.end(), m_clangOptions.cbegin(), m_clangOptions.cend());
|
|
clangArgs.insert(clangArgs.end(), m_sourceFiles.cbegin(), m_sourceFiles.cend());
|
|
return clangArgs;
|
|
}
|
|
|
|
/*!
|
|
* \brief Adds the specified \a decl to all underlying code generators. The generators might ignore irrelevant declarations.
|
|
* \remarks Supposed to be called by assigned generators inside readAST().
|
|
*/
|
|
void CodeFactory::addDeclaration(clang::Decl *decl)
|
|
{
|
|
for (const auto &generator : m_generators) {
|
|
generator->addDeclaration(decl);
|
|
}
|
|
}
|
|
|
|
/*!
|
|
* \brief Generates code based on the added declarations.
|
|
*/
|
|
bool CodeFactory::generate() const
|
|
{
|
|
for (const auto &generator : m_generators) {
|
|
generator->generate(m_os);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/*!
|
|
* \brief Reads (relevant) AST elements using Clang and generates code.
|
|
*/
|
|
bool CodeFactory::run()
|
|
{
|
|
// lazy initialize Clang tool invocation
|
|
if (!m_toolInvocation) {
|
|
m_toolInvocation = make_unique<ToolInvocation>(*this);
|
|
}
|
|
// run Clang
|
|
return m_toolInvocation->invocation.run();
|
|
}
|
|
|
|
} // namespace ReflectiveRapidJSON
|