ana::SAMProjectSource Class Reference

Fetch files from a pre-existing SAM project. More...

#include <SAMProjectSource.h>

Inheritance diagram for ana::SAMProjectSource:

Public Member Functions
	SAMProjectSource (const std::string &proj, int fileLimit=-1)
virtual	~SAMProjectSource ()
virtual TFile *	GetNextFile () override
	Returns the next file in sequence, ready for reading. More...
int	NFiles () const override
	May return -1 indicating the number of files is not known. More...
Public Member Functions inherited from ana::IFileSource
virtual	~IFileSource ()

Protected Attributes
ifdh_ns::ifdh *	fIFDH
std::string	fProjectURL
std::string	fProcessID
TFile *	fFile
	The most-recently-returned file. More...
int	fNFiles

Static Protected Attributes
static bool	fgGotTickets = false

Detailed Description

Fetch files from a pre-existing SAM project.

Definition at line 14 of file SAMProjectSource.h.

Constructor & Destructor Documentation

ana::SAMProjectSource::SAMProjectSource	(	const std::string &	proj,
		int	fileLimit = -1
	)

Parameters

proj	SAM project name (not URL)
fileLimit	Optional maximum number of files to process

Definition at line 20 of file SAMProjectSource.cxx.

    : fIFDH(new ifdh_ns::ifdh)
    , fFile(0)
    , fNFiles(fileLimit)
  {
    if(!fgGotTickets){
      // No kerberos ticket means no point trying to voms-proxy-init. It likely
      // also means we're in a grid job, where that would be counterproductive
      // anyway.
      if(system("klist -5 -s || klist -s") != 0) fgGotTickets = true;
    }

    if(!fgGotTickets){
      // This comes from NovaGridUtils, v02.10 onwards.
      system("setup_fnal_security -b");

      fgGotTickets = true;
    }

    // If X509_USER_PROXY isn't set, set it manually. Can help unconfuse IFDH
    // about what x509 it's supposed to be using.
    if(getenv("X509_USER_PROXY") == 0)
      setenv("X509_USER_PROXY", TString::Format("/tmp/x509up_u%d", getuid()).Data(), 0);

    fIFDH->set_debug("0"); // shut up

    fProjectURL = fIFDH->findProject(proj, SAMExperiment());

    fProcessID = fIFDH->establishProcess(fProjectURL, "CAFAna", "v0.9", getenv("HOSTNAME"), getenv("USER"), SAMExperiment(), "", fileLimit);
  }

ana::SAMProjectSource::~SAMProjectSource ( )

virtual

Definition at line 52 of file SAMProjectSource.cxx.

  {
    if(fFile){
      // Tidy up the final file
      const std::string fname = fFile->GetName();
      delete fFile;
      unlink(fname.c_str());
    }

    // End the process cleanly
    fIFDH->endProcess(fProjectURL, fProcessID);

    // certainly wrong for fileLimit case
    // status = fIFDH.endProject(fProjectURL);

    fIFDH->cleanup();
  }

Member Function Documentation

TFile * ana::SAMProjectSource::GetNextFile ( )

overridevirtual

Returns the next file in sequence, ready for reading.

A null return means that the end of the sequence has been reached. DO NOT close or delete the file that is returned.

Implements ana::IFileSource.

Definition at line 71 of file SAMProjectSource.cxx.

  {
    if(fFile){
      // Tidy up the previous file
      const std::string fname = fFile->GetName();
      delete fFile;
      fFile = 0;
      unlink(fname.c_str());

      // And let SAM know we're done with it
      fIFDH->updateFileStatus(fProjectURL, fProcessID, fname, "consumed");
    }

    std::string tmp;

    tmp = fIFDH->getNextFile(fProjectURL, fProcessID);
    const std::string uri(tmp);
    if(uri.empty()) return 0; // out of files

    tmp = fIFDH->fetchInput(uri);
    const std::string fname(tmp);
    assert(!fname.empty());

    // Let SAM know we got it OK
    fIFDH->updateFileStatus(fProjectURL, fProcessID, fname, "transferred");

    // Additional newlines because ifdh currently spams us with certificate
    // messages.
    if(fNFiles < 0) std::cout << std::endl << "Processing " << basename((char *)fname.c_str()) << std::endl << std::endl;

    fFile = new TFile(fname.c_str());
    return fFile;
  }

int ana::SAMProjectSource::NFiles ( ) const

inlineoverridevirtual

May return -1 indicating the number of files is not known.

Reimplemented from ana::IFileSource.

Definition at line 24 of file SAMProjectSource.h.

{return fNFiles;}

Member Data Documentation

TFile* ana::SAMProjectSource::fFile

protected

The most-recently-returned file.

Definition at line 31 of file SAMProjectSource.h.

bool ana::SAMProjectSource::fgGotTickets = false

staticprotected

Definition at line 35 of file SAMProjectSource.h.

ifdh_ns::ifdh* ana::SAMProjectSource::fIFDH

protected

Definition at line 26 of file SAMProjectSource.h.

int ana::SAMProjectSource::fNFiles

protected

Definition at line 33 of file SAMProjectSource.h.

std::string ana::SAMProjectSource::fProcessID

protected

Definition at line 29 of file SAMProjectSource.h.

std::string ana::SAMProjectSource::fProjectURL

protected

Definition at line 28 of file SAMProjectSource.h.

---

The documentation for this class was generated from the following files:

• SAMProjectSource.h
• SAMProjectSource.cxx