SpecViewer.h

Go to the documentation of this file.

#include "TH1.h"
#include "TH2I.h"
#include "TGraph.h"

template <typename T>
void read_1d_data(std::istream &in, TH1 *hist)
{
    std::vector<T> buffer(hist->GetNbinsX());
    in.read((char*)&buffer[0], buffer.size()*sizeof(T));
    for (unsigned i = 0; i < buffer.size(); ++i)
        hist->SetBinContent(i+1, buffer[i]);
}

template <typename T>
void read_1d_data_to_graph(std::istream &in, TGraph *graph)
{
    std::vector<T> buffer(graph->GetN());
    in.read((char*)&buffer[0], buffer.size()*sizeof(T));
    for (unsigned i = 0; i < buffer.size(); ++i)
        graph->SetPoint(i,i*10.,buffer[i]);
}

template <typename T>
void read_2d_data(std::istream &in, TH2I *hist)
{
    std::vector<T> buffer(hist->GetNbinsX()* hist->GetNbinsY());
    in.read((char*)&buffer[0], buffer.size()*sizeof(T));
    for (int j = 0; j < hist->GetNbinsY(); ++j)
        for (int i = 0; i < hist->GetNbinsX(); ++i)
        {
            hist->SetBinContent(i+1,j+1, buffer[j+i*hist->GetNbinsY()]);
        }
}

struct DinoFileInfo
{
    // basename__n1-n2-n3-n4-...-nN-datatype__specifier.type
    std::string basename;
    std::string specifier;
    std::string format;
    std::string histtype;
    std::string datatype;
    std::vector<int> number_stack;

    explicit DinoFileInfo(const std::string &filename)
    {
        // cut the proceeding directory information from the filename
        std::string workname(filename, filename.find_last_of("/")+1);
        for (uint i = 0; i < workname.size(); ++i)
            if (workname[i] == '_' || workname[i] == '.')
                workname[i] = ' ';

        std::istringstream workin(workname);

        workin >> basename >> format >> specifier >> histtype;

        std::istringstream format_in(format);
        // read the numbers from the format string
        for (;;)
        {
            int num;
            char c;
            format_in >> c;
            format_in.putback(c);
            if (!isdigit(c))
            {
                format_in >> datatype;
                break;
            }
            else
                format_in >> num >> c;
            if (!format_in || c != '-')
                break;
            number_stack.push_back(num);
        }
    }
};