/// helper class to make output HTML pages
#include "HTMLOutputGenerator.h"
/////////////////////////////////////////////////////////////////////////////
void HTMLOutputGenerator::write_html_header(ofstream &f, string thedir)
{
f << "" << endl
<< "" << endl
<< "" << endl
<< thedir << endl
<< "" << endl
<< "" << endl
<< "" << endl << endl;
string header_str = "";
vector dirs; // thedir.split("/");
istringstream thedir_ss( thedir );
while(!thedir_ss.eof()) {
string dir;
getline( thedir_ss, dir, '/' );
dirs.push_back( dir );
}
//cout << "printing out for " << thedir << " with "
// std::to_string(dirs.size()) << " elements" << endl;
for(unsigned int i=0; i" << endl << "" << endl;
}
void HTMLOutputGenerator::write_html_dirs(ofstream &f, vector &dirs)
{
f << "" << endl;
f << "- ../" << endl;
for( vector::const_iterator dir_it = dirs.begin();
dir_it != dirs.end(); dir_it++ ) {
f << "
- " + *dir_it + "/" << endl;
}
f << "
" << endl << endl;
}
void HTMLOutputGenerator::write_html_hists(ofstream &f, vector &hists)
{
for( vector::const_iterator hist_it = hists.begin();
hist_it != hists.end(); hist_it++ ) {
string hist_file = *hist_it + "." + IMG_SUFFIX;
f << "![\""](\"")
" << endl;
}
}
//static bool do_mkdir(const char *path, mode_t mode)
bool HTMLOutputGenerator::do_mkdir(string path, mode_t mode)
{
struct stat st;
bool status = true;
if (stat(path.c_str(), &st) != 0) {
// Directory does not exist. EEXIST for race condition
if (mkdir(path.c_str(), mode) != 0 && errno != EEXIST)
status = false;
}
else if (!S_ISDIR(st.st_mode)) {
errno = ENOTDIR;
status = false;
}
return status;
}
/////////////////////////////////////////////////////////////////////////////
// this should be recursive?
void HTMLOutputGenerator::GenerateOutput(TDirectory *root_dir, string basedir, string subdir)
{
vector dirs;
vector hists;
cout << "In HTMLOutputGenerator::GenerateOutput()..." << endl
<< " in directory " << basedir + "/" + subdir + "/index.html" << endl;
ofstream html_file;
html_file.open( (basedir + "/" + subdir + "/index.html").c_str());
if(!html_file.is_open())
return;
write_html_header(html_file, subdir);
// collect information on histograms and directories
TIter next(root_dir->GetListOfKeys());
TObject *keyobj;
while ((keyobj = next())) {
TKey *key = (TKey *)keyobj;
//const char *name = key.GetName()
string name = key->GetName();
TObject *obj = root_dir->Get( key->GetName() ); // should we be extracted TNamed?
//string obj_type = obj->Class_Name();
//cout << name << " is an object of type " << obj_type << endl;
// handle directories
TDirectory *the_dir = dynamic_cast(obj);
if(the_dir) {
cout << name << " is an object of type TDirectory" << endl;
dirs.push_back( name );
if(!do_mkdir(basedir + "/" + subdir + "/" + name, 0755))
continue;
GenerateOutput(the_dir, basedir, subdir+"/"+name); // make sure this works
}
// handle histograms - 1D
TH1 *the_hist = dynamic_cast(obj);
if(the_hist) {
cout << name << " is an object of type TH1" << endl;
hists.push_back( name );
the_hist->Draw();
string img_filename = basedir + "/" + subdir + "/" + name + "." + IMG_SUFFIX;
output_canvas->Print(img_filename.c_str());
cout << "outputting image: " << img_filename << endl;
}
// handle histograms - 2D
TH2 *the_hist2d = dynamic_cast(obj);
if(the_hist2d) {
cout << name << " is an object of type TH2" << endl;
hists.push_back( name );
the_hist->Draw("COLZ");
string img_filename = basedir + "/" + subdir + "/" + name + "." + IMG_SUFFIX;
output_canvas->Print(img_filename.c_str());
cout << "outputting image: " << img_filename << endl;
}
}
// write out data to files
write_html_dirs(html_file, dirs);
write_html_hists(html_file, hists);
write_html_footer(html_file);
html_file.close();
//delete output_canvas;
}