#include "/cvmfs/nova.opensciencegrid.org/externals/art/v2_13_00/source/art/Framework/Services/Optional/TFileService.h"

Inheritance diagram for art::TFileService:

Classes
struct	Config

Public Types
using	Parameters = ServiceTable< Config >

using	Callback_t = TFileDirectory::Callback_t

Public Member Functions
	TFileService (ServiceTable< Config > const &config, art::ActivityRegistry &r)

	~TFileService ()

void	registerFileSwitchCallback (Callback_t c)

template<typename T >
void	registerFileSwitchCallback (T provider, void(T::)())

TFile &	file () const

template<typename T , typename... ARGS>
T *	make (ARGS...args) const

template<typename T , typename... ARGS>
T *	makeAndRegister (char const name, char const title, ARGS...args) const

template<typename T , typename... ARGS>
T *	makeAndRegister (std::string const &name, std::string const &title, ARGS...args) const

TFileDirectory	mkdir (std::string const &dir, std::string const &descr="")

Static Public Attributes
static constexpr const char *	default_tmpDir = "<parent-path-of-filename>"

Protected Member Functions
void	invokeCallbacks ()

void	registerCallback (Callback_t)

std::string	fullPath () const

Protected Attributes
TFile *	file_

std::string	dir_

std::string	descr_

bool	requireCallback_ {false}

Private Member Functions
	TFileService (TFileService const &)=delete

TFileService	operator= (TFileService const &)=delete

void	setDirectoryName_ (art::ModuleDescription const &desc)

void	openFile_ ()

void	closeFile_ ()

void	maybeSwitchFiles_ ()

bool	requestsToCloseFile_ ()

std::string	fileNameAtOpen_ ()

std::string	fileNameAtClose_ (std::string const &)

Private Attributes
bool const	closeFileFast_

FileStatsCollector	fstats_

PostCloseFileRenamer	fRenamer_ {fstats_}

std::string	filePattern_

std::string	uniqueFilename_

std::string	tmpDir_

std::string	lastClosedFile_ {}

Granularity	currentGranularity_ {Granularity::Unset}

std::chrono::steady_clock::time_point	beginTime_ {}

ClosingCriteria	fileSwitchCriteria_ {ClosingCriteria::Config{}}

OutputFileStatus	status_ {OutputFileStatus::Closed}

FileProperties	fp_ {}

Detailed Description

Definition at line 31 of file TFileService.h.

Member Typedef Documentation

using art::TFileService::Callback_t = TFileDirectory::Callback_t

Definition at line 54 of file TFileService.h.

using art::TFileService::Parameters = ServiceTable<Config>

Definition at line 48 of file TFileService.h.

Constructor & Destructor Documentation

art::TFileService::TFileService ( TFileService const & )

privatedelete

TFileService::TFileService	(	ServiceTable< Config > const &	config,
		art::ActivityRegistry &	r
	)

Definition at line 37 of file TFileService_service.cc.

References closeFileFast_, gen_hdf5record::config, currentGranularity_, e, art::Granularity::Event, plotROC::fileName, filePattern_, fileSwitchCriteria_, fp_, fstats_, art::Run::id(), art::SubRun::id(), art::Event::id(), art::Granularity::InputFile, maybeSwitchFiles_(), openFile_(), r(), art::FileStatsCollector::recordEvent(), art::FileStatsCollector::recordInputFile(), art::FileStatsCollector::recordRun(), art::FileStatsCollector::recordSubRun(), requestsToCloseFile_(), art::TFileDirectory::requireCallback_, art::Granularity::Run, setDirectoryName_(), sr, status_, string, art::Granularity::SubRun, tmpDir_, art::FileProperties::update(), and art::detail::validateFileNamePattern().

   : TFileDirectory{"", "", nullptr, ""}
   , closeFileFast_{config().closeFileFast()}
   , fstats_{config.get_PSet().get<std::string>("service_type"),
             ServiceHandle<TriggerNamesService const>{}->getProcessName()}
   , filePattern_{config().fileName()}
   , tmpDir_{config().tmpDir()}
 {
   ClosingCriteria::Config fpConfig;
   requireCallback_ = config().fileProperties(fpConfig);
   if (requireCallback_) {
     detail::validateFileNamePattern(requireCallback_, filePattern_);
     fileSwitchCriteria_ = ClosingCriteria{fpConfig};
   }
   openFile_();
 
   // Activities to monitor in order to set the proper directory.
   r.sPostOpenFile.watch(
     [this](std::string const& fileName) { fstats_.recordInputFile(fileName); });
 
   r.sPreModuleBeginJob.watch(this, &TFileService::setDirectoryName_);
   r.sPreModuleEndJob.watch(this, &TFileService::setDirectoryName_);
 
   r.sPreModuleConstruction.watch(this, &TFileService::setDirectoryName_);
   r.sPreModuleRespondToOpenInputFile.watch(this,
                                            &TFileService::setDirectoryName_);
   r.sPreModuleRespondToCloseInputFile.watch(this,
                                             &TFileService::setDirectoryName_);
   r.sPreModuleRespondToOpenOutputFiles.watch(this,
                                              &TFileService::setDirectoryName_);
   r.sPreModuleRespondToCloseOutputFiles.watch(this,
                                               &TFileService::setDirectoryName_);
 
   r.sPreModuleBeginRun.watch(this, &TFileService::setDirectoryName_);
   r.sPreModuleEndRun.watch(this, &TFileService::setDirectoryName_);
 
   r.sPreModuleBeginSubRun.watch(this, &TFileService::setDirectoryName_);
   r.sPreModuleEndSubRun.watch(this, &TFileService::setDirectoryName_);
 
   r.sPreModule.watch(this, &TFileService::setDirectoryName_);
 
   // Activities to monitor to keep track of events, subruns and runs seen.
   r.sPostProcessEvent.watch([this](Event const& e) {
     currentGranularity_ = Granularity::Event;
     fp_.update<Granularity::Event>(status_);
     fstats_.recordEvent(e.id());
     if (requestsToCloseFile_()) {
       maybeSwitchFiles_();
     }
   });
   r.sPostEndSubRun.watch([this](SubRun const& sr) {
     currentGranularity_ = Granularity::SubRun;
     fp_.update<Granularity::SubRun>(status_);
     fstats_.recordSubRun(sr.id());
     if (requestsToCloseFile_()) {
       maybeSwitchFiles_();
     }
   });
   r.sPostEndRun.watch([this](Run const& r) {
     currentGranularity_ = Granularity::Run;
     fp_.update<Granularity::Run>(status_);
     fstats_.recordRun(r.id());
     if (requestsToCloseFile_()) {
       maybeSwitchFiles_();
     }
   });
   r.sPostCloseFile.watch([this] {
     currentGranularity_ = Granularity::InputFile;
     fp_.update<Granularity::InputFile>();
     if (requestsToCloseFile_()) {
       maybeSwitchFiles_();
     }
   });
 }

TFileService::~TFileService ( )

Definition at line 113 of file TFileService_service.cc.

References closeFile_().

 {
   closeFile_();
 }

Member Function Documentation

void TFileService::closeFile_ ( )

private

Definition at line 165 of file TFileService_service.cc.

References art::Closed, closeFileFast_, art::TFileDirectory::file_, fileNameAtClose_(), fstats_, lastClosedFile_, art::FileStatsCollector::recordFileClose(), status_, and uniqueFilename_.

Referenced by maybeSwitchFiles_(), and ~TFileService().

 {
   file_->Write();
   if (closeFileFast_) {
     gROOT->GetListOfFiles()->Remove(file_);
   }
   file_->Close();
   delete file_;
   file_ = nullptr;
   status_ = OutputFileStatus::Closed;
   fstats_.recordFileClose();
   lastClosedFile_ = fileNameAtClose_(uniqueFilename_);
 }

TFile& art::TFileService::file ( ) const

inline

Definition at line 63 of file TFileService.h.

Referenced by calib::BetheBlochFit::endJob(), and chaninfo::BadChanValidate::endSubRun().

   {
     return *file_;
   }

std::string TFileService::fileNameAtClose_ ( std::string const & filename )

private

Definition at line 146 of file TFileService_service.cc.

References filePattern_, fRenamer_, and art::PostCloseFileRenamer::maybeRenameFile().

Referenced by closeFile_().

 {
   return filePattern_ == dev_null ?
            dev_null :
            fRenamer_.maybeRenameFile(filename, filePattern_);
 }

std::string TFileService::fileNameAtOpen_ ( )

private

Definition at line 136 of file TFileService_service.cc.

References default_tmpDir, filePattern_, art::parent_path(), string, tmpDir_, and art::unique_filename().

Referenced by openFile_().

 {
   if (filePattern_ == dev_null)
     return dev_null;
   return unique_filename(
     (tmpDir_ == default_tmpDir ? parent_path(filePattern_) : tmpDir_) +
     "/TFileService");
 }

std::string art::TFileDirectory::fullPath ( ) const

protectedinherited

Return the full pathname of the current directory, formed from appending 'dir' to the end of 'path'.

void art::TFileDirectory::invokeCallbacks ( )

protectedinherited

Referenced by maybeSwitchFiles_().

template<typename T , typename... ARGS>

T * art::TFileDirectory::make ( ARGS... args ) const

inherited

make new ROOT object of type T, using constructor parameters args. It will be made in the 'current directory', as established with a call to 'cd'.

Definition at line 82 of file TFileDirectory.h.

References make_syst_table_plots::args, art::TFileDirectory::cd(), and T.

   {
     detail::RootDirectorySentry sentry;
     cd();
     return new T(args...);
   }

template<typename T , typename... ARGS>

T * art::TFileDirectory::makeAndRegister	(	char const *	name,
		char const *	title,
		ARGS...	args
	)		const

inherited

make and register a new ROOT object of type T, giving it the specified name and title, using constructor parameters 'args'. The type must be registerable, and must support naming and titling.

Definition at line 91 of file TFileDirectory.h.

References make_syst_table_plots::args, art::TFileDirectory::cd(), and T.

Referenced by tbana::DetectorRateShutOff::beginJob(), tbana::DSToFAna::DSToFAna(), cmf::PlotUtilities::Make1DPlot(), cmf::PlotUtilities::Make2DContours(), art::TFileDirectory::makeAndRegister(), cmf::ContourFromLibrary::MakeAndStorePlots(), and cmf::ContourFromGrid::MakeAndStorePlots().

   {
     detail::RootDirectorySentry sentry;
     cd();
     T* p = new T(args...);
     p->SetName(name);
     p->SetTitle(title);
     gDirectory->Append(p);
     return p;
   }

template<typename T , typename... ARGS>

T * art::TFileDirectory::makeAndRegister	(	std::string const &	name,
		std::string const &	title,
		ARGS...	args
	)		const

inherited

Definition at line 106 of file TFileDirectory.h.

References make_syst_table_plots::args, and art::TFileDirectory::makeAndRegister().

   {
     return makeAndRegister(name.c_str(), title.c_str(), args...);
   }

void TFileService::maybeSwitchFiles_ ( )

private

Definition at line 180 of file TFileService_service.cc.

References closeFile_(), currentGranularity_, filePattern_, fileSwitchCriteria_, fp_, art::ClosingCriteria::granularity(), art::TFileDirectory::invokeCallbacks(), lastClosedFile_, art::detail::logFileAction(), openFile_(), status_, and art::Switching.

Referenced by TFileService().

 {
   // FIXME: Should maybe include the granularity check in
   // requestsToCloseFile_().
   if (fileSwitchCriteria_.granularity() > currentGranularity_)
     return;
 
   status_ = OutputFileStatus::Switching;
   closeFile_();
   detail::logFileAction("Closed TFileService file ", lastClosedFile_);
   detail::logFileAction("Switching to new TFileService file with pattern ",
                         filePattern_);
   fp_ = FileProperties{};
   openFile_();
   invokeCallbacks();
 }

TFileDirectory art::TFileDirectory::mkdir	(	std::string const &	dir,
		std::string const &	descr = `""`
	)

inherited

Create a new TFileDirectory, sharing the same TFile as this one, but with an additional 'dir', and with 'path' being the fullPath() of this one.

Referenced by beamlinereco::EventWaveformDump::analyze(), beamlinereco::ToFSingleCounterAnalysis::beginJob(), beamlinereco::EventWaveformDump::beginJob(), beamlinereco::ToFAnalysisMini::beginJob(), beamlinereco::ToFPositionRecoAnalysis::beginJob(), cmf::CMFCappedEventList::endJob(), cmf::CMFSpectraMaker::endJob(), calib::CosmicCalib::endJob(), calib::MakeAttenuationProfiles::endJob(), cmf::SpectraFromPredictions::endJob(), beamlinereco::ToFPositionRecoAnalysis::endJob(), cmf::ContourFromGrid::MakeAndStoreMedianPlots(), cmf::GridPointsBestFit::MakeAndStorePlots(), cmf::ContourFromLibrary::MakeAndStorePlots(), cmf::ContourFromGrid::MakeAndStorePlots(), cmf::FitFeldmanCousinsPoint::MakeCanvases(), cmf::SpectrumPredictionMaker::MakeHiddenParameterHistograms(), cmf::CovarianceFitHelper::MakeResultPlots(), cmf::RandomUniverseUtility::MakeUniverseHistograms(), and cmf::SerializeEventListColl().

void TFileService::openFile_ ( )

private

Definition at line 154 of file TFileService_service.cc.

References ana::assert(), beginTime_, art::TFileDirectory::file_, fileNameAtOpen_(), fstats_, recentWatchdog::now, art::Open, art::FileStatsCollector::recordFileOpen(), status_, and uniqueFilename_.

Referenced by maybeSwitchFiles_(), and TFileService().

 {
   uniqueFilename_ = fileNameAtOpen_();
   assert(file_ == nullptr && "TFile pointer should always be zero here!");
   beginTime_ = std::chrono::steady_clock::now();
   file_ = new TFile{uniqueFilename_.c_str(), "RECREATE"};
   status_ = OutputFileStatus::Open;
   fstats_.recordFileOpen();
 }

TFileService art::TFileService::operator= ( TFileService const & )

privatedelete

void art::TFileDirectory::registerCallback ( Callback_t )

protectedinherited

Referenced by registerFileSwitchCallback().

void TFileService::registerFileSwitchCallback ( Callback_t c )

Definition at line 119 of file TFileService_service.cc.

References art::TFileDirectory::registerCallback().

 {
   registerCallback(cb);
 }

template<typename T >

void art::TFileService::registerFileSwitchCallback	(	T *	provider,
		void(T::*)()	f
	)

Definition at line 102 of file TFileService.h.

References DECLARE_ART_SERVICE, MakeMiniprodValidationCuts::f, and art::LEGACY.

 {
   auto cb = [provider, f] { return (provider->*f)(); };
   registerFileSwitchCallback(cb);
 }

bool TFileService::requestsToCloseFile_ ( )

private

Definition at line 198 of file TFileService_service.cc.

References beginTime_, DEFINE_ART_SERVICE, art::TFileDirectory::file_, fileSwitchCriteria_, fp_, recentWatchdog::now, art::ClosingCriteria::should_close(), submit_hadd::u, art::FileProperties::updateAge(), and art::FileProperties::updateSize().

Referenced by TFileService().

 {
   using namespace std::chrono;
   unsigned int constexpr oneK{1024u};
   fp_.updateSize(file_->GetSize() / oneK);
   fp_.updateAge(duration_cast<seconds>(steady_clock::now() - beginTime_));
   return fileSwitchCriteria_.should_close(fp_);
 }

void TFileService::setDirectoryName_ ( art::ModuleDescription const & desc )

private

Definition at line 125 of file TFileService_service.cc.

References CET_ASSERT_ONLY_ONE_THREAD, art::TFileDirectory::descr_, art::TFileDirectory::dir_, art::ModuleDescription::moduleLabel(), art::ModuleDescription::moduleName(), and string.

Referenced by TFileService().

 {
   CET_ASSERT_ONLY_ONE_THREAD();
   dir_ = desc.moduleLabel();
   descr_ = dir_;
   descr_ += " (";
   descr_ += desc.moduleName();
   descr_ += ") folder";
 }