CDCTimeWalks.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: CDC group                                                *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <map>
#include <iostream>
#include <fstream>
#include <iomanip>
#include <TObject.h>
 
namespace Belle2 {
  class CDCTimeWalks: public TObject {
  public:
 
    CDCTimeWalks() {}
 
    void setTwParamMode(unsigned short mode)
    {
      m_twParamMode = mode;
    }
 
    void setTimeWalkParams(unsigned short boardID, const std::vector<float>& params)
    {
      m_nTwParams = params.size();
      m_tws.insert(std::pair<unsigned short, std::vector<float>>(boardID, params));
    }
 
    void addTimeWalks(unsigned short boardID, const std::vector<float>& deltas)
    {
      std::map<unsigned short, std::vector<float>>::iterator it = m_tws.find(boardID);
      if (it != m_tws.end()) {
        for (unsigned short i = 0; i < m_nTwParams; ++i) {
          (it->second)[i] += deltas[i];
        }
      } else {
        B2FATAL("Specified tw params not found in addTimeWalks !");
      }
    }
 
 
    unsigned short getTwParamMode() const
    {
      return m_twParamMode;
    }
 
    unsigned short getEntries() const
    {
      return m_tws.size();
    }
 
    std::map<unsigned short, std::vector<float>> getTimeWalkParams() const
    {
      return m_tws;
    }
 
    const std::vector<float>& getTimeWalkParams(unsigned short boardID) const
    {
      std::map<unsigned short, std::vector<float>>::const_iterator it = m_tws.find(boardID);
      if (it != m_tws.end()) {
        return it->second;
      } else {
        B2FATAL("Specified tw params not found in getTimeWalks !");
      }
    }
 
    void dump() const
    {
      std::cout << " " << std::endl;
      std::cout << "Time walk coefficient list" << std::endl;
      std::cout << "#entries= " << m_tws.size() << std::endl;
      std::cout << m_twParamMode << "  " << m_nTwParams << std::endl;
      std::cout << "in order of board# and coefficients" << std::endl;
 
      for (auto const& ent : m_tws) {
        std::cout << ent.first;
        for (unsigned short i = 0; i < m_nTwParams; ++i) {
          std::cout << "  " << (ent.second)[i];
        }
        std::cout << std::endl;
      }
    }
 
    void outputToFile(std::string fileName) const
    {
      std::ofstream fout(fileName);
 
      if (fout.bad()) {
        B2ERROR("Specified output file could not be opened!");
      } else {
        fout << m_twParamMode << "  " << m_nTwParams << std::endl;
        for (auto const& ent : m_tws) {
          fout << std::setw(3) << std::right << ent.first;
          for (unsigned short i = 0; i < m_nTwParams; ++i) {
            fout << "  " << std::setw(15) << std::scientific << std::setprecision(8) << ent.second[i];
          }
          fout << std::endl;
        }
        fout.close();
      }
    }
 
    // ------------- Interface to global Millepede calibration ----------------
    static unsigned short getGlobalUniqueID() {return 26;}
    double getGlobalParam(unsigned short element, unsigned short i) const
    {
      return (getTimeWalkParams(element))[i];
    }
    void setGlobalParam(double value, unsigned short element, unsigned short i)
    {
      // TODO: this does not allow updates
      //setTimeWalkParam(element, value);
      // directly access the map
      //      m_tws[element] = value;
      std::map<unsigned short, std::vector<float>>::iterator it = m_tws.find(element);
      if (it != m_tws.end()) {
        it->second[i] = value;
      } else {
        B2FATAL("Specified tw params not found in setGlobalParam !");
      }
    }
    std::vector<std::pair<unsigned short, unsigned short>> listGlobalParams() const
    {
      std::vector<std::pair<unsigned short, unsigned short>> result;
      for (auto id_timewalk : m_tws) {
        result.push_back({id_timewalk.first, 0});
      }
      return result;
    }
    // ------------------------------------------------------------------------
 
  private:
    unsigned short m_twParamMode =
      0; 
    unsigned short m_nTwParams =
      1;   
    std::map<unsigned short, std::vector<float>> m_tws; 
    ClassDef(CDCTimeWalks, 2); 
  };
 
} // end namespace Belle2