TubeDataFromFile.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCALIB_TUBEDATAFROMFILE_H
#define MUONCALIB_TUBEDATAFROMFILE_H
 
#include <iostream>
#include <vector>
 
#include "AthenaKernel/getMessageSvc.h"
#include "GaudiKernel/MsgStream.h"
#include "MdtCalibData/MdtTubeFitContainer.h"
#include "MuonCalibIdentifier/MuonFixedId.h"
 
namespace MuonCalib {
 
    class TubeDataFromFile {
    public:
        using TubeData = std::vector<const MdtTubeFitContainer*>;
 
    public:
        TubeDataFromFile() = default;
        ~TubeDataFromFile() = default;

        unsigned int nRegions() const { return m_regions; }

        const MdtTubeFitContainer* getTubes(unsigned int regionId) const {
            if (regionId >= (unsigned int)m_regions) {
                MsgStream log(Athena::getMessageSvc(), "MdtTubeFitContainer");
                log << MSG::WARNING << "TubeDataFromFile::getTubes: <regionId out of range> " << regionId << " size " << m_regions
                    << endmsg;
                return nullptr;
            }
            return m_tubeData[regionId];
        }

        void setNRegions(unsigned int n) {
            m_regions = n;
            m_tubeData.resize(n);
        }

        bool addTubes(int regionId, const MdtTubeFitContainer* tubes) {
            if (regionId < 0 || regionId >= (int)m_regions) {
                MsgStream log(Athena::getMessageSvc(), "MdtTubeFitContainer");
                log << MSG::WARNING << "TubeDataFromFile::addTubes: <regionId out of range> " << regionId << " size " << m_regions
                    << endmsg;
                return false;
            }
            if (m_tubeData[regionId] != 0) {
                MsgStream log(Athena::getMessageSvc(), "MdtTubeFitContainer");
                log << MSG::WARNING << "TubeDataFromFile::addTubes: <tubes already set>" << endmsg;
                return false;
            }
 
            m_tubeData[regionId] = tubes;
 
            return true;
        }
 
        std::ostream& write(std::ostream& os) const;
 
        std::ostream& write_forDB(std::ostream& ftube, int mdt_head, int lowrun, int uprun) const;
 
    private:
        unsigned int m_regions{0};

        TubeData m_tubeData;
    };
 
}  // namespace MuonCalib
 
std::ostream& operator<<(std::ostream& os, const MuonCalib::TubeDataFromFile& data);
 
#endif