Geant4 Cross Reference |
1 // Copyright (C) 2010, Guy Barrand. All rights 1 // Copyright (C) 2010, Guy Barrand. All rights reserved. 2 // See the file tools.license for terms. 2 // See the file tools.license for terms. 3 3 4 #ifndef tools_ntuple_booking 4 #ifndef tools_ntuple_booking 5 #define tools_ntuple_booking 5 #define tools_ntuple_booking 6 6 7 // a little class to capture booking parameter 7 // a little class to capture booking parameters 8 // to create an ntuple. 8 // to create an ntuple. 9 9 10 #include "cids" 10 #include "cids" 11 11 12 #include <ostream> 12 #include <ostream> 13 #include "forit" 13 #include "forit" 14 14 15 namespace tools { 15 namespace tools { 16 16 17 class column_booking { 17 class column_booking { 18 public: 18 public: 19 column_booking(const std::string& a_name,cid 19 column_booking(const std::string& a_name,cid a_cid,void* a_user_obj) 20 :m_name(a_name) 20 :m_name(a_name) 21 ,m_cid(a_cid) 21 ,m_cid(a_cid) 22 ,m_user_obj(a_user_obj) //WARNING : not owne 22 ,m_user_obj(a_user_obj) //WARNING : not owner. 23 {} 23 {} 24 virtual ~column_booking() {} 24 virtual ~column_booking() {} 25 public: 25 public: 26 column_booking(const column_booking& a_from) 26 column_booking(const column_booking& a_from) 27 :m_name(a_from.m_name) 27 :m_name(a_from.m_name) 28 ,m_cid(a_from.m_cid) 28 ,m_cid(a_from.m_cid) 29 ,m_user_obj(a_from.m_user_obj) 29 ,m_user_obj(a_from.m_user_obj) 30 {} 30 {} 31 column_booking& operator=(const column_booki 31 column_booking& operator=(const column_booking& a_from) { 32 if(&a_from==this) return *this; 32 if(&a_from==this) return *this; 33 m_name = a_from.m_name; 33 m_name = a_from.m_name; 34 m_cid = a_from.m_cid; 34 m_cid = a_from.m_cid; 35 m_user_obj = a_from.m_user_obj; 35 m_user_obj = a_from.m_user_obj; 36 return *this; 36 return *this; 37 } 37 } 38 public: 38 public: 39 const std::string& name() const {return m_na 39 const std::string& name() const {return m_name;} 40 cid cls_id() const {return m_cid;} 40 cid cls_id() const {return m_cid;} 41 void* user_obj() const {return m_user_obj;} 41 void* user_obj() const {return m_user_obj;} 42 void set_user_obj(void* a_obj) {m_user_obj = 42 void set_user_obj(void* a_obj) {m_user_obj = a_obj;} 43 protected: 43 protected: 44 std::string m_name; 44 std::string m_name; 45 cid m_cid; 45 cid m_cid; 46 void* m_user_obj; 46 void* m_user_obj; 47 }; 47 }; 48 48 49 class ntuple_booking { 49 class ntuple_booking { 50 public: 50 public: 51 ntuple_booking(const std::string& a_name = " 51 ntuple_booking(const std::string& a_name = "",const std::string& a_title = "") 52 :m_name(a_name) 52 :m_name(a_name) 53 ,m_title(a_title) 53 ,m_title(a_title) 54 {} 54 {} 55 virtual ~ntuple_booking(){} 55 virtual ~ntuple_booking(){} 56 public: 56 public: 57 ntuple_booking(const ntuple_booking& a_from) 57 ntuple_booking(const ntuple_booking& a_from) 58 :m_name(a_from.m_name) 58 :m_name(a_from.m_name) 59 ,m_title(a_from.m_title) 59 ,m_title(a_from.m_title) 60 ,m_columns(a_from.m_columns) 60 ,m_columns(a_from.m_columns) 61 {} 61 {} 62 ntuple_booking& operator=(const ntuple_booki 62 ntuple_booking& operator=(const ntuple_booking& a_from){ 63 m_name = a_from.m_name; 63 m_name = a_from.m_name; 64 m_title = a_from.m_title; 64 m_title = a_from.m_title; 65 m_columns = a_from.m_columns; 65 m_columns = a_from.m_columns; 66 return *this; 66 return *this; 67 } 67 } 68 public: 68 public: 69 template <class T> 69 template <class T> 70 void add_column(const std::string& a_name) { 70 void add_column(const std::string& a_name) { 71 m_columns.push_back(column_booking(a_name, 71 m_columns.push_back(column_booking(a_name,_cid(T()),0)); 72 } 72 } 73 template <class T> 73 template <class T> 74 void add_column(const std::string& a_name,T& 74 void add_column(const std::string& a_name,T& a_user) { 75 m_columns.push_back(column_booking(a_name, 75 m_columns.push_back(column_booking(a_name,_cid(T()),(void*)&a_user)); 76 } 76 } 77 77 78 template <class T> 78 template <class T> 79 void add_column_vec(const std::string& a_nam 79 void add_column_vec(const std::string& a_name) { 80 m_columns.push_back(column_booking(a_name, 80 m_columns.push_back(column_booking(a_name,_cid_std_vector<T>(),0)); 81 } 81 } 82 template <class T> 82 template <class T> 83 void add_column_vec(const std::string& a_nam 83 void add_column_vec(const std::string& a_name,std::vector<T>& a_user_vec) { 84 m_columns.push_back(column_booking(a_name, 84 m_columns.push_back(column_booking(a_name,_cid_std_vector<T>(),(void*)&a_user_vec)); 85 } 85 } 86 template <class T> 86 template <class T> 87 void add_column(const std::string& a_name,st 87 void add_column(const std::string& a_name,std::vector<T>& a_user_vec) { 88 m_columns.push_back(column_booking(a_name, 88 m_columns.push_back(column_booking(a_name,_cid_std_vector<T>(),(void*)&a_user_vec)); 89 } 89 } 90 90 91 const std::string& name() const {return m_na 91 const std::string& name() const {return m_name;} 92 const std::string& title() const {return m_t 92 const std::string& title() const {return m_title;} 93 const std::vector<column_booking>& columns() 93 const std::vector<column_booking>& columns() const {return m_columns;} 94 std::vector<column_booking>& columns() {retu 94 std::vector<column_booking>& columns() {return m_columns;} 95 95 96 void set_name(const std::string& a_s) {m_nam 96 void set_name(const std::string& a_s) {m_name = a_s;} 97 void set_title(const std::string& a_s) {m_ti 97 void set_title(const std::string& a_s) {m_title = a_s;} 98 98 99 bool has_similar_layout(std::ostream& a_out, 99 bool has_similar_layout(std::ostream& a_out,const ntuple_booking& a_nbk) { 100 if(m_columns.size()!=a_nbk.m_columns.size( 100 if(m_columns.size()!=a_nbk.m_columns.size()) { 101 a_out << "tools::ntuple_booking::has_sim 101 a_out << "tools::ntuple_booking::has_similar_layout :" 102 << " bookings have not the same nu 102 << " bookings have not the same number of columns." 103 << " (" << m_columns.size() << " ! 103 << " (" << m_columns.size() << " != " << a_nbk.m_columns.size() << ")." 104 << std::endl; 104 << std::endl; 105 return false; 105 return false; 106 } 106 } 107 std::vector<column_booking>::const_iterato 107 std::vector<column_booking>::const_iterator ait = a_nbk.m_columns.begin(); 108 tools_vforit(tools::column_booking,m_colum 108 tools_vforit(tools::column_booking,m_columns,it) { 109 if((*it).name()!=(*ait).name()) { 109 if((*it).name()!=(*ait).name()) { 110 a_out << "tools::ntuple_booking::has_s 110 a_out << "tools::ntuple_booking::has_similar_layout :" 111 << " columns don't have same nam 111 << " columns don't have same name." 112 << " (" << (*it).name() << " != 112 << " (" << (*it).name() << " != " << (*ait).name() << ")." 113 << std::endl; 113 << std::endl; 114 return false; 114 return false; 115 } 115 } 116 if((*it).cls_id()!=(*ait).cls_id()) { 116 if((*it).cls_id()!=(*ait).cls_id()) { 117 a_out << "tools::ntuple_booking::has_s 117 a_out << "tools::ntuple_booking::has_similar_layout :" 118 << " columns don't have same cla 118 << " columns don't have same class id." 119 << " (" << (*it).cls_id() << " ! 119 << " (" << (*it).cls_id() << " != " << (*ait).cls_id() << ")." 120 << std::endl; 120 << std::endl; 121 return false; 121 return false; 122 } 122 } 123 123 124 ait++; 124 ait++; 125 } 125 } 126 return true; 126 return true; 127 } 127 } 128 128 129 protected: 129 protected: 130 std::string m_name; 130 std::string m_name; 131 std::string m_title; 131 std::string m_title; 132 std::vector<column_booking> m_columns; 132 std::vector<column_booking> m_columns; 133 }; 133 }; 134 134 135 } 135 } 136 136 137 #endif 137 #endif