288 lines
10 KiB
C++
Executable File
288 lines
10 KiB
C++
Executable File
|
|
|
|
|
|
|
|
#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),
|
|
m_logger(NGL::NGLLoggerBootstrap::getLogger()) {
|
|
|
|
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(m_logger) << 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(m_logger) << 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(m_logger) << 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;
|
|
}
|
|
|
|
|
|
}
|