Gabriele Cosmo
138 lines
4.0 KiB
Plaintext
138 lines
4.0 KiB
Plaintext
// Copyright (C) 2010, Guy Barrand. All rights reserved.
|
|
// See the file tools.license for terms.
|
|
|
|
#ifndef tools_ntuple_booking
|
|
#define tools_ntuple_booking
|
|
|
|
// a little class to capture booking parameters
|
|
// to create an ntuple.
|
|
|
|
#include "cids"
|
|
|
|
#include <ostream>
|
|
#include "forit"
|
|
|
|
namespace tools {
|
|
|
|
class column_booking {
|
|
public:
|
|
column_booking(const std::string& a_name,cid a_cid,void* a_user_obj)
|
|
:m_name(a_name)
|
|
,m_cid(a_cid)
|
|
,m_user_obj(a_user_obj) //WARNING : not owner.
|
|
{}
|
|
virtual ~column_booking() {}
|
|
public:
|
|
column_booking(const column_booking& a_from)
|
|
:m_name(a_from.m_name)
|
|
,m_cid(a_from.m_cid)
|
|
,m_user_obj(a_from.m_user_obj)
|
|
{}
|
|
column_booking& operator=(const column_booking& a_from) {
|
|
if(&a_from==this) return *this;
|
|
m_name = a_from.m_name;
|
|
m_cid = a_from.m_cid;
|
|
m_user_obj = a_from.m_user_obj;
|
|
return *this;
|
|
}
|
|
public:
|
|
const std::string& name() const {return m_name;}
|
|
cid cls_id() const {return m_cid;}
|
|
void* user_obj() const {return m_user_obj;}
|
|
void set_user_obj(void* a_obj) {m_user_obj = a_obj;}
|
|
protected:
|
|
std::string m_name;
|
|
cid m_cid;
|
|
void* m_user_obj;
|
|
};
|
|
|
|
class ntuple_booking {
|
|
public:
|
|
ntuple_booking(const std::string& a_name = "",const std::string& a_title = "")
|
|
:m_name(a_name)
|
|
,m_title(a_title)
|
|
{}
|
|
virtual ~ntuple_booking(){}
|
|
public:
|
|
ntuple_booking(const ntuple_booking& a_from)
|
|
:m_name(a_from.m_name)
|
|
,m_title(a_from.m_title)
|
|
,m_columns(a_from.m_columns)
|
|
{}
|
|
ntuple_booking& operator=(const ntuple_booking& a_from){
|
|
m_name = a_from.m_name;
|
|
m_title = a_from.m_title;
|
|
m_columns = a_from.m_columns;
|
|
return *this;
|
|
}
|
|
public:
|
|
template <class T>
|
|
void add_column(const std::string& a_name) {
|
|
m_columns.push_back(column_booking(a_name,_cid(T()),0));
|
|
}
|
|
template <class T>
|
|
void add_column(const std::string& a_name,T& a_user) {
|
|
m_columns.push_back(column_booking(a_name,_cid(T()),(void*)&a_user));
|
|
}
|
|
|
|
template <class T>
|
|
void add_column_vec(const std::string& a_name) {
|
|
m_columns.push_back(column_booking(a_name,_cid_std_vector<T>(),0));
|
|
}
|
|
template <class T>
|
|
void add_column_vec(const std::string& a_name,std::vector<T>& a_user_vec) {
|
|
m_columns.push_back(column_booking(a_name,_cid_std_vector<T>(),(void*)&a_user_vec));
|
|
}
|
|
template <class T>
|
|
void add_column(const std::string& a_name,std::vector<T>& a_user_vec) {
|
|
m_columns.push_back(column_booking(a_name,_cid_std_vector<T>(),(void*)&a_user_vec));
|
|
}
|
|
|
|
const std::string& name() const {return m_name;}
|
|
const std::string& title() const {return m_title;}
|
|
const std::vector<column_booking>& columns() const {return m_columns;}
|
|
std::vector<column_booking>& columns() {return m_columns;}
|
|
|
|
void set_name(const std::string& a_s) {m_name = a_s;}
|
|
void set_title(const std::string& a_s) {m_title = a_s;}
|
|
|
|
bool has_similar_layout(std::ostream& a_out,const ntuple_booking& a_nbk) {
|
|
if(m_columns.size()!=a_nbk.m_columns.size()) {
|
|
a_out << "tools::ntuple_booking::has_similar_layout :"
|
|
<< " bookings have not the same number of columns."
|
|
<< " (" << m_columns.size() << " != " << a_nbk.m_columns.size() << ")."
|
|
<< std::endl;
|
|
return false;
|
|
}
|
|
std::vector<column_booking>::const_iterator ait = a_nbk.m_columns.begin();
|
|
tools_vforit(tools::column_booking,m_columns,it) {
|
|
if((*it).name()!=(*ait).name()) {
|
|
a_out << "tools::ntuple_booking::has_similar_layout :"
|
|
<< " columns don't have same name."
|
|
<< " (" << (*it).name() << " != " << (*ait).name() << ")."
|
|
<< std::endl;
|
|
return false;
|
|
}
|
|
if((*it).cls_id()!=(*ait).cls_id()) {
|
|
a_out << "tools::ntuple_booking::has_similar_layout :"
|
|
<< " columns don't have same class id."
|
|
<< " (" << (*it).cls_id() << " != " << (*ait).cls_id() << ")."
|
|
<< std::endl;
|
|
return false;
|
|
}
|
|
|
|
ait++;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
protected:
|
|
std::string m_name;
|
|
std::string m_title;
|
|
std::vector<column_booking> m_columns;
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|