qt-migration-legatus/qt6-after/qt6-migrated-legatus/Lts/Plugins/PlanEditor/selection_worker.cpp

#include "selection_worker.h"
#include "plan_editor.h"

#include <Rubberbands/rubberband_factory.h>

#include <MathUtils/geometry_engine.h>

#include <GeometryModel/geometry_node_visitor.h>

namespace PlanEditor
{


SelectionWorker::SelectionWorker(PlanEditor                   *planEditor
                                 , SituationMap::GeometryModel *geometryModel
                                 , SituationMap::Map            *map
                                 , SelectionMode                 mode
                                 , const QPointF                &viewPortPosition)
  : QObject(planEditor)
  , m_planEditor(planEditor)
  , m_geometryModel(geometryModel)
  , m_map(map)
  , m_rubberband(nullptr)
{

  switch (mode)
  {
  case SelectionMode::Line :
  {
    SituationMap::RB::RubberbandMultiPointFeature *rubberband = m_map->rubberbandFactory().create<SituationMap::RB::RubberbandMultiPointFeature>(planEditor);

    if (rubberband != nullptr)
    {
      rubberband->setAutoDelete(true);
      rubberband->enableClosed(false);
      rubberband->maximumPointCount(2);
      rubberband->enableFinishAfterConstruction(true);

      

      rubberband->autoSelectAction(int(viewPortPosition.x()), int(viewPortPosition.y()));

      connect(rubberband, &SituationMap::RB::Rubberband::apply, this, &SelectionWorker::onLineSelect);
      connect(rubberband, &SituationMap::RB::Rubberband::abort, this, &SelectionWorker::onAbortSelect);
    }
    else ERROR_LOG() << Q_FUNC_INFO << "Creating rubberband for line selection failed.";
  } break;

  case SelectionMode::Lasso :
  {
    SituationMap::RB::RubberbandFreeHand *rubberband = m_map->rubberbandFactory().create<SituationMap::RB::RubberbandFreeHand>(planEditor);
    if (rubberband != nullptr)
    {
      rubberband->setAutoDelete(true);
      rubberband->enableClosed(true);
      rubberband->enableFinishAfterConstruction(true);
      

      rubberband->autoSelectAction(int(viewPortPosition.x()), int(viewPortPosition.y()));

      connect(rubberband, &SituationMap::RB::Rubberband::apply, this, &SelectionWorker::onLassoSelect);
      connect(rubberband, &SituationMap::RB::Rubberband::abort, this, &SelectionWorker::onAbortSelect);
    }
    else ERROR_LOG() << Q_FUNC_INFO << "Creating rubberband for lasso selection failed.";
  } break;

  case SelectionMode::Rectangle :
  {
    SituationMap::RB::RubberbandRectangle *rubberband = m_map->rubberbandFactory().create<SituationMap::RB::RubberbandRectangle>(planEditor);
    if (rubberband != nullptr)
    {
      rubberband->setAutoDelete(true);
      rubberband->enableFinishAfterConstruction(true);
      

      rubberband->autoSelectAction(int(viewPortPosition.x()), int(viewPortPosition.y()));

      connect(rubberband, &SituationMap::RB::Rubberband::apply, this, &SelectionWorker::onRectSelect);
      connect(rubberband, &SituationMap::RB::Rubberband::abort, this, &SelectionWorker::onAbortSelect);
    }
    else ERROR_LOG() << Q_FUNC_INFO << "Creating rubberband for rectangle selection failed.";
  } break;

  case SelectionMode::Picking        :
  case SelectionMode::PickingAndEdit :
  default                            : delete this;
  }
} 


SelectionWorker::~SelectionWorker()
{
  

  m_planEditor = nullptr;
  m_map        = nullptr;
  m_rubberband = nullptr;
} 



void SelectionWorker::onAbortSelect(SituationMap::RB::RubberbandBase *)
{
  m_planEditor->onClearSelection();
  delete this;
} 


class IntersectLineVisitor : public SituationMap::GeometryNodeVisitor
{
  PlanEditor                                         *m_planEditor;
  SituationMap::Map                                  *m_map;
  SituationMap::MathUtils::GeometryEngine::Geometry *m_intersector;
  unsigned int                                        m_hits;

public:

  IntersectLineVisitor(PlanEditor                   *planEditor
                       , SituationMap::Map            *map
                       , SituationMap::GeometryModel *geoModel
  , const QVector<QPointF>       &linePoints) : GeometryNodeVisitor(geoModel), m_planEditor(planEditor), m_map(map), m_hits(0)
  {
    m_intersector = SituationMap::MathUtils::GeometryEngine::createLine(map, linePoints[0], linePoints[1]);
  }

  virtual ~IntersectLineVisitor() { SituationMap::MathUtils::GeometryEngine::deleteGeometry(m_intersector); m_intersector = nullptr; }

  unsigned int hits() const { return m_hits; }

  bool apply(SituationMap::GeometryModel                 *geoModel
             , const SituationMap::GeometryModel::Node     &node
             , SituationMap::GeometryModel::PrimitiveType   primitiveType) override
  {
    if (SituationMap::GeometryModel::PrimitiveType::RootNode == primitiveType) return true;
    if (SituationMap::GeometryModel::PrimitiveType::Node     == primitiveType) return true;
    if (SituationMap::GeometryModel::PrimitiveType::Unknown  == primitiveType) return false;

    SituationMap::UniqueMapId uid = m_planEditor->getIdFromNode(node);
    if (SituationMap::isUniqueMapIdValid(uid))
    {
      SituationMap::ObjectFactory      &objFac = m_map->objectFactory();
      SituationMap::AbstractMapObject *mapObj = objFac.idToMapObject(uid);

      if (SituationMap::MathUtils::GeometryEngine::intersect(mapObj, *m_intersector))
      {
        m_planEditor->setSelected(node, true);
        m_hits++;
      }
      else
      {
        
        if ((SituationMap::GeometryModel::PrimitiveType::DictionarySymbol  == geoModel->getType(node))
            || (SituationMap::GeometryModel::PrimitiveType::ParticipantSymbol == geoModel->getType(node))
            || (SituationMap::GeometryModel::PrimitiveType::Text              == geoModel->getType(node)))
        {
          const QList<QPointF>                                     &points = geoModel->getPrimitivePoints(node);
          const SituationMap::MathUtils::GeometryEngine::Geometry *geom   = SituationMap::MathUtils::GeometryEngine::createPolygon(m_map, points);
          if (SituationMap::MathUtils::GeometryEngine::intersect(*geom, *m_intersector))
          {
            m_planEditor->setSelected(node, true);
            m_hits++;
          }
        }
      }
    }
    return true;
  } 

}; 


void SelectionWorker::onLineSelect(SituationMap::RB::RubberbandBase *rubberband)
{
  if (rubberband)
  {
    const QVector<QPointF> &linePoints = rubberband->coordinateVector();
    if (linePoints.size() != 2) return;

    IntersectLineVisitor selectionVisitor(m_planEditor, m_map, m_geometryModel, linePoints);
    selectionVisitor.traverse();
    if (selectionVisitor.hits() < 1) m_planEditor->onClearSelection();
  }
  m_planEditor->onTransformRubberband();
  delete this;
} 



class IsInPolygonVisitor : public SituationMap::GeometryNodeVisitor
{
  PlanEditor                                         *m_planEditor;
  SituationMap::Map                                  *m_map;
  SituationMap::MathUtils::GeometryEngine::Geometry *m_isInHull;
  unsigned int                                        m_hits;

public:

  IsInPolygonVisitor(PlanEditor                   *planEditor
                     , SituationMap::Map            *map
                     , SituationMap::GeometryModel* geoModel
  , const QVector<QPointF>       &hullPoints) : GeometryNodeVisitor(geoModel), m_planEditor(planEditor), m_map(map), m_hits(0)
  {
    m_isInHull = SituationMap::MathUtils::GeometryEngine::createPolygon(map, hullPoints);
  }

  virtual ~IsInPolygonVisitor() { SituationMap::MathUtils::GeometryEngine::deleteGeometry(m_isInHull); m_isInHull = nullptr; }

  unsigned int hits() const { return m_hits; }

  bool apply(SituationMap::GeometryModel                 *geoModel
             , const SituationMap::GeometryModel::Node     &node
             , SituationMap::GeometryModel::PrimitiveType   primitiveType) override
  {
    if (SituationMap::GeometryModel::PrimitiveType::RootNode == primitiveType) return true;
    if (SituationMap::GeometryModel::PrimitiveType::Node     == primitiveType) return true;
    if (SituationMap::GeometryModel::PrimitiveType::Unknown  == primitiveType) return false;

    SituationMap::UniqueMapId uid = m_planEditor->getIdFromNode(node);
    if (SituationMap::isUniqueMapIdValid(uid))
    {
      SituationMap::ObjectFactory      &objFac = m_map->objectFactory();
      SituationMap::AbstractMapObject *mapObj = objFac.idToMapObject(uid);

      if (SituationMap::MathUtils::GeometryEngine::contains(mapObj, *m_isInHull))
      {
        m_planEditor->setSelected(node, true);
        m_hits++;
      }
      else
      {
        
        if ((SituationMap::GeometryModel::PrimitiveType::DictionarySymbol  == geoModel    -> getType(node))
            || (SituationMap::GeometryModel::PrimitiveType::ParticipantSymbol == geoModel -> getType(node))
            || (SituationMap::GeometryModel::PrimitiveType::Text              == geoModel -> getType(node)))
        {
          const QList<QPointF>                                     &points = geoModel->getPrimitivePoints(node);
          const SituationMap::MathUtils::GeometryEngine::Geometry *geom   = SituationMap::MathUtils::GeometryEngine::createPolygon(m_map, points);
          if (SituationMap::MathUtils::GeometryEngine::contains(*geom, *m_isInHull))
          {
            m_planEditor->setSelected(node, true);
            m_hits++;
          }
        }
      }
    }
    return true;
  } 

}; 


void SelectionWorker::onLassoSelect(SituationMap::RB::RubberbandBase *rubberband)
{
  if (rubberband)
  {
    const QVector<QPointF> &linePoints = rubberband->coordinateVector();
    if (linePoints.size() < 3) return;

    IsInPolygonVisitor selectionVisitor(m_planEditor, m_map, m_geometryModel, linePoints);
    selectionVisitor.traverse();
    if (selectionVisitor.hits() < 1) m_planEditor->onClearSelection();
  }
  m_planEditor->onTransformRubberband();
  delete this;
} 


void SelectionWorker::onRectSelect(SituationMap::RB::RubberbandBase *rubberband)
{
  if (rubberband)
  {
    const QVector<QPointF> &linePoints = rubberband->coordinateVector();
    if (linePoints.size() == 2) return;

    IsInPolygonVisitor selectionVisitor(m_planEditor, m_map, m_geometryModel, linePoints);
    selectionVisitor.traverse();
    if (selectionVisitor.hits() < 1) m_planEditor->onClearSelection();
  }
  m_planEditor->onTransformRubberband();
  delete this;
} 


}