SelectionFootprint.h

//=========================================================================
//  Copyright (c) Kitware, Inc.
//  All rights reserved.
//  See LICENSE.txt for details.
//
//  This software is distributed WITHOUT ANY WARRANTY; without even
//  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
//  PURPOSE.  See the above copyright notice for more information.
//=========================================================================
 
#ifndef smtk_geometry_SelectionFootprint_h
#define smtk_geometry_SelectionFootprint_h
 
#include "smtk/CoreExports.h"
 
#include "smtk/geometry/Geometry.h"
#include "smtk/geometry/Resource.h"
 
#include "smtk/resource/Component.h"
#include "smtk/resource/query/DerivedFrom.h"
#include "smtk/resource/query/Query.h"
 
#include <unordered_set>
 
namespace smtk
{
namespace geometry
{
 
struct SMTKCORE_EXPORT SelectionFootprint
  : public smtk::resource::query::DerivedFrom<SelectionFootprint, smtk::resource::query::Query>
{
  virtual bool operator()(
    smtk::resource::PersistentObject& selectedObject,
    std::unordered_set<smtk::resource::PersistentObject*>& footprint,
    const smtk::geometry::Backend& backend) const = 0;
 
  bool hasGeometry(smtk::resource::PersistentObject& object, const smtk::geometry::Backend& backend)
    const
  {
    auto* resource = dynamic_cast<smtk::geometry::Resource*>(&object);
    if (!resource)
    {
      auto* component = dynamic_cast<smtk::resource::Component*>(&object);
      if (component)
      {
        resource = dynamic_cast<smtk::geometry::Resource*>(component->resource().get());
      }
    }
    if (!resource)
    {
      return false;
    }
    auto& geom = resource->geometry(backend);
    return geom &&
      geom->generationNumber(object.shared_from_this()) != smtk::geometry::Geometry::Invalid;
  }
 
  bool addAllComponentsIfResource(
    smtk::resource::PersistentObject& object,
    std::unordered_set<smtk::resource::PersistentObject*>& footprint,
    const smtk::geometry::Backend& backend) const
  {
    auto* resource = dynamic_cast<smtk::geometry::Resource*>(&object);
    if (!resource)
    {
      return false;
    }
    auto& geom = resource->geometry(backend);
    if (!geom)
    {
      return false;
    }
    if (geom->generationNumber(object.shared_from_this()) == smtk::geometry::Geometry::Invalid)
    {
      bool hasFootprint = false;
      auto visitor = [&hasFootprint, &footprint, &geom](
                       const resource::PersistentObject::Ptr& obj,
                       smtk::geometry::Geometry::GenerationNumber) -> bool {
        if (geom->generationNumber(obj) != smtk::geometry::Geometry::Invalid)
        {
          hasFootprint = true;
          footprint.insert(obj.get());
        }
        return true;
      };
      geom->visit(visitor);
      return hasFootprint;
    }
    return true; // Resource has geometry.
  }
};
} // namespace geometry
} // namespace smtk
 
#endif