Looper.h

Go to the documentation of this file.

#pragma once
 
#include <algorithm>
#include <memory>
#include <iostream>
#include <string>
 
#include <chrono>
#include <ctime>
 
#include <boost/exception/all.hpp>
 
#include <TChain.h>
#include <TTree.h>
 
#include "exceptions.h"
 
namespace Darwin::Tools {
 
typedef std::pair<unsigned, unsigned> Slice; 
 
inline std::ostream& operator<< (std::ostream& Stream, const Darwin::Tools::Slice& slice)
{
    Stream << slice.second << '/' << slice.first;
    return Stream;
}
 
class Looper {
    TTree * tree; 
    const Slice slice; 
 
    const long long nEvTot, 
                    start, 
                    stop, 
                    nEvSlice; // number of entries covered by the current process
    long long iEvSlice; // current entry
    long long percent; // percentage of progress for current entry
 
    const std::chrono::time_point<std::chrono::system_clock> start_t; 
 
    Looper (TTree * t, 
            long long nEvents, 
            Slice s = {1,0} 
            ) : tree(t), slice(s),
                nEvTot(nEvents),
                start(nEvTot*((s.second+0.)/s.first)),
                stop (nEvTot*((s.second+1.)/s.first)),
                nEvSlice(stop - start), iEvSlice(0), percent(-1),
                start_t(std::chrono::system_clock::now())
    {
        using namespace std;
 
        if (slice.first == 0 || slice.first <= slice.second)
            BOOST_THROW_EXCEPTION(invalid_argument("The number of slices " + to_string(slice.first)
                        + " must be larger than the index of the current slice " + to_string(slice.second)));
 
        if (nEvSlice <= 0)
            BOOST_THROW_EXCEPTION(invalid_argument("The number of events cannot be " + to_string(nEvSlice)));
 
        cout << slice << '\t' << start << '-' << stop << endl;
    }
 
public:
 
    static long long division; 
 
    template<typename TTreePtr>
    [[deprecated("Pass the slice to GetChain instead")]]
    Looper (const TTreePtr& t, 
            Slice s 
            ) : Looper(&*t, t->GetEntries(), s)
    {
        using namespace std;
        namespace DE = Darwin::Exceptions;
        string root_log = DE::intercept_printf([this]() { printf("loading first entry"); tree->GetEntry(start); });
        tree->GetEntry(start);
        if (root_log.find("Error") != string::npos)
            BOOST_THROW_EXCEPTION(runtime_error("Error while loading a TTree entry:\n" + root_log));
    }
 
    template<typename TTreePtr>
    Looper (const TTreePtr& t 
            ) : Looper(&*t, t->GetEntries(), {1, 0})
    {
        using namespace std;
        namespace DE = Darwin::Exceptions;
        string root_log = DE::intercept_printf([this]() { printf("loading first entry"); tree->GetEntry(start); });
        tree->GetEntry(start);
        if (root_log.find("Error") != string::npos)
            BOOST_THROW_EXCEPTION(runtime_error("Error while loading a TTree entry:\n" + root_log));
    }
 
    Looper (long long nEvents, 
            Slice s = {1,0} 
            ) : Looper(nullptr, nEvents, s)
    {
    }
 
    ~Looper ()
    {
        if (percent < 100ll)
            std::cerr << red << "Warning: the event loop has stopped at entry "
                      << iEvSlice << def << '\n';
    }
 
    void operator++ ()
    {
        ++iEvSlice;
        if (tree) tree->GetEntry(start + iEvSlice);
    }
 
    bool operator() ()
    {
        using namespace std;
        long long test_percent = (100ll*iEvSlice)/nEvSlice;
 
        if (test_percent != percent && test_percent % division == 0) {
            // display current slice
            cout << slice.second << '/' << slice.first;
 
            // display total number of seconds
            auto now_t = chrono::system_clock::now();
            auto elapsed_time {now_t - start_t};
            auto secs = chrono::duration_cast<chrono::seconds>(elapsed_time);
            cout << '\t' << secs.count() << 's';
 
            // display current time in human readable format
            time_t now = chrono::system_clock::to_time_t(now_t);
            cout << '\t' << put_time(localtime(&now), "%Y-%m-%d %H:%M:%S %Z");
 
            // display progress
            percent = test_percent;
            cout << '\t' << percent << '%' << endl;
        }
 
        return iEvSlice < nEvSlice;
    }
 
    inline long long operator* () const
    {
        return start + iEvSlice;
    }
};
 
} // end of Darwin::Tools namespace
 
using Darwin::Tools::operator<<;