From b5ae770f0f397918c1121d7ceb06ee9a56a508a0 Mon Sep 17 00:00:00 2001
From: =?utf8?q?L=C3=A9on=20Victor?= <leon.victor@kitware.com>
Date: Mon, 9 Sep 2024 13:58:40 +0200
Subject: [PATCH] WIP: better selection

---
 src/SPV3D/SPV3D_CADSelection.cxx | 49 +++++++++++++++++++++++++++++++-
 1 file changed, 48 insertions(+), 1 deletion(-)

diff --git a/src/SPV3D/SPV3D_CADSelection.cxx b/src/SPV3D/SPV3D_CADSelection.cxx
index 5d999ff60..85c0e8375 100644
--- a/src/SPV3D/SPV3D_CADSelection.cxx
+++ b/src/SPV3D/SPV3D_CADSelection.cxx
@@ -37,8 +37,14 @@
 #include <vtkSMStringVectorProperty.h>
 #include <vtkSMRepresentationProxy.h>
 #include <vtkPVDataSetAttributesInformation.h>
-
+#include <vtkSMSelectionHelper.h>
+#include <vtkSMOutputPort.h>
 #include <QDebug>
+#include <vtkSelectionSource.h>
+#include <vtkSelectionNode.h>
+#include <pqPVApplicationCore.h>
+#include <pqSelectionManager.h>
+#include <pqServerManagerModel.h>
 
 #include "SPV3D_Prs.h"
 
@@ -540,6 +546,17 @@ void SPV3D_CADSelection::SetSelectionFromEntrySet(std::set<std::string> EntryLis
 
   std::cout << "(Client) Sending selected entries to server:" << std::endl;
 
+  auto producerPort = vtkSMPropertyHelper(proxyRepresentation, "Input").GetAsOutputPort();
+  if (producerPort)
+  {
+    std::cout << producerPort->GetDataClassName() << std::endl;
+  }
+
+  vtkNew<vtkSelectionSource> selectionSource;
+  selectionSource->SetFieldType(vtkSelectionNode::CELL);
+  selectionSource->SetContentType(vtkSelectionNode::BLOCKS);
+  selectionSource->SetArrayName("name"); // TODO 
+
   // Retrieve entry ids and put them in a vector
   std::vector<int> selectedIds;
   selectedIds.reserve(EntryList.size());
@@ -548,8 +565,38 @@ void SPV3D_CADSelection::SetSelectionFromEntrySet(std::set<std::string> EntryLis
     vtkIdType id = SPV3D_Prs::FromEntryToVtkId(entryName.c_str());
     std::cout << "(Client) " << id << std::endl;
     selectedIds.push_back(id);
+
+    // TODO: I'm not sure what ids reference
+    selectionSource->AddBlock(id);
+
+
   }
 
+  selectionSource->Update();
+
+  auto producer = producerPort->GetSourceProxy();
+  
+  auto selectionSourceProxy = vtk::TakeSmartPointer(
+    vtkSMSourceProxy::SafeDownCast(vtkSMSelectionHelper::NewSelectionSourceFromSelection(
+      producer->GetSession(), selectionSource->GetOutput())));
+  // create append selection proxy from selection source proxy
+  auto appendSelections = vtk::TakeSmartPointer(vtkSMSourceProxy::SafeDownCast(
+    vtkSMSelectionHelper::NewAppendSelectionsFromSelectionSource(selectionSourceProxy)));
+  if (appendSelections)
+  {
+    producer->SetSelectionInput(producerPort->GetPortIndex(), appendSelections, 0);
+
+    auto smmodel = pqApplicationCore::instance()->getServerManagerModel();
+    auto selManager = pqPVApplicationCore::instance()->selectionManager();
+    selManager->select(smmodel->findItem<pqOutputPort*>(producerPort));
+  }
+
+
+
+  // vtkSMSelectionHelper::NewSelectionSourceFromSelection(, vtkSelection *selection);
+  vtkSMProperty* selectionProperty = proxyRepresentation->GetProperty("Selection");
+  // vtkSMPropertyHelper(selectionProperty)
+
   vtkSMProperty* selectedIDsProperty = proxyRepresentation->GetProperty("SelectedIDs");
   vtkSMPropertyHelper(selectedIDsProperty).Set(selectedIds.data(), selectedIds.size());
   proxyRepresentation->UpdateVTKObjects();
-- 
2.39.2