RtDataFromFile.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCALIB_RTDATAFROMFILE_H
#define MUONCALIB_RTDATAFROMFILE_H
 
#include <iostream>
#include <vector>
 
#include "AthenaKernel/getMessageSvc.h"
#include "GaudiKernel/MsgStream.h"
#include "MdtCalibData/RtFullInfo.h"
#include "MdtCalibUtils/RtData_t_r_reso.h"
 
namespace MuonCalib {

    class RtDataFromFile {
    public:
        typedef RtData_t_r_reso RtRelation;
        typedef std::vector<RtRelation*> RtRelations;
 
    public:
        RtDataFromFile() : m_rts(0), m_major_version(1), m_minor_version(0) {}
        ~RtDataFromFile() = default;

        unsigned int nRts() const { return m_rts; }

        RtRelation* getRt(unsigned int regionId) const {
            if (regionId >= (unsigned int)m_rts) {
                MsgStream log(Athena::getMessageSvc(), "RtDataFromFile");
                log << MSG::WARNING << "getRt() <regionId out of range> " << regionId << " size " << m_rts << endmsg;
                return 0;
            }
            return m_rtRelations[regionId];
        }

        void setNRts(unsigned int nrts) {
            m_rts = nrts;
            m_rtRelations.resize(nrts);
            m_fullInfo.resize(nrts);
        }

        bool addRt(int regionId, RtRelation* rt) {
            if (regionId < 0 || regionId >= (int)m_rts) {
                MsgStream log(Athena::getMessageSvc(), "RtDataFromFile");
                log << MSG::WARNING << "addRt() <regionId out of range> " << regionId << " size " << m_rts << endmsg;
                return false;
            }
            if (m_rtRelations[regionId] != 0) {
                MsgStream log(Athena::getMessageSvc(), "RtDataFromFile");
                log << MSG::WARNING << "addRt() <rt already set>" << endmsg;
                return false;
            }
 
            m_rtRelations[regionId] = rt;
 
            return true;
        }
        bool addRt(int regionId, RtRelation* rt, const RtFullInfo* info) {
            bool result = addRt(regionId, rt);
            if (result) { m_fullInfo[regionId] = info; }
            return result;
        }
        std::istream& read(std::istream& is);
        std::ostream& write(std::ostream& os, int region) const;
        std::ostream& write(std::ostream& os) const;
        void write_forDB(FILE* frt, FILE* frtt, FILE* frtr, FILE* frts, int region) const;
        void write_forDB(FILE* frt, FILE* frtt, FILE* frtr, FILE* frts) const;
        inline void setVersion(int major, int minor) {
            m_major_version = major;
            m_minor_version = minor;
        }
 
    private:
        unsigned int m_rts;

        RtRelations m_rtRelations;
        std::vector<const RtFullInfo*> m_fullInfo;
        int m_major_version, m_minor_version;
    };
 
}  // namespace MuonCalib
 
std::istream& operator>>(std::istream& is, MuonCalib::RtDataFromFile& data);
 
std::ostream& operator<<(std::ostream& os, const MuonCalib::RtDataFromFile& data);
 
#endif