From 78dff3b236f91b50dac873167906846eb4e074af Mon Sep 17 00:00:00 2001
From: tremblap <info@pierrealexandretremblay.com>
Date: Sat, 10 Dec 2022 16:34:13 +0000
Subject: [PATCH] feature/kdtree-distance-with-optional-k

---
 release-packaging/Classes/FluidKDTree.sc | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/release-packaging/Classes/FluidKDTree.sc b/release-packaging/Classes/FluidKDTree.sc
index 2f688a7..f850d7f 100644
--- a/release-packaging/Classes/FluidKDTree.sc
+++ b/release-packaging/Classes/FluidKDTree.sc
@@ -36,13 +36,16 @@ FluidKDTree : FluidModelObject
 		this.prSendMsg(this.kNearestMsg(buffer,k));
 	}
 
-	kNearestDistMsg {|buffer|
-		^this.prMakeMsg(\kNearestDist,id,this.prEncodeBuffer(buffer));
+	kNearestDistMsg {|buffer, k|
+		k !?
+		{^this.prMakeMsg(\kNearestDist,id,this.prEncodeBuffer(buffer),k);}
+		??
+		{^this.prMakeMsg(\kNearestDist,id,this.prEncodeBuffer(buffer));}
 	}
 
-	kNearestDist { |buffer, action|
+	kNearestDist { |buffer, k, action|
 		actions[\kNearestDist] = [numbers(FluidMessageResponse,_,nil,_),action];
-		this.prSendMsg(this.kNearestDistMsg(buffer));
+		this.prSendMsg(this.kNearestDistMsg(buffer,k));
 	}
 
 	kr{|trig, inputBuffer,outputBuffer, numNeighbours = 1, lookupDataSet|