TOPAlignmentAlgorithm.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 <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