SGAudSvc.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// SGAudSvc.h 
// Header file for class SGAudSvc
// Author: Ilija Vukotic <ivukotic@cern.ch>
#ifndef SGMONITORING_SGAUDSVC_H 
#define SGMONITORING_SGAUDSVC_H 
 
// STL includes
#include <string>
#include <iosfwd>
#include <map>
#include <set>
#include <fstream>
 
 
// FrameWork includes
#include "AthenaBaseComps/AthService.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/IIncidentListener.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/Algorithm.h"
#include "GaudiKernel/IClassIDSvc.h"
 
#include "SGAudCore/ISGAudSvc.h"
 
// Forward declaration
class ISvcLocator;
class IChronoStatSvc;
class IAlgContextSvc;
class AlgContextSvc;
 
class SGAudSvc : public extends<AthService,
                                ISGAudSvc, IIncidentListener>
{
public:
 
  SGAudSvc( const std::string& name, ISvcLocator* pSvcLocator );
 
  virtual ~SGAudSvc(); 
 
 
  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
 
  virtual void handle( const Incident& incident ) override;
 
  // do the auditing, called from DataStore.cxx 
  virtual void SGAudit(const std::string& key, const CLID& id, 
               const int& fnc, const int& store_id) override;
 
  
  bool SGGetCurrentAlg();
  virtual void setFakeCurrentAlg(const std::string&) override;
  virtual void clearFakeCurrentAlg() override;
 
  private: 
 
  SGAudSvc();
 
  void SGAudRETRIEVE(std::string SGobject);
  void SGAudRECORD(std::string SGobject);
 
  void getNobj(const std::string& name);
  void addRead();
  void addWrite();
 
  void monitor();
 
  void writeJSON();
 
  // Private data: 
 private: 
 
  MsgStream m_msg;
 
  ServiceHandle<IAlgContextSvc> p_algCtxSvc;
  ServiceHandle<IClassIDSvc> m_pCID;
  
  std::string m_outFileName, m_allFileName, m_sumFileName;  
 
  bool m_ignoreFakeAlgs;
 
  bool m_useCLID;
 
  std::vector<std::string> m_vObj;
  std::vector<std::string> m_vAlg;
  std::map<int,int> m_timesRead;
  std::map<int,int> m_timesWritten;
  std::string m_currAlg;
  std::string m_currObj;
  std::string m_fakeCurrAlg;
  int m_nCurrAlg;
  int m_nCurrObj;
  int m_nEvents;
  int m_startEvent;
 
  // map<"alg_name", set<"cid/key"> >
  typedef std::map<std::string, std::set<std::string> > DataMap;
  DataMap m_read;
  DataMap m_write;
 
  std::ofstream m_ofa, m_ofs;
  bool m_inExec;
}; 
 
#endif