From 496019ae4eb36ffac2065b36b46b904d5dae3251 Mon Sep 17 00:00:00 2001
From: Andreas Fabri <andreas.fabri@geometryfactory.com>
Date: Thu, 22 Aug 2024 13:25:31 +0100
Subject: [PATCH] Add a selection item for edges marked in the om file

---
 Lab/demo/Lab/Plugins/IO/CMakeLists.txt   |  2 +-
 Lab/demo/Lab/Plugins/IO/OM_io_plugin.cpp | 20 ++++++++++++++++----
 Stream_support/include/CGAL/IO/OM.h      | 15 +++++++++++++--
 3 files changed, 30 insertions(+), 7 deletions(-)

diff --git a/Lab/demo/Lab/Plugins/IO/CMakeLists.txt b/Lab/demo/Lab/Plugins/IO/CMakeLists.txt
index 8729366809f..2b1b56be430 100644
--- a/Lab/demo/Lab/Plugins/IO/CMakeLists.txt
+++ b/Lab/demo/Lab/Plugins/IO/CMakeLists.txt
@@ -47,7 +47,7 @@ find_package(OpenMesh)
 if(OpenMesh_FOUND)
   include(UseOpenMesh)
   cgal_lab_plugin(om_plugin OM_io_plugin KEYWORDS Viewer PMP)
-  target_link_libraries(om_plugin PUBLIC scene_surface_mesh_item scene_polygon_soup_item)
+  target_link_libraries(om_plugin PUBLIC scene_surface_mesh_item scene_polygon_soup_item scene_selection_item)
   target_link_libraries(om_plugin PRIVATE ${OPENMESH_LIBRARIES})
 else()
   message(STATUS "NOTICE: the OM IO plugin needs OpenMesh libraries and will not be compiled.")
diff --git a/Lab/demo/Lab/Plugins/IO/OM_io_plugin.cpp b/Lab/demo/Lab/Plugins/IO/OM_io_plugin.cpp
index 6ca5110a96b..20cda20b526 100644
--- a/Lab/demo/Lab/Plugins/IO/OM_io_plugin.cpp
+++ b/Lab/demo/Lab/Plugins/IO/OM_io_plugin.cpp
@@ -6,10 +6,10 @@
 
 #include <CGAL/Three/CGAL_Lab_io_plugin_interface.h>
 #include <CGAL/Three/Three.h>
+#include "Scene_polyhedron_selection_item.h"
 
 #include <CGAL/IO/OM.h>
 #include <CGAL/boost/graph/io.h>
-#include <CGAL/Polygon_mesh_processing/polygon_soup_to_polygon_mesh.h>
 
 #include <QColor>
 #include <QString>
@@ -90,7 +90,8 @@ load(QFileInfo fileinfo, bool& ok, bool add_to_scene){
     // Try building a surface_mesh
     SMesh* SM = new SMesh();
     if (CGAL::IO::read_OM((const char*)fileinfo.filePath().toUtf8(), *SM, sm_selected_pmap, sm_feature_pmap))
-    {/*
+    {
+      /*
       std::cout << "vertex selection values:\n";
       for(auto v : vertices(*SM)){
         std::cout << std::boolalpha << get(sm_selected_pmap, v) << std::endl;
@@ -108,11 +109,22 @@ load(QFileInfo fileinfo, bool& ok, bool add_to_scene){
     }
     else{
       Scene_surface_mesh_item* item = new Scene_surface_mesh_item(SM);
+
+      Scene_polyhedron_selection_item* selection_item = new Scene_polyhedron_selection_item(item, CGAL::Three::Three::mainWindow());
+      bool selected = false;
+      for(auto e : edges(*SM)){
+        if(get(sm_feature_pmap, e)){
+          selection_item->selected_edges.insert(e);
+          selected = true;
+        }
+      }
       item->setName(fileinfo.completeBaseName());
       ok = true;
-      if(add_to_scene)
+      if(add_to_scene){
         CGAL::Three::Three::scene()->addItem(item);
-      return QList<Scene_item*>()<<item;
+        CGAL::Three::Three::scene()->addItem(selection_item);
+      }
+      return QList<Scene_item*>()<<item << selection_item;
     }
   }
   catch(...){}
diff --git a/Stream_support/include/CGAL/IO/OM.h b/Stream_support/include/CGAL/IO/OM.h
index 6c350bad03f..4390be1ced0 100644
--- a/Stream_support/include/CGAL/IO/OM.h
+++ b/Stream_support/include/CGAL/IO/OM.h
@@ -1,4 +1,15 @@
-#pragma once
+// Copyright (c) 2024 GeometryFactory
+//
+// This file is part of CGAL (www.cgal.org);
+//
+// $URL$
+// $Id$
+// SPDX-License-Identifier: LGPL-3.0-or-later OR LicenseRef-Commercial
+//
+// Author(s)     : Andreas Fabri
+
+#ifndef CGAL_IO_OM_H
+#define CGAL_IO_OM_H
 
 #include <OpenMesh/Core/IO/MeshIO.hh>
 #include <OpenMesh/Core/Mesh/TriMesh_ArrayKernelT.hh>
@@ -59,4 +70,4 @@ bool read_OM(std::string fname, SM& sm, VSelectionPM vspm, EFeaturePM efpm)
 } // namespace IO
 } // namespace CGAL
 
-
+#endif // CGAL_IO_OM
\ No newline at end of file