From c1cedd50a39e6d2f6e7039cba9bcdd94f26bda0a Mon Sep 17 00:00:00 2001
From: Theo <theo.leplomb@mikrosimage.com>
Date: Thu, 23 Jan 2020 12:26:17 +0100
Subject: [PATCH] [nodes] ImageMatching: Add methods and option nbNeighbors

---
 meshroom/nodes/aliceVision/ImageMatching.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/meshroom/nodes/aliceVision/ImageMatching.py b/meshroom/nodes/aliceVision/ImageMatching.py
index 8d4ec51536..54344d6fba 100644
--- a/meshroom/nodes/aliceVision/ImageMatching.py
+++ b/meshroom/nodes/aliceVision/ImageMatching.py
@@ -1,4 +1,4 @@
-__version__ = "1.0"
+__version__ = "1.1"
 
 import os
 from meshroom.core import desc
@@ -28,6 +28,15 @@ class ImageMatching(desc.CommandLineNode):
             label="Features Folders",
             description="Folder(s) containing the extracted features and descriptors."
         ),
+        desc.ChoiceParam(
+            name='method',
+            label='Method',
+            description='Method used to select the image pairs to match.',
+            value='VocabularyTree',
+            values=['VocabularyTree', 'Sequential', 'SequentialAndVocabularyTree','Exhaustive','Frustum'],
+            exclusive=True,
+            uid=[0],
+        ),
         desc.File(
             name='tree',
             label='Tree',
@@ -70,6 +79,15 @@ class ImageMatching(desc.CommandLineNode):
             uid=[0],
             advanced=True,
         ),
+        desc.IntParam(
+            name='nbNeighbors',
+            label='Nb Neighbors',
+            description='The number of neighbors to retrieve for each image (If 0 it will retrieve all the neighbors).',
+            value=50,
+            range=(0, 1000, 1),
+            uid=[0],
+            advanced=True,
+        ),
         desc.ChoiceParam(
             name='verboseLevel',
             label='Verbose Level',