dataset.h

Go to the documentation of this file.

/* emacs: this is -*- c++ -*- */
 
 
#ifndef ANALYSIS_DATASET_H
#define ANALYSIS_DATASET_H
 
#include <cstdio>
#include <cstdlib>
 
#include <dirent.h>
 
#include <string>
#include <iostream>
#include <fstream>
#include <vector>
 
// #include <XrdPosix/XrdPosixXrootd.hh>
 
class dataset : public std::vector<std::string> {
  
public:
  
  dataset(const std::string& st) : m_directory(st) {
    std::cout << "dataset::dataset() reading files from " << st << std::endl;
 
    std::string s = st;
 
    if ( s.find("root://eos")!=std::string::npos ) { 

      //      std::string  xcmd = "/afs/cern.ch/project/eos/installation/0.3.84-aquamarine/bin/eos.select ls "+s+"/";
      size_t pos=s.find("//");
      while( pos!=std::string::npos ) { 
    s = s.substr(pos+1,s.length()); 
    pos=s.find("//");
      }
      //      std::string  xcmd = "eos ls "+bases+"/";
      std::string  xcmd = "eos ls "+s+"/";
      std::system( (xcmd+" > .eosfiles-$$.log").c_str() );
 
      std::string   cmd = "export EOSFILES=$("+xcmd+")";
      std::system(  cmd.c_str() );
 
      std::ifstream infile(".eosfiles-$$.log");
 
      while( !infile.fail() ) {
    std::string file;
    infile >>  file;
    if ( !infile.fail() && 
         file.find(".root")!=std::string::npos && 
         file.find(".part")==std::string::npos &&
         file.find(".bak")==std::string::npos &&
         file.find(".skimmed")==std::string::npos ) push_back( s+"/"+file );
      }
 
      //      for ( unsigned i=size() ; i-- ; ) std::cout << at(i) << std::endl;

      //      std::string files = std::getenv("EOSFILES");
      //      std::cout << "files " << files << std::endl;
 
    }
    else { 
      DIR *dirp;
      struct dirent *entry;
 
      if( ( dirp = opendir(s.c_str()) ) )  {
    while( ( entry = readdir(dirp) ) ) {
      std::string file = entry->d_name;
      if ( file.find(".root")!=std::string::npos ) { 
        push_back( s+"/"+file );
        //   std::cout << " read file: " << entry->d_name << std::endl;
      }
    }
    closedir(dirp);
      }   
      std::cout << "dataset::dataset() " << size() << " files in dataset" << std::endl;
    } 
  }
 
  ~dataset() { } 
 
  const std::string& directory() const { return m_directory; }
 
  const std::vector<std::string>& datafiles() { return *this; } 
 
private: 
 
  std::string m_directory;
 
};
 
inline std::ostream& operator<<( std::ostream& s, const dataset& ds ) {
  return s << ds.directory() << " " << ds.size();
}
 
 
#endif  // ANALYSIS_DATASET_H