Files
geant4/source/analysis/g4tools/include/tools/wroot/to
T
2016-06-10 14:11:04 +02:00

269 lines
8.7 KiB
Plaintext

// Copyright (C) 2010, Guy Barrand. All rights reserved.
// See the file tools.license for terms.
#ifndef tools_wroot_to
#define tools_wroot_to
#include "directory"
#include "streamers"
#include "bufobj"
namespace tools {
namespace wroot {
inline bool to(directory& a_dir,const histo::h1d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,a_name,a_histo.title(),"TH1D");
if(!TH1D_stream(*bo,a_histo,a_name)) {
a_dir.file().out() << "tools::wroot::to :"
<< " TH1D_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to(directory& a_dir,const histo::h1df& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,a_name,a_histo.title(),"TH1F");
if(!TH1F_stream(*bo,a_histo,a_name)) {
a_dir.file().out() << "tools::wroot::to :"
<< " TH1F_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to(directory& a_dir,const histo::h2d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,
a_name,a_histo.title(),"TH2D");
if(!TH2D_stream(*bo,a_histo,a_name)) {
a_dir.file().out() << "tools::wroot::to :"
<< " TH2D_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to(directory& a_dir,const histo::h2df& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,a_name,a_histo.title(),"TH2F");
if(!TH2F_stream(*bo,a_histo,a_name)) {
a_dir.file().out() << "tools::wroot::to :"
<< " TH2F_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to(directory& a_dir,const histo::h3d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,
a_name,a_histo.title(),"TH3D");
if(!TH3D_stream(*bo,a_histo,a_name)) {
a_dir.file().out() << "tools::wroot::to :"
<< " TH3D_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to(directory& a_dir,const histo::h3df& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,a_name,a_histo.title(),"TH3F");
if(!TH3F_stream(*bo,a_histo,a_name)) {
a_dir.file().out() << "tools::wroot::to :"
<< " TH3F_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to(directory& a_dir,const histo::p1d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,
a_name,a_histo.title(),"TProfile");
if(!TProfile_stream(*bo,a_histo,a_name)) {
a_dir.file().out() << "tools::wroot::to :"
<< " TProfile_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to(directory& a_dir,const histo::p2d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,
a_name,a_histo.title(),"TProfile2D");
if(!TProfile2D_stream(*bo,a_histo,a_name)) {
a_dir.file().out() << "tools::wroot::to :"
<< " TProfile2D_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool write_histos(directory& a_dir,const std::vector< std::pair<std::string,void*> >& a_hists) {
typedef std::pair<std::string,void*> class_pointer;
tools_vforcit(class_pointer,a_hists,it) {
const std::string& scls = (*it).first;
void* p = (*it).second;
if(scls==histo::h1d::s_class()) {
histo::h1d& h = *((histo::h1d*)p);
if(!to(a_dir,h,h.title())) return false;
} else if(scls==histo::h2d::s_class()) {
histo::h2d& h = *((histo::h2d*)p);
if(!to(a_dir,h,h.title())) return false;
} else if(scls==histo::h3d::s_class()) {
histo::h3d& h = *((histo::h3d*)p);
if(!to(a_dir,h,h.title())) return false;
} else if(scls==histo::p1d::s_class()) {
histo::p1d& h = *((histo::p1d*)p);
if(!to(a_dir,h,h.title())) return false;
} else if(scls==histo::p2d::s_class()) {
histo::p2d& h = *((histo::p2d*)p);
if(!to(a_dir,h,h.title())) return false;
} else {
a_dir.file().out() << "tools::wroot::write_histos :"
<< " WARNING : class " << scls << " not handled."
<< std::endl;
}
}
return true;
}
}}
#include "member_writer"
#include "../store/osc_streamers"
namespace tools {
namespace wroot {
inline bool to_osc(directory& a_dir,const histo::h1d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,
a_name,a_histo.title(),osc::s_h1d());
//Stream as in a BatchLab/Rio/Data.h :
if(!bo->write_version(1)) return false;
if(!Named_stream(*bo,a_name,a_histo.title())) return false;
member_writer mw(*bo);
if(!osc::visit(mw,a_histo)) {
a_dir.file().out() << "tools::wroot::to_osc :"
<< " Histogram1D_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to_osc(directory& a_dir,const histo::h2d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,
a_name,a_histo.title(),osc::s_h2d());
//Stream as in a BatchLab/Rio/Data.h :
if(!bo->write_version(1)) return false;
if(!Named_stream(*bo,a_name,a_histo.title())) return false;
member_writer mw(*bo);
if(!osc::visit(mw,a_histo)) {
a_dir.file().out() << "tools::wroot::to_osc :"
<< " Histogram2D_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to_osc(directory& a_dir,const histo::h3d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,
a_name,a_histo.title(),osc::s_h3d());
//Stream as in a BatchLab/Rio/Data.h :
if(!bo->write_version(1)) return false;
if(!Named_stream(*bo,a_name,a_histo.title())) return false;
member_writer mw(*bo);
if(!osc::visit(mw,a_histo)) {
a_dir.file().out() << "tools::wroot::to_osc :"
<< " Histogram3D_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to_osc(directory& a_dir,const histo::p1d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,
a_name,a_histo.title(),osc::s_p1d());
//Stream as in a BatchLab/Rio/Data.h :
if(!bo->write_version(1)) return false;
if(!Named_stream(*bo,a_name,a_histo.title())) return false;
member_writer mw(*bo);
if(!osc::visit(mw,a_histo)) {
a_dir.file().out() << "tools::wroot::to_osc :"
<< " Profile1D_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
inline bool to_osc(directory& a_dir,const histo::p2d& a_histo,const std::string& a_name) {
bufobj* bo = new bufobj(a_dir.file().out(),a_dir.file().byte_swap(),256,
a_name,a_histo.title(),osc::s_p2d());
//Stream as in a BatchLab/Rio/Data.h :
if(!bo->write_version(1)) return false;
if(!Named_stream(*bo,a_name,a_histo.title())) return false;
member_writer mw(*bo);
if(!osc::visit(mw,a_histo)) {
a_dir.file().out() << "tools::wroot::to_osc :"
<< " Profile2D_stream failed."
<< std::endl;
delete bo;
return false;
}
a_dir.append_object(bo); //a_dir takes ownership of bo.
return true;
}
}}
#endif