cafObjectCollector.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 "cafVisitor.h"

#include <list>

namespace caffa
{
class ObjectHandle;

template <typename ObjectType = ObjectHandle>
class ConstObjectCollector : public Inspector
{
public:
    using Selector = std::function<bool( const ObjectType* )>;

    ConstObjectCollector( Selector selector = nullptr )
        : m_selector( selector )
    {
    }

    void visitObject( const ObjectHandle* object ) override
    {
        CAFFA_ASSERT( object );
        CAFFA_TRACE( "Visiting OBJECT " << object->uuid() );
        auto typedObject = dynamic_cast<const ObjectType*>( object );
        if ( typedObject && ( !m_selector || m_selector( typedObject ) ) )
        {
            m_objects.push_back( typedObject );
        }
    }

    void visitField( const FieldHandle* field ) override {}

    const std::list<const ObjectType*>& objects() const { return m_objects; }

private:
    Selector                     m_selector;
    std::list<const ObjectType*> m_objects;
};

template <typename ObjectType = ObjectHandle>
class ObjectCollector : public Editor
{
public:
    using Selector = std::function<bool( const ObjectType* )>;

    ObjectCollector( Selector selector = nullptr )
        : m_selector( selector )
    {
    }
    const std::list<ObjectType*>& objects() { return m_objects; }

private:
    void visitObject( ObjectHandle* object ) override
    {
        auto typedObject = dynamic_cast<ObjectType*>( object );
        if ( typedObject && ( !m_selector || m_selector( typedObject ) ) )
        {
            m_objects.push_back( typedObject );
        }
    }

    void visitField( FieldHandle* field ) override {}

private:
    Selector               m_selector;
    std::list<ObjectType*> m_objects;
};
} // namespace caffa