Algorithm-QuadTree
view release on metacpan or search on metacpan
lib/Algorithm/QuadTree/PP.pm view on Meta::CPAN
? $cxmin
: $coords[0] > $cxmax
? $cxmax
: $coords[0]
;
my $cy = $coords[1] < $cymin
? $cymin
: $coords[1] > $cymax
? $cymax
: $coords[1]
;
# first check if obj overlaps current segment.
next if ($coords[0] - $cx) ** 2 + ($coords[1] - $cy) ** 2
> $radius_squared;
$current->{HAS_OBJECTS} = 1 if !$finding;
if ($current->{CHILDREN}) {
push @loopargs, @{$current->{CHILDREN}};
} else {
# segment is a leaf and overlaps the obj
push @nodes, $current;
}
}
return \@nodes;
}
# this private method executes $code on every leaf node of the tree
# which is within the rectangular shape
sub _loopOnNodesInRectangle
{
my ($self, $finding, @coords) = @_;
my @nodes;
my @loopargs = $self->{ROOT};
while (my $current = shift @loopargs) {
next if $finding && !$current->{HAS_OBJECTS};
# first check if obj overlaps current segment.
next if
$coords[0] > $current->{AREA}[2] ||
$coords[2] < $current->{AREA}[0] ||
$coords[1] > $current->{AREA}[3] ||
$coords[3] < $current->{AREA}[1];
$current->{HAS_OBJECTS} = 1 if !$finding;
if ($current->{CHILDREN}) {
push @loopargs, @{$current->{CHILDREN}};
} else {
# segment is a leaf and overlaps the obj
push @nodes, $current;
}
}
return \@nodes;
}
# choose the right function based on argument count
# first argument is always $self, second is $finding, the rest are coords
sub _loopOnNodes
{
goto \&_loopOnNodesInCircle if @_ == 5;
goto \&_loopOnNodesInRectangle;
}
sub _clearHasObjects
{
my $node = shift;
if ($node->{CHILDREN}) {
for my $child (@{$node->{CHILDREN}}) {
return if $child->{HAS_OBJECTS};
}
}
$node->{HAS_OBJECTS} = 0;
if ($node->{PARENT}) {
_clearHasObjects($node->{PARENT});
}
}
sub _AQT_init
{
my $obj = shift;
$obj->{BACKREF} = {};
$obj->{ROOT} = _addLevel(
$obj,
1, #current depth
undef, # parent - none
$obj->{XMIN},
$obj->{YMIN},
$obj->{XMAX},
$obj->{YMAX},
);
}
sub _AQT_deinit
{
# do nothing in PP implementation
}
sub _AQT_addObject
{
my ($self, $object, @coords) = @_;
for my $node (@{_loopOnNodes($self, 0, @coords)}) {
push @{$node->{OBJECTS}}, $object;
push @{$self->{BACKREF}{$object}}, $node;
}
}
sub _AQT_findObjects
{
my ($self, @coords) = @_;
# map returned nodes to an array containing all of
# their objects
return [
( run in 0.514 second using v1.01-cache-2.11-cpan-39bf76dae61 )