Pair.hh

Go to the documentation of this file.

#pragma once

#include "heavyhiggs/analysis/HeavyHiggs.hh"
#include "boca/generic/DEBUG_MACROS.hh"

namespace heavyhiggs
{

using namespace boca;

enum class Production
{
    DYP, VBF, Associated
};

std::string Name(Production production_channel);

namespace analysis
{

template <typename Tagger_>
class Pair : public boca::Analysis<Tagger_>
{

public:

    void SetFiles(Phase const& phase) override {
        switch (phase.Tag()) {
        case Tag::signal :
            this->AddSignal(Process::bb, Production::VBF);
            break;
        case Tag::background :
            this->AddBackground(Process::bb, Production::DYP);
            this->AddBackground(Process::cc, Production::DYP);
            this->AddBackground(Process::cc, Production::VBF);
            this->AddBackground(Process::jj, Production::DYP);
            this->AddBackground(Process::jj, Production::VBF);
            break;
        }
    }

    std::string Name() const override {
        return boca::Name(Collider()) + "-eta3.5";
        //         return  ProductionChannelName(ProductionChannel()) + DetectorName(Collider())  + "_" + Name(Mass());
    }

private:

    long TrainNumberMax() const override {
        return 1000000;
        return 100000;
        return 10000;
        return 1000;
        return 100;
    };

    Production ProductionChannel() const {
        //         return Production::DYP;
        return Production::VBF;
        //         return Production::Associated;
    }

    boca::Collider Collider() const {
        return boca::Collider::lhc;
        return boca::Collider::future;
    }

    Id MotherId(Production production) const {
        switch (production) {
        case Production::DYP : return Id::Z;
        case Production::VBF : return Id::bottom;
        case Production::Associated : return Id::gluon;
            DEFAULT(to_int(production), Id::none);
        }
    }

    int BackgroundFileNumber() const override {
        return 1;
    }

    Crosssection SignalCrosssection() const {
        return pb;
    }

    boca::Mass Mass() const {
        return 400_GeV;
        return 600_GeV;
        return 1000_GeV;
        return 2000_GeV;
        return 3000_GeV;
        return 4000_GeV;
        return 5000_GeV;
        return 6000_GeV;
        return 7000_GeV;
        return 8000_GeV;
        return 9000_GeV;
        return 10000_GeV;
    }

    void AddSignal(Process process, Production production) {
        boca::analysis::Base::AddSignal(NameString(process, production));
    }

    void AddBackground(Process process, Production production) {
        boca::analysis::Base::AddBackground(NameString(process, production));
    }

    void NewFile(Tag tag, Process process, Production production) {
        this->NewFile(tag, NameString(process, production));
    }


//     std::string NameString(Process process) const {
//         return heavyhiggs::Name(ProductionChannel()) + heavyhiggs::Name(process) + "_" + boca::Name(Collider());
//     }

    std::string NameString(Process process, Production production) const {
        return heavyhiggs::Name(production) + heavyhiggs::Name(process) + "_" + boca::Name(Collider());
    }

//     FileInfo BackgroundFile(Process process, Production production) const {
//         return BackgroundFile(process, BackgroundFileNumber(), production);
//     }
//
//     FileInfo BackgroundFile(Process process) const {
//         return BackgroundFile(process, BackgroundFileNumber());
//     }
//
//     FileInfo BackgroundFile(Process process, int) const {
//         std::vector<std::string> names;
//         names.emplace_back(NameString(process));
//         return FileInfo(names , BackgroundCrosssection(process));
//     }
//
//     FileInfo BackgroundFile(Process process, int, Production production) const {
//         std::vector<std::string> names;
//         names.emplace_back(NameString(process, production));
//         return FileInfo(names , BackgroundCrosssection(process));
//     }


//     std:: string SignalName(Process process) {
//         return  NameString(process) + "_" + boca::units::Name(Mass());
//     }

    Crosssection BackgroundCrosssection(Process) const {
        return pb;
    }

};

}

}