Gabriele Cosmo
323 lines
10 KiB
Plaintext
323 lines
10 KiB
Plaintext
// Copyright (C) 2010, Guy Barrand. All rights reserved.
|
|
// See the file tools.license for terms.
|
|
|
|
#ifndef tools_wroot_streamers
|
|
#define tools_wroot_streamers
|
|
|
|
#include "named"
|
|
#include "date"
|
|
#include "directory"
|
|
#include "file"
|
|
#include "../vmanip" //convert
|
|
|
|
#include "../histo/h1d"
|
|
#include "../histo/h2d"
|
|
//#include "../histo/h3d"
|
|
#include "../histo/p1d"
|
|
//#include "../histo/p2d"
|
|
|
|
namespace tools {
|
|
namespace wroot {
|
|
|
|
typedef histo::histo_data<double,unsigned int,double> hd_data;
|
|
typedef histo::profile_data<double,unsigned int,double,double> pd_data;
|
|
|
|
inline bool AttAxis_stream(buffer& a_buffer) {
|
|
int fNdivisions = 510; //Number of divisions(10000*n3 + 100*n2 + n1)
|
|
short fAxisColor = 1; //color of the line axis
|
|
short fLabelColor = 1; //color of labels
|
|
short fLabelFont = 62; //font for labels
|
|
float fLabelOffset = 0.005F; //offset of labels
|
|
float fLabelSize = 0.04F; //size of labels
|
|
float fTickLength = 0.03F; //length of tick marks
|
|
float fTitleOffset = 1; //offset of axis title
|
|
float fTitleSize = 0.04F; //size of axis title
|
|
short fTitleColor = 1; //color of axis title
|
|
short fTitleFont = 62; //font for axis title
|
|
|
|
// Version 4 streaming (ROOT/v3-00-6).
|
|
unsigned int beg;
|
|
if(!a_buffer.write_version(4,beg)) return false;
|
|
|
|
if(!a_buffer.write(fNdivisions)) return false;
|
|
if(!a_buffer.write(fAxisColor)) return false;
|
|
if(!a_buffer.write(fLabelColor)) return false;
|
|
if(!a_buffer.write(fLabelFont)) return false;
|
|
if(!a_buffer.write(fLabelOffset)) return false;
|
|
if(!a_buffer.write(fLabelSize)) return false;
|
|
if(!a_buffer.write(fTickLength)) return false;
|
|
if(!a_buffer.write(fTitleOffset)) return false;
|
|
if(!a_buffer.write(fTitleSize)) return false;
|
|
if(!a_buffer.write(fTitleColor)) return false;
|
|
if(!a_buffer.write(fTitleFont)) return false;
|
|
|
|
if(!a_buffer.set_byte_count(beg)) return false;
|
|
return true;
|
|
}
|
|
|
|
inline bool axis_stream(buffer& a_buffer,
|
|
const histo::axis<double>& a_axis,
|
|
const std::string& a_name,
|
|
const std::string& a_title) {
|
|
// Version 6 streaming (ROOT/v3-00-6).
|
|
|
|
unsigned int beg;
|
|
if(!a_buffer.write_version(6,beg)) return false;
|
|
|
|
if(!Named_stream(a_buffer,a_name,a_title)) return false;
|
|
|
|
if(!AttAxis_stream(a_buffer)) return false;
|
|
|
|
if(!a_buffer.write(a_axis.bins())) return false;
|
|
if(!a_buffer.write(a_axis.lower_edge())) return false;
|
|
if(!a_buffer.write(a_axis.upper_edge())) return false;
|
|
|
|
// fXbins
|
|
//if(a_axis.m_fixed) {
|
|
// std::vector<double> v;
|
|
// ArrayT<double> dummy(v);
|
|
// if(!dummy.stream(a_buffer)) return false; //TArrayD
|
|
//} else {
|
|
if(!a_buffer.write_array(a_axis.edges())) return false; //TArrayD
|
|
//}
|
|
|
|
if(!a_buffer.write((int)0)) return false; //fFirst
|
|
if(!a_buffer.write((int)0)) return false; //fLast
|
|
|
|
//Bool_t
|
|
if(!a_buffer.write((unsigned char)0)) return false; //TimeDisplay
|
|
|
|
//TString
|
|
if(!a_buffer.write(std::string())) return false; //TimeFormat
|
|
|
|
if(!a_buffer.set_byte_count(beg)) return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
inline bool List_empty_stream(buffer& a_buffer) {
|
|
unsigned int beg;
|
|
if(!a_buffer.write_version(4,beg)) return false;
|
|
if(!Object_stream(a_buffer)) return false;
|
|
std::string name;
|
|
if(!a_buffer.write(name)) return false;
|
|
int nobjects = 0;
|
|
if(!a_buffer.write(nobjects)) return false;
|
|
if(!a_buffer.set_byte_count(beg)) return false;
|
|
return true;
|
|
}
|
|
|
|
inline std::string axis_title(const hd_data& a_data,
|
|
const std::string& a_key) {
|
|
typedef std::map<std::string,std::string> annotations_t;
|
|
annotations_t::const_iterator it = a_data.m_annotations.find(a_key);
|
|
if(it==a_data.m_annotations.end()) return std::string();
|
|
return (*it).second;
|
|
}
|
|
|
|
inline bool TH_write_1D(buffer& a_buffer,
|
|
const hd_data& a_data,
|
|
const std::string& a_name) {
|
|
|
|
if(!a_buffer.write_version(3)) return false;
|
|
|
|
if(!Named_stream(a_buffer,a_name,a_data.m_title)) return false;
|
|
|
|
if(!AttLine_stream(a_buffer)) return false;
|
|
if(!AttFill_stream(a_buffer)) return false;
|
|
if(!AttMarker_stream(a_buffer)) return false;
|
|
|
|
if(!a_buffer.write((int)a_data.m_bin_number)) return false;
|
|
|
|
//fXAxis
|
|
if(a_data.m_dimension==2) {
|
|
|
|
{histo::axis<double> haxis(a_data.m_axes[0]);
|
|
if(!axis_stream(a_buffer,haxis,"xaxis",
|
|
axis_title(a_data,histo::key_axis_x_title())))
|
|
return false;}
|
|
|
|
{histo::axis<double> haxis(a_data.m_axes[1]);
|
|
if(!axis_stream(a_buffer,haxis,"yaxis",
|
|
axis_title(a_data,histo::key_axis_y_title())))
|
|
return false;}
|
|
|
|
{histo::axis<double> dummy;
|
|
dummy.configure(1,0,1);
|
|
if(!axis_stream(a_buffer,dummy,"zaxis",
|
|
axis_title(a_data,histo::key_axis_z_title())))
|
|
return false;}
|
|
|
|
} else if(a_data.m_dimension==1) {
|
|
|
|
{histo::axis<double> haxis(a_data.m_axes[0]);
|
|
if(!axis_stream(a_buffer,haxis,"xaxis",
|
|
axis_title(a_data,histo::key_axis_x_title())))
|
|
return false;}
|
|
|
|
{histo::axis<double> dummy;
|
|
dummy.configure(1,0,1);
|
|
if(!axis_stream(a_buffer,dummy,"yaxis",
|
|
axis_title(a_data,histo::key_axis_y_title())))
|
|
return false;}
|
|
|
|
{histo::axis<double> dummy;
|
|
dummy.configure(1,0,1);
|
|
if(!axis_stream(a_buffer,dummy,"zaxis",
|
|
axis_title(a_data,histo::key_axis_z_title())))
|
|
return false;}
|
|
|
|
} else {
|
|
return false;
|
|
}
|
|
|
|
if(!a_buffer.write((short)(1000 * 0.25))) return false; //fBarOffset
|
|
if(!a_buffer.write((short)(1000 * 0.5))) return false; //fBarWidth
|
|
|
|
if(!a_buffer.write((double)a_data.get_entries())) return false;
|
|
if(!a_buffer.write(a_data.get_Sw())) return false;
|
|
if(!a_buffer.write(a_data.get_Sw2())) return false;
|
|
|
|
{double value;
|
|
a_data.get_ith_axis_Sxw(0,value);
|
|
if(!a_buffer.write(value)) return false;}
|
|
|
|
{double value;
|
|
a_data.get_ith_axis_Sx2w(0,value);
|
|
if(!a_buffer.write(value)) return false;}
|
|
|
|
if(!a_buffer.write((double)-1111)) return false; //fMaximum
|
|
if(!a_buffer.write((double)-1111)) return false; //fMinimum
|
|
if(!a_buffer.write((double)0)) return false; //NormFactor
|
|
|
|
if(!a_buffer.write_array(std::vector<double>()))
|
|
return false; //fContour TArrayD
|
|
|
|
if(!a_buffer.write_array(a_data.m_bin_Sw2)) return false; //fSumw2 TArrayD
|
|
|
|
// store annotation on fOption
|
|
// but try to fool ROOT in order that it does not
|
|
// understand fOption as.. ROOT options !
|
|
//{std::string opt = " "+fAnnotation;
|
|
// opt[0] = 0; //awfull trick
|
|
// if(!a_buffer.write(opt)) return false;} //TString fOption
|
|
{std::string opt;
|
|
if(!a_buffer.write(opt)) return false;} //TString fOption
|
|
|
|
if(!List_empty_stream(a_buffer)) return false; //Functions
|
|
|
|
return true;
|
|
}
|
|
|
|
inline bool TH_write_2D(buffer& a_buffer,
|
|
const hd_data& a_data,
|
|
const std::string& a_name) {
|
|
if(!a_buffer.write_version(3)) return false;
|
|
if(!TH_write_1D(a_buffer,a_data,a_name)) return false;
|
|
if(!a_buffer.write((double)1)) return false; //ScaleFactor
|
|
|
|
{double value;
|
|
a_data.get_ith_axis_Sxw(1,value);
|
|
if(!a_buffer.write(value)) return false;}
|
|
|
|
{double value;
|
|
a_data.get_ith_axis_Sx2w(1,value);
|
|
if(!a_buffer.write(value)) return false;}
|
|
|
|
if(!a_buffer.write((double)0)) return false; //Tsumwxy //FIXME.
|
|
|
|
return true;
|
|
}
|
|
|
|
inline bool TH1F_stream(buffer& a_buffer,
|
|
const histo::h1d& a_h,
|
|
const std::string& a_name) {
|
|
if(!a_buffer.write_version(1)) return false;
|
|
hd_data data = a_h.get_histo_data();
|
|
if(!TH_write_1D(a_buffer,data,a_name)) return false;
|
|
if(!a_buffer.write_array(convert<double,float>(data.m_bin_Sw)))
|
|
return false; //TH1D::TArrayD::fArray
|
|
return true;
|
|
}
|
|
|
|
inline bool TH1D_stream(buffer& a_buffer,
|
|
const histo::h1d& a_h,
|
|
const std::string& a_name) {
|
|
if(!a_buffer.write_version(1)) return false;
|
|
hd_data data = a_h.get_histo_data();
|
|
if(!TH_write_1D(a_buffer,data,a_name)) return false;
|
|
if(!a_buffer.write_array(data.m_bin_Sw)) return false; //fArray TArrayD
|
|
return true;
|
|
}
|
|
|
|
inline bool TH2F_stream(buffer& a_buffer,
|
|
const histo::h2d& a_h,
|
|
const std::string& a_name){
|
|
if(!a_buffer.write_version(3)) return false;
|
|
hd_data data = a_h.get_histo_data();
|
|
if(!TH_write_2D(a_buffer,data,a_name)) return false;
|
|
if(!a_buffer.write_array(convert<double,float>(data.m_bin_Sw)))
|
|
return false; //fArray TArrayD
|
|
return true;
|
|
}
|
|
|
|
inline bool TH2D_stream(buffer& a_buffer,
|
|
const histo::h2d& a_h,
|
|
const std::string& a_name){
|
|
if(!a_buffer.write_version(3)) return false;
|
|
hd_data data = a_h.get_histo_data();
|
|
if(!TH_write_2D(a_buffer,data,a_name)) return false;
|
|
if(!a_buffer.write_array(data.m_bin_Sw)) return false; //fArray TArrayD
|
|
return true;
|
|
}
|
|
|
|
inline bool TProfile_stream(buffer& a_buffer,
|
|
const histo::p1d& a_p,
|
|
const std::string& a_name){
|
|
if(!a_buffer.write_version(3)) return false;
|
|
|
|
//WARNING : the mapping histo::p1d / TProfile is not obvious.
|
|
//HCL::m_bin_Svw <---> TProfile::fArray
|
|
//HCL::m_bin_Sv2w <---> TProfile::fSumw2
|
|
//HCL::m_bin_Sw <---> TProfile::fBinEntries
|
|
|
|
pd_data data = a_p.get_histo_data();
|
|
//NOTE : histo.m_bin_Sw <---> TH1D::TArrayD::fArray
|
|
// then have to copy histo.m_bin_Svw on histo.m_bin_Sw
|
|
// before streaming.
|
|
|
|
//save histo.m_bin_Sw on a ::Rio::Array_double :
|
|
std::vector<double> vbins;
|
|
{unsigned int binn = data.m_bin_number;
|
|
vbins.resize(binn);
|
|
for(unsigned int index=0;index<binn;index++){
|
|
vbins[index] = data.m_bin_Sw[index];
|
|
}}
|
|
|
|
// copy histo.m_bin_Svw on histo.m_bin_Sw :
|
|
data.m_bin_Sw = data.m_bin_Svw;
|
|
|
|
//WARNING : should create a valid Rio::TH1D::fHistogram
|
|
// being a histo::h1d.
|
|
// But the histo::p1d does not inherit histo::h1d.
|
|
|
|
histo::h1d h("",10,0,1);
|
|
h.copy_from_data(data);
|
|
h.update_fast_getters();
|
|
if(!TH1D_stream(a_buffer,h,a_name)) return false;
|
|
|
|
//TProfile specific :
|
|
if(!a_buffer.write_array(vbins)) return false; //fBinEntries TArrayD
|
|
|
|
int errorMode = 0;
|
|
if(!a_buffer.write(errorMode)) return false;
|
|
if(!a_buffer.write(data.m_min_v)) return false;
|
|
if(!a_buffer.write(data.m_max_v)) return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
}}
|
|
|
|
#endif
|