first try transformation matrix

Author: zippy84

Contact.cxx

@@ -164,18 +164,43 @@ Contact::Contact (vtkPolyData *newPdA, vtkPolyData *newPdB) : newPdA(newPdA), ne
 
     GetNonManifoldEdges(newPdA, edgesA);
     GetNonManifoldEdges(newPdB, edgesB);
-}
 
-vtkSmartPointer<vtkPolyData> Contact::GetLines () {
-    auto treeA = vtkSmartPointer<vtkOBBTree>::New();
+    treeA = vtkSmartPointer<vtkOBBTree>::New();
     treeA->SetDataSet(newPdA);
     treeA->BuildLocator();
 
-    auto treeB = vtkSmartPointer<vtkOBBTree>::New();
+    treeB = vtkSmartPointer<vtkOBBTree>::New();
     treeB->SetDataSet(newPdB);
     treeB->BuildLocator();
+}
+
+vtkSmartPointer<vtkPolyData> Contact::GetLines (vtkPolyData *pdA, vtkLinearTransform *transA, vtkPolyData *pdB, vtkLinearTransform *transB) {
+
+    auto matrix = vtkSmartPointer<vtkMatrix4x4>::New();
+
+    if (pdA != nullptr) {
+        newPdA = pdA;
+    }
+
+    if (pdB != nullptr) {
+        newPdB = pdB;
+    }
 
-    treeA->IntersectWithOBBTree(treeB, nullptr, InterNodes, this);
+    if (transA != nullptr || transB != nullptr) {
+        auto tmpMatrix = vtkSmartPointer<vtkMatrix4x4>::New();
+
+        vtkMatrix4x4::Invert(transA->GetMatrix(), tmpMatrix);
+        vtkMatrix4x4::Multiply4x4(tmpMatrix, transB->GetMatrix(), matrix);
+    }
+
+    sourcesA->Reset();
+    sourcesB->Reset();
+    contA->Reset();
+    contB->Reset();
+
+    lines->Reset();
+
+    treeA->IntersectWithOBBTree(treeB, matrix, InterNodes, this);
 
     IntersectReplacements();
 
@@ -572,13 +597,8 @@ void Contact::InterPolys (vtkIdType idA, vtkIdType idB) {
         return;
     }
 
-    double ptA[3], ptB[3];
-
-    newPdA->GetPoint(ptsA[0], ptA);
-    newPdB->GetPoint(ptsB[0], ptB);
-
-    double dA = vtkMath::Dot(nA, ptA);
-    double dB = vtkMath::Dot(nB, ptB);
+    double dA = polyA[0].x*nA[0]+polyA[0].y*nA[1]+polyA[0].z*nA[2];
+    double dB = polyB[0].x*nB[0]+polyB[0].y*nB[1]+polyB[0].z*nB[2];
 
     vtkMath::Cross(nA, nB, r);
     vtkMath::Normalize(r);

Contact.h

@@ -21,6 +21,7 @@ limitations under the License.
 
 #include <vtkOBBTree.h>
 #include <vtkMatrix4x4.h>
+#include <vtkLinearTransform.h>
 
 enum class Src {
     A,
@@ -123,7 +124,9 @@ class Contact {
 
     NonManifoldEdgesType edgesA, edgesB;
 
-    vtkSmartPointer<vtkPolyData> GetLines ();
+    vtkSmartPointer<vtkOBBTree> treeA, treeB;
+
+    vtkSmartPointer<vtkPolyData> GetLines (vtkPolyData *pdA = nullptr, vtkLinearTransform *transA = nullptr, vtkPolyData *pdB = nullptr, vtkLinearTransform *transB = nullptr);
 
     void GetNonManifoldEdges (vtkPolyData *pd, NonManifoldEdgesType &edges);
 

Utilities.h

@@ -90,6 +90,8 @@ class Point3d {
 
         return dx*dx+dy*dy+dz*dz;
     }
+
+
 };
 
 typedef std::vector<vtkIdType> IdsType;

transform/bone.vtk

@@ -0,0 +1,57 @@
+#!/usr/bin/env python
+# *-* coding: UTF-8 *-*
+
+import sys
+sys.path.extend(['/home/zippy/vtkbool/build/lib/python3.13/site-packages/vtkbool'])
+
+from vtkmodules.vtkIOLegacy import vtkPolyDataReader, vtkPolyDataWriter
+from vtkmodules.vtkCommonMath import vtkMatrix4x4
+from vtkmodules.vtkCommonTransforms import vtkTransform
+from vtkmodules.vtkFiltersGeneral import vtkTransformPolyDataFilter
+
+from vtkbool import vtkPolyDataBooleanFilter
+
+mandibleReader = vtkPolyDataReader()
+mandibleReader.SetFileName('bone.vtk')
+mandibleReader.Update()
+
+boxReader = vtkPolyDataReader()
+boxReader.SetFileName('box.vtk')
+boxReader.Update()
+
+moveBoxTransform = vtkTransform()
+moveBoxTransform.Translate(20.076142063093318, -8.288617300972092, -53.29349952031101)
+moveBoxMatrix = moveBoxTransform.GetMatrix()
+
+identityMatrix = vtkMatrix4x4()
+
+bf = vtkPolyDataBooleanFilter()
+bf.SetInputData(0, mandibleReader.GetOutput())
+bf.SetInputData(1, boxReader.GetOutput())
+
+bf.SetMatrix(0, identityMatrix)
+bf.SetMatrix(1, moveBoxMatrix)
+
+for i in range(10):
+    moveBoxTransform.Translate(0, -10, 0)
+    moveBoxMatrix = moveBoxTransform.GetMatrix()
+
+    print(moveBoxMatrix.GetElement(1, 3))
+
+    transformFilter = vtkTransformPolyDataFilter()
+    transformFilter.SetInputData(boxReader.GetOutput())
+    transformFilter.SetTransform(moveBoxTransform)
+    transformFilter.Update()
+
+    _writer = vtkPolyDataWriter()
+    _writer.SetFileName(f'out/pdB_{i}.vtk')
+    _writer.SetInputConnection(transformFilter.GetOutputPort())
+    _writer.Update()
+
+    bf.SetMatrix(1, moveBoxMatrix)
+
+    writer = vtkPolyDataWriter()
+    writer.SetFileName(f'out/transformed_{i}.vtk')
+    writer.SetInputConnection(bf.GetOutputPort())
+
+    writer.Update()

transform/box.vtk

@@ -47,6 +47,7 @@ limitations under the License.
 #include <vtkCellArrayIterator.h>
 #include <vtkKdTree.h>
 #include <vtkCellIterator.h>
+#include <vtkTransformPolyDataFilter.h>
 
 #include "vtkPolyDataBooleanFilter.h"
 
@@ -78,10 +79,30 @@ vtkPolyDataBooleanFilter::vtkPolyDataBooleanFilter () {
 
     OperMode = OPER_UNION;
 
+    matrices[0] = nullptr;
+    matrices[1] = nullptr;
+
+    transforms[0] = nullptr;
+    transforms[1] = nullptr;
+
 }
 
 vtkPolyDataBooleanFilter::~vtkPolyDataBooleanFilter () {
-    // nix mehr
+    if (matrices[0] != nullptr) {
+        matrices[0]->Delete();
+    }
+
+    if (matrices[1] != nullptr) {
+        matrices[1]->Delete();
+    }
+
+    if (transforms[0] != nullptr) {
+        transforms[0]->Delete();
+    }
+
+    if (transforms[1] != nullptr) {
+        transforms[1]->Delete();
+    }
 }
 
 int vtkPolyDataBooleanFilter::RequestData(vtkInformation *request, vtkInformationVector **inputVector, vtkInformationVector *outputVector) {
@@ -106,36 +127,63 @@ int vtkPolyDataBooleanFilter::RequestData(vtkInformation *request, vtkInformatio
         std::vector<clock::duration> times;
         clock::time_point start;
 
-        if (pdA->GetMTime() > timePdA || pdB->GetMTime() > timePdB) {
+        // if (pdA->GetMTime() > timePdA || pdB->GetMTime() > timePdB)
+        {
             // CellData sichern
 
-            cellDataA->DeepCopy(pdA->GetCellData());
-            cellDataB->DeepCopy(pdB->GetCellData());
+            if (contact == nullptr) {
+                cellDataA->DeepCopy(pdA->GetCellData());
+                cellDataB->DeepCopy(pdB->GetCellData());
 
-            modPdA = Clean(pdA);
-            modPdB = Clean(pdB);
+                cleanA = Clean(pdA);
+                cleanB = Clean(pdB);
 
-            modPdA->EditableOn();
-            modPdB->EditableOn();
+    #ifdef DEBUG
+                WriteVTK("modPdA.vtk", cleanA);
+                WriteVTK("modPdB.vtk", cleanB);
+    #endif
 
-#ifdef DEBUG
-            WriteVTK("modPdA.vtk", modPdA);
-            WriteVTK("modPdB.vtk", modPdB);
-#endif
+                try {
+                    PreventEqualCaptPoints(cleanA, cleanB).Run();
+                } catch (const std::runtime_error &e) {
+                    vtkErrorMacro("Cannot prevent equal capture points.");
+                    return 1;
+                }
 
-            try {
-                PreventEqualCaptPoints(modPdA, modPdB).Run();
-            } catch (const std::runtime_error &e) {
-                vtkErrorMacro("Cannot prevent equal capture points.");
-                return 1;
+                contact = std::make_shared<Contact>(cleanA, cleanB);
             }
 
             start = clock::now();
 
-            Contact contact(modPdA, modPdB);
+            if (transforms[0] != nullptr) {
+                auto tfA = vtkSmartPointer<vtkTransformPolyDataFilter>::New();
+                tfA->SetInputData(cleanA);
+                tfA->SetTransform(transforms[0]);
+                tfA->Update();
+
+                modPdA = vtkSmartPointer<vtkPolyData>::New();
+                modPdA->DeepCopy(tfA->GetOutput());
+            } else {
+                modPdA = cleanA;
+            }
+
+            if (transforms[1] != nullptr) {
+                auto tfB = vtkSmartPointer<vtkTransformPolyDataFilter>::New();
+                tfB->SetInputData(cleanB);
+                tfB->SetTransform(transforms[1]);
+                tfB->Update();
+
+                modPdB = vtkSmartPointer<vtkPolyData>::New();
+                modPdB->DeepCopy(tfB->GetOutput());
+            } else {
+                modPdB = cleanB;
+            }
+
+            modPdA->EditableOn();
+            modPdB->EditableOn();
 
             try {
-                contLines = contact.GetLines();
+                contLines = contact->GetLines(modPdA, transforms[0], modPdB, transforms[1]);
             } catch (const std::runtime_error &e) {
                 std::stringstream ss;
                 ss << std::quoted(e.what());
@@ -154,7 +202,7 @@ int vtkPolyDataBooleanFilter::RequestData(vtkInformation *request, vtkInformatio
 #endif
 
             if (contLines->GetNumberOfCells() == 0) {
-                vtkErrorMacro("There is no contact.");
+                vtkErrorMacro("There is no contact 1.");
                 return 1;
             }
 
@@ -210,7 +258,7 @@ int vtkPolyDataBooleanFilter::RequestData(vtkInformation *request, vtkInformatio
             // löscht bestimmte strips
 
             if (CleanStrips()) {
-                vtkErrorMacro("There is no contact.");
+                vtkErrorMacro("There is no contact 2.");
                 return 1;
             }
 
@@ -3024,3 +3072,33 @@ bool vtkPolyDataBooleanFilter::CombineRegions () {
     return false;
 
 }
+
+void vtkPolyDataBooleanFilter::SetMatrix (int i, vtkMatrix4x4 *matrix) {
+    if (transforms[i]) {
+        transforms[i]->Delete();
+        transforms[i] = nullptr;
+    }
+
+    if (matrices[i]) {
+        matrices[i]->Delete();
+        matrices[i] = nullptr;
+    }
+
+    matrices[i] = matrix;
+    matrix->Register(this);
+
+    auto transform = vtkMatrixToLinearTransform::New();
+
+    transform->Register(this);
+    transform->Delete();
+
+    transform->SetInput(matrix);
+
+    transforms[i] = transform;
+
+    Modified();
+}
+
+vtkMatrix4x4* vtkPolyDataBooleanFilter::GetMatrix (int i) {
+    return matrices[i];
+}

transform/test.py

@@ -28,6 +28,10 @@ limitations under the License.
 #include <vtkPolyDataAlgorithm.h>
 #include <vtkKdTree.h>
 #include <vtkModifiedBSPTree.h>
+#include <vtkMatrixToLinearTransform.h>
+#include <vtkMatrix4x4.h>
+
+#include "Contact.h"
 
 #include "Utilities.h"
 
@@ -190,6 +194,13 @@ class VTK_EXPORT vtkPolyDataBooleanFilter : public vtkPolyDataAlgorithm {
 
     int OperMode;
 
+    vtkMatrix4x4 *matrices[2];
+    vtkLinearTransform *transforms[2];
+
+    vtkSmartPointer<vtkPolyData> cleanA, cleanB;
+
+    std::shared_ptr<Contact> contact;
+
 public:
     vtkTypeMacro(vtkPolyDataBooleanFilter, vtkPolyDataAlgorithm);
     static vtkPolyDataBooleanFilter* New ();
@@ -203,6 +214,9 @@ class VTK_EXPORT vtkPolyDataBooleanFilter : public vtkPolyDataAlgorithm {
     void SetOperModeToDifference () { OperMode = OPER_DIFFERENCE; Modified(); }
     void SetOperModeToDifference2 () { OperMode = OPER_DIFFERENCE2; Modified(); }
 
+    void SetMatrix (int i, vtkMatrix4x4 *matrix);
+    vtkMatrix4x4* GetMatrix (int i);
+
 protected:
     vtkPolyDataBooleanFilter ();
     ~vtkPolyDataBooleanFilter ();