Selector.cpp

Go to the documentation of this file.

//
// Class Selector
//   Set selection flags for a given range in a beam line.
//
// Copyright (c) 200x - 2020, Paul Scherrer Institut, Villigen PSI, Switzerland
// All rights reserved
//
// This file is part of OPAL.
//
// OPAL is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// You should have received a copy of the GNU General Public License
// along with OPAL. If not, see <https://www.gnu.org/licenses/>.
//
#include "Tables/Selector.h"
#include "AbstractObjects/Attribute.h"
#include "AbstractObjects/OpalData.h"
#include "AbstractObjects/Object.h"
#include "AbstractObjects/Table.h"
#include "Beamlines/FlaggedElmPtr.h"
#include "Elements/OpalElement.h"
#include "Utilities/Options.h"
#include "Utilities/RegularExpression.h"
#include <iostream>


Selector::Selector(const Beamline &bl, const RangeRep &range,
                   const std::string &clsName, const std::string &typName,
                   const std::string &pattern):
    RangeSelector(bl, range),
    itsClass(0), itsType(typName), itsPattern(0), itsCount(0) {
    if(! clsName.empty()  && (itsClass = Element::find(clsName)) == 0) {
        if(Options::warn) {
            std::cerr << "\n### Warning ### Unknown class name \""
                      << clsName << "\"; will select all classes.\n" << std::endl;
        }
    }

    if(! pattern.empty()) {
        itsPattern = new RegularExpression(pattern);
    }
}


Selector::~Selector() {
    delete itsPattern;
}


void Selector::execute() {
    itsCount = 0;
    RangeSelector::execute();
}


void Selector::handleElement(const FlaggedElmPtr &fep) {
    // Skip elements which are not in range.
    if(itsRange.isActive()) {
        const std::string &name = fep.getElement()->getName();
        if(name[0] != '[') {
            bool set = true;
            OpalElement &elem = dynamic_cast<OpalElement &>(*Element::find(name));

            // If class exists and element is not class member, then skip.
            if(itsClass != 0  &&  ! elem.isTreeMember(itsClass)) set = false;

            // If pattern does exists and element name does not match, then skip.
            if(itsPattern != 0  &&  ! itsPattern->match(name)) set = false;

            // If type name is not blank and element type is different, then skip.
            if(! itsType.empty() && itsType != elem.getTypeName()) set = false;

            // The current element matches all conditions.
            if(set) {
                fep.setSelectionFlag(true);
                ++itsCount;
            }
        }
    }
}


int Selector::getCount() const {
    return itsCount;
}