Import Geant4 11.0.0.beta source tree
This commit is contained in:
Gabriele Cosmo
+249
@@ -0,0 +1,249 @@
|
||||
// Copyright (C) 2010, Guy Barrand. All rights reserved.
|
||||
// See the file tools.license for terms.
|
||||
|
||||
#ifndef tools_wroot_tree
|
||||
#define tools_wroot_tree
|
||||
|
||||
#include "itree"
|
||||
#include "iobject"
|
||||
#include "idir"
|
||||
|
||||
#include "branch_element"
|
||||
#include "branch_object"
|
||||
|
||||
namespace tools {
|
||||
namespace wroot {
|
||||
|
||||
class tree : public virtual iobject, public virtual itree {
|
||||
#ifdef TOOLS_MEM
|
||||
static const std::string& s_class() {
|
||||
static const std::string s_v("tools::wroot::tree");
|
||||
return s_v;
|
||||
}
|
||||
#endif
|
||||
public: //iobject
|
||||
virtual const std::string& name() const {return m_name;}
|
||||
virtual const std::string& title() const {return m_title;}
|
||||
virtual const std::string& store_class_name() const {
|
||||
static const std::string s_v("TTree");
|
||||
return s_v;
|
||||
}
|
||||
virtual bool stream(buffer& a_buffer) const {
|
||||
unsigned int c;
|
||||
if(!a_buffer.write_version(5,c)) return false;
|
||||
|
||||
if(!Named_stream(a_buffer,m_name,m_title)) return false;
|
||||
|
||||
// Beurk.
|
||||
if(!AttLine_stream(a_buffer)) return false;
|
||||
if(!AttFill_stream(a_buffer)) return false;
|
||||
if(!AttMarker_stream(a_buffer)) return false;
|
||||
|
||||
double fEntries = (double)m_entries;
|
||||
if(!a_buffer.write(fEntries)) return false;
|
||||
|
||||
double fTotBytes = (double)m_tot_bytes;
|
||||
double fZipBytes = (double)m_zip_bytes;
|
||||
if(!a_buffer.write(fTotBytes)) return false;
|
||||
if(!a_buffer.write(fZipBytes)) return false;
|
||||
if(!a_buffer.write((double)0)) return false; //fSavedBytes
|
||||
if(!a_buffer.write((int)0)) return false; //fTimerInterval
|
||||
if(!a_buffer.write((int)25)) return false; //fScanField (25)
|
||||
if(!a_buffer.write((int)0)) return false; //fUpdate
|
||||
if(!a_buffer.write((int)1000000000)) return false; //fMaxEntryLoop
|
||||
int fMaxVirtualSize = 0;
|
||||
int fAutoSave = 100000000;
|
||||
if(!a_buffer.write(fMaxVirtualSize)) return false;
|
||||
if(!a_buffer.write(fAutoSave)) return false;
|
||||
if(!a_buffer.write((int)1000000)) return false; //fEstimate;
|
||||
|
||||
if(!m_branches.stream(a_buffer)) return false;
|
||||
|
||||
{obj_array<base_leaf> m_leaves;
|
||||
tools_vforcit(branch*,m_branches,itb) {
|
||||
const std::vector<base_leaf*>& leaves = (*itb)->leaves();
|
||||
tools_vforcit(base_leaf*,leaves,itl) {
|
||||
m_leaves.push_back(*itl); //WARNING : ownership touchy.
|
||||
}
|
||||
}
|
||||
if(!m_leaves.stream(a_buffer)) return false;
|
||||
m_leaves.clear();} //WARNING : important.
|
||||
|
||||
// fIndexValues (TArrayD).
|
||||
if(!a_buffer.write_array(std::vector<double>())) return false; //TArrayD
|
||||
// fIndex (TArrayI).
|
||||
if(!a_buffer.write_array(std::vector<int>())) return false; //TArrayI
|
||||
|
||||
if(!a_buffer.set_byte_count(c)) return false;
|
||||
return true;
|
||||
}
|
||||
public: //itree
|
||||
//virtual void add_tot_bytes(uint32 a_n) {m_tot_bytes += a_n;}
|
||||
//virtual void add_zip_bytes(uint32 a_n) {m_zip_bytes += a_n;}
|
||||
virtual idir& dir() {return m_dir;}
|
||||
virtual const idir& dir() const {return m_dir;}
|
||||
public:
|
||||
tree(idir& a_dir,const std::string& a_name,const std::string& a_title,bool a_managed = true)
|
||||
:m_dir(a_dir)
|
||||
,m_out(a_dir.file().out())
|
||||
,m_name(a_name)
|
||||
,m_title(a_title)
|
||||
,m_entries(0)
|
||||
,m_tot_bytes(0)
|
||||
,m_zip_bytes(0)
|
||||
{
|
||||
#ifdef TOOLS_MEM
|
||||
mem::increment(s_class().c_str());
|
||||
#endif
|
||||
if(a_managed) a_dir.append_object(this); //a_dir takes ownership of tree.
|
||||
}
|
||||
virtual ~tree(){
|
||||
#ifdef TOOLS_MEM
|
||||
mem::decrement(s_class().c_str());
|
||||
#endif
|
||||
}
|
||||
protected:
|
||||
tree(const tree& a_from)
|
||||
:iobject(a_from),itree(a_from)
|
||||
,m_dir(a_from.m_dir)
|
||||
,m_out(a_from.m_out)
|
||||
{}
|
||||
tree& operator=(const tree&){return *this;}
|
||||
public:
|
||||
std::ostream& out() const {return m_out;}
|
||||
const std::vector<branch*>& branches() const {return m_branches;}
|
||||
//uint64 tot_bytes() const {return m_tot_bytes;}
|
||||
//uint64 zip_bytes() const {return m_zip_bytes;}
|
||||
uint64 entries() const {return m_entries;}
|
||||
|
||||
branch* create_branch(const std::string& a_name){
|
||||
const ifile& _file = m_dir.file();
|
||||
branch* br = new branch(m_out,_file.byte_swap(),_file.compression(),
|
||||
m_dir.seek_directory(),a_name,m_name,_file.verbose());
|
||||
if(!br) return 0;
|
||||
m_branches.push_back(br);
|
||||
return br;
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////
|
||||
/// ref : //////////////////////////////////////
|
||||
////////////////////////////////////////////////
|
||||
template <class TYPE>
|
||||
leaf_ref<TYPE>* create_leaf_ref(const std::string& a_name,const TYPE& a_ref){
|
||||
branch* br = create_branch(a_name);
|
||||
if(!br) return 0;
|
||||
return br->create_leaf_ref<TYPE>(a_name,a_ref);
|
||||
}
|
||||
|
||||
leaf_string_ref* create_leaf_string_ref(const std::string& a_name,const std::string& a_ref){
|
||||
branch* br = create_branch(a_name);
|
||||
if(!br) return 0;
|
||||
return br->create_leaf_string_ref(a_name,a_ref);
|
||||
}
|
||||
|
||||
template <class T>
|
||||
leaf_element* create_std_vector_leaf_ref(const std::string& a_name,const std::vector<T>& a_ref){
|
||||
const ifile& _file = m_dir.file();
|
||||
std_vector_be_ref<T>* br = new std_vector_be_ref<T>(m_out,_file.byte_swap(),_file.compression(),
|
||||
m_dir.seek_directory(),a_name,m_name,a_ref,_file.verbose());
|
||||
leaf_element* le = br->create_leaf_element(a_name);
|
||||
m_branches.push_back(br);
|
||||
return le;
|
||||
}
|
||||
////////////////////////////////////////////////
|
||||
////////////////////////////////////////////////
|
||||
////////////////////////////////////////////////
|
||||
template <class T>
|
||||
std_vector_be_ref<T>* create_std_vector_be_ref(const std::string& a_name,const std::vector<T>& a_ref){
|
||||
const ifile& _file = m_dir.file();
|
||||
std_vector_be_ref<T>* br = new std_vector_be_ref<T>(m_out,_file.byte_swap(),_file.compression(),
|
||||
m_dir.seek_directory(),a_name,m_name,a_ref,_file.verbose());
|
||||
m_branches.push_back(br);
|
||||
return br;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
std_vector_be<T>* create_std_vector_be(const std::string& a_name,const std::vector<T>& a_def = std::vector<T>()){
|
||||
const ifile& _file = m_dir.file();
|
||||
std_vector_be<T>* br = new std_vector_be<T>(m_out,_file.byte_swap(),_file.compression(),
|
||||
m_dir.seek_directory(),a_name,m_name,a_def,_file.verbose());
|
||||
m_branches.push_back(br);
|
||||
return br;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
std_vector_be_pointer<T>* create_std_vector_be_pointer(const std::string& a_name,std::vector<T>* a_pointer){
|
||||
const ifile& _file = m_dir.file();
|
||||
std_vector_be_pointer<T>* br = new std_vector_be_pointer<T>(m_out,_file.byte_swap(),_file.compression(),
|
||||
m_dir.seek_directory(),a_name,m_name,a_pointer,_file.verbose());
|
||||
m_branches.push_back(br);
|
||||
return br;
|
||||
}
|
||||
|
||||
template <class TYPE>
|
||||
leaf<TYPE>* create_leaf(const std::string& a_name){
|
||||
branch* br = create_branch(a_name);
|
||||
if(!br) return 0;
|
||||
return br->create_leaf<TYPE>(a_name);
|
||||
}
|
||||
|
||||
leaf_object* create_leaf(const std::string& a_name,const iobject& a_obj){
|
||||
const ifile& _file = m_dir.file();
|
||||
branch_object* br = new branch_object(m_out,_file.byte_swap(),_file.compression(),
|
||||
m_dir.seek_directory(),a_name,m_name,_file.verbose());
|
||||
m_branches.push_back(br);
|
||||
return br->create_leaf(a_name,a_obj);
|
||||
}
|
||||
|
||||
bool fill(uint32& a_nbytes) {
|
||||
// Fill all branches of a Tree :
|
||||
// This function loops on all the branches of this tree.
|
||||
// For each branch, it copies to the branch buffer (basket) the current
|
||||
// values of the leaves data types.
|
||||
// If a leaf is a simple data type, a simple conversion to a machine
|
||||
// independent format has to be done.
|
||||
a_nbytes = 0;
|
||||
tools_vforcit(branch*,m_branches,it) {
|
||||
//FIXME if ((*it)->testBit(kDoNotProcess)) continue;
|
||||
uint32 n,add_bytes,nout;
|
||||
if(!(*it)->fill(m_dir.file(),n,add_bytes,nout)) {a_nbytes = 0;return false;}
|
||||
a_nbytes += n;
|
||||
m_tot_bytes += add_bytes;
|
||||
m_zip_bytes += nout;
|
||||
}
|
||||
|
||||
m_entries++;
|
||||
|
||||
//if (fTotBytes - fSavedBytes > fAutoSave) {
|
||||
// if(!autoSave()) return false;
|
||||
//}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void reset() {
|
||||
// Reset buffers and entries count in all branches/leaves
|
||||
m_entries = 0;
|
||||
m_tot_bytes = 0;
|
||||
m_zip_bytes = 0;
|
||||
//fSavedBytes = 0;
|
||||
//fTotalBuffers = 0;
|
||||
//fChainOffset = 0;
|
||||
{tools_vforcit(branch*,m_branches,it) (*it)->reset();}
|
||||
}
|
||||
protected:
|
||||
idir& m_dir;
|
||||
std::ostream& m_out;
|
||||
//Named
|
||||
std::string m_name;
|
||||
std::string m_title;
|
||||
|
||||
obj_array<branch> m_branches;
|
||||
uint64 m_entries; // Number of entries
|
||||
uint64 m_tot_bytes; // Total number of bytes in branches before compression
|
||||
uint64 m_zip_bytes; // Total number of bytes in branches after compression
|
||||
};
|
||||
|
||||
}}
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user