private KDNode findIn(KDNode click, KDNode n, int dim)
{
if (n.isLeaf)
{
return n;
}
if (dim == 0)
{
if (n.getSplit() >= click.getPoint().X)
{
return findIn(click, n.getLeft(), 1 - dim);
} else {
return findIn(click, n.getRight(), 1 - dim);
}
} else {
if (n.getSplit() >= click.getPoint().Y)
{
return findIn(click, n.getLeft(), 1 - dim);
} else {
return findIn(click, n.getRight(), 1 - dim);
}
}
}
// najde numNearest najblizsich vrcholov ku target
// na konci budu v usporiadanom zozname nearestK
public void findNearest(KDNode subTree, KDNode target)
{
// ak je to list, zistime, ci je blizsie ako doposial najdene, ak ich je dost
if (subTree.isLeaf) {
subTree.dist = distance(target.getPoint(), subTree.getPoint());
if (nearestK.Count < numNearest)
{
nearestK.Add(subTree);
} else {
if (nearestK.Last().dist > subTree.dist)
{
nearestK.Remove(nearestK.Last());
nearestK.Add(subTree);
}
}
return;
}
int dim = subTree.getDim();
// ak je target v lavom podstrome, nearer je lavy a further je pravy, inak opacne
KDNode nearer;
KDNode further;
if (dim == 0)
{
if (subTree.getSplit() >= target.getPoint().X)
{
nearer = subTree.getLeft();
further = subTree.getRight();
} else {
further = subTree.getLeft();
nearer = subTree.getRight();
}
} else {
if (subTree.getSplit() >= target.getPoint().Y)
{
nearer = subTree.getLeft();
further = subTree.getRight();
}else {
further = subTree.getLeft();
nearer = subTree.getRight();
}
}
// najprv prehladavame blizsi podstrom
// (prvy prehladany blok bude ten, v kt. je target)
// teda si ohranicime v priemere celkom slusne najblizsie vrcholy
findNearest(nearer, target);
// zrata najblizsi bod od target na bunke further
// ak je dalej ako doposial najdeny k-ty nejmensi, tak vo further
// nemoze byt blizsi bod => nevnara sa
Point nearestF = nearestTo(further, target);
double dist = distance(nearestF, target.getPoint());
if ((nearestK.Last().dist > dist) || (nearestK.Count < numNearest))
{
findNearest(further, target);
}
}
// pre kazdy vrchol podstromu n updatne pocet listov,
// ktore su pod nim
public int updateNumLeafsDown(KDNode n)
{
if (n.isLeaf) {
n.setNumLeafs(1);
Console.WriteLine("update " + n.getPoint() + " numLeafs = 1");
return 1;
}
int r = updateNumLeafsDown(n.getRight());
int l = updateNumLeafsDown(n.getLeft());
n.setNumLeafs(r + l);
Console.WriteLine("update " + n.getSplit() + " numLeafs = " + n.getNumLeafs());
return r + l;
}
// pre kazdeho predchodcu n updatne pocet listov,
// ktore su nad nim
public void updateNumLeafsUp(KDNode n)
{
n.setNumLeafs(n.getLeft().getNumLeafs() + n.getRight().getNumLeafs());
Console.WriteLine("update to root " + n.getSplit() + " numLeafs = " + n.getNumLeafs());
Console.WriteLine("update to root right " + n.getRight().getNumLeafs() + " left " + n.getLeft().getNumLeafs());
if (n.Equals(root))
{
return;
}
updateNumLeafsUp(n.getParent());
}
//vrati podstrom, do ktoreho syna ma ist novy vrchol a bola by v nom narusena rovnovaha
public KDNode findPlace(KDNode n, KDNode leaf)
{
// ak n je list, potom sa rovnovaha nenarusi a iba sa novy vrchol vlozi.
// z listu spravime otca dvoch listov v procedure rebuilt
if(n.isLeaf)
{
//Console.Write(n.getPoint());
//Console.WriteLine(" numLeafs " + n.getNumLeafs());
return n;
}
//Console.Write(n.getSplit());
//Console.WriteLine(" numLeafs right" + n.getRight().getNumLeafs());
//Console.WriteLine(" numLeafs left" + n.getLeft().getNumLeafs());
// ak to nie je list, potom hladame podstrom, ktoremu sa narusi rovnovaha.
if (n.getDim() == 0) // split je v x-ovej suradnici
{
if (n.getSplit() >= leaf.getPoint().X) // ak ma ist do laveho podstromu
{
// a uz teraz je vlavo viac listov ako vpravo, narusi sa rovnovaha n
if (n.getLeft().getNumLeafs() > n.getRight().getNumLeafs())
{
return n;
} else { // inak sa vrati podstrom z laveho podstromu, v ktorom je narusena rovnovaha
return findPlace(n.getLeft(), leaf);
}
} else { // ak ma ist do praveho podstromu
// a uz teraz je vpravo viac listov ako vlavo, narusi sa rovnovaha n
if (n.getLeft().getNumLeafs() < n.getRight().getNumLeafs())
{
return n;
} else { // inak sa vrati podstrom z praveho podstromu, v ktorom je narusena rovnovaha
return findPlace(n.getRight(), leaf);
}
}
} else { // split je v y-ovej suradnici
if (n.getSplit() >= leaf.getPoint().Y) // ak ma ist do laveho podstromu (= patri hore, nad split)
{
// a uz teraz je vlavo viac listov ako vpravo, narusi sa rovnovaha n
if (n.getLeft().getNumLeafs() > n.getRight().getNumLeafs())
{
return n;
}
else
{ // inak sa vrati podstrom z laveho podstromu, v ktorom je narusena rovnovaha
return findPlace(n.getLeft(), leaf);
}
}
else
{ // ak ma ist do praveho podstromu (= patri dole, pod split)
// a uz teraz je vpravo viac listov ako vlavo, narusi sa rovnovaha n
if (n.getLeft().getNumLeafs() < n.getRight().getNumLeafs())
{
return n;
}
else
{ // inak sa vrati podstrom z praveho podstromu, v ktorom je narusena rovnovaha
return findPlace(n.getRight(), leaf);
}
}
}
}
