Di.cc File Reference

#include <boost/test/included/unit_test.hpp>
#include <iostream>
#include <map>
#include <TFile.h>
#include <TTree.h>
#include <Di.h>
#include <Jet.h>

Include dependency graph for Di.cc:

Macros
#define	BOOST_TEST_MODULE   Di

Functions
vector< GenJet >	GetDummyGenJets ()
	BOOST_AUTO_TEST_CASE (dijet)
	BOOST_TEST (gendijet.Rapidity()==1.3573785791881385)
	BOOST_AUTO_TEST_CASE (variations)

Variables
auto	gendijet = genjets.at(0) + genjets.at(1)

Macro Definition Documentation

BOOST_TEST_MODULE

#define BOOST_TEST_MODULE   Di

Function Documentation

BOOST_AUTO_TEST_CASE() [1/2]

BOOST_AUTO_TEST_CASE

(

dijet

)

{
    vector<GenJet> * genjets = new vector<GenJet>;
    auto gendijet = new Di<GenJet,GenJet>;
    bool valid = static_cast<bool>(*gendijet);
    BOOST_TEST(!valid);
 
    *genjets = GetDummyGenJets();
    cout << *gendijet << endl;
    *gendijet = genjets->at(0) + genjets->at(1);
    valid = static_cast<bool>(*gendijet);
    BOOST_TEST(valid);
 
    auto f = make_unique<TFile>("di.root", "RECREATE");
 
    cout << "Creating tree" << endl;
    {
        auto t = make_unique<TTree>("tree", "tree");
        t->Branch("genJets", &genjets);
        t->Branch("genDijet", &gendijet);
        t->Fill();
        t->Write();
        // Note: calling `TTree:Scan()` at this stage leads to a seg
    }
 
    delete genjets;
    delete gendijet;
    genjets = nullptr;
    gendijet = nullptr;
 
    cout << "Reading tree" << endl;
    {
        auto t = unique_ptr<TTree>(f->Get<TTree>("tree"));
        t->Show(0);
        cout << "Checking genJets" << endl;
        t->Scan("genJets.CorrPt()");
        cout << "Checking genDijet" << endl;
        t->Scan("genDijet.first.CorrPt():genDijet.second.CorrPt():genDijet.HT()");
        cout << "Setting branch addresses" << endl;
        t->SetBranchAddress("genJets", &genjets);
        t->SetBranchAddress("genDijet", &gendijet);
        t->GetEntry(0);
        BOOST_TEST( genjets->at(0).CorrPt() == 1. );
        BOOST_TEST( genjets->at(1).CorrPt() == 2. );
        BOOST_TEST( gendijet->HT() == 1.5 );
        gendijet->clear();
        BOOST_TEST( gendijet->first  == nullptr );
        BOOST_TEST( gendijet->second == nullptr );
    }
}

BOOST_AUTO_TEST_CASE() [2/2]

BOOST_AUTO_TEST_CASE

(

variations

)

{
    auto genjets = GetDummyGenJets();
    auto gendijet = genjets.at(0) + genjets.at(1);
 
    using namespace Darwin::Physics;
    cout << genjets.at(0).Weight(nominal) << ' ' << genjets.at(1).Weight(nominal) << ' ' << gendijet.Weight(nominal) << endl;
 
    map<int, double> bits_results {{1,1.5}, {2,1.5}, {3,1.0}};
    for (auto bit_result: bits_results) {
        int bit = bit_result.first;
        double result = bit_result.second;
        Variation v(GenJet::WeightVar, "wgtUp_stat" + to_string(bit), 1, bit);
        cout << "1st-jet weight: " << genjets.at(0).Weight(v) << '\n'
             << "2nd-jet weight: " << genjets.at(1).Weight(v) << '\n'
             << "dijet weight: " << gendijet.Weight(v) << endl;
        BOOST_TEST( gendijet.Weight(v) == result );
    }
 
    for (int i = 1; i <= 2; ++i) {
        Variation v(GenJet::ScaleVar, "scale", i);
        cout << "1st-jet corrected pt: " << genjets.at(0).CorrPt(v) << '\n'
             << "2nd-jet corrected pt: " << genjets.at(1).CorrPt(v) << '\n'
             << "dijet corrected pt: " << gendijet.CorrPt(v) << '\n'
             << "dijet corrected HT: " << gendijet.HT(v) << endl;
        BOOST_TEST( genjets.at(0).CorrPt(i) == genjets.at(0).CorrPt(v) );
        BOOST_TEST( genjets.at(1).CorrPt(i) == genjets.at(1).CorrPt(v) );
        BOOST_TEST( 2*gendijet.HT(v) == genjets.at(0).CorrPt(i) + genjets.at(1).CorrPt(i) );
    }
}

BOOST_TEST()

BOOST_TEST

(

gendijet.

Rapidity() = =1.3573785791881385

)

GetDummyGenJets()

vector<GenJet> GetDummyGenJets

(

)

{
    GenJet genjet1, genjet2;
 
    // 4-vectors
    genjet1.p4 = {1,1,1,1};
    genjet2.p4 = {2,2,2,2};
 
    // JES
    genjet1.scales.push_back(1.5);
    genjet1.scales.push_back(0.5);
    genjet2.scales.push_back(1.5);
    genjet2.scales.push_back(0.5);
 
    // weights
    genjet1.weights.push_back({1.5,1});
    genjet1.weights.push_back({0.5,1});
    genjet2.weights.push_back({1.5,2});
    genjet2.weights.push_back({0.5,2});
 
    // collection
    vector<GenJet> genjets;
    genjets.push_back(genjet1);
    genjets.push_back(genjet2);
    return genjets;
}

Variable Documentation

gendijet

auto gendijet = genjets.at(0) + genjets.at(1)