CastorHitCorrection.h

Go to the documentation of this file.

#ifndef CastorSim_CastorHitCorrection_h
#define CastorSim_CastorHitCorrection_h

#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/HcalDetId/interface/HcalDetId.h"
#include "SimCalorimetry/CaloSimAlgos/interface/CaloVHitCorrection.h"
#include "SimCalorimetry/CaloSimAlgos/interface/CaloVSimParameterMap.h"
#include "SimDataFormats/CrossingFrame/interface/MixCollection.h"
#include <map>
#include <vector>

namespace CLHEP {
  class HepRandomEngine;
}

class CastorHitCorrection : public CaloVHitCorrection {
public:
  typedef std::map<DetId, double> ChargeSumsByChannel;

  CastorHitCorrection(const CaloVSimParameterMap *parameterMap);
  ~CastorHitCorrection() override {}

  void fillChargeSums(MixCollection<PCaloHit> &hits);

  void fillChargeSums(const std::vector<PCaloHit> &hits);

  void clear();

  double charge(const PCaloHit &hit) const;

  double delay(const PCaloHit &hit, CLHEP::HepRandomEngine *) const override;

  int timeBin(const PCaloHit &hit) const;

  double timeOfFlight(const DetId &id) const;

private:
  const CaloVSimParameterMap *theParameterMap;

  ChargeSumsByChannel theChargeSumsForTimeBin[10];
};

#endif