BCAux.h

#ifndef __BCAUX__H
#define __BCAUX__H

/*
 * Copyright (C) 2007-2018, the BAT core developer team
 * All rights reserved.
 *
 * For the licensing terms see doc/COPYING.
 * For documentation see http://mpp.mpg.de/bat
 */

// ---------------------------------------------------------

#include <string>

#include "BCH1D.h"
#include "BCH2D.h"

#include <TROOT.h>

class TH2;

// ---------------------------------------------------------

namespace BCAux
{

template <typename T>
class BCTrash
{
public:
    BCTrash() {}
    BCTrash(const BCTrash<T>&) {}
    BCTrash<T>& operator=(const BCTrash<T>&)
    {
        return *this;
    }

    ~BCTrash()
    {
        for (unsigned i = 0; i < fStorage.size(); ++i)
            delete fStorage[i];
    }

    void Put(T* object)
    {
        fStorage.push_back(object);
    }

private:
    std::vector<T*> fStorage;
};

void SetStyle();

void DefaultToPDF(std::string& filename);

TH2* Transpose(const TH2* const h, const std::string& name = "");

enum BCRange {
    kFiniteRange           = 1, 
    kNegativeInfiniteRange = 2, 
    kPositiveInfiniteRange = 3, 
    kInfiniteRange         = 4, 
    kEmptyRange            = 5, 
    kReverseRange          = 6  
};

BCAux::BCRange RangeType(double xmin, double xmax);

void MakeFinite(double& xmin, double& xmax);

std::string SafeName(const std::string& name);

bool AllowedCharacter(char c);

enum BCKnowledgeUpdateDrawingStyle {
    kKnowledgeUpdateDefaultStyle      = 0, 
    kKnowledgeUpdateDetailedPosterior = 1, 
    kKnowledgeUpdateDetailedPrior     = 2  
};

void SetKnowledgeUpdateDrawingStyle(BCH1D& prior, BCH1D& posterior, BCAux::BCKnowledgeUpdateDrawingStyle style = BCAux::kKnowledgeUpdateDefaultStyle);

void SetKnowledgeUpdateDrawingStyle(BCH2D& prior, BCH2D& posterior, BCAux::BCKnowledgeUpdateDrawingStyle style = BCAux::kKnowledgeUpdateDefaultStyle);

void DrawKnowledgeUpdate(BCHistogramBase& prior, BCHistogramBase& posterior, bool draw_prior_first, BCTrash<TObject>& trash);

unsigned PrintPlots(std::vector<BCH1D>& h1, std::vector<BCH2D>& h2, const std::string& filename, unsigned hdiv = 1, unsigned vdiv = 1);

class RootSideEffectGuard
{
public:
    RootSideEffectGuard();
    ~RootSideEffectGuard();

private:
    // not implemented
    RootSideEffectGuard(const RootSideEffectGuard&);
    RootSideEffectGuard& operator=(const RootSideEffectGuard&);

    TDirectory* fDirectory;
};

template <class T>
T* OwnClone(const T* o)
{
    return static_cast<T*>(gROOT->CloneObject(o, false));
}

template <class T>
T* OwnClone(const T* o, const std::string& name)
{
    T* res = OwnClone(o);
    res->SetName(name.c_str());
    return res;
}

}

// ---------------------------------------------------------

#endif