CinematicExtractor.h

Go to the documentation of this file.

#ifndef CINEMATIC_EXTRACTOR_H
#define CINEMATIC_EXTRACTOR_H
 
#include <string>
#include <utility>
#include <regex>
#include <fstream>
#include <vector>
#include <unordered_map>
#include <optional>
 
int countMatchInRegex(std::string s, std::string re);
 
struct CinematicProcess {
    std::vector<std::string> incoming;
    std::vector<std::string> outgoing;
 
    bool empty() const { return incoming.empty() && outgoing.empty(); }
    std::size_t incoming_count() const { return incoming.size(); }
    std::size_t outgoing_count() const { return outgoing.size(); }
 
    std::vector<std::string> ordered_particles() const {
        std::vector<std::string> out;
        out.reserve(incoming.size() + outgoing.size());
        out.insert(out.end(), incoming.begin(), incoming.end());
        out.insert(out.end(), outgoing.begin(), outgoing.end());
        return out;
    }
};
 
class CinematicExtractor {
public:
    std::pair<int, int> extract(const std::string& filename);
 
    CinematicProcess extract_process(const std::string& filename) const;
 
    static std::string normalize_particle_name(std::string particle_name);
 
    static std::optional<std::string> mass_symbol_for_particle(const std::string& particle_name);
 
    static const std::unordered_map<std::string, std::string>& default_mass_symbols();
};
 
#endif // CINEMATIC_EXTRACTOR_H