TPSVariables.h

Go to the documentation of this file.

#pragma once
 
#include <string>
#include <vector>
#include <functional>
#include "TMVA/DataLoader.h"
#include "TMVA/Reader.h"
#include "TString.h"
 
namespace DAS::TPS {
 
class TPSVariables {
public:
    // Variable information structure
    struct VariableInfo {
        TString name;               
        char type;                  
        TString description;        
        VariableInfo(const TString& n, char t, const TString& desc = ""):
            name(n), type(t), description(desc) { }
    };
 
    std::vector<VariableInfo> variables_;
    TPSVariables(TMVA::DataLoader* dataloader = nullptr) {
 
        // Basic jet variables
        variables_.emplace_back("nJets", 'I', "Number of jets");
        variables_.emplace_back("jet_pt[0]", 'F', "Leading jet pT");
        variables_.emplace_back("jet_pt[1]", 'F', "Second jet pT");
        variables_.emplace_back("jet_eta[0]", 'F', "Leading jet eta");
        // variables_.emplace_back("jet_eta[1]", 'F', "Second jet eta");
        variables_.emplace_back("jet_phi[0]", 'F', "Leading jet phi");
        // variables_.emplace_back("jet_phi[1]", 'F', "Second jet phi");
        
        // Pair asymmetry variables
        for (int j = 1; j <= 2; j++) {
            variables_.emplace_back(Form("pt_pair_asymm_abs_0[%d]", j), 'F', 
                                  Form("PT pair asymmetry absolute for jet pair 0-%d", j));
            variables_.emplace_back(Form("Delta_eta_pair_0[%d]", j), 'F', 
                                  Form("Delta eta for jet pair 0-%d", j));
            variables_.emplace_back(Form("Delta_phi_pair_0[%d]", j), 'F', 
                                  Form("Delta phi for jet pair 0-%d", j));
        }
        
        // S observables
        variables_.emplace_back("S_pt", 'D', "S_pt observable");
        variables_.emplace_back("S_pt_min", 'D', "S_pt_min observable");
        variables_.emplace_back("S_pt_4[0]", 'D', "S_pt_4[0] observable");
        // variables_.emplace_back("S_pt_4[1]", 'D', "S_pt_4[1] observable");
        variables_.emplace_back("S_pt_4_min[0]", 'D', "S_pt_4_min observable");
 
        variables_.emplace_back("S_phi", 'D', "S_phi observable");
        variables_.emplace_back("S_phi_min", 'D', "S_phi_min observable");
        // variables_.emplace_back("S_phi_min", 'D', "S_phi_min observable");
        variables_.emplace_back("S_phi_4[0]", 'D', "S_phi_4[0] observable");
        // variables_.emplace_back("S_phi_4[1]", 'D', "S_phi_4[1] observable");
        variables_.emplace_back("S_phi_4_min[0]", 'D', "S_phi_4_min observable");
        
        // Add variables to TMVA DataLoader for training
        if (dataloader) {
            for (const auto& var : variables_) {
                dataloader->AddVariable(var.name, var.type);
            }
        }
    }
 
    void printVariables() const {
        std::cout << "TPS Variables (" << variables_.size() << " total):" << std::endl;
        for (size_t i = 0; i < variables_.size(); ++i) {
            const auto& var = variables_[i];
            std::cout << "  " << i+1 << ". " << var.name.Data() << " [" << var.type << "]";
            if (!var.description.IsNull()) {
                std::cout << " - " << var.description.Data();
            }
            std::cout << std::endl;
        }
    }
};
 
} // namespace DAS::TPS