xAODChargedTracksWeightFilter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef GENERATORFILTERS_XAODCHARGEDTRACKSWEIGHTFILTER_H
#define GENERATORFILTERS_XAODCHARGEDTRACKSWEIGHTFILTER_H
 
#include "GeneratorModules/GenFilter.h"
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaKernel/IAthRNGSvc.h"
 
#include "xAODTruth/TruthEvent.h"
#include "xAODTruth/TruthEventContainer.h"
 
namespace CLHEP {
  class HepRandomEngine;
}
 
class xAODChargedTracksWeightFilter : public GenFilter {
public:
 
  xAODChargedTracksWeightFilter(const std::string& name, ISvcLocator* pSvcLocator);
  StatusCode filterInitialize();
  StatusCode filterEvent();
  StatusCode filterFinalize();
 
  struct Spline {
 
     struct Point {
        Point(double _x, double _y, double _slope) :
           x(_x), y(_y), slope(_slope) {};
 
        double x;
        double y;
        double slope;
     };
 
     std::vector<Point> points;
 
     StatusCode initialize( std::vector<double> & x, std::vector<double> & y);
 
     double get_minimum(double min, double max) const;
 
     double value(int nch) const;
  };
 
private:
 
  CLHEP::HepRandomEngine* getRandomEngine(const std::string& streamName,
                                          const EventContext& ctx) const;
 
  double get_nch_weight(int nch) const;
  
  StatusCode event_weight(double & event_weight) const;
 
  void weight_event(double weight);
 
  ServiceHandle<IAthRNGSvc> m_rndmSvc{this, "RndmSvc", "AthRNGSvc"};
 
  // Minimum pT for a track to count
  double m_Ptmin;
 
  // Maximum |pseudorapidity| for a track to count
  double m_EtaRange;
 
  // Minimum number of tracks
  int m_nchmin;
 
  // Maximum number of tracks
  int m_nchmax;
 
  Spline m_spline;
 
  std::vector<double> m_weight_fun_x;
  std::vector<double> m_weight_fun_y;
 
  double m_min_weight = 0;
 
  double m_nevents_selected = 0;
  double m_nevents_accepted = 0;
};
 
#endif