cafObjectPerformer.h

// ##################################################################################################
//
//    Copyright (C) 2023- Kontur AS
//
//    GNU Lesser General Public License Usage
//    This library is free software; you can redistribute it and/or modify
//    it under the terms of the GNU Lesser General Public License as published by
//    the Free Software Foundation; either version 2.1 of the License, or
//    (at your option) any later version.
//
//    This library is distributed in the hope that it will be useful, but WITHOUT ANY
//    WARRANTY; without even the implied warranty of MERCHANTABILITY or
//    FITNESS FOR A PARTICULAR PURPOSE.
//
//    See the GNU Lesser General Public License at <<http://www.gnu.org/licenses/lgpl-2.1.html>>
//    for more details.
//
// ##################################################################################################
#pragma once
 
#include "cafChildFieldHandle.h"
#include "cafObjectHandle.h"
#include "cafVisitor.h"
 
#include <list>
 
namespace caffa
{
template <typename ObjectType = ObjectHandle>
class ObjectPerformer : public Editor
{
public:
    using Callback = std::function<void( ObjectType* )>;
    using Selector = std::function<bool( const ObjectType* )>;
 
    ObjectPerformer( Callback callback, Selector selector = nullptr )
        : m_selector( selector )
        , m_callback( callback )
    {
    }
 
    void visitObject( ObjectHandle* object ) override
    {
        auto typedObject = dynamic_cast<ObjectType*>( object );
        if ( typedObject && ( !m_selector || m_selector( typedObject ) ) )
        {
            m_callback( typedObject );
        }
 
        for ( auto field : object->fields() )
        {
            field->accept( this );
        }
    }
 
    void visitField( FieldHandle* field ) override {}
 
    void visitChildField( ChildFieldBaseHandle* childField ) override
    {
        for ( auto object : childField->childObjects() )
        {
            object->accept( this );
        }
    }
 
private:
    Selector m_selector;
    Callback m_callback;
};
} // namespace caffa