AlgResourcePool.h

Go to the documentation of this file.

#ifndef GAUDIHIVE_ALGRESOURCEPOOL_H
#define GAUDIHIVE_ALGRESOURCEPOOL_H

// Include files
#include "GaudiKernel/IAlgResourcePool.h"
#include "GaudiKernel/IAlgManager.h"
#include "GaudiKernel/Service.h"
#include "GaudiKernel/IAlgorithm.h"
#include "GaudiKernel/Algorithm.h"

// TODO: include here is only a workaround
#include "ExecutionFlowGraph.h"

// std includes
#include <string>
#include <list>
#include <vector>
#include <bitset>
#include <atomic>
#include <mutex>

// External libs
#include "boost/dynamic_bitset.hpp"
#include "tbb/concurrent_queue.h"

class AlgResourcePool: public extends<Service,
                                      IAlgResourcePool> {
public:
  // Standard constructor
  AlgResourcePool(  const std::string& name, ISvcLocator* svc );
  // Standard destructor
  ~AlgResourcePool();

  virtual StatusCode start();
  virtual StatusCode initialize();
  virtual StatusCode acquireAlgorithm(const std::string& name, IAlgorithm*& algo, bool blocking = false);
  virtual StatusCode releaseAlgorithm(const std::string& name, IAlgorithm*& algo);
  virtual StatusCode acquireResource(const std::string& name);
  virtual StatusCode releaseResource(const std::string& name);

  virtual std::list<IAlgorithm*> getFlatAlgList();
  virtual std::list<IAlgorithm*> getTopAlgList();

  virtual StatusCode beginRun();
  virtual StatusCode endRun();

  virtual StatusCode stop();

  virtual concurrency::ExecutionFlowGraph* getExecutionFlowGraph() const {return m_EFGraph;}

private:
  typedef tbb::concurrent_bounded_queue<IAlgorithm*> concurrentQueueIAlgPtr;
  typedef std::list<SmartIF<IAlgorithm> > ListAlg;
  typedef boost::dynamic_bitset<> state_type;

  std::mutex m_resource_mutex;
  bool m_lazyCreation;
  state_type m_available_resources;
  std::map<size_t,concurrentQueueIAlgPtr*> m_algqueue_map;
  std::map<size_t,state_type> m_resource_requirements;
  std::map<size_t,size_t> m_n_of_allowed_instances;
  std::map<size_t,unsigned int> m_n_of_created_instances;
  std::map<std::string,unsigned int> m_resource_indices;

  StatusCode decodeTopAlgs();

  StatusCode flattenSequencer(Algorithm* sequencer, ListAlg& alglist, const std::string& parentName, unsigned int recursionDepth=0);

  StringArrayProperty m_topAlgNames;

  ListAlg m_algList;

  ListAlg m_topAlgList;

  ListAlg m_flatUniqueAlgList;

  std::list<IAlgorithm*> m_flatUniqueAlgPtrList;

  std::list<IAlgorithm*> m_topAlgPtrList;

  concurrency::ExecutionFlowGraph* m_EFGraph;
};

#endif  // GAUDIHIVE_ALGRESOURCEPOOL_H