TOPAlignmentAlgorithm.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 *                                                                        *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
#include <calibration/CalibrationAlgorithm.h>
#include <vector>
#include <map>
 
namespace Belle2 {
  namespace TOP {
 
    class TOPAlignmentAlgorithm : public CalibrationAlgorithm {
    public:
 
      TOPAlignmentAlgorithm();
 
      virtual ~TOPAlignmentAlgorithm() {}
 
      void setSpatialPrecision(double precision) {m_spatialPrecision = precision;}
 
      void setAngularPrecision(double precision) {m_angularPrecision = precision;}
 
    private:
 
      struct AlignData {
        int iter = 0;  
        int ntrk = 0;  
        std::vector<float> alignPars;  
        std::vector<float> alignErrs;  
        bool valid = false;  
        void add(const AlignData& data);
        void finalize(double scaleFact);
      };
 
      virtual EResult calibrate() final;
 
      void mergeData();
 
      // algorithm parameters
      double m_spatialPrecision = 0.1; 
      double m_angularPrecision = 0.001; 
      // input tree variables
      int m_moduleID = 0; 
      int m_iter = 0;  
      int m_ntrk = 0;  
      int m_errorCode = 0;  
      std::vector<float>* m_vAlignPars = 0;     
      std::vector<float>* m_vAlignParsErr = 0;  
      bool m_valid = false;  
      TBranch* m_bAlignPars = 0; 
      TBranch* m_bAlignParsErr = 0; 
      // maps of moduleID and AlignData
      std::multimap<int, AlignData> m_inputData; 
      std::map<int, AlignData> m_mergedData; 
    };
 
  } // end namespace TOP
} // end namespace Belle2