PXDClusterPositionCalibrationAlgorithm.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
#include <calibration/CalibrationAlgorithm.h>
#include <pxd/dbobjects/PXDClusterShapeIndexPar.h>
#include <pxd/dbobjects/PXDClusterPositionEstimatorPar.h>
 
#include <vector>
#include <set>
 
namespace Belle2 {
  class  PXDClusterPositionCalibrationAlgorithm : public CalibrationAlgorithm {
  public:
 
    PXDClusterPositionCalibrationAlgorithm();
 
    int minClusterForShapeLikelyhood;
 
    int minClusterForPositionOffset;
 
    int maxEtaBins;
 
    std::vector<int> clusterKinds;
 
  protected:
 
    virtual EResult calibrate() override;
 
  private:
 
    void createShapeClassifier(std::string treename, PXDClusterShapeClassifierPar* shapeClassifier,
                               PXDClusterShapeIndexPar* shapeIndexer);
 
    PXDClusterShapeClassifierPar mirrorShapeClassifier(PXDClusterShapeClassifierPar* shapeClassifier,
                                                       PXDClusterShapeIndexPar* shapeIndexer, int clusterKind);
 
    PXDClusterShapeClassifierPar localToGlobal(PXDClusterShapeClassifierPar* localShapeClassifier,
                                               PXDClusterShapeIndexPar* localShapeIndexer, PXDClusterShapeIndexPar* globalShapeIndexer);
 
    std::string m_shapeName;
    std::string m_mirroredShapeName;
    float m_clusterEta;
    float m_positionOffsetU;
    float m_positionOffsetV;
    int m_sizeV;
 
    float m_pitchV;
    int m_clusterKind;
 
 
    std::map<int, float> m_pitchMap;
    std::map<std::string, std::string> m_mirrorMap;
    std::map<std::string, int> m_sizeMap;
    std::set<std::string> m_shapeSet;
  };
} // namespace Belle2