Warning, /include/Geant4/tools/rroot/directory is written in an unsupported language. File is not indexed.
0001 // Copyright (C) 2010, Guy Barrand. All rights reserved.
0002 // See the file tools.license for terms.
0003
0004 #ifndef tools_rroot_directory
0005 #define tools_rroot_directory
0006
0007 #include "seek"
0008 #include "date"
0009 #include "key"
0010 #include "ifile"
0011 #include "../forit"
0012 #include "../vmanip"
0013
0014 namespace tools {
0015 namespace rroot {
0016
0017 class directory {
0018 public:
0019 static const std::string& s_class() {
0020 static const std::string s_v("tools::rroot::directory");
0021 return s_v;
0022 }
0023 public:
0024 directory(ifile& a_file)
0025 :m_file(a_file)
0026 {
0027 m_date_C = 0; //.set();
0028 m_date_M = 0; //.set();
0029 m_nbytes_keys = 0;
0030 m_nbytes_name = 0;
0031 m_seek_directory = 0;
0032 m_seek_parent = 0;
0033 m_seek_keys = 0;
0034 }
0035 virtual ~directory(){
0036 clear_keys();
0037 }
0038 protected:
0039 directory(const directory& a_from):m_file(a_from.m_file){
0040 }
0041 directory& operator=(const directory &){return *this;}
0042 public:
0043 ifile& file() {return m_file;}
0044
0045 const std::vector<key*>& keys() const {return m_keys;}
0046 std::vector<key*>& keys() {return m_keys;}
0047
0048 key* find_key(const std::string& a_name) {
0049 if(m_file.verbose()) {
0050 m_file.out() << "tools::rroot::directory::find_key :"
0051 << " " << sout(a_name) << " ..."
0052 << std::endl;
0053 }
0054 tools_vforcit(key*,m_keys,it) {
0055 if((*it)->object_name()==a_name) return *it;
0056 }
0057
0058 return 0;
0059 }
0060
0061 key* find_key_from_class(const std::string& a_class) {
0062 if(m_file.verbose()) {
0063 m_file.out() << "tools::rroot::directory::find_key_from_class :"
0064 << " " << sout(a_class) << " ..."
0065 << std::endl;
0066 }
0067 tools_vforcit(key*,m_keys,it) {
0068 if((*it)->object_class()==a_class) return *it;
0069 }
0070
0071 return 0;
0072 }
0073
0074 uint32 nbytes_name() const {return m_nbytes_name;}
0075 seek seek_keys() const {return m_seek_keys;}
0076 uint32 record_size(uint32 a_version) const {
0077 uint32 nbytes = sizeof(short);
0078 nbytes += sizeof(date); //m_date_C.record_size();
0079 nbytes += sizeof(date); //m_date_M.record_size();
0080 nbytes += sizeof(m_nbytes_keys);
0081 nbytes += sizeof(m_nbytes_name);
0082 if(a_version>=40000) {
0083 nbytes += sizeof(seek);
0084 nbytes += sizeof(seek);
0085 nbytes += sizeof(seek);
0086 } else {
0087 nbytes += sizeof(seek32);
0088 nbytes += sizeof(seek32);
0089 nbytes += sizeof(seek32);
0090 }
0091 return nbytes;
0092 }
0093 bool from_buffer(const char* aEOB,char*& a_buffer){
0094 // Decode input buffer.
0095 // (Name, title) are stored in the (name, title) of the associated key.
0096 rbuf rb(m_file.out(),m_file.byte_swap(),aEOB,a_buffer);
0097 short versiondir;
0098 if(!rb.read(versiondir)) return false;
0099 unsigned int _date;
0100 if(!rb.read(_date)) return false;
0101 if(!rb.read(_date)) return false;
0102 {int v;
0103 if(!rb.read(v)) return false;
0104 m_nbytes_keys = v;}
0105 {int v;
0106 if(!rb.read(v)) return false;
0107 m_nbytes_name = v;}
0108 if(versiondir>(short)big_file_version_tag()) {
0109 if(!rb.read(m_seek_directory)) return false;
0110 if(!rb.read(m_seek_parent)) return false;
0111 if(!rb.read(m_seek_keys)) return false;
0112 } else {
0113 {seek32 i;
0114 if(!rb.read(i)) return false;
0115 m_seek_directory = i;}
0116
0117 {seek32 i;
0118 if(!rb.read(i)) return false;
0119 m_seek_parent = i;}
0120
0121 {seek32 i;
0122 if(!rb.read(i)) return false;
0123 m_seek_keys = i;}
0124 }
0125 if(m_file.verbose()) {
0126 m_file.out() << "tools::rroot::key::from_buffer :"
0127 << " nbytes keys : " << m_nbytes_keys
0128 << ", pos keys : " << m_seek_keys
0129 << std::endl;
0130 }
0131 return true;
0132 }
0133 bool read_keys(uint32& a_number) {
0134 // Read the KEYS :
0135 // Every directory has a list (fKeys). This list has been
0136 // written on the file via CERN-ROOT::TDirectory::writeKeys
0137 // as a single data record.
0138 a_number = 0;
0139
0140 clear_keys();
0141
0142 key headerkey(m_file.out(),m_seek_keys,m_nbytes_keys);
0143 if(!headerkey.read_file(m_file)) return false;
0144 char* buffer = headerkey.data_buffer();
0145 if(!headerkey.from_buffer(m_file.byte_swap(),headerkey.eob(),buffer,m_file.verbose())) return false;
0146 int nkeys = 0;
0147 rbuf rb(m_file.out(),m_file.byte_swap(),headerkey.eob(),buffer);
0148 if(!rb.read(nkeys)) return false;
0149 if(m_file.verbose()) {
0150 m_file.out() << "tools::rroot::directory::read_keys :"
0151 << " nkeys " << nkeys
0152 << "."
0153 << std::endl;
0154 }
0155 for(int i=0;i<nkeys;i++) {
0156 key* k = new key(m_file.out());
0157 if(!k->from_buffer(m_file.byte_swap(),headerkey.eob(),buffer,m_file.verbose())) {
0158 delete k;
0159 return false;
0160 }
0161 m_keys.push_back(k);
0162 }
0163 a_number = nkeys;
0164 return true;
0165 }
0166 void clear_keys() {
0167 safe_clear<key>(m_keys);
0168 }
0169 protected:
0170 ifile& m_file;
0171 std::vector<key*> m_keys;
0172 // Record (stored in file):
0173 date m_date_C; //Date and time when directory is created
0174 date m_date_M; //Date and time of last modification
0175 uint32 m_nbytes_keys; //Number of bytes for the keys
0176 uint32 m_nbytes_name; //Number of bytes in TNamed at creation time
0177 seek m_seek_directory; //Location of directory on file
0178 seek m_seek_parent; //Location of parent directory on file
0179 seek m_seek_keys; //Location of Keys record on file
0180 };
0181
0182 }}
0183
0184 #endif