CDCEDepToADCConversions.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 <map>
#include <iostream>
#include <fstream>
#include <iomanip>
#include <TObject.h>
 
namespace Belle2 {
  class CDCEDepToADCConversions: public TObject {
  public:
 
    CDCEDepToADCConversions() {}
 
    void setParamMode(unsigned short mode)
    {
      m_paramMode = mode;
    }
 
    void setGroupID(unsigned short mode)
    {
      m_groupID = mode;
    }
 
    void setParams(unsigned short id, const std::vector<float>& params)
    {
      m_cvs.insert(std::pair<unsigned short, std::vector<float>>(id, params));
    }
 
    unsigned short getParamMode() const
    {
      return m_paramMode;
    }
 
    unsigned short getGroupID() const
    {
      return m_groupID;
    }
 
    unsigned short getEntries() const
    {
      return m_cvs.size();
    }
 
    std::map<unsigned short, std::vector<float>> getParams() const
    {
      return m_cvs;
    }
 
    const std::vector<float>& getParams(unsigned short id) const
    {
      std::map<unsigned short, std::vector<float>>::const_iterator it = m_cvs.find(id);
      if (it != m_cvs.end()) {
        return it->second;
      } else {
        B2FATAL("Specified id not found in getParams !");
      }
    }
 
    void dump() const
    {
      std::cout << " " << std::endl;
      std::cout << "Edep-to-ADC conversion list" << std::endl;
      std::cout << "#entries= " << m_cvs.size() << std::endl;
      std::cout << m_paramMode << std::endl;
      std::cout << "in order of id and parameters" << std::endl;
 
      for (auto const& ent : m_cvs) {
        std::cout << ent.first;
        unsigned short np = (ent.second).size();
        for (unsigned short i = 0; i < np; ++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 {
        std::map<unsigned short, std::vector<float>>::const_iterator it = m_cvs.find(0);
        int nParams = (it->second).size();
        fout << m_paramMode << "  " << nParams << std::endl;
        fout << m_groupID << std::endl;
        for (auto const& ent : m_cvs) {
          fout << std::setw(3) << std::right << ent.first;
          for (unsigned short i = 0; i < nParams; ++i) {
            fout << "  " << std::setw(15) << std::scientific << std::setprecision(8) << (ent.second)[i];
          }
          fout << std::endl;
        }
        fout.close();
      }
    }
 
  private:
    unsigned short m_paramMode = 0; 
    unsigned short m_groupID = 0;   
    std::map<unsigned short, std::vector<float>> m_cvs; 
    ClassDef(CDCEDepToADCConversions, 3); 
    // Version histroy:
    // v2: original: paramMode=0, nParams=6.
    // v3: paramMode=1, nParams=7; added sigma for gaussian smearing;
    //     main-factor is now of order(1), while was of order(10) before;
    //     activated outputToFile().
  };
 
} // end namespace Belle2