StationSelector.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MuonGeoModel/StationSelector.h"
 
#include "AthenaKernel/getMessageSvc.h"
#include "GaudiKernel/MsgStream.h"
#include "MuonGeoModel/MYSQL.h"
#include "MuonGeoModel/Station.h"
 
#include <fstream>
#include <sstream>
#include <utility>
 
namespace MuonGM {
 
    std::atomic<int> StationSelector::m_selectType = 0;
 
    StationSelector::StationSelector(const MYSQL& mysql, const std::string& filename) {
        std::ifstream from;
        from.open(filename.c_str());
        char buffer[200];
        while (from.getline(buffer, 200)) {
            std::istringstream line(buffer);
            std::string key;
            line >> key;
            m_selector.push_back(key);
        }
 
        for (const auto& [name, station] :  mysql.stationMap()) {
            if (select(name))
                m_theMap[name] = station.get();
        }
    }
 
    StationSelector::StationSelector(const MYSQL& mysql, std::vector<std::string> s) : m_selector(std::move(s)) {
        StationIterator it;
        for (const auto& [name, station] :  mysql.stationMap()) {
            if (select(name))
                m_theMap[name] = station.get();
        }
    }
 
    StationSelector::StationIterator StationSelector::begin() { return m_theMap.begin(); }
    StationSelector::StationIterator StationSelector::end() { return m_theMap.end(); }
 
    bool StationSelector::select(const std::string& name ) {
        MsgStream log(Athena::getMessageSvc(), "MuonGeoModel");
        int selFlag = m_selectType;
        if (m_selector[0] == "*") {
            selFlag = 1; // override JO choice for general configuration
        }
 
        if (selFlag == 1) {
            for (unsigned int k = 0; k < m_selector.size(); k++) {
                std::string sel = m_selector[k];
 
                if (sel.size() <= name.size()) {
                    bool myflag = true;
                    for (unsigned int i = 0; i < sel.size(); i++) {
                        if (sel[i] == '*')
                            continue;
                        if (sel[i] != name[i]) {
                            myflag = false;
                            break;
                        }
                    }
 
                    if (myflag)
                        return myflag;
                }
            }
            return false;
        } else if (selFlag == 2) {
            for (unsigned int k = 0; k < m_selector.size(); k++) {
                std::string sel = m_selector[k];
 
                if (sel.size() != name.size())
                    continue; // require exact match
 
                bool myflag = true;
                for (unsigned int i = 0; i < sel.size(); i++) {
                    if (sel[i] == '*')
                        continue;
                    if (sel[i] != name[i]) {
                        myflag = false;
                        break;
                    }
                }
 
                if (myflag) {
                    log << MSG::INFO << " StationSelector: removing station " << name << endmsg;
                    return false;
                }
            }
 
            return true;
        }
 
        return true;
    }
 
    void StationSelector::SetSelectionType(int t) { m_selectType = t; }
 
} // namespace MuonGM