SGAudSvc.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2017 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;
template <class TYPE> class SvcFactory;
class IChronoStatSvc;
class IAlgContextSvc;
class AlgContextSvc;
 
class SGAudSvc : virtual public ISGAudSvc,
         virtual public IIncidentListener,
         public AthService
{ 
 
protected:
    
  friend class SvcFactory<SGAudSvc>;
 
 
public: 
 
  SGAudSvc( const std::string& name, ISvcLocator* pSvcLocator );
 
  virtual ~SGAudSvc(); 
 
 
  StatusCode initialize();
  StatusCode finalize();
  virtual StatusCode queryInterface( const InterfaceID& riid, void** ppvInterface );
 
  static const InterfaceID& interfaceID();
  
 
  void handle( const Incident& incident );
 
  // do the auditing, called from DataStore.cxx 
  virtual void SGAudit(const std::string& key, const CLID& id, 
               const int& fnc, const int& store_id);
 
  
  bool SGGetCurrentAlg();
  void setFakeCurrentAlg(const std::string&);
  void clearFakeCurrentAlg();
 
  private: 
 
  SGAudSvc();
 
  void SGAudRETRIEVE(std::string SGobject);
  void SGAudRECORD(std::string SGobject);
 
  void getNobj(std::string name);
  void addRead();
  void addWrite();
 
  void monitor();
 
  void writeJSON();
 
  // Private data: 
 private: 
 
  MsgStream m_msg;
 
  IAlgContextSvc *p_algCtxSvc;
  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;
}; 
 
 
 
inline const InterfaceID& SGAudSvc::interfaceID() 
{ 
   return ISGAudSvc::interfaceID(); 
}
 
#endif