StreamToFileTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include <JiveXML/StreamToFileTool.h>
 
#include <fstream>
#include <iomanip>
 
namespace JiveXML {
 
  StreamToFileTool::StreamToFileTool( const std::string& type , const std::string& name, const IInterface* p):
    AthAlgTool(type,name,p){
 
    //Provide interface
    declareInterface<IStreamTool>(this);
 
    declareProperty("FileNamePrefix", m_FileNamePrefix = "JiveXML");
    declareProperty("FileNameSuffix", m_FileNameSuffix = ".xml");
 
  }
 
  StatusCode StreamToFileTool::initialize(){
    if(m_isOnline){
      if(m_onlineEDsvc.retrieve().isFailure()){
        ATH_MSG_ERROR("Could not locate the online event displays service");
        return StatusCode::FAILURE;
      }
    }
    return StatusCode::SUCCESS;
  }
 
  StatusCode StreamToFileTool::finalize(){
    return StatusCode::SUCCESS;
  }
 
   StatusCode StreamToFileTool::StreamEvent( const unsigned long EventNumber, const unsigned int RunNumber, const std::ostringstream* EventBuffer ) {
     if(m_isOnline){
       m_FileNamePrefix = m_onlineEDsvc->getFileNamePrefix();
       ATH_MSG_DEBUG("m_FileNamePrefix: " << m_FileNamePrefix << " EventNumber: " << EventNumber);
     }
     std::ofstream* outFile;
     StatusCode sc = NewFile(EventNumber,RunNumber,outFile);
     if (sc.isFailure()){
       ATH_MSG_WARNING("Could not open file for event " << EventNumber << " from run " << RunNumber);
       return sc;
     }
     (*outFile) << EventBuffer->str();
     outFile->flush();
     if (!outFile->good()){
       ATH_MSG_WARNING("Could not open file for event " << EventNumber << " from run " << RunNumber);
       return StatusCode::FAILURE;
     }
 
     sc = CloseFile(outFile);
     return sc;
 
   }
 
   std::string StreamToFileTool::MakeFileName(const unsigned long EventNumber, const unsigned int RunNumber) const {
 
     //Generate a the return string with file prefix
     std::ostringstream name;
 
     //Assemble file name
     name << m_FileNamePrefix << std::setfill('0');
     name << "_" << std::setw(5) << RunNumber;
     name << "_" << std::setw(5) << EventNumber;
     name << m_FileNameSuffix;
     return name.str();
   }
 
   StatusCode StreamToFileTool::NewFile( const unsigned long EventNumber, const unsigned int RunNumber, std::ofstream *& outputFile) const {
 
      // Generate the file name
      std::string filename = MakeFileName(EventNumber,RunNumber);
 
      // create a new output stream
      outputFile = new std::ofstream(filename.c_str());
      ATH_MSG_DEBUG("outputFile "<< filename);
      // check if it worked
      if ( !(outputFile->good()) ){
        ATH_MSG_WARNING("Unable to create output file with name " << filename);
        return StatusCode::FAILURE;
      }
 
      return StatusCode::SUCCESS;
    }
 
   StatusCode StreamToFileTool::CloseFile( std::ofstream *& outputFile ) const {
 
    //Try to close the file
    outputFile->close();
 
    //See if it worked
    if (!outputFile->good()){
      ATH_MSG_WARNING("Unable to close file");
    }
 
    //In any case delete object
    delete outputFile;
 
    return StatusCode::SUCCESS;
  }
 
} //namespace