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 <iostream>
#include "AbstractObjects/Attribute.h"
#include "AbstractObjects/Object.h"
#include "AbstractObjects/OpalData.h"
#include "AbstractObjects/Table.h"
#include "Beamlines/FlaggedElmPtr.h"
#include "Elements/OpalElement.h"
#include "Utilities/Options.h"
#include "Utilities/RegularExpression.h"
 
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; }