ObjectTree.hpp

Go to the documentation of this file.

 
#ifndef H_POTTU_OBJECTTREE
#define H_POTTU_OBJECTTREE
 
 
#include "Histogram1D.hpp"
#include "Histogram2D.hpp"
#include "binners.hpp"
#include "OTObject.hpp"
 
#include <sys/stat.h>
#include <sys/types.h>
 
#include <stdexcept>
#include <memory>
#include <set>
#include <string>
#include <cstring>
// Keeping still c++14, filesystem requires c++17
//#include <filesystem>
 
 
namespace pottu {
 
 
    static inline bool operator<( const std::unique_ptr<OTObject> &o1, const std::unique_ptr<OTObject> &o2 ) {
    return( o1->name() < o2->name() );
    }
    static inline bool operator<( const std::string &name, const std::unique_ptr<OTObject> &oto ) {
    return( name < oto->name() );
    }
    static inline bool operator<( const std::unique_ptr<OTObject> &oto, const std::string &name ) {
    return( oto->name() < name );
    }
 
    
    class ObjectTree : public OTObject {
    public:
    typedef Histogram1D< BinnerFloat<double>, uint64_t > default_h1_type;
    typedef Histogram1D< BinnerFloat<double>, double > default_h1_doublecell_type;
    typedef Histogram1D< BinnerPow2<uint32_t>, uint64_t > pow2_h1_type;
    
    typedef Histogram2D< BinnerFloat<double>, BinnerFloat<double>, uint64_t > default_h2_type;
    typedef Histogram2D< BinnerFloat<double>, BinnerFloat<double>, double > default_h2_doublecell_type;
 
    ObjectTree( const std::string &name ) : OTObject(name) {}
    virtual ~ObjectTree() {}
    
    ObjectTree *createSubTree( const std::string &name ) {
        auto it = _objects.find<std::string>( name );
        if( it != _objects.end() )
        throw std::runtime_error( std::string()+"Object with name " + name + " already in the tree." );
    
        auto p = new ObjectTree( name );
 
        _objects.emplace( p );
        return( p );
    }
 
 
    default_h1_type *
    createHistogram1D( const std::string &name,
               double low, double high, unsigned int binc,
               const std::string &title = "",
               const std::string &xlabel = "",
               const std::string &ylabel = "" ) {
        auto it = _objects.find<std::string>( name );
        if( it != _objects.end() )
        throw std::runtime_error( std::string()+"Object with name " + name + " already in the tree." );
    
        auto p = new OTOH1D_double_64( name, low, high, binc, title, xlabel, ylabel );
 
        _objects.emplace( p );
        return( &(p->_h) );
    }
 
    default_h1_doublecell_type *
    createHistogram1D_doublecell( const std::string &name,
                      double low, double high, unsigned int binc,
                      const std::string &title = "",
                      const std::string &xlabel = "",
                      const std::string &ylabel = "" ) {
        auto it = _objects.find<std::string>( name );
        if( it != _objects.end() )
        throw std::runtime_error( std::string()+"Object with name " + name + " already in the tree." );
    
        auto p = new OTOH1D_double_double( name, low, high, binc, title, xlabel, ylabel );
 
        _objects.emplace( p );
        return( &(p->_h) );
    }
 
    pow2_h1_type *
    createHistogram1D_pow2( const std::string &name,
                uint32_t high_bits, uint32_t bins_bits,
                const std::string &title = "",
                const std::string &xlabel = "",
                const std::string &ylabel = "" ) {
        auto it = _objects.find<std::string>( name );
        if( it != _objects.end() )
        throw std::runtime_error( std::string()+"Object with name " + name + " already in the tree." );
    
        auto p = new OTOH1D_pow2_64( name, high_bits, bins_bits, title, xlabel, ylabel );
 
        _objects.emplace( p );
        return( &(p->_h) );
    }
 
    
 
    default_h2_type *
    createHistogram2D( const std::string &name,
               double xlow, double xhigh, unsigned int xbinc,
               double ylow, double yhigh, unsigned int ybinc,
               const std::string &title = "",
               const std::string &xlabel = "",
               const std::string &ylabel = "",
               const std::string &zlabel = "" ) {
        auto it = _objects.find<std::string>( name );
        if( it != _objects.end() )
        throw std::runtime_error( std::string()+"Object with name " + name + " already in the tree." );
        
        auto p = new OTOH2D_double_64( name,
                       xlow, xhigh, xbinc,
                       ylow, yhigh, ybinc,
                       title, xlabel, ylabel, zlabel );
        
        _objects.emplace( p );
        return( &(p->_h) );
    }
 
 
    default_h2_doublecell_type *
    createHistogram2D_doublecell( const std::string &name,
                      double xlow, double xhigh, unsigned int xbinc,
                      double ylow, double yhigh, unsigned int ybinc,
                      const std::string &title = "",
                      const std::string &xlabel = "",
                      const std::string &ylabel = "",
                      const std::string &zlabel = "" ) {
        auto it = _objects.find<std::string>( name );
        if( it != _objects.end() )
        throw std::runtime_error( std::string()+"Object with name " + name + " already in the tree." );
        
        auto p = new OTOH2D_double_double( name,
                           xlow, xhigh, xbinc,
                           ylow, yhigh, ybinc,
                           title, xlabel, ylabel, zlabel );
        
        _objects.emplace( p );
        return( &(p->_h) );
    }
 
 
    
    virtual void exportToDisk( const std::string &prefix, export_context_t &ectx ) const {
        
        // First trying to make a directory "prefixname/" where name is
        // the name of this tree.
        std::string newprefix=prefix;
        newprefix += name();
        newprefix += "/";
    
        std::cout << "Trying to create \'" << newprefix << "\'\n";
 
        int ret = mkdir( newprefix.c_str(), 0777 );
        if( ret == -1 && errno != EEXIST ) {
        throw std::runtime_error( strerror(errno) );
        }
 
        /*
          try {
          std::filesystem::create_directory( newprefix );
          } catch( std::exception &e ) {
          std::cerr << "Failed to create \'" << newprefix << "\'\n";
          }
        */
        
        for( auto &obj : _objects ) {
        //cout << "Calling export for subobject " << obj->name() << "\n";
        obj->exportToDisk( newprefix, ectx );
        }
    }
 
        
 
 
    
    std::set< std::unique_ptr<OTObject>, std::less<> >    _objects;
    };
    
}
 
#endif