Files
clang-uml/src/main.cc
2021-02-15 23:59:20 +01:00

73 lines
2.1 KiB
C++

#include "config/config.h"
#include "puml/class_diagram_generator.h"
#include "puml/sequence_diagram_generator.h"
#include "uml/class_diagram_model.h"
#include "uml/class_diagram_visitor.h"
#include "cx/compilation_database.h"
#include "uml/sequence_diagram_visitor.h"
#include <cli11/CLI11.hpp>
#include <glob/glob.hpp>
#include <spdlog/spdlog.h>
#include <filesystem>
#include <fstream>
#include <iostream>
#include <limits.h>
#include <malloc.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
using namespace clanguml;
using config::config;
using cx::compilation_database;
int main(int argc, const char *argv[])
{
spdlog::set_pattern("[%l] %v");
CLI::App app{"Clang-based PlantUML generator from C++ sources"};
std::string config_path{".clanguml"};
std::string compilation_database_dir{'.'};
bool verbose{false};
app.add_option(
"-c,--config", config_path, "Location of configuration file");
app.add_option("-d,--compile-database", compilation_database_dir,
"Location of configuration file");
app.add_flag("-v,--verbose", verbose, "Verbose logging");
CLI11_PARSE(app, argc, argv);
if (verbose)
spdlog::set_level(spdlog::level::debug);
spdlog::info("Loading clang-uml config from {}", config_path);
auto config = config::load(config_path);
spdlog::info("Loading compilation database from {} directory",
config.compilation_database_dir);
auto db =
compilation_database::from_directory(config.compilation_database_dir);
for (const auto &[name, diagram] : config.diagrams) {
using config::class_diagram;
using config::sequence_diagram;
if (std::dynamic_pointer_cast<class_diagram>(diagram)) {
generators::class_diagram::generate(
db, name, dynamic_cast<class_diagram &>(*diagram));
}
else if (std::dynamic_pointer_cast<sequence_diagram>(diagram)) {
generators::sequence_diagram::generate(
db, name, dynamic_cast<sequence_diagram &>(*diagram));
}
}
return 0;
}