1 |
amagnan |
1.1 |
#include <ostream>
|
2 |
|
|
#include <iostream>
|
3 |
|
|
#include <iomanip>
|
4 |
|
|
#include <stdlib.h>
|
5 |
|
|
|
6 |
|
|
#include "TROOT.h"
|
7 |
|
|
#include "TFile.h"
|
8 |
|
|
|
9 |
|
|
#include "UserCode/HbbAnalysis/interface/TFileDirectory.hh"
|
10 |
|
|
|
11 |
|
|
using namespace std;
|
12 |
|
|
|
13 |
|
|
namespace HbbAnalysis{
|
14 |
|
|
|
15 |
|
|
TFileDirectory::TFileDirectory(TFile * file,
|
16 |
|
|
const std::string & path) :
|
17 |
|
|
file_( file ), path_( path )
|
18 |
|
|
{
|
19 |
|
|
|
20 |
|
|
//std::cout << "[TFileDirectory] --- TFileDirectory::TFileDirectory fullPath = " << fullPath() << std::endl;
|
21 |
|
|
|
22 |
|
|
}
|
23 |
|
|
|
24 |
|
|
void TFileDirectory::cd() const {
|
25 |
|
|
string fpath = fullPath();
|
26 |
|
|
TDirectory * dir = file_->GetDirectory( fpath.c_str() );
|
27 |
|
|
if ( dir == 0 ) {
|
28 |
|
|
if ( ! path_.empty() ) {
|
29 |
|
|
dir = file_->GetDirectory( path_.c_str() );
|
30 |
|
|
if ( dir == 0 ){
|
31 |
|
|
std::cerr << "[TFileDirectory] --- cd() : Can't change directory to path: " << path_ << std::endl;
|
32 |
|
|
exit(1);
|
33 |
|
|
}
|
34 |
|
|
} else {
|
35 |
|
|
dir = file_;
|
36 |
|
|
}
|
37 |
|
|
dir = dir->mkdir( dir_.c_str(), descr_.c_str() );
|
38 |
|
|
if ( dir == 0 ){
|
39 |
|
|
std::cerr << "[TFileDirectory] --- cd() : Can't create directory " << dir_ << " in path: " << path_ << std::endl;
|
40 |
|
|
exit(1);
|
41 |
|
|
}
|
42 |
|
|
}
|
43 |
|
|
bool ok = file_->cd( fpath.c_str() );
|
44 |
|
|
if ( ! ok ){
|
45 |
|
|
std::cerr << "[TFileDirectory] --- cd() : Can't change directory to path: " << fpath << std::endl;
|
46 |
|
|
exit(1);
|
47 |
|
|
}
|
48 |
|
|
}
|
49 |
|
|
|
50 |
|
|
std::string TFileDirectory::fullPath() const {
|
51 |
|
|
return string( path_.empty() ? dir_ : path_ + "/" + dir_ );
|
52 |
|
|
}
|
53 |
|
|
|
54 |
|
|
TFileDirectory TFileDirectory::mkdir( const std::string & dir, const std::string & descr ) {
|
55 |
|
|
TH1AddDirectorySentry sentry;
|
56 |
|
|
cd();
|
57 |
|
|
//std::cout << "[TFileDirectory] --- mkdir: fullPath = " << fullPath() << std::endl;
|
58 |
|
|
return TFileDirectory( dir, descr, file_, fullPath() );
|
59 |
|
|
}
|
60 |
|
|
|
61 |
|
|
}//namespace
|