CPUCruncher.h

Go to the documentation of this file.

#include "GaudiKernel/AlgFactory.h"
#include "GaudiKernel/IRndmGenSvc.h"
#include "GaudiKernel/RndmGenerators.h"
#include "GaudiAlg/GaudiAlgorithm.h"
#include "GaudiKernel/RegistryEntry.h"

#include <tbb/concurrent_hash_map.h>

//------------------------------------------------------------------------------

  class CPUCruncher : public GaudiAlgorithm
  {

    friend class AlgFactory<CPUCruncher> ;

  public:

    typedef tbb::concurrent_hash_map<std::string,unsigned int> CHM;

    virtual StatusCode execute  () ; // the execution of the algorithm
    virtual StatusCode initialize();
    virtual StatusCode finalize () ; // the finalization of the algorithm
    virtual const std::vector<std::string> get_inputs();
    virtual const std::vector<std::string> get_outputs();

    double get_runtime() const { return m_avg_runtime; };

  protected:

    CPUCruncher
    ( const std::string& name , // the algorithm instance name
      ISvcLocator*       pSvc ); // the Service Locator

    virtual ~CPUCruncher();     // virtual & protected desctrustor

  private:

    CPUCruncher () ;                              // no default constructor
    CPUCruncher            ( const CPUCruncher& ) ; // no copy constructor
    CPUCruncher& operator= ( const CPUCruncher& ) ; // no assignement
    void findPrimes (const unsigned long int ) ;

    void calibrate();
    long unsigned int getNCaliIters(double);

    double m_avg_runtime ; //Avg Runtime
    double m_var_runtime ; //Variance of Runtime
    bool m_local_rndm_gen; //Decide what random number generation to use
    bool m_shortCalib;

    // To calib only once
    static std::vector<unsigned int>m_niters_vect;
    static std::vector<double> m_times_vect;

    // For the concurrency

    const uint MAX_INPUTS = 40;
    const uint MAX_OUTPUTS = 10;

    std::vector<DataObjectHandle<DataObject> *> m_inputHandles;
    std::vector<DataObjectHandle<DataObject> *> m_outputHandles;

    unsigned int m_rwRepetitions;

    static CHM m_name_ncopies_map;

    // Sleep during execution instead of real CPU crunching
    bool m_sleepyExecution;

  };