CDCTimeZeros.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>
#include <cdc/dataobjects/WireID.h>
 
namespace Belle2 {

  class CDCTimeZeros: public TObject {
  public:

    CDCTimeZeros() {}

    void setT0(unsigned short iCLayer, unsigned short iWire, float t0)
    {
      m_t0s.insert(std::pair<unsigned short, float>(WireID(iCLayer, iWire).getEWire(), t0));
    }

    void setT0(const WireID& wid, float t0)
    {
      m_t0s.insert(std::pair<unsigned short, float>(wid.getEWire(), t0));
    }

    void addT0(unsigned short iCLayer, unsigned short iWire, float delta)
    {
      WireID wid(iCLayer, iWire);
      std::map<unsigned short, float>::iterator it = m_t0s.find(wid.getEWire());
      it->second += delta;
    }

    void addT0(const WireID& wid, float delta)
    {
      std::map<unsigned short, float>::iterator it = m_t0s.find(wid.getEWire());
      it->second += delta;
    }

    unsigned short getEntries() const
    {
      return m_t0s.size();
    }

    std::map<unsigned short, float> getT0s() const
    {
      return m_t0s;
    }

    float getT0(const WireID& wid) const
    {
      std::map<unsigned short, float>::const_iterator it = m_t0s.find(wid.getEWire());
      return it->second;
    }

    void dump() const
    {
      std::cout << " " << std::endl;
      std::cout << "t0 list" << std::endl;
      std::cout << "# of entries= " << m_t0s.size() << std::endl;
      std::cout << "in order of clayer#, wire#, t0" << std::endl;
      for (auto const& ent : m_t0s) {
        std::cout << WireID(ent.first).getICLayer() << " " << WireID(ent.first).getIWire() << " " << ent.second << std::endl;
      }
    }

    void outputToFile(std::string fileName) const
    {
      std::ofstream fout(fileName);
 
      if (fout.bad()) {
        B2ERROR("Specified output file could not be opened!");
      } else {
        for (auto const& ent : m_t0s) {
          fout << std::setw(2) << std::right << WireID(ent.first).getICLayer() << "  " << std::setw(3) << WireID(
                 ent.first).getIWire() << "  " << std::setw(15) << std::scientific << std::setprecision(8) << ent.second << std::endl;
        }
        fout.close();
      }
    }
 
    // ------------- Interface to global Millepede calibration ----------------
    static unsigned short getGlobalUniqueID() {return 25;}
    double getGlobalParam(unsigned short element, unsigned short)
    {
      return getT0(WireID(element));
    }

    void setGlobalParam(double value, unsigned short element, unsigned short)
    {
      WireID wire(element);
      //This does not work, tries to insert, but we need an update
      //setT0(wire.getICLayer(), wire.getIWire(), value);
      m_t0s[wire.getEWire()] = value;
    }

    std::vector<std::pair<unsigned short, unsigned short>> listGlobalParams()
    {
      std::vector<std::pair<unsigned short, unsigned short>> result;
      for (auto id_t0 : getT0s()) {
        result.push_back({id_t0.first, 0});
      }
      return result;
    }
    // ------------------------------------------------------------------------
 
  private:
    std::map<unsigned short, float> m_t0s; 
 
    ClassDef(CDCTimeZeros, 2); 
  };

} // end namespace Belle2