Application.h

Go to the documentation of this file.

#pragma once

#include <csignal>

#include <CLI/CLI.hpp>

#include <goofit/Color.h>
#include <goofit/Error.h>
#include <goofit/Log.h>

#define GOOFIT_PARSE(app, ...)                                                                                         \
    try {                                                                                                              \
        app.run();                                                                                                     \
    } catch(const GooFit::ParseError &e) {                                                                             \
        return app.exit(e);                                                                                            \
    }

namespace GooFit {

void signal_handler(int s);

// Importing into the GooFit namespace the main classes from CLI11
using CLI::App;
using CLI::ExistingDirectory;
using CLI::ExistingFile;
using CLI::ExitCodes;
using CLI::FileError;
using CLI::NonexistentPath;
using CLI::Option;
using CLI::ParseError;
using CLI::Success;

void print_splash();

void print_goofit_info(int gpuDev_ = 0);

class Application : public CLI::App {
  protected:
    int gpuDev_ = 0;
    bool show_gpus_;
    bool quiet_;
    bool splash_;
    int argc_;
    char **argv_;

    struct sigaction sigIntHandler;

  public:
    Application(std::string discription, int argc, char **argv);

    Application(int argc, char **argv)
        : Application("", argc, argv) {}

    int get_device() const { return gpuDev_; }

    void run() { parse(argc_, argv_); }

    void pre_callback() override;

    int exit(const CLI::Error &e);

    void set_device() const;

    ~Application() override;

    std::string get_filename(const std::string &input_str, std::string base = "") const;
};
} // namespace GooFit