forked from sascha/godot
Update RVO2 to git 2022.09
parent
72b59325cf
commit
c920881105
@ -0,0 +1,693 @@
|
||||
/*
|
||||
* Agent2d.cpp
|
||||
* RVO2 Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
/**
|
||||
* @file Agent2d.cpp
|
||||
* @brief Defines the Agent2D class.
|
||||
*/
|
||||
|
||||
#include "Agent2d.h"
|
||||
|
||||
#include <algorithm>
|
||||
#include <cmath>
|
||||
#include <limits>
|
||||
|
||||
#include "KdTree2d.h"
|
||||
#include "Obstacle2d.h"
|
||||
|
||||
namespace RVO2D {
|
||||
namespace {
|
||||
/**
|
||||
* @relates Agent2D
|
||||
* @brief Solves a one-dimensional linear program on a specified line
|
||||
* subject to linear constraints defined by lines and a circular
|
||||
* constraint.
|
||||
* @param[in] lines Lines defining the linear constraints.
|
||||
* @param[in] lineNo The specified line constraint.
|
||||
* @param[in] radius The radius of the circular constraint.
|
||||
* @param[in] optVelocity The optimization velocity.
|
||||
* @param[in] directionOpt True if the direction should be optimized.
|
||||
* @param[in, out] result A reference to the result of the linear program.
|
||||
* @return True if successful.
|
||||
*/
|
||||
bool linearProgram1(const std::vector<Line> &lines, std::size_t lineNo,
|
||||
float radius, const Vector2 &optVelocity, bool directionOpt,
|
||||
Vector2 &result) { /* NOLINT(runtime/references) */
|
||||
const float dotProduct = lines[lineNo].point * lines[lineNo].direction;
|
||||
const float discriminant =
|
||||
dotProduct * dotProduct + radius * radius - absSq(lines[lineNo].point);
|
||||
|
||||
if (discriminant < 0.0F) {
|
||||
/* Max speed circle fully invalidates line lineNo. */
|
||||
return false;
|
||||
}
|
||||
|
||||
const float sqrtDiscriminant = std::sqrt(discriminant);
|
||||
float tLeft = -dotProduct - sqrtDiscriminant;
|
||||
float tRight = -dotProduct + sqrtDiscriminant;
|
||||
|
||||
for (std::size_t i = 0U; i < lineNo; ++i) {
|
||||
const float denominator = det(lines[lineNo].direction, lines[i].direction);
|
||||
const float numerator =
|
||||
det(lines[i].direction, lines[lineNo].point - lines[i].point);
|
||||
|
||||
if (std::fabs(denominator) <= RVO2D_EPSILON) {
|
||||
/* Lines lineNo and i are (almost) parallel. */
|
||||
if (numerator < 0.0F) {
|
||||
return false;
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
const float t = numerator / denominator;
|
||||
|
||||
if (denominator >= 0.0F) {
|
||||
/* Line i bounds line lineNo on the right. */
|
||||
tRight = std::min(tRight, t);
|
||||
} else {
|
||||
/* Line i bounds line lineNo on the left. */
|
||||
tLeft = std::max(tLeft, t);
|
||||
}
|
||||
|
||||
if (tLeft > tRight) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (directionOpt) {
|
||||
/* Optimize direction. */
|
||||
if (optVelocity * lines[lineNo].direction > 0.0F) {
|
||||
/* Take right extreme. */
|
||||
result = lines[lineNo].point + tRight * lines[lineNo].direction;
|
||||
} else {
|
||||
/* Take left extreme. */
|
||||
result = lines[lineNo].point + tLeft * lines[lineNo].direction;
|
||||
}
|
||||
} else {
|
||||
/* Optimize closest point. */
|
||||
const float t =
|
||||
lines[lineNo].direction * (optVelocity - lines[lineNo].point);
|
||||
|
||||
if (t < tLeft) {
|
||||
result = lines[lineNo].point + tLeft * lines[lineNo].direction;
|
||||
} else if (t > tRight) {
|
||||
result = lines[lineNo].point + tRight * lines[lineNo].direction;
|
||||
} else {
|
||||
result = lines[lineNo].point + t * lines[lineNo].direction;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* @relates Agent2D
|
||||
* @brief Solves a two-dimensional linear program subject to linear
|
||||
* constraints defined by lines and a circular constraint.
|
||||
* @param[in] lines Lines defining the linear constraints.
|
||||
* @param[in] radius The radius of the circular constraint.
|
||||
* @param[in] optVelocity The optimization velocity.
|
||||
* @param[in] directionOpt True if the direction should be optimized.
|
||||
* @param[in, out] result A reference to the result of the linear program.
|
||||
* @return The number of the line it fails on, and the number of lines
|
||||
* if successful.
|
||||
*/
|
||||
std::size_t linearProgram2(const std::vector<Line> &lines, float radius,
|
||||
const Vector2 &optVelocity, bool directionOpt,
|
||||
Vector2 &result) { /* NOLINT(runtime/references) */
|
||||
if (directionOpt) {
|
||||
/* Optimize direction. Note that the optimization velocity is of unit length
|
||||
* in this case.
|
||||
*/
|
||||
result = optVelocity * radius;
|
||||
} else if (absSq(optVelocity) > radius * radius) {
|
||||
/* Optimize closest point and outside circle. */
|
||||
result = normalize(optVelocity) * radius;
|
||||
} else {
|
||||
/* Optimize closest point and inside circle. */
|
||||
result = optVelocity;
|
||||
}
|
||||
|
||||
for (std::size_t i = 0U; i < lines.size(); ++i) {
|
||||
if (det(lines[i].direction, lines[i].point - result) > 0.0F) {
|
||||
/* Result does not satisfy constraint i. Compute new optimal result. */
|
||||
const Vector2 tempResult = result;
|
||||
|
||||
if (!linearProgram1(lines, i, radius, optVelocity, directionOpt,
|
||||
result)) {
|
||||
result = tempResult;
|
||||
|
||||
return i;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return lines.size();
|
||||
}
|
||||
|
||||
/**
|
||||
* @relates Agent2D
|
||||
* @brief Solves a two-dimensional linear program subject to linear
|
||||
* constraints defined by lines and a circular constraint.
|
||||
* @param[in] lines Lines defining the linear constraints.
|
||||
* @param[in] numObstLines Count of obstacle lines.
|
||||
* @param[in] beginLine The line on which the 2-d linear program failed.
|
||||
* @param[in] radius The radius of the circular constraint.
|
||||
* @param[in, out] result A reference to the result of the linear program.
|
||||
*/
|
||||
void linearProgram3(const std::vector<Line> &lines, std::size_t numObstLines,
|
||||
std::size_t beginLine, float radius,
|
||||
Vector2 &result) { /* NOLINT(runtime/references) */
|
||||
float distance = 0.0F;
|
||||
|
||||
for (std::size_t i = beginLine; i < lines.size(); ++i) {
|
||||
if (det(lines[i].direction, lines[i].point - result) > distance) {
|
||||
/* Result does not satisfy constraint of line i. */
|
||||
std::vector<Line> projLines(
|
||||
lines.begin(),
|
||||
lines.begin() + static_cast<std::ptrdiff_t>(numObstLines));
|
||||
|
||||
for (std::size_t j = numObstLines; j < i; ++j) {
|
||||
Line line;
|
||||
|
||||
const float determinant = det(lines[i].direction, lines[j].direction);
|
||||
|
||||
if (std::fabs(determinant) <= RVO2D_EPSILON) {
|
||||
/* Line i and line j are parallel. */
|
||||
if (lines[i].direction * lines[j].direction > 0.0F) {
|
||||
/* Line i and line j point in the same direction. */
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Line i and line j point in opposite direction. */
|
||||
line.point = 0.5F * (lines[i].point + lines[j].point);
|
||||
} else {
|
||||
line.point = lines[i].point + (det(lines[j].direction,
|
||||
lines[i].point - lines[j].point) /
|
||||
determinant) *
|
||||
lines[i].direction;
|
||||
}
|
||||
|
||||
line.direction = normalize(lines[j].direction - lines[i].direction);
|
||||
projLines.push_back(line);
|
||||
}
|
||||
|
||||
const Vector2 tempResult = result;
|
||||
|
||||
if (linearProgram2(
|
||||
projLines, radius,
|
||||
Vector2(-lines[i].direction.y(), lines[i].direction.x()), true,
|
||||
result) < projLines.size()) {
|
||||
/* This should in principle not happen. The result is by definition
|
||||
* already in the feasible region of this linear program. If it fails,
|
||||
* it is due to small floating point error, and the current result is
|
||||
* kept. */
|
||||
result = tempResult;
|
||||
}
|
||||
|
||||
distance = det(lines[i].direction, lines[i].point - result);
|
||||
}
|
||||
}
|
||||
}
|
||||
} /* namespace */
|
||||
|
||||
Agent2D::Agent2D()
|
||||
: id_(0U),
|
||||
maxNeighbors_(0U),
|
||||
maxSpeed_(0.0F),
|
||||
neighborDist_(0.0F),
|
||||
radius_(0.0F),
|
||||
timeHorizon_(0.0F),
|
||||
timeHorizonObst_(0.0F) {}
|
||||
|
||||
Agent2D::~Agent2D() {}
|
||||
|
||||
void Agent2D::computeNeighbors(const KdTree2D *kdTree) {
|
||||
obstacleNeighbors_.clear();
|
||||
const float range = timeHorizonObst_ * maxSpeed_ + radius_;
|
||||
kdTree->computeObstacleNeighbors(this, range * range);
|
||||
|
||||
agentNeighbors_.clear();
|
||||
|
||||
if (maxNeighbors_ > 0U) {
|
||||
float rangeSq = neighborDist_ * neighborDist_;
|
||||
kdTree->computeAgentNeighbors(this, rangeSq);
|
||||
}
|
||||
}
|
||||
|
||||
/* Search for the best new velocity. */
|
||||
void Agent2D::computeNewVelocity(float timeStep) {
|
||||
orcaLines_.clear();
|
||||
|
||||
const float invTimeHorizonObst = 1.0F / timeHorizonObst_;
|
||||
|
||||
/* Create obstacle ORCA lines. */
|
||||
for (std::size_t i = 0U; i < obstacleNeighbors_.size(); ++i) {
|
||||
const Obstacle2D *obstacle1 = obstacleNeighbors_[i].second;
|
||||
const Obstacle2D *obstacle2 = obstacle1->next_;
|
||||
|
||||
const Vector2 relativePosition1 = obstacle1->point_ - position_;
|
||||
const Vector2 relativePosition2 = obstacle2->point_ - position_;
|
||||
|
||||
/* Check if velocity obstacle of obstacle is already taken care of by
|
||||
* previously constructed obstacle ORCA lines. */
|
||||
bool alreadyCovered = false;
|
||||
|
||||
for (std::size_t j = 0U; j < orcaLines_.size(); ++j) {
|
||||
if (det(invTimeHorizonObst * relativePosition1 - orcaLines_[j].point,
|
||||
orcaLines_[j].direction) -
|
||||
invTimeHorizonObst * radius_ >=
|
||||
-RVO2D_EPSILON &&
|
||||
det(invTimeHorizonObst * relativePosition2 - orcaLines_[j].point,
|
||||
orcaLines_[j].direction) -
|
||||
invTimeHorizonObst * radius_ >=
|
||||
-RVO2D_EPSILON) {
|
||||
alreadyCovered = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (alreadyCovered) {
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Not yet covered. Check for collisions. */
|
||||
const float distSq1 = absSq(relativePosition1);
|
||||
const float distSq2 = absSq(relativePosition2);
|
||||
|
||||
const float radiusSq = radius_ * radius_;
|
||||
|
||||
const Vector2 obstacleVector = obstacle2->point_ - obstacle1->point_;
|
||||
const float s =
|
||||
(-relativePosition1 * obstacleVector) / absSq(obstacleVector);
|
||||
const float distSqLine = absSq(-relativePosition1 - s * obstacleVector);
|
||||
|
||||
Line line;
|
||||
|
||||
if (s < 0.0F && distSq1 <= radiusSq) {
|
||||
/* Collision with left vertex. Ignore if non-convex. */
|
||||
if (obstacle1->isConvex_) {
|
||||
line.point = Vector2(0.0F, 0.0F);
|
||||
line.direction =
|
||||
normalize(Vector2(-relativePosition1.y(), relativePosition1.x()));
|
||||
orcaLines_.push_back(line);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (s > 1.0F && distSq2 <= radiusSq) {
|
||||
/* Collision with right vertex. Ignore if non-convex or if it will be
|
||||
* taken care of by neighoring obstace */
|
||||
if (obstacle2->isConvex_ &&
|
||||
det(relativePosition2, obstacle2->direction_) >= 0.0F) {
|
||||
line.point = Vector2(0.0F, 0.0F);
|
||||
line.direction =
|
||||
normalize(Vector2(-relativePosition2.y(), relativePosition2.x()));
|
||||
orcaLines_.push_back(line);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
if (s >= 0.0F && s <= 1.0F && distSqLine <= radiusSq) {
|
||||
/* Collision with obstacle segment. */
|
||||
line.point = Vector2(0.0F, 0.0F);
|
||||
line.direction = -obstacle1->direction_;
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
|
||||
/* No collision. Compute legs. When obliquely viewed, both legs can come
|
||||
* from a single vertex. Legs extend cut-off line when nonconvex vertex. */
|
||||
Vector2 leftLegDirection;
|
||||
Vector2 rightLegDirection;
|
||||
|
||||
if (s < 0.0F && distSqLine <= radiusSq) {
|
||||
/* Obstacle2D viewed obliquely so that left vertex defines velocity
|
||||
* obstacle. */
|
||||
if (!obstacle1->isConvex_) {
|
||||
/* Ignore obstacle. */
|
||||
continue;
|
||||
}
|
||||
|
||||
obstacle2 = obstacle1;
|
||||
|
||||
const float leg1 = std::sqrt(distSq1 - radiusSq);
|
||||
leftLegDirection =
|
||||
Vector2(
|
||||
relativePosition1.x() * leg1 - relativePosition1.y() * radius_,
|
||||
relativePosition1.x() * radius_ + relativePosition1.y() * leg1) /
|
||||
distSq1;
|
||||
rightLegDirection =
|
||||
Vector2(
|
||||
relativePosition1.x() * leg1 + relativePosition1.y() * radius_,
|
||||
-relativePosition1.x() * radius_ + relativePosition1.y() * leg1) /
|
||||
distSq1;
|
||||
} else if (s > 1.0F && distSqLine <= radiusSq) {
|
||||
/* Obstacle2D viewed obliquely so that right vertex defines velocity
|
||||
* obstacle. */
|
||||
if (!obstacle2->isConvex_) {
|
||||
/* Ignore obstacle. */
|
||||
continue;
|
||||
}
|
||||
|
||||
obstacle1 = obstacle2;
|
||||
|
||||
const float leg2 = std::sqrt(distSq2 - radiusSq);
|
||||
leftLegDirection =
|
||||
Vector2(
|
||||
relativePosition2.x() * leg2 - relativePosition2.y() * radius_,
|
||||
relativePosition2.x() * radius_ + relativePosition2.y() * leg2) /
|
||||
distSq2;
|
||||
rightLegDirection =
|
||||
Vector2(
|
||||
relativePosition2.x() * leg2 + relativePosition2.y() * radius_,
|
||||
-relativePosition2.x() * radius_ + relativePosition2.y() * leg2) /
|
||||
distSq2;
|
||||
} else {
|
||||
/* Usual situation. */
|
||||
if (obstacle1->isConvex_) {
|
||||
const float leg1 = std::sqrt(distSq1 - radiusSq);
|
||||
leftLegDirection = Vector2(relativePosition1.x() * leg1 -
|
||||
relativePosition1.y() * radius_,
|
||||
relativePosition1.x() * radius_ +
|
||||
relativePosition1.y() * leg1) /
|
||||
distSq1;
|
||||
} else {
|
||||
/* Left vertex non-convex; left leg extends cut-off line. */
|
||||
leftLegDirection = -obstacle1->direction_;
|
||||
}
|
||||
|
||||
if (obstacle2->isConvex_) {
|
||||
const float leg2 = std::sqrt(distSq2 - radiusSq);
|
||||
rightLegDirection = Vector2(relativePosition2.x() * leg2 +
|
||||
relativePosition2.y() * radius_,
|
||||
-relativePosition2.x() * radius_ +
|
||||
relativePosition2.y() * leg2) /
|
||||
distSq2;
|
||||
} else {
|
||||
/* Right vertex non-convex; right leg extends cut-off line. */
|
||||
rightLegDirection = obstacle1->direction_;
|
||||
}
|
||||
}
|
||||
|
||||
/* Legs can never point into neighboring edge when convex vertex, take
|
||||
* cutoff-line of neighboring edge instead. If velocity projected on
|
||||
* "foreign" leg, no constraint is added. */
|
||||
const Obstacle2D *const leftNeighbor = obstacle1->previous_;
|
||||
|
||||
bool isLeftLegForeign = false;
|
||||
bool isRightLegForeign = false;
|
||||
|
||||
if (obstacle1->isConvex_ &&
|
||||
det(leftLegDirection, -leftNeighbor->direction_) >= 0.0F) {
|
||||
/* Left leg points into obstacle. */
|
||||
leftLegDirection = -leftNeighbor->direction_;
|
||||
isLeftLegForeign = true;
|
||||
}
|
||||
|
||||
if (obstacle2->isConvex_ &&
|
||||
det(rightLegDirection, obstacle2->direction_) <= 0.0F) {
|
||||
/* Right leg points into obstacle. */
|
||||
rightLegDirection = obstacle2->direction_;
|
||||
isRightLegForeign = true;
|
||||
}
|
||||
|
||||
/* Compute cut-off centers. */
|
||||
const Vector2 leftCutoff =
|
||||
invTimeHorizonObst * (obstacle1->point_ - position_);
|
||||
const Vector2 rightCutoff =
|
||||
invTimeHorizonObst * (obstacle2->point_ - position_);
|
||||
const Vector2 cutoffVector = rightCutoff - leftCutoff;
|
||||
|
||||
/* Project current velocity on velocity obstacle. */
|
||||
|
||||
/* Check if current velocity is projected on cutoff circles. */
|
||||
const float t =
|
||||
obstacle1 == obstacle2
|
||||
? 0.5F
|
||||
: (velocity_ - leftCutoff) * cutoffVector / absSq(cutoffVector);
|
||||
const float tLeft = (velocity_ - leftCutoff) * leftLegDirection;
|
||||
const float tRight = (velocity_ - rightCutoff) * rightLegDirection;
|
||||
|
||||
if ((t < 0.0F && tLeft < 0.0F) ||
|
||||
(obstacle1 == obstacle2 && tLeft < 0.0F && tRight < 0.0F)) {
|
||||
/* Project on left cut-off circle. */
|
||||
const Vector2 unitW = normalize(velocity_ - leftCutoff);
|
||||
|
||||
line.direction = Vector2(unitW.y(), -unitW.x());
|
||||
line.point = leftCutoff + radius_ * invTimeHorizonObst * unitW;
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
|
||||
if (t > 1.0F && tRight < 0.0F) {
|
||||
/* Project on right cut-off circle. */
|
||||
const Vector2 unitW = normalize(velocity_ - rightCutoff);
|
||||
|
||||
line.direction = Vector2(unitW.y(), -unitW.x());
|
||||
line.point = rightCutoff + radius_ * invTimeHorizonObst * unitW;
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Project on left leg, right leg, or cut-off line, whichever is closest to
|
||||
* velocity. */
|
||||
const float distSqCutoff =
|
||||
(t < 0.0F || t > 1.0F || obstacle1 == obstacle2)
|
||||
? std::numeric_limits<float>::infinity()
|
||||
: absSq(velocity_ - (leftCutoff + t * cutoffVector));
|
||||
const float distSqLeft =
|
||||
tLeft < 0.0F
|
||||
? std::numeric_limits<float>::infinity()
|
||||
: absSq(velocity_ - (leftCutoff + tLeft * leftLegDirection));
|
||||
const float distSqRight =
|
||||
tRight < 0.0F
|
||||
? std::numeric_limits<float>::infinity()
|
||||
: absSq(velocity_ - (rightCutoff + tRight * rightLegDirection));
|
||||
|
||||
if (distSqCutoff <= distSqLeft && distSqCutoff <= distSqRight) {
|
||||
/* Project on cut-off line. */
|
||||
line.direction = -obstacle1->direction_;
|
||||
line.point =
|
||||
leftCutoff + radius_ * invTimeHorizonObst *
|
||||
Vector2(-line.direction.y(), line.direction.x());
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
|
||||
if (distSqLeft <= distSqRight) {
|
||||
/* Project on left leg. */
|
||||
if (isLeftLegForeign) {
|
||||
continue;
|
||||
}
|
||||
|
||||
line.direction = leftLegDirection;
|
||||
line.point =
|
||||
leftCutoff + radius_ * invTimeHorizonObst *
|
||||
Vector2(-line.direction.y(), line.direction.x());
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Project on right leg. */
|
||||
if (isRightLegForeign) {
|
||||
continue;
|
||||
}
|
||||
|
||||
line.direction = -rightLegDirection;
|
||||
line.point =
|
||||
rightCutoff + radius_ * invTimeHorizonObst *
|
||||
Vector2(-line.direction.y(), line.direction.x());
|
||||
orcaLines_.push_back(line);
|
||||
}
|
||||
|
||||
const std::size_t numObstLines = orcaLines_.size();
|
||||
|
||||
const float invTimeHorizon = 1.0F / timeHorizon_;
|
||||
|
||||
/* Create agent ORCA lines. */
|
||||
for (std::size_t i = 0U; i < agentNeighbors_.size(); ++i) {
|
||||
const Agent2D *const other = agentNeighbors_[i].second;
|
||||
|
||||
const Vector2 relativePosition = other->position_ - position_;
|
||||
const Vector2 relativeVelocity = velocity_ - other->velocity_;
|
||||
const float distSq = absSq(relativePosition);
|
||||
const float combinedRadius = radius_ + other->radius_;
|
||||
const float combinedRadiusSq = combinedRadius * combinedRadius;
|
||||
|
||||
Line line;
|
||||
Vector2 u;
|
||||
|
||||
if (distSq > combinedRadiusSq) {
|
||||
/* No collision. */
|
||||
const Vector2 w = relativeVelocity - invTimeHorizon * relativePosition;
|
||||
/* Vector from cutoff center to relative velocity. */
|
||||
const float wLengthSq = absSq(w);
|
||||
|
||||
const float dotProduct = w * relativePosition;
|
||||
|
||||
if (dotProduct < 0.0F &&
|
||||
dotProduct * dotProduct > combinedRadiusSq * wLengthSq) {
|
||||
/* Project on cut-off circle. */
|
||||
const float wLength = std::sqrt(wLengthSq);
|
||||
const Vector2 unitW = w / wLength;
|
||||
|
||||
line.direction = Vector2(unitW.y(), -unitW.x());
|
||||
u = (combinedRadius * invTimeHorizon - wLength) * unitW;
|
||||
} else {
|
||||
/* Project on legs. */
|
||||
const float leg = std::sqrt(distSq - combinedRadiusSq);
|
||||
|
||||
if (det(relativePosition, w) > 0.0F) {
|
||||
/* Project on left leg. */
|
||||
line.direction = Vector2(relativePosition.x() * leg -
|
||||
relativePosition.y() * combinedRadius,
|
||||
relativePosition.x() * combinedRadius +
|
||||
relativePosition.y() * leg) /
|
||||
distSq;
|
||||
} else {
|
||||
/* Project on right leg. */
|
||||
line.direction = -Vector2(relativePosition.x() * leg +
|
||||
relativePosition.y() * combinedRadius,
|
||||
-relativePosition.x() * combinedRadius +
|
||||
relativePosition.y() * leg) /
|
||||
distSq;
|
||||
}
|
||||
|
||||
u = (relativeVelocity * line.direction) * line.direction -
|
||||
relativeVelocity;
|
||||
}
|
||||
} else {
|
||||
/* Collision. Project on cut-off circle of time timeStep. */
|
||||
const float invTimeStep = 1.0F / timeStep;
|
||||
|
||||
/* Vector from cutoff center to relative velocity. */
|
||||
const Vector2 w = relativeVelocity - invTimeStep * relativePosition;
|
||||
|
||||
const float wLength = abs(w);
|
||||
const Vector2 unitW = w / wLength;
|
||||
|
||||
line.direction = Vector2(unitW.y(), -unitW.x());
|
||||
u = (combinedRadius * invTimeStep - wLength) * unitW;
|
||||
}
|
||||
|
||||
line.point = velocity_ + 0.5F * u;
|
||||
orcaLines_.push_back(line);
|
||||
}
|
||||
|
||||
const std::size_t lineFail =
|
||||
linearProgram2(orcaLines_, maxSpeed_, prefVelocity_, false, newVelocity_);
|
||||
|
||||
if (lineFail < orcaLines_.size()) {
|
||||
linearProgram3(orcaLines_, numObstLines, lineFail, maxSpeed_, newVelocity_);
|
||||
}
|
||||
}
|
||||
|
||||
void Agent2D::insertAgentNeighbor(const Agent2D *agent, float &rangeSq) {
|
||||
// no point processing same agent
|
||||
if (this == agent) {
|
||||
return;
|
||||
}
|
||||
// ignore other agent if layers/mask bitmasks have no matching bit
|
||||
if ((avoidance_mask_ & agent->avoidance_layers_) == 0) {
|
||||
return;
|
||||
}
|
||||
// ignore other agent if this agent is below or above
|
||||
if ((elevation_ > agent->elevation_ + agent->height_) || (elevation_ + height_ < agent->elevation_)) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (avoidance_priority_ > agent->avoidance_priority_) {
|
||||
return;
|
||||
}
|
||||
const float distSq = absSq(position_ - agent->position_);
|
||||
|
||||
if (distSq < rangeSq) {
|
||||
if (agentNeighbors_.size() < maxNeighbors_) {
|
||||
agentNeighbors_.push_back(std::make_pair(distSq, agent));
|
||||
}
|
||||
|
||||
std::size_t i = agentNeighbors_.size() - 1U;
|
||||
|
||||
while (i != 0U && distSq < agentNeighbors_[i - 1U].first) {
|
||||
agentNeighbors_[i] = agentNeighbors_[i - 1U];
|
||||
--i;
|
||||
}
|
||||
|
||||
agentNeighbors_[i] = std::make_pair(distSq, agent);
|
||||
|
||||
if (agentNeighbors_.size() == maxNeighbors_) {
|
||||
rangeSq = agentNeighbors_.back().first;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void Agent2D::insertObstacleNeighbor(const Obstacle2D *obstacle, float rangeSq) {
|
||||
const Obstacle2D *const nextObstacle = obstacle->next_;
|
||||
|
||||
float distSq = 0.0F;
|
||||
const float r = ((position_ - obstacle->point_) *
|
||||
(nextObstacle->point_ - obstacle->point_)) /
|
||||
absSq(nextObstacle->point_ - obstacle->point_);
|
||||
|
||||
if (r < 0.0F) {
|
||||
distSq = absSq(position_ - obstacle->point_);
|
||||
} else if (r > 1.0F) {
|
||||
distSq = absSq(position_ - nextObstacle->point_);
|
||||
} else {
|
||||
distSq = absSq(position_ - (obstacle->point_ +
|
||||
r * (nextObstacle->point_ - obstacle->point_)));
|
||||
}
|
||||
|
||||
if (distSq < rangeSq) {
|
||||
obstacleNeighbors_.push_back(std::make_pair(distSq, obstacle));
|
||||
|
||||
std::size_t i = obstacleNeighbors_.size() - 1U;
|
||||
|
||||
while (i != 0U && distSq < obstacleNeighbors_[i - 1U].first) {
|
||||
obstacleNeighbors_[i] = obstacleNeighbors_[i - 1U];
|
||||
--i;
|
||||
}
|
||||
|
||||
obstacleNeighbors_[i] = std::make_pair(distSq, obstacle);
|
||||
}
|
||||
}
|
||||
|
||||
void Agent2D::update(float timeStep) {
|
||||
velocity_ = newVelocity_;
|
||||
position_ += velocity_ * timeStep;
|
||||
}
|
||||
} /* namespace RVO2D */
|
||||
@ -1,594 +0,0 @@
|
||||
/*
|
||||
* Agent2d.cpp
|
||||
* RVO2 Library
|
||||
*
|
||||
* Copyright 2008 University of North Carolina at Chapel Hill
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <http://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "Agent2d.h"
|
||||
|
||||
#include "KdTree2d.h"
|
||||
#include "Obstacle2d.h"
|
||||
|
||||
namespace RVO2D {
|
||||
Agent2D::Agent2D() : maxNeighbors_(0), maxSpeed_(0.0f), neighborDist_(0.0f), radius_(0.0f), timeHorizon_(0.0f), timeHorizonObst_(0.0f), id_(0) { }
|
||||
|
||||
void Agent2D::computeNeighbors(RVOSimulator2D *sim_)
|
||||
{
|
||||
obstacleNeighbors_.clear();
|
||||
float rangeSq = sqr(timeHorizonObst_ * maxSpeed_ + radius_);
|
||||
sim_->kdTree_->computeObstacleNeighbors(this, rangeSq);
|
||||
|
||||
agentNeighbors_.clear();
|
||||
|
||||
if (maxNeighbors_ > 0) {
|
||||
rangeSq = sqr(neighborDist_);
|
||||
sim_->kdTree_->computeAgentNeighbors(this, rangeSq);
|
||||
}
|
||||
}
|
||||
|
||||
/* Search for the best new velocity. */
|
||||
void Agent2D::computeNewVelocity(RVOSimulator2D *sim_)
|
||||
{
|
||||
orcaLines_.clear();
|
||||
|
||||
const float invTimeHorizonObst = 1.0f / timeHorizonObst_;
|
||||
|
||||
/* Create obstacle ORCA lines. */
|
||||
for (size_t i = 0; i < obstacleNeighbors_.size(); ++i) {
|
||||
|
||||
const Obstacle2D *obstacle1 = obstacleNeighbors_[i].second;
|
||||
const Obstacle2D *obstacle2 = obstacle1->nextObstacle_;
|
||||
|
||||
const Vector2 relativePosition1 = obstacle1->point_ - position_;
|
||||
const Vector2 relativePosition2 = obstacle2->point_ - position_;
|
||||
|
||||
/*
|
||||
* Check if velocity obstacle of obstacle is already taken care of by
|
||||
* previously constructed obstacle ORCA lines.
|
||||
*/
|
||||
bool alreadyCovered = false;
|
||||
|
||||
for (size_t j = 0; j < orcaLines_.size(); ++j) {
|
||||
if (det(invTimeHorizonObst * relativePosition1 - orcaLines_[j].point, orcaLines_[j].direction) - invTimeHorizonObst * radius_ >= -RVO_EPSILON && det(invTimeHorizonObst * relativePosition2 - orcaLines_[j].point, orcaLines_[j].direction) - invTimeHorizonObst * radius_ >= -RVO_EPSILON) {
|
||||
alreadyCovered = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (alreadyCovered) {
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Not yet covered. Check for collisions. */
|
||||
|
||||
const float distSq1 = absSq(relativePosition1);
|
||||
const float distSq2 = absSq(relativePosition2);
|
||||
|
||||
const float radiusSq = sqr(radius_);
|
||||
|
||||
const Vector2 obstacleVector = obstacle2->point_ - obstacle1->point_;
|
||||
const float s = (-relativePosition1 * obstacleVector) / absSq(obstacleVector);
|
||||
const float distSqLine = absSq(-relativePosition1 - s * obstacleVector);
|
||||
|
||||
Line line;
|
||||
|
||||
if (s < 0.0f && distSq1 <= radiusSq) {
|
||||
/* Collision with left vertex. Ignore if non-convex. */
|
||||
if (obstacle1->isConvex_) {
|
||||
line.point = Vector2(0.0f, 0.0f);
|
||||
line.direction = normalize(Vector2(-relativePosition1.y(), relativePosition1.x()));
|
||||
orcaLines_.push_back(line);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
else if (s > 1.0f && distSq2 <= radiusSq) {
|
||||
/* Collision with right vertex. Ignore if non-convex
|
||||
* or if it will be taken care of by neighoring obstace */
|
||||
if (obstacle2->isConvex_ && det(relativePosition2, obstacle2->unitDir_) >= 0.0f) {
|
||||
line.point = Vector2(0.0f, 0.0f);
|
||||
line.direction = normalize(Vector2(-relativePosition2.y(), relativePosition2.x()));
|
||||
orcaLines_.push_back(line);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
else if (s >= 0.0f && s < 1.0f && distSqLine <= radiusSq) {
|
||||
/* Collision with obstacle segment. */
|
||||
line.point = Vector2(0.0f, 0.0f);
|
||||
line.direction = -obstacle1->unitDir_;
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
|
||||
/*
|
||||
* No collision.
|
||||
* Compute legs. When obliquely viewed, both legs can come from a single
|
||||
* vertex. Legs extend cut-off line when nonconvex vertex.
|
||||
*/
|
||||
|
||||
Vector2 leftLegDirection, rightLegDirection;
|
||||
|
||||
if (s < 0.0f && distSqLine <= radiusSq) {
|
||||
/*
|
||||
* Obstacle viewed obliquely so that left vertex
|
||||
* defines velocity obstacle.
|
||||
*/
|
||||
if (!obstacle1->isConvex_) {
|
||||
/* Ignore obstacle. */
|
||||
continue;
|
||||
}
|
||||
|
||||
obstacle2 = obstacle1;
|
||||
|
||||
const float leg1 = std::sqrt(distSq1 - radiusSq);
|
||||
leftLegDirection = Vector2(relativePosition1.x() * leg1 - relativePosition1.y() * radius_, relativePosition1.x() * radius_ + relativePosition1.y() * leg1) / distSq1;
|
||||
rightLegDirection = Vector2(relativePosition1.x() * leg1 + relativePosition1.y() * radius_, -relativePosition1.x() * radius_ + relativePosition1.y() * leg1) / distSq1;
|
||||
}
|
||||
else if (s > 1.0f && distSqLine <= radiusSq) {
|
||||
/*
|
||||
* Obstacle viewed obliquely so that
|
||||
* right vertex defines velocity obstacle.
|
||||
*/
|
||||
if (!obstacle2->isConvex_) {
|
||||
/* Ignore obstacle. */
|
||||
continue;
|
||||
}
|
||||
|
||||
obstacle1 = obstacle2;
|
||||
|
||||
const float leg2 = std::sqrt(distSq2 - radiusSq);
|
||||
leftLegDirection = Vector2(relativePosition2.x() * leg2 - relativePosition2.y() * radius_, relativePosition2.x() * radius_ + relativePosition2.y() * leg2) / distSq2;
|
||||
rightLegDirection = Vector2(relativePosition2.x() * leg2 + relativePosition2.y() * radius_, -relativePosition2.x() * radius_ + relativePosition2.y() * leg2) / distSq2;
|
||||
}
|
||||
else {
|
||||
/* Usual situation. */
|
||||
if (obstacle1->isConvex_) {
|
||||
const float leg1 = std::sqrt(distSq1 - radiusSq);
|
||||
leftLegDirection = Vector2(relativePosition1.x() * leg1 - relativePosition1.y() * radius_, relativePosition1.x() * radius_ + relativePosition1.y() * leg1) / distSq1;
|
||||
}
|
||||
else {
|
||||
/* Left vertex non-convex; left leg extends cut-off line. */
|
||||
leftLegDirection = -obstacle1->unitDir_;
|
||||
}
|
||||
|
||||
if (obstacle2->isConvex_) {
|
||||
const float leg2 = std::sqrt(distSq2 - radiusSq);
|
||||
rightLegDirection = Vector2(relativePosition2.x() * leg2 + relativePosition2.y() * radius_, -relativePosition2.x() * radius_ + relativePosition2.y() * leg2) / distSq2;
|
||||
}
|
||||
else {
|
||||
/* Right vertex non-convex; right leg extends cut-off line. */
|
||||
rightLegDirection = obstacle1->unitDir_;
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
* Legs can never point into neighboring edge when convex vertex,
|
||||
* take cutoff-line of neighboring edge instead. If velocity projected on
|
||||
* "foreign" leg, no constraint is added.
|
||||
*/
|
||||
|
||||
const Obstacle2D *const leftNeighbor = obstacle1->prevObstacle_;
|
||||
|
||||
bool isLeftLegForeign = false;
|
||||
bool isRightLegForeign = false;
|
||||
|
||||
if (obstacle1->isConvex_ && det(leftLegDirection, -leftNeighbor->unitDir_) >= 0.0f) {
|
||||
/* Left leg points into obstacle. */
|
||||
leftLegDirection = -leftNeighbor->unitDir_;
|
||||
isLeftLegForeign = true;
|
||||
}
|
||||
|
||||
if (obstacle2->isConvex_ && det(rightLegDirection, obstacle2->unitDir_) <= 0.0f) {
|
||||
/* Right leg points into obstacle. */
|
||||
rightLegDirection = obstacle2->unitDir_;
|
||||
isRightLegForeign = true;
|
||||
}
|
||||
|
||||
/* Compute cut-off centers. */
|
||||
const Vector2 leftCutoff = invTimeHorizonObst * (obstacle1->point_ - position_);
|
||||
const Vector2 rightCutoff = invTimeHorizonObst * (obstacle2->point_ - position_);
|
||||
const Vector2 cutoffVec = rightCutoff - leftCutoff;
|
||||
|
||||
/* Project current velocity on velocity obstacle. */
|
||||
|
||||
/* Check if current velocity is projected on cutoff circles. */
|
||||
const float t = (obstacle1 == obstacle2 ? 0.5f : ((velocity_ - leftCutoff) * cutoffVec) / absSq(cutoffVec));
|
||||
const float tLeft = ((velocity_ - leftCutoff) * leftLegDirection);
|
||||
const float tRight = ((velocity_ - rightCutoff) * rightLegDirection);
|
||||
|
||||
if ((t < 0.0f && tLeft < 0.0f) || (obstacle1 == obstacle2 && tLeft < 0.0f && tRight < 0.0f)) {
|
||||
/* Project on left cut-off circle. */
|
||||
const Vector2 unitW = normalize(velocity_ - leftCutoff);
|
||||
|
||||
line.direction = Vector2(unitW.y(), -unitW.x());
|
||||
line.point = leftCutoff + radius_ * invTimeHorizonObst * unitW;
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
else if (t > 1.0f && tRight < 0.0f) {
|
||||
/* Project on right cut-off circle. */
|
||||
const Vector2 unitW = normalize(velocity_ - rightCutoff);
|
||||
|
||||
line.direction = Vector2(unitW.y(), -unitW.x());
|
||||
line.point = rightCutoff + radius_ * invTimeHorizonObst * unitW;
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
|
||||
/*
|
||||
* Project on left leg, right leg, or cut-off line, whichever is closest
|
||||
* to velocity.
|
||||
*/
|
||||
const float distSqCutoff = ((t < 0.0f || t > 1.0f || obstacle1 == obstacle2) ? std::numeric_limits<float>::infinity() : absSq(velocity_ - (leftCutoff + t * cutoffVec)));
|
||||
const float distSqLeft = ((tLeft < 0.0f) ? std::numeric_limits<float>::infinity() : absSq(velocity_ - (leftCutoff + tLeft * leftLegDirection)));
|
||||
const float distSqRight = ((tRight < 0.0f) ? std::numeric_limits<float>::infinity() : absSq(velocity_ - (rightCutoff + tRight * rightLegDirection)));
|
||||
|
||||
if (distSqCutoff <= distSqLeft && distSqCutoff <= distSqRight) {
|
||||
/* Project on cut-off line. */
|
||||
line.direction = -obstacle1->unitDir_;
|
||||
line.point = leftCutoff + radius_ * invTimeHorizonObst * Vector2(-line.direction.y(), line.direction.x());
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
else if (distSqLeft <= distSqRight) {
|
||||
/* Project on left leg. */
|
||||
if (isLeftLegForeign) {
|
||||
continue;
|
||||
}
|
||||
|
||||
line.direction = leftLegDirection;
|
||||
line.point = leftCutoff + radius_ * invTimeHorizonObst * Vector2(-line.direction.y(), line.direction.x());
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
else {
|
||||
/* Project on right leg. */
|
||||
if (isRightLegForeign) {
|
||||
continue;
|
||||
}
|
||||
|
||||
line.direction = -rightLegDirection;
|
||||
line.point = rightCutoff + radius_ * invTimeHorizonObst * Vector2(-line.direction.y(), line.direction.x());
|
||||
orcaLines_.push_back(line);
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
const size_t numObstLines = orcaLines_.size();
|
||||
|
||||
const float invTimeHorizon = 1.0f / timeHorizon_;
|
||||
|
||||
/* Create agent ORCA lines. */
|
||||
for (size_t i = 0; i < agentNeighbors_.size(); ++i) {
|
||||
const Agent2D *const other = agentNeighbors_[i].second;
|
||||
|
||||
//const float timeHorizon_mod = (avoidance_priority_ - other->avoidance_priority_ + 1.0f) * 0.5f;
|
||||
//const float invTimeHorizon = (1.0f / timeHorizon_) * timeHorizon_mod;
|
||||
|
||||
const Vector2 relativePosition = other->position_ - position_;
|
||||
const Vector2 relativeVelocity = velocity_ - other->velocity_;
|
||||
const float distSq = absSq(relativePosition);
|
||||
const float combinedRadius = radius_ + other->radius_;
|
||||
const float combinedRadiusSq = sqr(combinedRadius);
|
||||
|
||||
Line line;
|
||||
Vector2 u;
|
||||
|
||||
if (distSq > combinedRadiusSq) {
|
||||
/* No collision. */
|
||||
const Vector2 w = relativeVelocity - invTimeHorizon * relativePosition;
|
||||
/* Vector from cutoff center to relative velocity. */
|
||||
const float wLengthSq = absSq(w);
|
||||
|
||||
const float dotProduct1 = w * relativePosition;
|
||||
|
||||
if (dotProduct1 < 0.0f && sqr(dotProduct1) > combinedRadiusSq * wLengthSq) {
|
||||
/* Project on cut-off circle. */
|
||||
const float wLength = std::sqrt(wLengthSq);
|
||||
const Vector2 unitW = w / wLength;
|
||||
|
||||
line.direction = Vector2(unitW.y(), -unitW.x());
|
||||
u = (combinedRadius * invTimeHorizon - wLength) * unitW;
|
||||
}
|
||||
else {
|
||||
/* Project on legs. */
|
||||
const float leg = std::sqrt(distSq - combinedRadiusSq);
|
||||
|
||||
if (det(relativePosition, w) > 0.0f) {
|
||||
/* Project on left leg. */
|
||||
line.direction = Vector2(relativePosition.x() * leg - relativePosition.y() * combinedRadius, relativePosition.x() * combinedRadius + relativePosition.y() * leg) / distSq;
|
||||
}
|
||||
else {
|
||||
/* Project on right leg. */
|
||||
line.direction = -Vector2(relativePosition.x() * leg + relativePosition.y() * combinedRadius, -relativePosition.x() * combinedRadius + relativePosition.y() * leg) / distSq;
|
||||
}
|
||||
|
||||
const float dotProduct2 = relativeVelocity * line.direction;
|
||||
|
||||
u = dotProduct2 * line.direction - relativeVelocity;
|
||||
}
|
||||
}
|
||||
else {
|
||||
/* Collision. Project on cut-off circle of time timeStep. */
|
||||
const float invTimeStep = 1.0f / sim_->timeStep_;
|
||||
|
||||
/* Vector from cutoff center to relative velocity. */
|
||||
const Vector2 w = relativeVelocity - invTimeStep * relativePosition;
|
||||
|
||||
const float wLength = abs(w);
|
||||
const Vector2 unitW = w / wLength;
|
||||
|
||||
line.direction = Vector2(unitW.y(), -unitW.x());
|
||||
u = (combinedRadius * invTimeStep - wLength) * unitW;
|
||||
}
|
||||
|
||||
line.point = velocity_ + 0.5f * u;
|
||||
orcaLines_.push_back(line);
|
||||
}
|
||||
|
||||
size_t lineFail = linearProgram2(orcaLines_, maxSpeed_, prefVelocity_, false, newVelocity_);
|
||||
|
||||
if (lineFail < orcaLines_.size()) {
|
||||
linearProgram3(orcaLines_, numObstLines, lineFail, maxSpeed_, newVelocity_);
|
||||
}
|
||||
}
|
||||
|
||||
void Agent2D::insertAgentNeighbor(const Agent2D *agent, float &rangeSq)
|
||||
{
|
||||
// no point processing same agent
|
||||
if (this == agent) {
|
||||
return;
|
||||
}
|
||||
// ignore other agent if layers/mask bitmasks have no matching bit
|
||||
if ((avoidance_mask_ & agent->avoidance_layers_) == 0) {
|
||||
return;
|
||||
}
|
||||
// ignore other agent if this agent is below or above
|
||||
if ((elevation_ > agent->elevation_ + agent->height_) || (elevation_ + height_ < agent->elevation_)) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (avoidance_priority_ > agent->avoidance_priority_) {
|
||||
return;
|
||||
}
|
||||
|
||||
const float distSq = absSq(position_ - agent->position_);
|
||||
|
||||
if (distSq < rangeSq) {
|
||||
if (agentNeighbors_.size() < maxNeighbors_) {
|
||||
agentNeighbors_.push_back(std::make_pair(distSq, agent));
|
||||
}
|
||||
|
||||
size_t i = agentNeighbors_.size() - 1;
|
||||
|
||||
while (i != 0 && distSq < agentNeighbors_[i - 1].first) {
|
||||
agentNeighbors_[i] = agentNeighbors_[i - 1];
|
||||
--i;
|
||||
}
|
||||
|
||||
agentNeighbors_[i] = std::make_pair(distSq, agent);
|
||||
|
||||
if (agentNeighbors_.size() == maxNeighbors_) {
|
||||
rangeSq = agentNeighbors_.back().first;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void Agent2D::insertObstacleNeighbor(const Obstacle2D *obstacle, float rangeSq)
|
||||
{
|
||||
const Obstacle2D *const nextObstacle = obstacle->nextObstacle_;
|
||||
|
||||
// ignore obstacle if no matching layer/mask
|
||||
if ((avoidance_mask_ & nextObstacle->avoidance_layers_) == 0) {
|
||||
return;
|
||||
}
|
||||
// ignore obstacle if below or above
|
||||
if ((elevation_ > obstacle->elevation_ + obstacle->height_) || (elevation_ + height_ < obstacle->elevation_)) {
|
||||
return;
|
||||
}
|
||||
|
||||
const float distSq = distSqPointLineSegment(obstacle->point_, nextObstacle->point_, position_);
|
||||
|
||||
if (distSq < rangeSq) {
|
||||
obstacleNeighbors_.push_back(std::make_pair(distSq, obstacle));
|
||||
|
||||
size_t i = obstacleNeighbors_.size() - 1;
|
||||
|
||||
while (i != 0 && distSq < obstacleNeighbors_[i - 1].first) {
|
||||
obstacleNeighbors_[i] = obstacleNeighbors_[i - 1];
|
||||
--i;
|
||||
}
|
||||
|
||||
obstacleNeighbors_[i] = std::make_pair(distSq, obstacle);
|
||||
}
|
||||
//}
|
||||
}
|
||||
|
||||
void Agent2D::update(RVOSimulator2D *sim_)
|
||||
{
|
||||
velocity_ = newVelocity_;
|
||||
position_ += velocity_ * sim_->timeStep_;
|
||||
}
|
||||
|
||||
bool linearProgram1(const std::vector<Line> &lines, size_t lineNo, float radius, const Vector2 &optVelocity, bool directionOpt, Vector2 &result)
|
||||
{
|
||||
const float dotProduct = lines[lineNo].point * lines[lineNo].direction;
|
||||
const float discriminant = sqr(dotProduct) + sqr(radius) - absSq(lines[lineNo].point);
|
||||
|
||||
if (discriminant < 0.0f) {
|
||||
/* Max speed circle fully invalidates line lineNo. */
|
||||
return false;
|
||||
}
|
||||
|
||||
const float sqrtDiscriminant = std::sqrt(discriminant);
|
||||
float tLeft = -dotProduct - sqrtDiscriminant;
|
||||
float tRight = -dotProduct + sqrtDiscriminant;
|
||||
|
||||
for (size_t i = 0; i < lineNo; ++i) {
|
||||
const float denominator = det(lines[lineNo].direction, lines[i].direction);
|
||||
const float numerator = det(lines[i].direction, lines[lineNo].point - lines[i].point);
|
||||
|
||||
if (std::fabs(denominator) <= RVO_EPSILON) {
|
||||
/* Lines lineNo and i are (almost) parallel. */
|
||||
if (numerator < 0.0f) {
|
||||
return false;
|
||||
}
|
||||
else {
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
const float t = numerator / denominator;
|
||||
|
||||
if (denominator >= 0.0f) {
|
||||
/* Line i bounds line lineNo on the right. */
|
||||
tRight = std::min(tRight, t);
|
||||
}
|
||||
else {
|
||||
/* Line i bounds line lineNo on the left. */
|
||||
tLeft = std::max(tLeft, t);
|
||||
}
|
||||
|
||||
if (tLeft > tRight) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (directionOpt) {
|
||||
/* Optimize direction. */
|
||||
if (optVelocity * lines[lineNo].direction > 0.0f) {
|
||||
/* Take right extreme. */
|
||||
result = lines[lineNo].point + tRight * lines[lineNo].direction;
|
||||
}
|
||||
else {
|
||||
/* Take left extreme. */
|
||||
result = lines[lineNo].point + tLeft * lines[lineNo].direction;
|
||||
}
|
||||
}
|
||||
else {
|
||||
/* Optimize closest point. */
|
||||
const float t = lines[lineNo].direction * (optVelocity - lines[lineNo].point);
|
||||
|
||||
if (t < tLeft) {
|
||||
result = lines[lineNo].point + tLeft * lines[lineNo].direction;
|
||||
}
|
||||
else if (t > tRight) {
|
||||
result = lines[lineNo].point + tRight * lines[lineNo].direction;
|
||||
}
|
||||
else {
|
||||
result = lines[lineNo].point + t * lines[lineNo].direction;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
size_t linearProgram2(const std::vector<Line> &lines, float radius, const Vector2 &optVelocity, bool directionOpt, Vector2 &result)
|
||||
{
|
||||
if (directionOpt) {
|
||||
/*
|
||||
* Optimize direction. Note that the optimization velocity is of unit
|
||||
* length in this case.
|
||||
*/
|
||||
result = optVelocity * radius;
|
||||
}
|
||||
else if (absSq(optVelocity) > sqr(radius)) {
|
||||
/* Optimize closest point and outside circle. */
|
||||
result = normalize(optVelocity) * radius;
|
||||
}
|
||||
else {
|
||||
/* Optimize closest point and inside circle. */
|
||||
result = optVelocity;
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < lines.size(); ++i) {
|
||||
if (det(lines[i].direction, lines[i].point - result) > 0.0f) {
|
||||
/* Result does not satisfy constraint i. Compute new optimal result. */
|
||||
const Vector2 tempResult = result;
|
||||
|
||||
if (!linearProgram1(lines, i, radius, optVelocity, directionOpt, result)) {
|
||||
result = tempResult;
|
||||
return i;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return lines.size();
|
||||
}
|
||||
|
||||
void linearProgram3(const std::vector<Line> &lines, size_t numObstLines, size_t beginLine, float radius, Vector2 &result)
|
||||
{
|
||||
float distance = 0.0f;
|
||||
|
||||
for (size_t i = beginLine; i < lines.size(); ++i) {
|
||||
if (det(lines[i].direction, lines[i].point - result) > distance) {
|
||||
/* Result does not satisfy constraint of line i. */
|
||||
std::vector<Line> projLines(lines.begin(), lines.begin() + static_cast<ptrdiff_t>(numObstLines));
|
||||
|
||||
for (size_t j = numObstLines; j < i; ++j) {
|
||||
Line line;
|
||||
|
||||
float determinant = det(lines[i].direction, lines[j].direction);
|
||||
|
||||
if (std::fabs(determinant) <= RVO_EPSILON) {
|
||||
/* Line i and line j are parallel. */
|
||||
if (lines[i].direction * lines[j].direction > 0.0f) {
|
||||
/* Line i and line j point in the same direction. */
|
||||
continue;
|
||||
}
|
||||
else {
|
||||
/* Line i and line j point in opposite direction. */
|
||||
line.point = 0.5f * (lines[i].point + lines[j].point);
|
||||
}
|
||||
}
|
||||
else {
|
||||
line.point = lines[i].point + (det(lines[j].direction, lines[i].point - lines[j].point) / determinant) * lines[i].direction;
|
||||
}
|
||||
|
||||
line.direction = normalize(lines[j].direction - lines[i].direction);
|
||||
projLines.push_back(line);
|
||||
}
|
||||
|
||||
const Vector2 tempResult = result;
|
||||
|
||||
if (linearProgram2(projLines, radius, Vector2(-lines[i].direction.y(), lines[i].direction.x()), true, result) < projLines.size()) {
|
||||
/* This should in principle not happen. The result is by definition
|
||||
* already in the feasible region of this linear program. If it fails,
|
||||
* it is due to small floating point error, and the current result is
|
||||
* kept.
|
||||
*/
|
||||
result = tempResult;
|
||||
}
|
||||
|
||||
distance = det(lines[i].direction, lines[i].point - result);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -1,110 +0,0 @@
|
||||
/*
|
||||
* Definitions.h
|
||||
* RVO2 Library
|
||||
*
|
||||
* Copyright 2008 University of North Carolina at Chapel Hill
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <http://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#ifndef RVO2D_DEFINITIONS_H_
|
||||
#define RVO2D_DEFINITIONS_H_
|
||||
|
||||
/**
|
||||
* \file Definitions.h
|
||||
* \brief Contains functions and constants used in multiple classes.
|
||||
*/
|
||||
|
||||
#include <algorithm>
|
||||
#include <cmath>
|
||||
#include <cstddef>
|
||||
#include <cstdint>
|
||||
#include <limits>
|
||||
#include <vector>
|
||||
|
||||
#include "Vector2.h"
|
||||
|
||||
/**
|
||||
* \brief A sufficiently small positive number.
|
||||
*/
|
||||
const float RVO_EPSILON = 0.00001f;
|
||||
|
||||
namespace RVO2D {
|
||||
class Agent2D;
|
||||
class Obstacle2D;
|
||||
class RVOSimulator2D;
|
||||
|
||||
/**
|
||||
* \brief Computes the squared distance from a line segment with the
|
||||
* specified endpoints to a specified point.
|
||||
* \param a The first endpoint of the line segment.
|
||||
* \param b The second endpoint of the line segment.
|
||||
* \param c The point to which the squared distance is to
|
||||
* be calculated.
|
||||
* \return The squared distance from the line segment to the point.
|
||||
*/
|
||||
inline float distSqPointLineSegment(const Vector2 &a, const Vector2 &b,
|
||||
const Vector2 &c)
|
||||
{
|
||||
const float r = ((c - a) * (b - a)) / absSq(b - a);
|
||||
|
||||
if (r < 0.0f) {
|
||||
return absSq(c - a);
|
||||
}
|
||||
else if (r > 1.0f) {
|
||||
return absSq(c - b);
|
||||
}
|
||||
else {
|
||||
return absSq(c - (a + r * (b - a)));
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Computes the signed distance from a line connecting the
|
||||
* specified points to a specified point.
|
||||
* \param a The first point on the line.
|
||||
* \param b The second point on the line.
|
||||
* \param c The point to which the signed distance is to
|
||||
* be calculated.
|
||||
* \return Positive when the point c lies to the left of the line ab.
|
||||
*/
|
||||
inline float leftOf(const Vector2 &a, const Vector2 &b, const Vector2 &c)
|
||||
{
|
||||
return det(a - c, b - a);
|
||||
}
|
||||
|
||||
/**
|
||||
* \brief Computes the square of a float.
|
||||
* \param a The float to be squared.
|
||||
* \return The square of the float.
|
||||
*/
|
||||
inline float sqr(float a)
|
||||
{
|
||||
return a * a;
|
||||
}
|
||||
}
|
||||
|
||||
#endif /* RVO2D_DEFINITIONS_H_ */
|
||||
@ -0,0 +1,517 @@
|
||||
/*
|
||||
* KdTree2d.cpp
|
||||
* RVO2 Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
/**
|
||||
* @file KdTree2d.cpp
|
||||
* @brief Defines the KdTree2D class.
|
||||
*/
|
||||
|
||||
#include "KdTree2d.h"
|
||||
|
||||
#include <algorithm>
|
||||
#include <utility>
|
||||
|
||||
#include "Agent2d.h"
|
||||
#include "Obstacle2d.h"
|
||||
#include "RVOSimulator2d.h"
|
||||
#include "Vector2.h"
|
||||
|
||||
namespace RVO2D {
|
||||
namespace {
|
||||
/**
|
||||
* @relates KdTree2D
|
||||
* @brief The maximum k-D tree node leaf size.
|
||||
*/
|
||||
const std::size_t RVO_MAX_LEAF_SIZE = 10U;
|
||||
} /* namespace */
|
||||
|
||||
/**
|
||||
* @brief Defines an agent k-D tree node.
|
||||
*/
|
||||
class KdTree2D::AgentTreeNode {
|
||||
public:
|
||||
/**
|
||||
* @brief Constructs an agent k-D tree node instance.
|
||||
*/
|
||||
AgentTreeNode();
|
||||
|
||||
/**
|
||||
* @brief The beginning node number.
|
||||
*/
|
||||
std::size_t begin;
|
||||
|
||||
/**
|
||||
* @brief The ending node number.
|
||||
*/
|
||||
std::size_t end;
|
||||
|
||||
/**
|
||||
* @brief The left node number.
|
||||
*/
|
||||
std::size_t left;
|
||||
|
||||
/**
|
||||
* @brief The right node number.
|
||||
*/
|
||||
std::size_t right;
|
||||
|
||||
/**
|
||||
* @brief The maximum x-coordinate.
|
||||
*/
|
||||
float maxX;
|
||||
|
||||
/**
|
||||
* @brief The maximum y-coordinate.
|
||||
*/
|
||||
float maxY;
|
||||
|
||||
/**
|
||||
* @brief The minimum x-coordinate.
|
||||
*/
|
||||
float minX;
|
||||
|
||||
/**
|
||||
* @brief The minimum y-coordinate.
|
||||
*/
|
||||
float minY;
|
||||
};
|
||||
|
||||
KdTree2D::AgentTreeNode::AgentTreeNode()
|
||||
: begin(0U),
|
||||
end(0U),
|
||||
left(0U),
|
||||
right(0U),
|
||||
maxX(0.0F),
|
||||
maxY(0.0F),
|
||||
minX(0.0F),
|
||||
minY(0.0F) {}
|
||||
|
||||
/**
|
||||
* @brief Defines an obstacle k-D tree node.
|
||||
*/
|
||||
class KdTree2D::ObstacleTreeNode {
|
||||
public:
|
||||
/**
|
||||
* @brief Constructs an obstacle k-D tree node instance.
|
||||
*/
|
||||
ObstacleTreeNode();
|
||||
|
||||
/**
|
||||
* @brief Destroys this obstacle k-D tree node instance.
|
||||
*/
|
||||
~ObstacleTreeNode();
|
||||
|
||||
/**
|
||||
* @brief The obstacle number.
|
||||
*/
|
||||
const Obstacle2D *obstacle;
|
||||
|
||||
/**
|
||||
* @brief The left obstacle tree node.
|
||||
*/
|
||||
ObstacleTreeNode *left;
|
||||
|
||||
/**
|
||||
* @brief The right obstacle tree node.
|
||||
*/
|
||||
ObstacleTreeNode *right;
|
||||
|
||||
private:
|
||||
/* Not implemented. */
|
||||
ObstacleTreeNode(const ObstacleTreeNode &other);
|
||||
|
||||
/* Not implemented. */
|
||||
ObstacleTreeNode &operator=(const ObstacleTreeNode &other);
|
||||
};
|
||||
|
||||
KdTree2D::ObstacleTreeNode::ObstacleTreeNode()
|
||||
: obstacle(NULL), left(NULL), right(NULL) {}
|
||||
|
||||
KdTree2D::ObstacleTreeNode::~ObstacleTreeNode() {}
|
||||
|
||||
KdTree2D::KdTree2D(RVOSimulator2D *simulator)
|
||||
: obstacleTree_(NULL), simulator_(simulator) {}
|
||||
|
||||
KdTree2D::~KdTree2D() { deleteObstacleTree(obstacleTree_); }
|
||||
|
||||
void KdTree2D::buildAgentTree(std::vector<Agent2D *> agents) {
|
||||
agents_.swap(agents);
|
||||
|
||||
if (!agents_.empty()) {
|
||||
agentTree_.resize(2 * agents_.size() - 1);
|
||||
buildAgentTreeRecursive(0, agents_.size(), 0);
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree2D::buildAgentTreeRecursive(std::size_t begin, std::size_t end,
|
||||
std::size_t node) {
|
||||
agentTree_[node].begin = begin;
|
||||
agentTree_[node].end = end;
|
||||
agentTree_[node].minX = agentTree_[node].maxX = agents_[begin]->position_.x();
|
||||
agentTree_[node].minY = agentTree_[node].maxY = agents_[begin]->position_.y();
|
||||
|
||||
for (std::size_t i = begin + 1U; i < end; ++i) {
|
||||
agentTree_[node].maxX =
|
||||
std::max(agentTree_[node].maxX, agents_[i]->position_.x());
|
||||
agentTree_[node].minX =
|
||||
std::min(agentTree_[node].minX, agents_[i]->position_.x());
|
||||
agentTree_[node].maxY =
|
||||
std::max(agentTree_[node].maxY, agents_[i]->position_.y());
|
||||
agentTree_[node].minY =
|
||||
std::min(agentTree_[node].minY, agents_[i]->position_.y());
|
||||
}
|
||||
|
||||
if (end - begin > RVO_MAX_LEAF_SIZE) {
|
||||
/* No leaf node. */
|
||||
const bool isVertical = agentTree_[node].maxX - agentTree_[node].minX >
|
||||
agentTree_[node].maxY - agentTree_[node].minY;
|
||||
const float splitValue =
|
||||
0.5F * (isVertical ? agentTree_[node].maxX + agentTree_[node].minX
|
||||
: agentTree_[node].maxY + agentTree_[node].minY);
|
||||
|
||||
std::size_t left = begin;
|
||||
std::size_t right = end;
|
||||
|
||||
while (left < right) {
|
||||
while (left < right &&
|
||||
(isVertical ? agents_[left]->position_.x()
|
||||
: agents_[left]->position_.y()) < splitValue) {
|
||||
++left;
|
||||
}
|
||||
|
||||
while (right > left &&
|
||||
(isVertical ? agents_[right - 1U]->position_.x()
|
||||
: agents_[right - 1U]->position_.y()) >= splitValue) {
|
||||
--right;
|
||||
}
|
||||
|
||||
if (left < right) {
|
||||
std::swap(agents_[left], agents_[right - 1U]);
|
||||
++left;
|
||||
--right;
|
||||
}
|
||||
}
|
||||
|
||||
if (left == begin) {
|
||||
++left;
|
||||
++right;
|
||||
}
|
||||
|
||||
agentTree_[node].left = node + 1U;
|
||||
agentTree_[node].right = node + 2U * (left - begin);
|
||||
|
||||
buildAgentTreeRecursive(begin, left, agentTree_[node].left);
|
||||
buildAgentTreeRecursive(left, end, agentTree_[node].right);
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree2D::buildObstacleTree(std::vector<Obstacle2D *> obstacles) {
|
||||
deleteObstacleTree(obstacleTree_);
|
||||
|
||||
obstacleTree_ = buildObstacleTreeRecursive(obstacles);
|
||||
}
|
||||
|
||||
KdTree2D::ObstacleTreeNode *KdTree2D::buildObstacleTreeRecursive(
|
||||
const std::vector<Obstacle2D *> &obstacles) {
|
||||
if (!obstacles.empty()) {
|
||||
ObstacleTreeNode *const node = new ObstacleTreeNode();
|
||||
|
||||
std::size_t optimalSplit = 0U;
|
||||
std::size_t minLeft = obstacles.size();
|
||||
std::size_t minRight = obstacles.size();
|
||||
|
||||
for (std::size_t i = 0U; i < obstacles.size(); ++i) {
|
||||
std::size_t leftSize = 0U;
|
||||
std::size_t rightSize = 0U;
|
||||
|
||||
const Obstacle2D *const obstacleI1 = obstacles[i];
|
||||
const Obstacle2D *const obstacleI2 = obstacleI1->next_;
|
||||
|
||||
/* Compute optimal split node. */
|
||||
for (std::size_t j = 0U; j < obstacles.size(); ++j) {
|
||||
if (i != j) {
|
||||
const Obstacle2D *const obstacleJ1 = obstacles[j];
|
||||
const Obstacle2D *const obstacleJ2 = obstacleJ1->next_;
|
||||
|
||||
const float j1LeftOfI = leftOf(obstacleI1->point_, obstacleI2->point_,
|
||||
obstacleJ1->point_);
|
||||
const float j2LeftOfI = leftOf(obstacleI1->point_, obstacleI2->point_,
|
||||
obstacleJ2->point_);
|
||||
|
||||
if (j1LeftOfI >= -RVO2D_EPSILON && j2LeftOfI >= -RVO2D_EPSILON) {
|
||||
++leftSize;
|
||||
} else if (j1LeftOfI <= RVO2D_EPSILON && j2LeftOfI <= RVO2D_EPSILON) {
|
||||
++rightSize;
|
||||
} else {
|
||||
++leftSize;
|
||||
++rightSize;
|
||||
}
|
||||
|
||||
if (std::make_pair(std::max(leftSize, rightSize),
|
||||
std::min(leftSize, rightSize)) >=
|
||||
std::make_pair(std::max(minLeft, minRight),
|
||||
std::min(minLeft, minRight))) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (std::make_pair(std::max(leftSize, rightSize),
|
||||
std::min(leftSize, rightSize)) <
|
||||
std::make_pair(std::max(minLeft, minRight),
|
||||
std::min(minLeft, minRight))) {
|
||||
minLeft = leftSize;
|
||||
minRight = rightSize;
|
||||
optimalSplit = i;
|
||||
}
|
||||
}
|
||||
|
||||
/* Build split node. */
|
||||
std::vector<Obstacle2D *> leftObstacles(minLeft);
|
||||
std::vector<Obstacle2D *> rightObstacles(minRight);
|
||||
|
||||
std::size_t leftCounter = 0U;
|
||||
std::size_t rightCounter = 0U;
|
||||
const std::size_t i = optimalSplit;
|
||||
|
||||
const Obstacle2D *const obstacleI1 = obstacles[i];
|
||||
const Obstacle2D *const obstacleI2 = obstacleI1->next_;
|
||||
|
||||
for (std::size_t j = 0U; j < obstacles.size(); ++j) {
|
||||
if (i != j) {
|
||||
Obstacle2D *const obstacleJ1 = obstacles[j];
|
||||
Obstacle2D *const obstacleJ2 = obstacleJ1->next_;
|
||||
|
||||
const float j1LeftOfI =
|
||||
leftOf(obstacleI1->point_, obstacleI2->point_, obstacleJ1->point_);
|
||||
const float j2LeftOfI =
|
||||
leftOf(obstacleI1->point_, obstacleI2->point_, obstacleJ2->point_);
|
||||
|
||||
if (j1LeftOfI >= -RVO2D_EPSILON && j2LeftOfI >= -RVO2D_EPSILON) {
|
||||
leftObstacles[leftCounter++] = obstacles[j];
|
||||
} else if (j1LeftOfI <= RVO2D_EPSILON && j2LeftOfI <= RVO2D_EPSILON) {
|
||||
rightObstacles[rightCounter++] = obstacles[j];
|
||||
} else {
|
||||
/* Split obstacle j. */
|
||||
const float t = det(obstacleI2->point_ - obstacleI1->point_,
|
||||
obstacleJ1->point_ - obstacleI1->point_) /
|
||||
det(obstacleI2->point_ - obstacleI1->point_,
|
||||
obstacleJ1->point_ - obstacleJ2->point_);
|
||||
|
||||
const Vector2 splitPoint =
|
||||
obstacleJ1->point_ +
|
||||
t * (obstacleJ2->point_ - obstacleJ1->point_);
|
||||
|
||||
Obstacle2D *const newObstacle = new Obstacle2D();
|
||||
newObstacle->direction_ = obstacleJ1->direction_;
|
||||
newObstacle->point_ = splitPoint;
|
||||
newObstacle->next_ = obstacleJ2;
|
||||
newObstacle->previous_ = obstacleJ1;
|
||||
newObstacle->id_ = simulator_->obstacles_.size();
|
||||
newObstacle->isConvex_ = true;
|
||||
simulator_->obstacles_.push_back(newObstacle);
|
||||
|
||||
obstacleJ1->next_ = newObstacle;
|
||||
obstacleJ2->previous_ = newObstacle;
|
||||
|
||||
if (j1LeftOfI > 0.0F) {
|
||||
leftObstacles[leftCounter++] = obstacleJ1;
|
||||
rightObstacles[rightCounter++] = newObstacle;
|
||||
} else {
|
||||
rightObstacles[rightCounter++] = obstacleJ1;
|
||||
leftObstacles[leftCounter++] = newObstacle;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
node->obstacle = obstacleI1;
|
||||
node->left = buildObstacleTreeRecursive(leftObstacles);
|
||||
node->right = buildObstacleTreeRecursive(rightObstacles);
|
||||
|
||||
return node;
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void KdTree2D::computeAgentNeighbors(Agent2D *agent, float &rangeSq) const {
|
||||
queryAgentTreeRecursive(agent, rangeSq, 0U);
|
||||
}
|
||||
|
||||
void KdTree2D::computeObstacleNeighbors(Agent2D *agent, float rangeSq) const {
|
||||
queryObstacleTreeRecursive(agent, rangeSq, obstacleTree_);
|
||||
}
|
||||
|
||||
void KdTree2D::deleteObstacleTree(ObstacleTreeNode *node) {
|
||||
if (node != NULL) {
|
||||
deleteObstacleTree(node->left);
|
||||
deleteObstacleTree(node->right);
|
||||
delete node;
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree2D::queryAgentTreeRecursive(Agent2D *agent, float &rangeSq,
|
||||
std::size_t node) const {
|
||||
if (agentTree_[node].end - agentTree_[node].begin <= RVO_MAX_LEAF_SIZE) {
|
||||
for (std::size_t i = agentTree_[node].begin; i < agentTree_[node].end;
|
||||
++i) {
|
||||
agent->insertAgentNeighbor(agents_[i], rangeSq);
|
||||
}
|
||||
} else {
|
||||
const float distLeftMinX = std::max(
|
||||
0.0F, agentTree_[agentTree_[node].left].minX - agent->position_.x());
|
||||
const float distLeftMaxX = std::max(
|
||||
0.0F, agent->position_.x() - agentTree_[agentTree_[node].left].maxX);
|
||||
const float distLeftMinY = std::max(
|
||||
0.0F, agentTree_[agentTree_[node].left].minY - agent->position_.y());
|
||||
const float distLeftMaxY = std::max(
|
||||
0.0F, agent->position_.y() - agentTree_[agentTree_[node].left].maxY);
|
||||
|
||||
const float distSqLeft =
|
||||
distLeftMinX * distLeftMinX + distLeftMaxX * distLeftMaxX +
|
||||
distLeftMinY * distLeftMinY + distLeftMaxY * distLeftMaxY;
|
||||
|
||||
const float distRightMinX = std::max(
|
||||
0.0F, agentTree_[agentTree_[node].right].minX - agent->position_.x());
|
||||
const float distRightMaxX = std::max(
|
||||
0.0F, agent->position_.x() - agentTree_[agentTree_[node].right].maxX);
|
||||
const float distRightMinY = std::max(
|
||||
0.0F, agentTree_[agentTree_[node].right].minY - agent->position_.y());
|
||||
const float distRightMaxY = std::max(
|
||||
0.0F, agent->position_.y() - agentTree_[agentTree_[node].right].maxY);
|
||||
|
||||
const float distSqRight =
|
||||
distRightMinX * distRightMinX + distRightMaxX * distRightMaxX +
|
||||
distRightMinY * distRightMinY + distRightMaxY * distRightMaxY;
|
||||
|
||||
if (distSqLeft < distSqRight) {
|
||||
if (distSqLeft < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].left);
|
||||
|
||||
if (distSqRight < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].right);
|
||||
}
|
||||
}
|
||||
} else if (distSqRight < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].right);
|
||||
|
||||
if (distSqLeft < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].left);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree2D::queryObstacleTreeRecursive(Agent2D *agent, float rangeSq,
|
||||
const ObstacleTreeNode *node) const {
|
||||
if (node != NULL) {
|
||||
const Obstacle2D *const obstacle1 = node->obstacle;
|
||||
const Obstacle2D *const obstacle2 = obstacle1->next_;
|
||||
|
||||
const float agentLeftOfLine =
|
||||
leftOf(obstacle1->point_, obstacle2->point_, agent->position_);
|
||||
|
||||
queryObstacleTreeRecursive(
|
||||
agent, rangeSq, agentLeftOfLine >= 0.0F ? node->left : node->right);
|
||||
|
||||
const float distSqLine = agentLeftOfLine * agentLeftOfLine /
|
||||
absSq(obstacle2->point_ - obstacle1->point_);
|
||||
|
||||
if (distSqLine < rangeSq) {
|
||||
if (agentLeftOfLine < 0.0F) {
|
||||
/* Try obstacle at this node only if agent is on right side of obstacle
|
||||
* and can see obstacle. */
|
||||
agent->insertObstacleNeighbor(node->obstacle, rangeSq);
|
||||
}
|
||||
|
||||
/* Try other side of line. */
|
||||
queryObstacleTreeRecursive(
|
||||
agent, rangeSq, agentLeftOfLine >= 0.0F ? node->right : node->left);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
bool KdTree2D::queryVisibility(const Vector2 &vector1, const Vector2 &vector2,
|
||||
float radius) const {
|
||||
return queryVisibilityRecursive(vector1, vector2, radius, obstacleTree_);
|
||||
}
|
||||
|
||||
bool KdTree2D::queryVisibilityRecursive(const Vector2 &vector1,
|
||||
const Vector2 &vector2, float radius,
|
||||
const ObstacleTreeNode *node) const {
|
||||
if (node != NULL) {
|
||||
const Obstacle2D *const obstacle1 = node->obstacle;
|
||||
const Obstacle2D *const obstacle2 = obstacle1->next_;
|
||||
|
||||
const float q1LeftOfI =
|
||||
leftOf(obstacle1->point_, obstacle2->point_, vector1);
|
||||
const float q2LeftOfI =
|
||||
leftOf(obstacle1->point_, obstacle2->point_, vector2);
|
||||
const float invLengthI =
|
||||
1.0F / absSq(obstacle2->point_ - obstacle1->point_);
|
||||
|
||||
if (q1LeftOfI >= 0.0F && q2LeftOfI >= 0.0F) {
|
||||
return queryVisibilityRecursive(vector1, vector2, radius, node->left) &&
|
||||
((q1LeftOfI * q1LeftOfI * invLengthI >= radius * radius &&
|
||||
q2LeftOfI * q2LeftOfI * invLengthI >= radius * radius) ||
|
||||
queryVisibilityRecursive(vector1, vector2, radius, node->right));
|
||||
}
|
||||
|
||||
if (q1LeftOfI <= 0.0F && q2LeftOfI <= 0.0F) {
|
||||
return queryVisibilityRecursive(vector1, vector2, radius, node->right) &&
|
||||
((q1LeftOfI * q1LeftOfI * invLengthI >= radius * radius &&
|
||||
q2LeftOfI * q2LeftOfI * invLengthI >= radius * radius) ||
|
||||
queryVisibilityRecursive(vector1, vector2, radius, node->left));
|
||||
}
|
||||
|
||||
if (q1LeftOfI >= 0.0F && q2LeftOfI <= 0.0F) {
|
||||
/* One can see through obstacle from left to right. */
|
||||
return queryVisibilityRecursive(vector1, vector2, radius, node->left) &&
|
||||
queryVisibilityRecursive(vector1, vector2, radius, node->right);
|
||||
}
|
||||
|
||||
const float point1LeftOfQ = leftOf(vector1, vector2, obstacle1->point_);
|
||||
const float point2LeftOfQ = leftOf(vector1, vector2, obstacle2->point_);
|
||||
const float invLengthQ = 1.0F / absSq(vector2 - vector1);
|
||||
|
||||
return point1LeftOfQ * point2LeftOfQ >= 0.0F &&
|
||||
point1LeftOfQ * point1LeftOfQ * invLengthQ > radius * radius &&
|
||||
point2LeftOfQ * point2LeftOfQ * invLengthQ > radius * radius &&
|
||||
queryVisibilityRecursive(vector1, vector2, radius, node->left) &&
|
||||
queryVisibilityRecursive(vector1, vector2, radius, node->right);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
} /* namespace RVO2D */
|
||||
@ -1,357 +0,0 @@
|
||||
/*
|
||||
* KdTree2d.cpp
|
||||
* RVO2 Library
|
||||
*
|
||||
* Copyright 2008 University of North Carolina at Chapel Hill
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <http://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "KdTree2d.h"
|
||||
|
||||
#include "Agent2d.h"
|
||||
#include "RVOSimulator2d.h"
|
||||
#include "Obstacle2d.h"
|
||||
|
||||
namespace RVO2D {
|
||||
KdTree2D::KdTree2D(RVOSimulator2D *sim) : obstacleTree_(NULL), sim_(sim) { }
|
||||
|
||||
KdTree2D::~KdTree2D()
|
||||
{
|
||||
deleteObstacleTree(obstacleTree_);
|
||||
}
|
||||
|
||||
void KdTree2D::buildAgentTree(std::vector<Agent2D *> agents)
|
||||
{
|
||||
agents_.swap(agents);
|
||||
|
||||
if (!agents_.empty()) {
|
||||
agentTree_.resize(2 * agents_.size() - 1);
|
||||
buildAgentTreeRecursive(0, agents_.size(), 0);
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree2D::buildAgentTreeRecursive(size_t begin, size_t end, size_t node)
|
||||
{
|
||||
agentTree_[node].begin = begin;
|
||||
agentTree_[node].end = end;
|
||||
agentTree_[node].minX = agentTree_[node].maxX = agents_[begin]->position_.x();
|
||||
agentTree_[node].minY = agentTree_[node].maxY = agents_[begin]->position_.y();
|
||||
|
||||
for (size_t i = begin + 1; i < end; ++i) {
|
||||
agentTree_[node].maxX = std::max(agentTree_[node].maxX, agents_[i]->position_.x());
|
||||
agentTree_[node].minX = std::min(agentTree_[node].minX, agents_[i]->position_.x());
|
||||
agentTree_[node].maxY = std::max(agentTree_[node].maxY, agents_[i]->position_.y());
|
||||
agentTree_[node].minY = std::min(agentTree_[node].minY, agents_[i]->position_.y());
|
||||
}
|
||||
|
||||
if (end - begin > MAX_LEAF_SIZE) {
|
||||
/* No leaf node. */
|
||||
const bool isVertical = (agentTree_[node].maxX - agentTree_[node].minX > agentTree_[node].maxY - agentTree_[node].minY);
|
||||
const float splitValue = (isVertical ? 0.5f * (agentTree_[node].maxX + agentTree_[node].minX) : 0.5f * (agentTree_[node].maxY + agentTree_[node].minY));
|
||||
|
||||
size_t left = begin;
|
||||
size_t right = end;
|
||||
|
||||
while (left < right) {
|
||||
while (left < right && (isVertical ? agents_[left]->position_.x() : agents_[left]->position_.y()) < splitValue) {
|
||||
++left;
|
||||
}
|
||||
|
||||
while (right > left && (isVertical ? agents_[right - 1]->position_.x() : agents_[right - 1]->position_.y()) >= splitValue) {
|
||||
--right;
|
||||
}
|
||||
|
||||
if (left < right) {
|
||||
std::swap(agents_[left], agents_[right - 1]);
|
||||
++left;
|
||||
--right;
|
||||
}
|
||||
}
|
||||
|
||||
if (left == begin) {
|
||||
++left;
|
||||
++right;
|
||||
}
|
||||
|
||||
agentTree_[node].left = node + 1;
|
||||
agentTree_[node].right = node + 2 * (left - begin);
|
||||
|
||||
buildAgentTreeRecursive(begin, left, agentTree_[node].left);
|
||||
buildAgentTreeRecursive(left, end, agentTree_[node].right);
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree2D::buildObstacleTree(std::vector<Obstacle2D *> obstacles)
|
||||
{
|
||||
deleteObstacleTree(obstacleTree_);
|
||||
|
||||
obstacleTree_ = buildObstacleTreeRecursive(obstacles);
|
||||
}
|
||||
|
||||
|
||||
KdTree2D::ObstacleTreeNode *KdTree2D::buildObstacleTreeRecursive(const std::vector<Obstacle2D *> &obstacles)
|
||||
{
|
||||
if (obstacles.empty()) {
|
||||
return NULL;
|
||||
}
|
||||
else {
|
||||
ObstacleTreeNode *const node = new ObstacleTreeNode;
|
||||
|
||||
size_t optimalSplit = 0;
|
||||
size_t minLeft = obstacles.size();
|
||||
size_t minRight = obstacles.size();
|
||||
|
||||
for (size_t i = 0; i < obstacles.size(); ++i) {
|
||||
size_t leftSize = 0;
|
||||
size_t rightSize = 0;
|
||||
|
||||
const Obstacle2D *const obstacleI1 = obstacles[i];
|
||||
const Obstacle2D *const obstacleI2 = obstacleI1->nextObstacle_;
|
||||
|
||||
/* Compute optimal split node. */
|
||||
for (size_t j = 0; j < obstacles.size(); ++j) {
|
||||
if (i == j) {
|
||||
continue;
|
||||
}
|
||||
|
||||
const Obstacle2D *const obstacleJ1 = obstacles[j];
|
||||
const Obstacle2D *const obstacleJ2 = obstacleJ1->nextObstacle_;
|
||||
|
||||
const float j1LeftOfI = leftOf(obstacleI1->point_, obstacleI2->point_, obstacleJ1->point_);
|
||||
const float j2LeftOfI = leftOf(obstacleI1->point_, obstacleI2->point_, obstacleJ2->point_);
|
||||
|
||||
if (j1LeftOfI >= -RVO_EPSILON && j2LeftOfI >= -RVO_EPSILON) {
|
||||
++leftSize;
|
||||
}
|
||||
else if (j1LeftOfI <= RVO_EPSILON && j2LeftOfI <= RVO_EPSILON) {
|
||||
++rightSize;
|
||||
}
|
||||
else {
|
||||
++leftSize;
|
||||
++rightSize;
|
||||
}
|
||||
|
||||
if (std::make_pair(std::max(leftSize, rightSize), std::min(leftSize, rightSize)) >= std::make_pair(std::max(minLeft, minRight), std::min(minLeft, minRight))) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (std::make_pair(std::max(leftSize, rightSize), std::min(leftSize, rightSize)) < std::make_pair(std::max(minLeft, minRight), std::min(minLeft, minRight))) {
|
||||
minLeft = leftSize;
|
||||
minRight = rightSize;
|
||||
optimalSplit = i;
|
||||
}
|
||||
}
|
||||
|
||||
/* Build split node. */
|
||||
std::vector<Obstacle2D *> leftObstacles(minLeft);
|
||||
std::vector<Obstacle2D *> rightObstacles(minRight);
|
||||
|
||||
size_t leftCounter = 0;
|
||||
size_t rightCounter = 0;
|
||||
const size_t i = optimalSplit;
|
||||
|
||||
const Obstacle2D *const obstacleI1 = obstacles[i];
|
||||
const Obstacle2D *const obstacleI2 = obstacleI1->nextObstacle_;
|
||||
|
||||
for (size_t j = 0; j < obstacles.size(); ++j) {
|
||||
if (i == j) {
|
||||
continue;
|
||||
}
|
||||
|
||||
Obstacle2D *const obstacleJ1 = obstacles[j];
|
||||
Obstacle2D *const obstacleJ2 = obstacleJ1->nextObstacle_;
|
||||
|
||||
const float j1LeftOfI = leftOf(obstacleI1->point_, obstacleI2->point_, obstacleJ1->point_);
|
||||
const float j2LeftOfI = leftOf(obstacleI1->point_, obstacleI2->point_, obstacleJ2->point_);
|
||||
|
||||
if (j1LeftOfI >= -RVO_EPSILON && j2LeftOfI >= -RVO_EPSILON) {
|
||||
leftObstacles[leftCounter++] = obstacles[j];
|
||||
}
|
||||
else if (j1LeftOfI <= RVO_EPSILON && j2LeftOfI <= RVO_EPSILON) {
|
||||
rightObstacles[rightCounter++] = obstacles[j];
|
||||
}
|
||||
else {
|
||||
/* Split obstacle j. */
|
||||
const float t = det(obstacleI2->point_ - obstacleI1->point_, obstacleJ1->point_ - obstacleI1->point_) / det(obstacleI2->point_ - obstacleI1->point_, obstacleJ1->point_ - obstacleJ2->point_);
|
||||
|
||||
const Vector2 splitpoint = obstacleJ1->point_ + t * (obstacleJ2->point_ - obstacleJ1->point_);
|
||||
|
||||
Obstacle2D *const newObstacle = new Obstacle2D();
|
||||
newObstacle->point_ = splitpoint;
|
||||
newObstacle->prevObstacle_ = obstacleJ1;
|
||||
newObstacle->nextObstacle_ = obstacleJ2;
|
||||
newObstacle->isConvex_ = true;
|
||||
newObstacle->unitDir_ = obstacleJ1->unitDir_;
|
||||
|
||||
newObstacle->id_ = sim_->obstacles_.size();
|
||||
|
||||
sim_->obstacles_.push_back(newObstacle);
|
||||
|
||||
obstacleJ1->nextObstacle_ = newObstacle;
|
||||
obstacleJ2->prevObstacle_ = newObstacle;
|
||||
|
||||
if (j1LeftOfI > 0.0f) {
|
||||
leftObstacles[leftCounter++] = obstacleJ1;
|
||||
rightObstacles[rightCounter++] = newObstacle;
|
||||
}
|
||||
else {
|
||||
rightObstacles[rightCounter++] = obstacleJ1;
|
||||
leftObstacles[leftCounter++] = newObstacle;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
node->obstacle = obstacleI1;
|
||||
node->left = buildObstacleTreeRecursive(leftObstacles);
|
||||
node->right = buildObstacleTreeRecursive(rightObstacles);
|
||||
return node;
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree2D::computeAgentNeighbors(Agent2D *agent, float &rangeSq) const
|
||||
{
|
||||
queryAgentTreeRecursive(agent, rangeSq, 0);
|
||||
}
|
||||
|
||||
void KdTree2D::computeObstacleNeighbors(Agent2D *agent, float rangeSq) const
|
||||
{
|
||||
queryObstacleTreeRecursive(agent, rangeSq, obstacleTree_);
|
||||
}
|
||||
|
||||
void KdTree2D::deleteObstacleTree(ObstacleTreeNode *node)
|
||||
{
|
||||
if (node != NULL) {
|
||||
deleteObstacleTree(node->left);
|
||||
deleteObstacleTree(node->right);
|
||||
delete node;
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree2D::queryAgentTreeRecursive(Agent2D *agent, float &rangeSq, size_t node) const
|
||||
{
|
||||
if (agentTree_[node].end - agentTree_[node].begin <= MAX_LEAF_SIZE) {
|
||||
for (size_t i = agentTree_[node].begin; i < agentTree_[node].end; ++i) {
|
||||
agent->insertAgentNeighbor(agents_[i], rangeSq);
|
||||
}
|
||||
}
|
||||
else {
|
||||
const float distSqLeft = sqr(std::max(0.0f, agentTree_[agentTree_[node].left].minX - agent->position_.x())) + sqr(std::max(0.0f, agent->position_.x() - agentTree_[agentTree_[node].left].maxX)) + sqr(std::max(0.0f, agentTree_[agentTree_[node].left].minY - agent->position_.y())) + sqr(std::max(0.0f, agent->position_.y() - agentTree_[agentTree_[node].left].maxY));
|
||||
|
||||
const float distSqRight = sqr(std::max(0.0f, agentTree_[agentTree_[node].right].minX - agent->position_.x())) + sqr(std::max(0.0f, agent->position_.x() - agentTree_[agentTree_[node].right].maxX)) + sqr(std::max(0.0f, agentTree_[agentTree_[node].right].minY - agent->position_.y())) + sqr(std::max(0.0f, agent->position_.y() - agentTree_[agentTree_[node].right].maxY));
|
||||
|
||||
if (distSqLeft < distSqRight) {
|
||||
if (distSqLeft < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].left);
|
||||
|
||||
if (distSqRight < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].right);
|
||||
}
|
||||
}
|
||||
}
|
||||
else {
|
||||
if (distSqRight < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].right);
|
||||
|
||||
if (distSqLeft < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].left);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree2D::queryObstacleTreeRecursive(Agent2D *agent, float rangeSq, const ObstacleTreeNode *node) const
|
||||
{
|
||||
if (node == NULL) {
|
||||
return;
|
||||
}
|
||||
else {
|
||||
const Obstacle2D *const obstacle1 = node->obstacle;
|
||||
const Obstacle2D *const obstacle2 = obstacle1->nextObstacle_;
|
||||
|
||||
const float agentLeftOfLine = leftOf(obstacle1->point_, obstacle2->point_, agent->position_);
|
||||
|
||||
queryObstacleTreeRecursive(agent, rangeSq, (agentLeftOfLine >= 0.0f ? node->left : node->right));
|
||||
|
||||
const float distSqLine = sqr(agentLeftOfLine) / absSq(obstacle2->point_ - obstacle1->point_);
|
||||
|
||||
if (distSqLine < rangeSq) {
|
||||
if (agentLeftOfLine < 0.0f) {
|
||||
/*
|
||||
* Try obstacle at this node only if agent is on right side of
|
||||
* obstacle (and can see obstacle).
|
||||
*/
|
||||
agent->insertObstacleNeighbor(node->obstacle, rangeSq);
|
||||
}
|
||||
|
||||
/* Try other side of line. */
|
||||
queryObstacleTreeRecursive(agent, rangeSq, (agentLeftOfLine >= 0.0f ? node->right : node->left));
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
bool KdTree2D::queryVisibility(const Vector2 &q1, const Vector2 &q2, float radius) const
|
||||
{
|
||||
return queryVisibilityRecursive(q1, q2, radius, obstacleTree_);
|
||||
}
|
||||
|
||||
bool KdTree2D::queryVisibilityRecursive(const Vector2 &q1, const Vector2 &q2, float radius, const ObstacleTreeNode *node) const
|
||||
{
|
||||
if (node == NULL) {
|
||||
return true;
|
||||
}
|
||||
else {
|
||||
const Obstacle2D *const obstacle1 = node->obstacle;
|
||||
const Obstacle2D *const obstacle2 = obstacle1->nextObstacle_;
|
||||
|
||||
const float q1LeftOfI = leftOf(obstacle1->point_, obstacle2->point_, q1);
|
||||
const float q2LeftOfI = leftOf(obstacle1->point_, obstacle2->point_, q2);
|
||||
const float invLengthI = 1.0f / absSq(obstacle2->point_ - obstacle1->point_);
|
||||
|
||||
if (q1LeftOfI >= 0.0f && q2LeftOfI >= 0.0f) {
|
||||
return queryVisibilityRecursive(q1, q2, radius, node->left) && ((sqr(q1LeftOfI) * invLengthI >= sqr(radius) && sqr(q2LeftOfI) * invLengthI >= sqr(radius)) || queryVisibilityRecursive(q1, q2, radius, node->right));
|
||||
}
|
||||
else if (q1LeftOfI <= 0.0f && q2LeftOfI <= 0.0f) {
|
||||
return queryVisibilityRecursive(q1, q2, radius, node->right) && ((sqr(q1LeftOfI) * invLengthI >= sqr(radius) && sqr(q2LeftOfI) * invLengthI >= sqr(radius)) || queryVisibilityRecursive(q1, q2, radius, node->left));
|
||||
}
|
||||
else if (q1LeftOfI >= 0.0f && q2LeftOfI <= 0.0f) {
|
||||
/* One can see through obstacle from left to right. */
|
||||
return queryVisibilityRecursive(q1, q2, radius, node->left) && queryVisibilityRecursive(q1, q2, radius, node->right);
|
||||
}
|
||||
else {
|
||||
const float point1LeftOfQ = leftOf(q1, q2, obstacle1->point_);
|
||||
const float point2LeftOfQ = leftOf(q1, q2, obstacle2->point_);
|
||||
const float invLengthQ = 1.0f / absSq(q2 - q1);
|
||||
|
||||
return (point1LeftOfQ * point2LeftOfQ >= 0.0f && sqr(point1LeftOfQ) * invLengthQ > sqr(radius) && sqr(point2LeftOfQ) * invLengthQ > sqr(radius) && queryVisibilityRecursive(q1, q2, radius, node->left) && queryVisibilityRecursive(q1, q2, radius, node->right));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,67 @@
|
||||
/*
|
||||
* Line.h
|
||||
* RVO2 Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#ifndef RVO2D_LINE_H_
|
||||
#define RVO2D_LINE_H_
|
||||
|
||||
/**
|
||||
* @file Line.h
|
||||
* @brief Declares the Line class.
|
||||
*/
|
||||
|
||||
#include "Vector2.h"
|
||||
|
||||
namespace RVO2D {
|
||||
/**
|
||||
* @brief Defines a directed line.
|
||||
*/
|
||||
class Line {
|
||||
public:
|
||||
/**
|
||||
* @brief Constructs a directed line instance.
|
||||
*/
|
||||
Line();
|
||||
|
||||
/**
|
||||
* @brief The direction of the directed line.
|
||||
*/
|
||||
Vector2 direction;
|
||||
|
||||
/**
|
||||
* @brief A point on the directed line.
|
||||
*/
|
||||
Vector2 point;
|
||||
};
|
||||
} /* namespace RVO2D */
|
||||
|
||||
#endif /* RVO2D_LINE_H_ */
|
||||
@ -0,0 +1,46 @@
|
||||
/*
|
||||
* Obstacle2d.cpp
|
||||
* RVO2 Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
/**
|
||||
* @file Obstacle2d.cpp
|
||||
* @brief Defines the Obstacle2D class.
|
||||
*/
|
||||
|
||||
#include "Obstacle2d.h"
|
||||
|
||||
namespace RVO2D {
|
||||
Obstacle2D::Obstacle2D()
|
||||
: next_(NULL), previous_(NULL), id_(0U), isConvex_(false) {}
|
||||
|
||||
Obstacle2D::~Obstacle2D() {}
|
||||
} /* namespace RVO2D */
|
||||
@ -0,0 +1,371 @@
|
||||
/*
|
||||
* RVOSimulator2d.cpp
|
||||
* RVO2 Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
/**
|
||||
* @file RVOSimulator2d.cpp
|
||||
* @brief Defines the RVOSimulator2D class.
|
||||
*/
|
||||
|
||||
#include "RVOSimulator2d.h"
|
||||
|
||||
#include <limits>
|
||||
#include <utility>
|
||||
|
||||
#include "Agent2d.h"
|
||||
#include "KdTree2d.h"
|
||||
#include "Line.h"
|
||||
#include "Obstacle2d.h"
|
||||
#include "Vector2.h"
|
||||
|
||||
#ifdef _OPENMP
|
||||
#include <omp.h>
|
||||
#endif /* _OPENMP */
|
||||
|
||||
namespace RVO2D {
|
||||
const std::size_t RVO2D_ERROR = std::numeric_limits<std::size_t>::max();
|
||||
|
||||
RVOSimulator2D::RVOSimulator2D()
|
||||
: defaultAgent_(NULL),
|
||||
kdTree_(new KdTree2D(this)),
|
||||
globalTime_(0.0F),
|
||||
timeStep_(0.0F) {}
|
||||
|
||||
RVOSimulator2D::RVOSimulator2D(float timeStep, float neighborDist,
|
||||
std::size_t maxNeighbors, float timeHorizon,
|
||||
float timeHorizonObst, float radius, float maxSpeed)
|
||||
: defaultAgent_(new Agent2D()),
|
||||
kdTree_(new KdTree2D(this)),
|
||||
globalTime_(0.0F),
|
||||
timeStep_(timeStep) {
|
||||
defaultAgent_->maxNeighbors_ = maxNeighbors;
|
||||
defaultAgent_->maxSpeed_ = maxSpeed;
|
||||
defaultAgent_->neighborDist_ = neighborDist;
|
||||
defaultAgent_->radius_ = radius;
|
||||
defaultAgent_->timeHorizon_ = timeHorizon;
|
||||
defaultAgent_->timeHorizonObst_ = timeHorizonObst;
|
||||
}
|
||||
|
||||
RVOSimulator2D::RVOSimulator2D(float timeStep, float neighborDist,
|
||||
std::size_t maxNeighbors, float timeHorizon,
|
||||
float timeHorizonObst, float radius, float maxSpeed,
|
||||
const Vector2 &velocity)
|
||||
: defaultAgent_(new Agent2D()),
|
||||
kdTree_(new KdTree2D(this)),
|
||||
globalTime_(0.0F),
|
||||
timeStep_(timeStep) {
|
||||
defaultAgent_->velocity_ = velocity;
|
||||
defaultAgent_->maxNeighbors_ = maxNeighbors;
|
||||
defaultAgent_->maxSpeed_ = maxSpeed;
|
||||
defaultAgent_->neighborDist_ = neighborDist;
|
||||
defaultAgent_->radius_ = radius;
|
||||
defaultAgent_->timeHorizon_ = timeHorizon;
|
||||
defaultAgent_->timeHorizonObst_ = timeHorizonObst;
|
||||
}
|
||||
|
||||
RVOSimulator2D::~RVOSimulator2D() {
|
||||
delete defaultAgent_;
|
||||
delete kdTree_;
|
||||
|
||||
for (std::size_t i = 0U; i < agents_.size(); ++i) {
|
||||
delete agents_[i];
|
||||
}
|
||||
|
||||
for (std::size_t i = 0U; i < obstacles_.size(); ++i) {
|
||||
delete obstacles_[i];
|
||||
}
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::addAgent(const Vector2 &position) {
|
||||
if (defaultAgent_ != NULL) {
|
||||
Agent2D *const agent = new Agent2D();
|
||||
agent->position_ = position;
|
||||
agent->velocity_ = defaultAgent_->velocity_;
|
||||
agent->id_ = agents_.size();
|
||||
agent->maxNeighbors_ = defaultAgent_->maxNeighbors_;
|
||||
agent->maxSpeed_ = defaultAgent_->maxSpeed_;
|
||||
agent->neighborDist_ = defaultAgent_->neighborDist_;
|
||||
agent->radius_ = defaultAgent_->radius_;
|
||||
agent->timeHorizon_ = defaultAgent_->timeHorizon_;
|
||||
agent->timeHorizonObst_ = defaultAgent_->timeHorizonObst_;
|
||||
agents_.push_back(agent);
|
||||
|
||||
return agents_.size() - 1U;
|
||||
}
|
||||
|
||||
return RVO2D_ERROR;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::addAgent(const Vector2 &position, float neighborDist,
|
||||
std::size_t maxNeighbors, float timeHorizon,
|
||||
float timeHorizonObst, float radius,
|
||||
float maxSpeed) {
|
||||
return addAgent(position, neighborDist, maxNeighbors, timeHorizon,
|
||||
timeHorizonObst, radius, maxSpeed, Vector2());
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::addAgent(const Vector2 &position, float neighborDist,
|
||||
std::size_t maxNeighbors, float timeHorizon,
|
||||
float timeHorizonObst, float radius,
|
||||
float maxSpeed, const Vector2 &velocity) {
|
||||
Agent2D *const agent = new Agent2D();
|
||||
agent->position_ = position;
|
||||
agent->velocity_ = velocity;
|
||||
agent->id_ = agents_.size();
|
||||
agent->maxNeighbors_ = maxNeighbors;
|
||||
agent->maxSpeed_ = maxSpeed;
|
||||
agent->neighborDist_ = neighborDist;
|
||||
agent->radius_ = radius;
|
||||
agent->timeHorizon_ = timeHorizon;
|
||||
agent->timeHorizonObst_ = timeHorizonObst;
|
||||
agents_.push_back(agent);
|
||||
|
||||
return agents_.size() - 1U;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::addObstacle(const std::vector<Vector2> &vertices) {
|
||||
if (vertices.size() > 1U) {
|
||||
const std::size_t obstacleNo = obstacles_.size();
|
||||
|
||||
for (std::size_t i = 0U; i < vertices.size(); ++i) {
|
||||
Obstacle2D *const obstacle = new Obstacle2D();
|
||||
obstacle->point_ = vertices[i];
|
||||
|
||||
if (i != 0U) {
|
||||
obstacle->previous_ = obstacles_.back();
|
||||
obstacle->previous_->next_ = obstacle;
|
||||
}
|
||||
|
||||
if (i == vertices.size() - 1U) {
|
||||
obstacle->next_ = obstacles_[obstacleNo];
|
||||
obstacle->next_->previous_ = obstacle;
|
||||
}
|
||||
|
||||
obstacle->direction_ = normalize(
|
||||
vertices[(i == vertices.size() - 1U ? 0U : i + 1U)] - vertices[i]);
|
||||
|
||||
if (vertices.size() == 2U) {
|
||||
obstacle->isConvex_ = true;
|
||||
} else {
|
||||
obstacle->isConvex_ =
|
||||
leftOf(vertices[i == 0U ? vertices.size() - 1U : i - 1U],
|
||||
vertices[i],
|
||||
vertices[i == vertices.size() - 1U ? 0U : i + 1U]) >= 0.0F;
|
||||
}
|
||||
|
||||
obstacle->id_ = obstacles_.size();
|
||||
|
||||
obstacles_.push_back(obstacle);
|
||||
}
|
||||
|
||||
return obstacleNo;
|
||||
}
|
||||
|
||||
return RVO2D_ERROR;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::doStep() {
|
||||
kdTree_->buildAgentTree(agents_);
|
||||
|
||||
#ifdef _OPENMP
|
||||
#pragma omp parallel for
|
||||
#endif /* _OPENMP */
|
||||
for (int i = 0; i < static_cast<int>(agents_.size()); ++i) {
|
||||
agents_[i]->computeNeighbors(kdTree_);
|
||||
agents_[i]->computeNewVelocity(timeStep_);
|
||||
}
|
||||
|
||||
#ifdef _OPENMP
|
||||
#pragma omp parallel for
|
||||
#endif /* _OPENMP */
|
||||
for (int i = 0; i < static_cast<int>(agents_.size()); ++i) {
|
||||
agents_[i]->update(timeStep_);
|
||||
}
|
||||
|
||||
globalTime_ += timeStep_;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::getAgentAgentNeighbor(std::size_t agentNo,
|
||||
std::size_t neighborNo) const {
|
||||
return agents_[agentNo]->agentNeighbors_[neighborNo].second->id_;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::getAgentMaxNeighbors(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->maxNeighbors_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentMaxSpeed(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->maxSpeed_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentNeighborDist(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->neighborDist_;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::getAgentNumAgentNeighbors(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->agentNeighbors_.size();
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::getAgentNumObstacleNeighbors(
|
||||
std::size_t agentNo) const {
|
||||
return agents_[agentNo]->obstacleNeighbors_.size();
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::getAgentNumORCALines(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->orcaLines_.size();
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::getAgentObstacleNeighbor(
|
||||
std::size_t agentNo, std::size_t neighborNo) const {
|
||||
return agents_[agentNo]->obstacleNeighbors_[neighborNo].second->id_;
|
||||
}
|
||||
|
||||
const Line &RVOSimulator2D::getAgentORCALine(std::size_t agentNo,
|
||||
std::size_t lineNo) const {
|
||||
return agents_[agentNo]->orcaLines_[lineNo];
|
||||
}
|
||||
|
||||
const Vector2 &RVOSimulator2D::getAgentPosition(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->position_;
|
||||
}
|
||||
|
||||
const Vector2 &RVOSimulator2D::getAgentPrefVelocity(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->prefVelocity_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentRadius(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->radius_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentTimeHorizon(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->timeHorizon_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentTimeHorizonObst(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->timeHorizonObst_;
|
||||
}
|
||||
|
||||
const Vector2 &RVOSimulator2D::getAgentVelocity(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->velocity_;
|
||||
}
|
||||
|
||||
const Vector2 &RVOSimulator2D::getObstacleVertex(std::size_t vertexNo) const {
|
||||
return obstacles_[vertexNo]->point_;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::getNextObstacleVertexNo(std::size_t vertexNo) const {
|
||||
return obstacles_[vertexNo]->next_->id_;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator2D::getPrevObstacleVertexNo(std::size_t vertexNo) const {
|
||||
return obstacles_[vertexNo]->previous_->id_;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::processObstacles() { kdTree_->buildObstacleTree(obstacles_); }
|
||||
|
||||
bool RVOSimulator2D::queryVisibility(const Vector2 &point1,
|
||||
const Vector2 &point2) const {
|
||||
return kdTree_->queryVisibility(point1, point2, 0.0F);
|
||||
}
|
||||
|
||||
bool RVOSimulator2D::queryVisibility(const Vector2 &point1, const Vector2 &point2,
|
||||
float radius) const {
|
||||
return kdTree_->queryVisibility(point1, point2, radius);
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentDefaults(float neighborDist,
|
||||
std::size_t maxNeighbors, float timeHorizon,
|
||||
float timeHorizonObst, float radius,
|
||||
float maxSpeed) {
|
||||
setAgentDefaults(neighborDist, maxNeighbors, timeHorizon, timeHorizonObst,
|
||||
radius, maxSpeed, Vector2());
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentDefaults(float neighborDist,
|
||||
std::size_t maxNeighbors, float timeHorizon,
|
||||
float timeHorizonObst, float radius,
|
||||
float maxSpeed, const Vector2 &velocity) {
|
||||
if (defaultAgent_ == NULL) {
|
||||
defaultAgent_ = new Agent2D();
|
||||
}
|
||||
|
||||
defaultAgent_->maxNeighbors_ = maxNeighbors;
|
||||
defaultAgent_->maxSpeed_ = maxSpeed;
|
||||
defaultAgent_->neighborDist_ = neighborDist;
|
||||
defaultAgent_->radius_ = radius;
|
||||
defaultAgent_->timeHorizon_ = timeHorizon;
|
||||
defaultAgent_->timeHorizonObst_ = timeHorizonObst;
|
||||
defaultAgent_->velocity_ = velocity;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentMaxNeighbors(std::size_t agentNo,
|
||||
std::size_t maxNeighbors) {
|
||||
agents_[agentNo]->maxNeighbors_ = maxNeighbors;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentMaxSpeed(std::size_t agentNo, float maxSpeed) {
|
||||
agents_[agentNo]->maxSpeed_ = maxSpeed;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentNeighborDist(std::size_t agentNo,
|
||||
float neighborDist) {
|
||||
agents_[agentNo]->neighborDist_ = neighborDist;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentPosition(std::size_t agentNo,
|
||||
const Vector2 &position) {
|
||||
agents_[agentNo]->position_ = position;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentPrefVelocity(std::size_t agentNo,
|
||||
const Vector2 &prefVelocity) {
|
||||
agents_[agentNo]->prefVelocity_ = prefVelocity;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentRadius(std::size_t agentNo, float radius) {
|
||||
agents_[agentNo]->radius_ = radius;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentTimeHorizon(std::size_t agentNo, float timeHorizon) {
|
||||
agents_[agentNo]->timeHorizon_ = timeHorizon;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentTimeHorizonObst(std::size_t agentNo,
|
||||
float timeHorizonObst) {
|
||||
agents_[agentNo]->timeHorizonObst_ = timeHorizonObst;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentVelocity(std::size_t agentNo,
|
||||
const Vector2 &velocity) {
|
||||
agents_[agentNo]->velocity_ = velocity;
|
||||
}
|
||||
} /* namespace RVO2D */
|
||||
@ -1,363 +0,0 @@
|
||||
/*
|
||||
* RVOSimulator2d.cpp
|
||||
* RVO2 Library
|
||||
*
|
||||
* Copyright 2008 University of North Carolina at Chapel Hill
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <http://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "RVOSimulator2d.h"
|
||||
|
||||
#include "Agent2d.h"
|
||||
#include "KdTree2d.h"
|
||||
#include "Obstacle2d.h"
|
||||
|
||||
#ifdef _OPENMP
|
||||
#include <omp.h>
|
||||
#endif
|
||||
|
||||
namespace RVO2D {
|
||||
RVOSimulator2D::RVOSimulator2D() : defaultAgent_(NULL), globalTime_(0.0f), kdTree_(NULL), timeStep_(0.0f)
|
||||
{
|
||||
kdTree_ = new KdTree2D(this);
|
||||
}
|
||||
|
||||
RVOSimulator2D::RVOSimulator2D(float timeStep, float neighborDist, size_t maxNeighbors, float timeHorizon, float timeHorizonObst, float radius, float maxSpeed, const Vector2 &velocity) : defaultAgent_(NULL), globalTime_(0.0f), kdTree_(NULL), timeStep_(timeStep)
|
||||
{
|
||||
kdTree_ = new KdTree2D(this);
|
||||
defaultAgent_ = new Agent2D();
|
||||
|
||||
defaultAgent_->maxNeighbors_ = maxNeighbors;
|
||||
defaultAgent_->maxSpeed_ = maxSpeed;
|
||||
defaultAgent_->neighborDist_ = neighborDist;
|
||||
defaultAgent_->radius_ = radius;
|
||||
defaultAgent_->timeHorizon_ = timeHorizon;
|
||||
defaultAgent_->timeHorizonObst_ = timeHorizonObst;
|
||||
defaultAgent_->velocity_ = velocity;
|
||||
}
|
||||
|
||||
RVOSimulator2D::~RVOSimulator2D()
|
||||
{
|
||||
if (defaultAgent_ != NULL) {
|
||||
delete defaultAgent_;
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < agents_.size(); ++i) {
|
||||
delete agents_[i];
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < obstacles_.size(); ++i) {
|
||||
delete obstacles_[i];
|
||||
}
|
||||
|
||||
delete kdTree_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::addAgent(const Vector2 &position)
|
||||
{
|
||||
if (defaultAgent_ == NULL) {
|
||||
return RVO2D_ERROR;
|
||||
}
|
||||
|
||||
Agent2D *agent = new Agent2D();
|
||||
|
||||
agent->position_ = position;
|
||||
agent->maxNeighbors_ = defaultAgent_->maxNeighbors_;
|
||||
agent->maxSpeed_ = defaultAgent_->maxSpeed_;
|
||||
agent->neighborDist_ = defaultAgent_->neighborDist_;
|
||||
agent->radius_ = defaultAgent_->radius_;
|
||||
agent->timeHorizon_ = defaultAgent_->timeHorizon_;
|
||||
agent->timeHorizonObst_ = defaultAgent_->timeHorizonObst_;
|
||||
agent->velocity_ = defaultAgent_->velocity_;
|
||||
|
||||
agent->id_ = agents_.size();
|
||||
|
||||
agents_.push_back(agent);
|
||||
|
||||
return agents_.size() - 1;
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::addAgent(const Vector2 &position, float neighborDist, size_t maxNeighbors, float timeHorizon, float timeHorizonObst, float radius, float maxSpeed, const Vector2 &velocity)
|
||||
{
|
||||
Agent2D *agent = new Agent2D();
|
||||
|
||||
agent->position_ = position;
|
||||
agent->maxNeighbors_ = maxNeighbors;
|
||||
agent->maxSpeed_ = maxSpeed;
|
||||
agent->neighborDist_ = neighborDist;
|
||||
agent->radius_ = radius;
|
||||
agent->timeHorizon_ = timeHorizon;
|
||||
agent->timeHorizonObst_ = timeHorizonObst;
|
||||
agent->velocity_ = velocity;
|
||||
|
||||
agent->id_ = agents_.size();
|
||||
|
||||
agents_.push_back(agent);
|
||||
|
||||
return agents_.size() - 1;
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::addObstacle(const std::vector<Vector2> &vertices)
|
||||
{
|
||||
if (vertices.size() < 2) {
|
||||
return RVO2D_ERROR;
|
||||
}
|
||||
|
||||
const size_t obstacleNo = obstacles_.size();
|
||||
|
||||
for (size_t i = 0; i < vertices.size(); ++i) {
|
||||
Obstacle2D *obstacle = new Obstacle2D();
|
||||
obstacle->point_ = vertices[i];
|
||||
|
||||
if (i != 0) {
|
||||
obstacle->prevObstacle_ = obstacles_.back();
|
||||
obstacle->prevObstacle_->nextObstacle_ = obstacle;
|
||||
}
|
||||
|
||||
if (i == vertices.size() - 1) {
|
||||
obstacle->nextObstacle_ = obstacles_[obstacleNo];
|
||||
obstacle->nextObstacle_->prevObstacle_ = obstacle;
|
||||
}
|
||||
|
||||
obstacle->unitDir_ = normalize(vertices[(i == vertices.size() - 1 ? 0 : i + 1)] - vertices[i]);
|
||||
|
||||
if (vertices.size() == 2) {
|
||||
obstacle->isConvex_ = true;
|
||||
}
|
||||
else {
|
||||
obstacle->isConvex_ = (leftOf(vertices[(i == 0 ? vertices.size() - 1 : i - 1)], vertices[i], vertices[(i == vertices.size() - 1 ? 0 : i + 1)]) >= 0.0f);
|
||||
}
|
||||
|
||||
obstacle->id_ = obstacles_.size();
|
||||
|
||||
obstacles_.push_back(obstacle);
|
||||
}
|
||||
|
||||
return obstacleNo;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::doStep()
|
||||
{
|
||||
kdTree_->buildAgentTree(agents_);
|
||||
|
||||
for (int i = 0; i < static_cast<int>(agents_.size()); ++i) {
|
||||
agents_[i]->computeNeighbors(this);
|
||||
agents_[i]->computeNewVelocity(this);
|
||||
}
|
||||
|
||||
for (int i = 0; i < static_cast<int>(agents_.size()); ++i) {
|
||||
agents_[i]->update(this);
|
||||
}
|
||||
|
||||
globalTime_ += timeStep_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getAgentAgentNeighbor(size_t agentNo, size_t neighborNo) const
|
||||
{
|
||||
return agents_[agentNo]->agentNeighbors_[neighborNo].second->id_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getAgentMaxNeighbors(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->maxNeighbors_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentMaxSpeed(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->maxSpeed_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentNeighborDist(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->neighborDist_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getAgentNumAgentNeighbors(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->agentNeighbors_.size();
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getAgentNumObstacleNeighbors(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->obstacleNeighbors_.size();
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getAgentNumORCALines(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->orcaLines_.size();
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getAgentObstacleNeighbor(size_t agentNo, size_t neighborNo) const
|
||||
{
|
||||
return agents_[agentNo]->obstacleNeighbors_[neighborNo].second->id_;
|
||||
}
|
||||
|
||||
const Line &RVOSimulator2D::getAgentORCALine(size_t agentNo, size_t lineNo) const
|
||||
{
|
||||
return agents_[agentNo]->orcaLines_[lineNo];
|
||||
}
|
||||
|
||||
const Vector2 &RVOSimulator2D::getAgentPosition(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->position_;
|
||||
}
|
||||
|
||||
const Vector2 &RVOSimulator2D::getAgentPrefVelocity(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->prefVelocity_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentRadius(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->radius_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentTimeHorizon(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->timeHorizon_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getAgentTimeHorizonObst(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->timeHorizonObst_;
|
||||
}
|
||||
|
||||
const Vector2 &RVOSimulator2D::getAgentVelocity(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->velocity_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getGlobalTime() const
|
||||
{
|
||||
return globalTime_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getNumAgents() const
|
||||
{
|
||||
return agents_.size();
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getNumObstacleVertices() const
|
||||
{
|
||||
return obstacles_.size();
|
||||
}
|
||||
|
||||
const Vector2 &RVOSimulator2D::getObstacleVertex(size_t vertexNo) const
|
||||
{
|
||||
return obstacles_[vertexNo]->point_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getNextObstacleVertexNo(size_t vertexNo) const
|
||||
{
|
||||
return obstacles_[vertexNo]->nextObstacle_->id_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator2D::getPrevObstacleVertexNo(size_t vertexNo) const
|
||||
{
|
||||
return obstacles_[vertexNo]->prevObstacle_->id_;
|
||||
}
|
||||
|
||||
float RVOSimulator2D::getTimeStep() const
|
||||
{
|
||||
return timeStep_;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::processObstacles()
|
||||
{
|
||||
kdTree_->buildObstacleTree(obstacles_);
|
||||
}
|
||||
|
||||
bool RVOSimulator2D::queryVisibility(const Vector2 &point1, const Vector2 &point2, float radius) const
|
||||
{
|
||||
return kdTree_->queryVisibility(point1, point2, radius);
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentDefaults(float neighborDist, size_t maxNeighbors, float timeHorizon, float timeHorizonObst, float radius, float maxSpeed, const Vector2 &velocity)
|
||||
{
|
||||
if (defaultAgent_ == NULL) {
|
||||
defaultAgent_ = new Agent2D();
|
||||
}
|
||||
|
||||
defaultAgent_->maxNeighbors_ = maxNeighbors;
|
||||
defaultAgent_->maxSpeed_ = maxSpeed;
|
||||
defaultAgent_->neighborDist_ = neighborDist;
|
||||
defaultAgent_->radius_ = radius;
|
||||
defaultAgent_->timeHorizon_ = timeHorizon;
|
||||
defaultAgent_->timeHorizonObst_ = timeHorizonObst;
|
||||
defaultAgent_->velocity_ = velocity;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentMaxNeighbors(size_t agentNo, size_t maxNeighbors)
|
||||
{
|
||||
agents_[agentNo]->maxNeighbors_ = maxNeighbors;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentMaxSpeed(size_t agentNo, float maxSpeed)
|
||||
{
|
||||
agents_[agentNo]->maxSpeed_ = maxSpeed;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentNeighborDist(size_t agentNo, float neighborDist)
|
||||
{
|
||||
agents_[agentNo]->neighborDist_ = neighborDist;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentPosition(size_t agentNo, const Vector2 &position)
|
||||
{
|
||||
agents_[agentNo]->position_ = position;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentPrefVelocity(size_t agentNo, const Vector2 &prefVelocity)
|
||||
{
|
||||
agents_[agentNo]->prefVelocity_ = prefVelocity;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentRadius(size_t agentNo, float radius)
|
||||
{
|
||||
agents_[agentNo]->radius_ = radius;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentTimeHorizon(size_t agentNo, float timeHorizon)
|
||||
{
|
||||
agents_[agentNo]->timeHorizon_ = timeHorizon;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentTimeHorizonObst(size_t agentNo, float timeHorizonObst)
|
||||
{
|
||||
agents_[agentNo]->timeHorizonObst_ = timeHorizonObst;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setAgentVelocity(size_t agentNo, const Vector2 &velocity)
|
||||
{
|
||||
agents_[agentNo]->velocity_ = velocity;
|
||||
}
|
||||
|
||||
void RVOSimulator2D::setTimeStep(float timeStep)
|
||||
{
|
||||
timeStep_ = timeStep;
|
||||
}
|
||||
}
|
||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,136 @@
|
||||
/*
|
||||
* Vector2.cpp
|
||||
* RVO2 Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
/**
|
||||
* @file Vector2.cc
|
||||
* @brief Defines the Vector2 class.
|
||||
*/
|
||||
|
||||
#include "Vector2.h"
|
||||
|
||||
#include <cmath>
|
||||
#include <ostream>
|
||||
|
||||
namespace RVO2D {
|
||||
const float RVO2D_EPSILON = 0.00001F;
|
||||
|
||||
Vector2::Vector2() : x_(0.0F), y_(0.0F) {}
|
||||
|
||||
Vector2::Vector2(float x, float y) : x_(x), y_(y) {}
|
||||
|
||||
Vector2 Vector2::operator-() const { return Vector2(-x_, -y_); }
|
||||
|
||||
float Vector2::operator*(const Vector2 &vector) const {
|
||||
return x_ * vector.x_ + y_ * vector.y_;
|
||||
}
|
||||
|
||||
Vector2 Vector2::operator*(float scalar) const {
|
||||
return Vector2(x_ * scalar, y_ * scalar);
|
||||
}
|
||||
|
||||
Vector2 Vector2::operator/(float scalar) const {
|
||||
const float invScalar = 1.0F / scalar;
|
||||
|
||||
return Vector2(x_ * invScalar, y_ * invScalar);
|
||||
}
|
||||
|
||||
Vector2 Vector2::operator+(const Vector2 &vector) const {
|
||||
return Vector2(x_ + vector.x_, y_ + vector.y_);
|
||||
}
|
||||
|
||||
Vector2 Vector2::operator-(const Vector2 &vector) const {
|
||||
return Vector2(x_ - vector.x_, y_ - vector.y_);
|
||||
}
|
||||
|
||||
bool Vector2::operator==(const Vector2 &vector) const {
|
||||
return x_ == vector.x_ && y_ == vector.y_;
|
||||
}
|
||||
|
||||
bool Vector2::operator!=(const Vector2 &vector) const {
|
||||
return x_ != vector.x_ || y_ != vector.y_;
|
||||
}
|
||||
|
||||
Vector2 &Vector2::operator*=(float scalar) {
|
||||
x_ *= scalar;
|
||||
y_ *= scalar;
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
Vector2 &Vector2::operator/=(float scalar) {
|
||||
const float invScalar = 1.0F / scalar;
|
||||
x_ *= invScalar;
|
||||
y_ *= invScalar;
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
Vector2 &Vector2::operator+=(const Vector2 &vector) {
|
||||
x_ += vector.x_;
|
||||
y_ += vector.y_;
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
Vector2 &Vector2::operator-=(const Vector2 &vector) {
|
||||
x_ -= vector.x_;
|
||||
y_ -= vector.y_;
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
Vector2 operator*(float scalar, const Vector2 &vector) {
|
||||
return Vector2(scalar * vector.x(), scalar * vector.y());
|
||||
}
|
||||
|
||||
std::ostream &operator<<(std::ostream &stream, const Vector2 &vector) {
|
||||
stream << "(" << vector.x() << "," << vector.y() << ")";
|
||||
|
||||
return stream;
|
||||
}
|
||||
|
||||
float abs(const Vector2 &vector) { return std::sqrt(vector * vector); }
|
||||
|
||||
float absSq(const Vector2 &vector) { return vector * vector; }
|
||||
|
||||
float det(const Vector2 &vector1, const Vector2 &vector2) {
|
||||
return vector1.x() * vector2.y() - vector1.y() * vector2.x();
|
||||
}
|
||||
|
||||
float leftOf(const Vector2 &vector1, const Vector2 &vector2,
|
||||
const Vector2 &vector3) {
|
||||
return det(vector1 - vector3, vector2 - vector1);
|
||||
}
|
||||
|
||||
Vector2 normalize(const Vector2 &vector) { return vector / abs(vector); }
|
||||
} /* namespace RVO */
|
||||
@ -0,0 +1,474 @@
|
||||
/*
|
||||
* Agent3d.cc
|
||||
* RVO2-3D Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "Agent3d.h"
|
||||
|
||||
#include <algorithm>
|
||||
#include <cmath>
|
||||
|
||||
#include "KdTree3d.h"
|
||||
#include "RVOSimulator3d.h"
|
||||
|
||||
namespace RVO3D {
|
||||
namespace {
|
||||
/**
|
||||
* @brief A sufficiently small positive number.
|
||||
*/
|
||||
const float RVO3D_EPSILON = 0.00001F;
|
||||
|
||||
/**
|
||||
* @brief Defines a directed line.
|
||||
*/
|
||||
class Line3D {
|
||||
public:
|
||||
/**
|
||||
* @brief Constructs a directed line.``
|
||||
*/
|
||||
Line3D();
|
||||
|
||||
/**
|
||||
* @brief The direction of the directed line.
|
||||
*/
|
||||
Vector3 direction;
|
||||
|
||||
/**
|
||||
* @brief A point on the directed line.
|
||||
*/
|
||||
Vector3 point;
|
||||
};
|
||||
|
||||
Line3D::Line3D() {}
|
||||
|
||||
/**
|
||||
* @brief Solves a one-dimensional linear program on a specified line
|
||||
* subject to linear constraints defined by planes and a spherical
|
||||
* constraint.
|
||||
* @param[in] planes Planes defining the linear constraints.
|
||||
* @param[in] planeNo The plane on which the line lies.
|
||||
* @param[in] line The line on which the one-dimensional linear program
|
||||
* is solved.
|
||||
* @param[in] radius The radius of the spherical constraint.
|
||||
* @param[in] optVelocity The optimization velocity.
|
||||
* @param[in] directionOpt True if the direction should be optimized.
|
||||
* @param[in] result A reference to the result of the linear program.
|
||||
* @return True if successful.
|
||||
*/
|
||||
bool linearProgram1(const std::vector<Plane> &planes, std::size_t planeNo,
|
||||
const Line3D &line, float radius, const Vector3 &optVelocity,
|
||||
bool directionOpt,
|
||||
Vector3 &result) { /* NOLINT(runtime/references) */
|
||||
const float dotProduct = line.point * line.direction;
|
||||
const float discriminant =
|
||||
dotProduct * dotProduct + radius * radius - absSq(line.point);
|
||||
|
||||
if (discriminant < 0.0F) {
|
||||
/* Max speed sphere fully invalidates line. */
|
||||
return false;
|
||||
}
|
||||
|
||||
const float sqrtDiscriminant = std::sqrt(discriminant);
|
||||
float tLeft = -dotProduct - sqrtDiscriminant;
|
||||
float tRight = -dotProduct + sqrtDiscriminant;
|
||||
|
||||
for (std::size_t i = 0U; i < planeNo; ++i) {
|
||||
const float numerator = (planes[i].point - line.point) * planes[i].normal;
|
||||
const float denominator = line.direction * planes[i].normal;
|
||||
|
||||
if (denominator * denominator <= RVO3D_EPSILON) {
|
||||
/* Lines line is (almost) parallel to plane i. */
|
||||
if (numerator > 0.0F) {
|
||||
return false;
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
const float t = numerator / denominator;
|
||||
|
||||
if (denominator >= 0.0F) {
|
||||
/* Plane i bounds line on the left. */
|
||||
tLeft = std::max(tLeft, t);
|
||||
} else {
|
||||
/* Plane i bounds line on the right. */
|
||||
tRight = std::min(tRight, t);
|
||||
}
|
||||
|
||||
if (tLeft > tRight) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (directionOpt) {
|
||||
/* Optimize direction. */
|
||||
if (optVelocity * line.direction > 0.0F) {
|
||||
/* Take right extreme. */
|
||||
result = line.point + tRight * line.direction;
|
||||
} else {
|
||||
/* Take left extreme. */
|
||||
result = line.point + tLeft * line.direction;
|
||||
}
|
||||
} else {
|
||||
/* Optimize closest point. */
|
||||
const float t = line.direction * (optVelocity - line.point);
|
||||
|
||||
if (t < tLeft) {
|
||||
result = line.point + tLeft * line.direction;
|
||||
} else if (t > tRight) {
|
||||
result = line.point + tRight * line.direction;
|
||||
} else {
|
||||
result = line.point + t * line.direction;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Solves a two-dimensional linear program on a specified plane
|
||||
* subject to linear constraints defined by planes and a spherical
|
||||
* constraint.
|
||||
* @param[in] planes Planes defining the linear constraints.
|
||||
* @param[in] planeNo The plane on which the two-dimensional linear
|
||||
* program is solved.
|
||||
* @param[in] radius The radius of the spherical constraint.
|
||||
* @param[in] optVelocity The optimization velocity.
|
||||
* @param[in] directionOpt True if the direction should be optimized.
|
||||
* @param[out] result A reference to the result of the linear program.
|
||||
* @return True if successful.
|
||||
*/
|
||||
bool linearProgram2(const std::vector<Plane> &planes, std::size_t planeNo,
|
||||
float radius, const Vector3 &optVelocity, bool directionOpt,
|
||||
Vector3 &result) { /* NOLINT(runtime/references) */
|
||||
const float planeDist = planes[planeNo].point * planes[planeNo].normal;
|
||||
const float planeDistSq = planeDist * planeDist;
|
||||
const float radiusSq = radius * radius;
|
||||
|
||||
if (planeDistSq > radiusSq) {
|
||||
/* Max speed sphere fully invalidates plane planeNo. */
|
||||
return false;
|
||||
}
|
||||
|
||||
const float planeRadiusSq = radiusSq - planeDistSq;
|
||||
|
||||
const Vector3 planeCenter = planeDist * planes[planeNo].normal;
|
||||
|
||||
if (directionOpt) {
|
||||
/* Project direction optVelocity on plane planeNo. */
|
||||
const Vector3 planeOptVelocity =
|
||||
optVelocity -
|
||||
(optVelocity * planes[planeNo].normal) * planes[planeNo].normal;
|
||||
const float planeOptVelocityLengthSq = absSq(planeOptVelocity);
|
||||
|
||||
if (planeOptVelocityLengthSq <= RVO3D_EPSILON) {
|
||||
result = planeCenter;
|
||||
} else {
|
||||
result =
|
||||
planeCenter + std::sqrt(planeRadiusSq / planeOptVelocityLengthSq) *
|
||||
planeOptVelocity;
|
||||
}
|
||||
} else {
|
||||
/* Project point optVelocity on plane planeNo. */
|
||||
result = optVelocity +
|
||||
((planes[planeNo].point - optVelocity) * planes[planeNo].normal) *
|
||||
planes[planeNo].normal;
|
||||
|
||||
/* If outside planeCircle, project on planeCircle. */
|
||||
if (absSq(result) > radiusSq) {
|
||||
const Vector3 planeResult = result - planeCenter;
|
||||
const float planeResultLengthSq = absSq(planeResult);
|
||||
result = planeCenter +
|
||||
std::sqrt(planeRadiusSq / planeResultLengthSq) * planeResult;
|
||||
}
|
||||
}
|
||||
|
||||
for (std::size_t i = 0U; i < planeNo; ++i) {
|
||||
if (planes[i].normal * (planes[i].point - result) > 0.0F) {
|
||||
/* Result does not satisfy constraint i. Compute new optimal result.
|
||||
* Compute intersection line of plane i and plane planeNo.
|
||||
*/
|
||||
Vector3 crossProduct = cross(planes[i].normal, planes[planeNo].normal);
|
||||
|
||||
if (absSq(crossProduct) <= RVO3D_EPSILON) {
|
||||
/* Planes planeNo and i are (almost) parallel, and plane i fully
|
||||
* invalidates plane planeNo.
|
||||
*/
|
||||
return false;
|
||||
}
|
||||
|
||||
Line3D line;
|
||||
line.direction = normalize(crossProduct);
|
||||
const Vector3 lineNormal = cross(line.direction, planes[planeNo].normal);
|
||||
line.point =
|
||||
planes[planeNo].point +
|
||||
(((planes[i].point - planes[planeNo].point) * planes[i].normal) /
|
||||
(lineNormal * planes[i].normal)) *
|
||||
lineNormal;
|
||||
|
||||
if (!linearProgram1(planes, i, line, radius, optVelocity, directionOpt,
|
||||
result)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Solves a three-dimensional linear program subject to linear
|
||||
* constraints defined by planes and a spherical constraint.
|
||||
* @param[in] planes Planes defining the linear constraints.
|
||||
* @param[in] radius The radius of the spherical constraint.
|
||||
* @param[in] optVelocity The optimization velocity.
|
||||
* @param[in] directionOpt True if the direction should be optimized.
|
||||
* @param[out] result A reference to the result of the linear program.
|
||||
* @return The number of the plane it fails on, and the number of planes if
|
||||
* successful.
|
||||
*/
|
||||
std::size_t linearProgram3(const std::vector<Plane> &planes, float radius,
|
||||
const Vector3 &optVelocity, bool directionOpt,
|
||||
Vector3 &result) { /* NOLINT(runtime/references) */
|
||||
if (directionOpt) {
|
||||
/* Optimize direction. Note that the optimization velocity is of unit length
|
||||
* in this case.
|
||||
*/
|
||||
result = optVelocity * radius;
|
||||
} else if (absSq(optVelocity) > radius * radius) {
|
||||
/* Optimize closest point and outside circle. */
|
||||
result = normalize(optVelocity) * radius;
|
||||
} else {
|
||||
/* Optimize closest point and inside circle. */
|
||||
result = optVelocity;
|
||||
}
|
||||
|
||||
for (std::size_t i = 0U; i < planes.size(); ++i) {
|
||||
if (planes[i].normal * (planes[i].point - result) > 0.0F) {
|
||||
/* Result does not satisfy constraint i. Compute new optimal result. */
|
||||
const Vector3 tempResult = result;
|
||||
|
||||
if (!linearProgram2(planes, i, radius, optVelocity, directionOpt,
|
||||
result)) {
|
||||
result = tempResult;
|
||||
return i;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return planes.size();
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Solves a four-dimensional linear program subject to linear
|
||||
* constraints defined by planes and a spherical constraint.
|
||||
* @param[in] planes Planes defining the linear constraints.
|
||||
* @param[in] beginPlane The plane on which the three-dimensional linear
|
||||
* program failed.
|
||||
* @param[in] radius The radius of the spherical constraint.
|
||||
* @param[out] result A reference to the result of the linear program.
|
||||
*/
|
||||
void linearProgram4(const std::vector<Plane> &planes, std::size_t beginPlane,
|
||||
float radius,
|
||||
Vector3 &result) { /* NOLINT(runtime/references) */
|
||||
float distance = 0.0F;
|
||||
|
||||
for (std::size_t i = beginPlane; i < planes.size(); ++i) {
|
||||
if (planes[i].normal * (planes[i].point - result) > distance) {
|
||||
/* Result does not satisfy constraint of plane i. */
|
||||
std::vector<Plane> projPlanes;
|
||||
|
||||
for (std::size_t j = 0U; j < i; ++j) {
|
||||
Plane plane;
|
||||
|
||||
const Vector3 crossProduct = cross(planes[j].normal, planes[i].normal);
|
||||
|
||||
if (absSq(crossProduct) <= RVO3D_EPSILON) {
|
||||
/* Plane i and plane j are (almost) parallel. */
|
||||
if (planes[i].normal * planes[j].normal > 0.0F) {
|
||||
/* Plane i and plane j point in the same direction. */
|
||||
continue;
|
||||
}
|
||||
|
||||
/* Plane i and plane j point in opposite direction. */
|
||||
plane.point = 0.5F * (planes[i].point + planes[j].point);
|
||||
} else {
|
||||
/* Plane.point is point on line of intersection between plane i and
|
||||
* plane j.
|
||||
*/
|
||||
const Vector3 lineNormal = cross(crossProduct, planes[i].normal);
|
||||
plane.point =
|
||||
planes[i].point +
|
||||
(((planes[j].point - planes[i].point) * planes[j].normal) /
|
||||
(lineNormal * planes[j].normal)) *
|
||||
lineNormal;
|
||||
}
|
||||
|
||||
plane.normal = normalize(planes[j].normal - planes[i].normal);
|
||||
projPlanes.push_back(plane);
|
||||
}
|
||||
|
||||
const Vector3 tempResult = result;
|
||||
|
||||
if (linearProgram3(projPlanes, radius, planes[i].normal, true, result) <
|
||||
projPlanes.size()) {
|
||||
/* This should in principle not happen. The result is by definition
|
||||
* already in the feasible region of this linear program. If it fails,
|
||||
* it is due to small floating point error, and the current result is
|
||||
* kept.
|
||||
*/
|
||||
result = tempResult;
|
||||
}
|
||||
|
||||
distance = planes[i].normal * (planes[i].point - result);
|
||||
}
|
||||
}
|
||||
}
|
||||
} /* namespace */
|
||||
|
||||
Agent3D::Agent3D()
|
||||
: id_(0U),
|
||||
maxNeighbors_(0U),
|
||||
maxSpeed_(0.0F),
|
||||
neighborDist_(0.0F),
|
||||
radius_(0.0F),
|
||||
timeHorizon_(0.0F) {}
|
||||
|
||||
Agent3D::~Agent3D() {}
|
||||
|
||||
void Agent3D::computeNeighbors(RVOSimulator3D *sim_) {
|
||||
agentNeighbors_.clear();
|
||||
|
||||
if (maxNeighbors_ > 0) {
|
||||
sim_->kdTree_->computeAgentNeighbors(this, neighborDist_ * neighborDist_);
|
||||
}
|
||||
}
|
||||
|
||||
void Agent3D::computeNewVelocity(RVOSimulator3D *sim_) {
|
||||
orcaPlanes_.clear();
|
||||
const float invTimeHorizon = 1.0F / timeHorizon_;
|
||||
|
||||
/* Create agent ORCA planes. */
|
||||
for (std::size_t i = 0U; i < agentNeighbors_.size(); ++i) {
|
||||
const Agent3D *const other = agentNeighbors_[i].second;
|
||||
const Vector3 relativePosition = other->position_ - position_;
|
||||
const Vector3 relativeVelocity = velocity_ - other->velocity_;
|
||||
const float distSq = absSq(relativePosition);
|
||||
const float combinedRadius = radius_ + other->radius_;
|
||||
const float combinedRadiusSq = combinedRadius * combinedRadius;
|
||||
|
||||
Plane plane;
|
||||
Vector3 u;
|
||||
|
||||
if (distSq > combinedRadiusSq) {
|
||||
/* No collision. */
|
||||
const Vector3 w = relativeVelocity - invTimeHorizon * relativePosition;
|
||||
/* Vector from cutoff center to relative velocity. */
|
||||
const float wLengthSq = absSq(w);
|
||||
|
||||
const float dotProduct = w * relativePosition;
|
||||
|
||||
if (dotProduct < 0.0F &&
|
||||
dotProduct * dotProduct > combinedRadiusSq * wLengthSq) {
|
||||
/* Project on cut-off circle. */
|
||||
const float wLength = std::sqrt(wLengthSq);
|
||||
const Vector3 unitW = w / wLength;
|
||||
|
||||
plane.normal = unitW;
|
||||
u = (combinedRadius * invTimeHorizon - wLength) * unitW;
|
||||
} else {
|
||||
/* Project on cone. */
|
||||
const float a = distSq;
|
||||
const float b = relativePosition * relativeVelocity;
|
||||
const float c = absSq(relativeVelocity) -
|
||||
absSq(cross(relativePosition, relativeVelocity)) /
|
||||
(distSq - combinedRadiusSq);
|
||||
const float t = (b + std::sqrt(b * b - a * c)) / a;
|
||||
const Vector3 ww = relativeVelocity - t * relativePosition;
|
||||
const float wwLength = abs(ww);
|
||||
const Vector3 unitWW = ww / wwLength;
|
||||
|
||||
plane.normal = unitWW;
|
||||
u = (combinedRadius * t - wwLength) * unitWW;
|
||||
}
|
||||
} else {
|
||||
/* Collision. */
|
||||
const float invTimeStep = 1.0F / sim_->timeStep_;
|
||||
const Vector3 w = relativeVelocity - invTimeStep * relativePosition;
|
||||
const float wLength = abs(w);
|
||||
const Vector3 unitW = w / wLength;
|
||||
|
||||
plane.normal = unitW;
|
||||
u = (combinedRadius * invTimeStep - wLength) * unitW;
|
||||
}
|
||||
|
||||
plane.point = velocity_ + 0.5F * u;
|
||||
orcaPlanes_.push_back(plane);
|
||||
}
|
||||
|
||||
const std::size_t planeFail = linearProgram3(
|
||||
orcaPlanes_, maxSpeed_, prefVelocity_, false, newVelocity_);
|
||||
|
||||
if (planeFail < orcaPlanes_.size()) {
|
||||
linearProgram4(orcaPlanes_, planeFail, maxSpeed_, newVelocity_);
|
||||
}
|
||||
}
|
||||
|
||||
void Agent3D::insertAgentNeighbor(const Agent3D *agent, float &rangeSq) {
|
||||
if (this != agent) {
|
||||
const float distSq = absSq(position_ - agent->position_);
|
||||
|
||||
if (distSq < rangeSq) {
|
||||
if (agentNeighbors_.size() < maxNeighbors_) {
|
||||
agentNeighbors_.push_back(std::make_pair(distSq, agent));
|
||||
}
|
||||
|
||||
std::size_t i = agentNeighbors_.size() - 1U;
|
||||
|
||||
while (i != 0U && distSq < agentNeighbors_[i - 1U].first) {
|
||||
agentNeighbors_[i] = agentNeighbors_[i - 1U];
|
||||
--i;
|
||||
}
|
||||
|
||||
agentNeighbors_[i] = std::make_pair(distSq, agent);
|
||||
|
||||
if (agentNeighbors_.size() == maxNeighbors_) {
|
||||
rangeSq = agentNeighbors_.back().first;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void Agent3D::update(RVOSimulator3D *sim_) {
|
||||
velocity_ = newVelocity_;
|
||||
position_ += velocity_ * sim_->timeStep_;
|
||||
}
|
||||
} /* namespace RVO3D */
|
||||
@ -1,449 +0,0 @@
|
||||
/*
|
||||
* Agent.cpp
|
||||
* RVO2-3D Library
|
||||
*
|
||||
* Copyright 2008 University of North Carolina at Chapel Hill
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "Agent3d.h"
|
||||
|
||||
#include <cmath>
|
||||
#include <algorithm>
|
||||
|
||||
#include "Definitions.h"
|
||||
#include "KdTree3d.h"
|
||||
|
||||
namespace RVO3D {
|
||||
/**
|
||||
* \brief A sufficiently small positive number.
|
||||
*/
|
||||
const float RVO3D_EPSILON = 0.00001f;
|
||||
|
||||
/**
|
||||
* \brief Defines a directed line.
|
||||
*/
|
||||
class Line3D {
|
||||
public:
|
||||
/**
|
||||
* \brief The direction of the directed line.
|
||||
*/
|
||||
Vector3 direction;
|
||||
|
||||
/**
|
||||
* \brief A point on the directed line.
|
||||
*/
|
||||
Vector3 point;
|
||||
};
|
||||
|
||||
/**
|
||||
* \brief Solves a one-dimensional linear program on a specified line subject to linear constraints defined by planes and a spherical constraint.
|
||||
* \param planes Planes defining the linear constraints.
|
||||
* \param planeNo The plane on which the line lies.
|
||||
* \param line The line on which the 1-d linear program is solved
|
||||
* \param radius The radius of the spherical constraint.
|
||||
* \param optVelocity The optimization velocity.
|
||||
* \param directionOpt True if the direction should be optimized.
|
||||
* \param result A reference to the result of the linear program.
|
||||
* \return True if successful.
|
||||
*/
|
||||
bool linearProgram1(const std::vector<Plane> &planes, size_t planeNo, const Line3D &line, float radius, const Vector3 &optVelocity, bool directionOpt, Vector3 &result);
|
||||
|
||||
/**
|
||||
* \brief Solves a two-dimensional linear program on a specified plane subject to linear constraints defined by planes and a spherical constraint.
|
||||
* \param planes Planes defining the linear constraints.
|
||||
* \param planeNo The plane on which the 2-d linear program is solved
|
||||
* \param radius The radius of the spherical constraint.
|
||||
* \param optVelocity The optimization velocity.
|
||||
* \param directionOpt True if the direction should be optimized.
|
||||
* \param result A reference to the result of the linear program.
|
||||
* \return True if successful.
|
||||
*/
|
||||
bool linearProgram2(const std::vector<Plane> &planes, size_t planeNo, float radius, const Vector3 &optVelocity, bool directionOpt, Vector3 &result);
|
||||
|
||||
/**
|
||||
* \brief Solves a three-dimensional linear program subject to linear constraints defined by planes and a spherical constraint.
|
||||
* \param planes Planes defining the linear constraints.
|
||||
* \param radius The radius of the spherical constraint.
|
||||
* \param optVelocity The optimization velocity.
|
||||
* \param directionOpt True if the direction should be optimized.
|
||||
* \param result A reference to the result of the linear program.
|
||||
* \return The number of the plane it fails on, and the number of planes if successful.
|
||||
*/
|
||||
size_t linearProgram3(const std::vector<Plane> &planes, float radius, const Vector3 &optVelocity, bool directionOpt, Vector3 &result);
|
||||
|
||||
/**
|
||||
* \brief Solves a four-dimensional linear program subject to linear constraints defined by planes and a spherical constraint.
|
||||
* \param planes Planes defining the linear constraints.
|
||||
* \param beginPlane The plane on which the 3-d linear program failed.
|
||||
* \param radius The radius of the spherical constraint.
|
||||
* \param result A reference to the result of the linear program.
|
||||
*/
|
||||
void linearProgram4(const std::vector<Plane> &planes, size_t beginPlane, float radius, Vector3 &result);
|
||||
|
||||
Agent3D::Agent3D() : id_(0), maxNeighbors_(0), maxSpeed_(0.0f), neighborDist_(0.0f), radius_(0.0f), timeHorizon_(0.0f) { }
|
||||
|
||||
void Agent3D::computeNeighbors(RVOSimulator3D *sim_)
|
||||
{
|
||||
agentNeighbors_.clear();
|
||||
|
||||
if (maxNeighbors_ > 0) {
|
||||
sim_->kdTree_->computeAgentNeighbors(this, neighborDist_ * neighborDist_);
|
||||
}
|
||||
}
|
||||
|
||||
void Agent3D::computeNewVelocity(RVOSimulator3D *sim_)
|
||||
{
|
||||
orcaPlanes_.clear();
|
||||
|
||||
const float invTimeHorizon = 1.0f / timeHorizon_;
|
||||
|
||||
/* Create agent ORCA planes. */
|
||||
for (size_t i = 0; i < agentNeighbors_.size(); ++i) {
|
||||
const Agent3D *const other = agentNeighbors_[i].second;
|
||||
|
||||
//const float timeHorizon_mod = (avoidance_priority_ - other->avoidance_priority_ + 1.0f) * 0.5f;
|
||||
//const float invTimeHorizon = (1.0f / timeHorizon_) * timeHorizon_mod;
|
||||
|
||||
const Vector3 relativePosition = other->position_ - position_;
|
||||
const Vector3 relativeVelocity = velocity_ - other->velocity_;
|
||||
const float distSq = absSq(relativePosition);
|
||||
const float combinedRadius = radius_ + other->radius_;
|
||||
const float combinedRadiusSq = sqr(combinedRadius);
|
||||
|
||||
Plane plane;
|
||||
Vector3 u;
|
||||
|
||||
if (distSq > combinedRadiusSq) {
|
||||
/* No collision. */
|
||||
const Vector3 w = relativeVelocity - invTimeHorizon * relativePosition;
|
||||
/* Vector from cutoff center to relative velocity. */
|
||||
const float wLengthSq = absSq(w);
|
||||
|
||||
const float dotProduct = w * relativePosition;
|
||||
|
||||
if (dotProduct < 0.0f && sqr(dotProduct) > combinedRadiusSq * wLengthSq) {
|
||||
/* Project on cut-off circle. */
|
||||
const float wLength = std::sqrt(wLengthSq);
|
||||
const Vector3 unitW = w / wLength;
|
||||
|
||||
plane.normal = unitW;
|
||||
u = (combinedRadius * invTimeHorizon - wLength) * unitW;
|
||||
}
|
||||
else {
|
||||
/* Project on cone. */
|
||||
const float a = distSq;
|
||||
const float b = relativePosition * relativeVelocity;
|
||||
const float c = absSq(relativeVelocity) - absSq(cross(relativePosition, relativeVelocity)) / (distSq - combinedRadiusSq);
|
||||
const float t = (b + std::sqrt(sqr(b) - a * c)) / a;
|
||||
const Vector3 w = relativeVelocity - t * relativePosition;
|
||||
const float wLength = abs(w);
|
||||
const Vector3 unitW = w / wLength;
|
||||
|
||||
plane.normal = unitW;
|
||||
u = (combinedRadius * t - wLength) * unitW;
|
||||
}
|
||||
}
|
||||
else {
|
||||
/* Collision. */
|
||||
const float invTimeStep = 1.0f / sim_->timeStep_;
|
||||
const Vector3 w = relativeVelocity - invTimeStep * relativePosition;
|
||||
const float wLength = abs(w);
|
||||
const Vector3 unitW = w / wLength;
|
||||
|
||||
plane.normal = unitW;
|
||||
u = (combinedRadius * invTimeStep - wLength) * unitW;
|
||||
}
|
||||
|
||||
plane.point = velocity_ + 0.5f * u;
|
||||
orcaPlanes_.push_back(plane);
|
||||
}
|
||||
|
||||
const size_t planeFail = linearProgram3(orcaPlanes_, maxSpeed_, prefVelocity_, false, newVelocity_);
|
||||
|
||||
if (planeFail < orcaPlanes_.size()) {
|
||||
linearProgram4(orcaPlanes_, planeFail, maxSpeed_, newVelocity_);
|
||||
}
|
||||
}
|
||||
|
||||
void Agent3D::insertAgentNeighbor(const Agent3D *agent, float &rangeSq)
|
||||
{
|
||||
// no point processing same agent
|
||||
if (this == agent) {
|
||||
return;
|
||||
}
|
||||
// ignore other agent if layers/mask bitmasks have no matching bit
|
||||
if ((avoidance_mask_ & agent->avoidance_layers_) == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (avoidance_priority_ > agent->avoidance_priority_) {
|
||||
return;
|
||||
}
|
||||
|
||||
const float distSq = absSq(position_ - agent->position_);
|
||||
|
||||
if (distSq < rangeSq) {
|
||||
if (agentNeighbors_.size() < maxNeighbors_) {
|
||||
agentNeighbors_.push_back(std::make_pair(distSq, agent));
|
||||
}
|
||||
|
||||
size_t i = agentNeighbors_.size() - 1;
|
||||
|
||||
while (i != 0 && distSq < agentNeighbors_[i - 1].first) {
|
||||
agentNeighbors_[i] = agentNeighbors_[i - 1];
|
||||
--i;
|
||||
}
|
||||
|
||||
agentNeighbors_[i] = std::make_pair(distSq, agent);
|
||||
|
||||
if (agentNeighbors_.size() == maxNeighbors_) {
|
||||
rangeSq = agentNeighbors_.back().first;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void Agent3D::update(RVOSimulator3D *sim_)
|
||||
{
|
||||
velocity_ = newVelocity_;
|
||||
position_ += velocity_ * sim_->timeStep_;
|
||||
}
|
||||
|
||||
bool linearProgram1(const std::vector<Plane> &planes, size_t planeNo, const Line3D &line, float radius, const Vector3 &optVelocity, bool directionOpt, Vector3 &result)
|
||||
{
|
||||
const float dotProduct = line.point * line.direction;
|
||||
const float discriminant = sqr(dotProduct) + sqr(radius) - absSq(line.point);
|
||||
|
||||
if (discriminant < 0.0f) {
|
||||
/* Max speed sphere fully invalidates line. */
|
||||
return false;
|
||||
}
|
||||
|
||||
const float sqrtDiscriminant = std::sqrt(discriminant);
|
||||
float tLeft = -dotProduct - sqrtDiscriminant;
|
||||
float tRight = -dotProduct + sqrtDiscriminant;
|
||||
|
||||
for (size_t i = 0; i < planeNo; ++i) {
|
||||
const float numerator = (planes[i].point - line.point) * planes[i].normal;
|
||||
const float denominator = line.direction * planes[i].normal;
|
||||
|
||||
if (sqr(denominator) <= RVO3D_EPSILON) {
|
||||
/* Lines3D line is (almost) parallel to plane i. */
|
||||
if (numerator > 0.0f) {
|
||||
return false;
|
||||
}
|
||||
else {
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
const float t = numerator / denominator;
|
||||
|
||||
if (denominator >= 0.0f) {
|
||||
/* Plane i bounds line on the left. */
|
||||
tLeft = std::max(tLeft, t);
|
||||
}
|
||||
else {
|
||||
/* Plane i bounds line on the right. */
|
||||
tRight = std::min(tRight, t);
|
||||
}
|
||||
|
||||
if (tLeft > tRight) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
if (directionOpt) {
|
||||
/* Optimize direction. */
|
||||
if (optVelocity * line.direction > 0.0f) {
|
||||
/* Take right extreme. */
|
||||
result = line.point + tRight * line.direction;
|
||||
}
|
||||
else {
|
||||
/* Take left extreme. */
|
||||
result = line.point + tLeft * line.direction;
|
||||
}
|
||||
}
|
||||
else {
|
||||
/* Optimize closest point. */
|
||||
const float t = line.direction * (optVelocity - line.point);
|
||||
|
||||
if (t < tLeft) {
|
||||
result = line.point + tLeft * line.direction;
|
||||
}
|
||||
else if (t > tRight) {
|
||||
result = line.point + tRight * line.direction;
|
||||
}
|
||||
else {
|
||||
result = line.point + t * line.direction;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool linearProgram2(const std::vector<Plane> &planes, size_t planeNo, float radius, const Vector3 &optVelocity, bool directionOpt, Vector3 &result)
|
||||
{
|
||||
const float planeDist = planes[planeNo].point * planes[planeNo].normal;
|
||||
const float planeDistSq = sqr(planeDist);
|
||||
const float radiusSq = sqr(radius);
|
||||
|
||||
if (planeDistSq > radiusSq) {
|
||||
/* Max speed sphere fully invalidates plane planeNo. */
|
||||
return false;
|
||||
}
|
||||
|
||||
const float planeRadiusSq = radiusSq - planeDistSq;
|
||||
|
||||
const Vector3 planeCenter = planeDist * planes[planeNo].normal;
|
||||
|
||||
if (directionOpt) {
|
||||
/* Project direction optVelocity on plane planeNo. */
|
||||
const Vector3 planeOptVelocity = optVelocity - (optVelocity * planes[planeNo].normal) * planes[planeNo].normal;
|
||||
const float planeOptVelocityLengthSq = absSq(planeOptVelocity);
|
||||
|
||||
if (planeOptVelocityLengthSq <= RVO3D_EPSILON) {
|
||||
result = planeCenter;
|
||||
}
|
||||
else {
|
||||
result = planeCenter + std::sqrt(planeRadiusSq / planeOptVelocityLengthSq) * planeOptVelocity;
|
||||
}
|
||||
}
|
||||
else {
|
||||
/* Project point optVelocity on plane planeNo. */
|
||||
result = optVelocity + ((planes[planeNo].point - optVelocity) * planes[planeNo].normal) * planes[planeNo].normal;
|
||||
|
||||
/* If outside planeCircle, project on planeCircle. */
|
||||
if (absSq(result) > radiusSq) {
|
||||
const Vector3 planeResult = result - planeCenter;
|
||||
const float planeResultLengthSq = absSq(planeResult);
|
||||
result = planeCenter + std::sqrt(planeRadiusSq / planeResultLengthSq) * planeResult;
|
||||
}
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < planeNo; ++i) {
|
||||
if (planes[i].normal * (planes[i].point - result) > 0.0f) {
|
||||
/* Result does not satisfy constraint i. Compute new optimal result. */
|
||||
/* Compute intersection line of plane i and plane planeNo. */
|
||||
Vector3 crossProduct = cross(planes[i].normal, planes[planeNo].normal);
|
||||
|
||||
if (absSq(crossProduct) <= RVO3D_EPSILON) {
|
||||
/* Planes planeNo and i are (almost) parallel, and plane i fully invalidates plane planeNo. */
|
||||
return false;
|
||||
}
|
||||
|
||||
Line3D line;
|
||||
line.direction = normalize(crossProduct);
|
||||
const Vector3 lineNormal = cross(line.direction, planes[planeNo].normal);
|
||||
line.point = planes[planeNo].point + (((planes[i].point - planes[planeNo].point) * planes[i].normal) / (lineNormal * planes[i].normal)) * lineNormal;
|
||||
|
||||
if (!linearProgram1(planes, i, line, radius, optVelocity, directionOpt, result)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
size_t linearProgram3(const std::vector<Plane> &planes, float radius, const Vector3 &optVelocity, bool directionOpt, Vector3 &result)
|
||||
{
|
||||
if (directionOpt) {
|
||||
/* Optimize direction. Note that the optimization velocity is of unit length in this case. */
|
||||
result = optVelocity * radius;
|
||||
}
|
||||
else if (absSq(optVelocity) > sqr(radius)) {
|
||||
/* Optimize closest point and outside circle. */
|
||||
result = normalize(optVelocity) * radius;
|
||||
}
|
||||
else {
|
||||
/* Optimize closest point and inside circle. */
|
||||
result = optVelocity;
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < planes.size(); ++i) {
|
||||
if (planes[i].normal * (planes[i].point - result) > 0.0f) {
|
||||
/* Result does not satisfy constraint i. Compute new optimal result. */
|
||||
const Vector3 tempResult = result;
|
||||
|
||||
if (!linearProgram2(planes, i, radius, optVelocity, directionOpt, result)) {
|
||||
result = tempResult;
|
||||
return i;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return planes.size();
|
||||
}
|
||||
|
||||
void linearProgram4(const std::vector<Plane> &planes, size_t beginPlane, float radius, Vector3 &result)
|
||||
{
|
||||
float distance = 0.0f;
|
||||
|
||||
for (size_t i = beginPlane; i < planes.size(); ++i) {
|
||||
if (planes[i].normal * (planes[i].point - result) > distance) {
|
||||
/* Result does not satisfy constraint of plane i. */
|
||||
std::vector<Plane> projPlanes;
|
||||
|
||||
for (size_t j = 0; j < i; ++j) {
|
||||
Plane plane;
|
||||
|
||||
const Vector3 crossProduct = cross(planes[j].normal, planes[i].normal);
|
||||
|
||||
if (absSq(crossProduct) <= RVO3D_EPSILON) {
|
||||
/* Plane i and plane j are (almost) parallel. */
|
||||
if (planes[i].normal * planes[j].normal > 0.0f) {
|
||||
/* Plane i and plane j point in the same direction. */
|
||||
continue;
|
||||
}
|
||||
else {
|
||||
/* Plane i and plane j point in opposite direction. */
|
||||
plane.point = 0.5f * (planes[i].point + planes[j].point);
|
||||
}
|
||||
}
|
||||
else {
|
||||
/* Plane.point is point on line of intersection between plane i and plane j. */
|
||||
const Vector3 lineNormal = cross(crossProduct, planes[i].normal);
|
||||
plane.point = planes[i].point + (((planes[j].point - planes[i].point) * planes[j].normal) / (lineNormal * planes[j].normal)) * lineNormal;
|
||||
}
|
||||
|
||||
plane.normal = normalize(planes[j].normal - planes[i].normal);
|
||||
projPlanes.push_back(plane);
|
||||
}
|
||||
|
||||
const Vector3 tempResult = result;
|
||||
|
||||
if (linearProgram3(projPlanes, radius, planes[i].normal, true, result) < projPlanes.size()) {
|
||||
/* This should in principle not happen. The result is by definition already in the feasible region of this linear program. If it fails, it is due to small floating point error, and the current result is kept. */
|
||||
result = tempResult;
|
||||
}
|
||||
|
||||
distance = planes[i].normal * (planes[i].point - result);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,264 @@
|
||||
/*
|
||||
* KdTree3d.cc
|
||||
* RVO2-3D Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "KdTree3d.h"
|
||||
|
||||
#include <algorithm>
|
||||
#include <utility>
|
||||
|
||||
#include "Agent3d.h"
|
||||
#include "RVOSimulator3d.h"
|
||||
#include "Vector3.h"
|
||||
|
||||
namespace RVO3D {
|
||||
namespace {
|
||||
/**
|
||||
* @brief The maximum size of a k-D leaf node.
|
||||
*/
|
||||
const std::size_t RVO3D_MAX_LEAF_SIZE = 10U;
|
||||
} /* namespace */
|
||||
|
||||
/**
|
||||
* @brief Defines an agent k-D tree node.
|
||||
*/
|
||||
class KdTree3D::AgentTreeNode {
|
||||
public:
|
||||
/**
|
||||
* @brief Constructs an agent k-D tree node.
|
||||
*/
|
||||
AgentTreeNode();
|
||||
|
||||
/**
|
||||
* @brief The beginning node number.
|
||||
*/
|
||||
std::size_t begin;
|
||||
|
||||
/**
|
||||
* @brief The ending node number.
|
||||
*/
|
||||
std::size_t end;
|
||||
|
||||
/**
|
||||
* @brief The left node number.
|
||||
*/
|
||||
std::size_t left;
|
||||
|
||||
/**
|
||||
* @brief The right node number.
|
||||
*/
|
||||
std::size_t right;
|
||||
|
||||
/**
|
||||
* @brief The maximum coordinates.
|
||||
*/
|
||||
Vector3 maxCoord;
|
||||
|
||||
/**
|
||||
* @brief The minimum coordinates.
|
||||
*/
|
||||
Vector3 minCoord;
|
||||
};
|
||||
|
||||
KdTree3D::AgentTreeNode::AgentTreeNode()
|
||||
: begin(0U), end(0U), left(0U), right(0U) {}
|
||||
|
||||
KdTree3D::KdTree3D(RVOSimulator3D *sim) : sim_(sim) {}
|
||||
|
||||
KdTree3D::~KdTree3D() {}
|
||||
|
||||
void KdTree3D::buildAgentTree(std::vector<Agent3D *> agents) {
|
||||
agents_.swap(agents_);
|
||||
|
||||
if (!agents_.empty()) {
|
||||
agentTree_.resize(2U * agents_.size() - 1U);
|
||||
buildAgentTreeRecursive(0U, agents_.size(), 0U);
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree3D::buildAgentTreeRecursive(std::size_t begin, std::size_t end,
|
||||
std::size_t node) {
|
||||
agentTree_[node].begin = begin;
|
||||
agentTree_[node].end = end;
|
||||
agentTree_[node].minCoord = agents_[begin]->position_;
|
||||
agentTree_[node].maxCoord = agents_[begin]->position_;
|
||||
|
||||
for (std::size_t i = begin + 1U; i < end; ++i) {
|
||||
agentTree_[node].maxCoord[0] =
|
||||
std::max(agentTree_[node].maxCoord[0], agents_[i]->position_.x());
|
||||
agentTree_[node].minCoord[0] =
|
||||
std::min(agentTree_[node].minCoord[0], agents_[i]->position_.x());
|
||||
agentTree_[node].maxCoord[1] =
|
||||
std::max(agentTree_[node].maxCoord[1], agents_[i]->position_.y());
|
||||
agentTree_[node].minCoord[1] =
|
||||
std::min(agentTree_[node].minCoord[1], agents_[i]->position_.y());
|
||||
agentTree_[node].maxCoord[2] =
|
||||
std::max(agentTree_[node].maxCoord[2], agents_[i]->position_.z());
|
||||
agentTree_[node].minCoord[2] =
|
||||
std::min(agentTree_[node].minCoord[2], agents_[i]->position_.z());
|
||||
}
|
||||
|
||||
if (end - begin > RVO3D_MAX_LEAF_SIZE) {
|
||||
/* No leaf node. */
|
||||
std::size_t coord = 0U;
|
||||
|
||||
if (agentTree_[node].maxCoord[0] - agentTree_[node].minCoord[0] >
|
||||
agentTree_[node].maxCoord[1] - agentTree_[node].minCoord[1] &&
|
||||
agentTree_[node].maxCoord[0] - agentTree_[node].minCoord[0] >
|
||||
agentTree_[node].maxCoord[2] - agentTree_[node].minCoord[2]) {
|
||||
coord = 0U;
|
||||
} else if (agentTree_[node].maxCoord[1] - agentTree_[node].minCoord[1] >
|
||||
agentTree_[node].maxCoord[2] - agentTree_[node].minCoord[2]) {
|
||||
coord = 1U;
|
||||
} else {
|
||||
coord = 2U;
|
||||
}
|
||||
|
||||
const float splitValue = 0.5F * (agentTree_[node].maxCoord[coord] +
|
||||
agentTree_[node].minCoord[coord]);
|
||||
|
||||
std::size_t left = begin;
|
||||
|
||||
std::size_t right = end;
|
||||
|
||||
while (left < right) {
|
||||
while (left < right && agents_[left]->position_[coord] < splitValue) {
|
||||
++left;
|
||||
}
|
||||
|
||||
while (right > left &&
|
||||
agents_[right - 1U]->position_[coord] >= splitValue) {
|
||||
--right;
|
||||
}
|
||||
|
||||
if (left < right) {
|
||||
std::swap(agents_[left], agents_[right - 1U]);
|
||||
++left;
|
||||
--right;
|
||||
}
|
||||
}
|
||||
|
||||
std::size_t leftSize = left - begin;
|
||||
|
||||
if (leftSize == 0U) {
|
||||
++leftSize;
|
||||
++left;
|
||||
}
|
||||
|
||||
agentTree_[node].left = node + 1U;
|
||||
agentTree_[node].right = node + 2U * leftSize;
|
||||
|
||||
buildAgentTreeRecursive(begin, left, agentTree_[node].left);
|
||||
buildAgentTreeRecursive(left, end, agentTree_[node].right);
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree3D::computeAgentNeighbors(Agent3D *agent, float rangeSq) const {
|
||||
queryAgentTreeRecursive(agent, rangeSq, 0U);
|
||||
}
|
||||
|
||||
void KdTree3D::queryAgentTreeRecursive(Agent3D *agent, float &rangeSq,
|
||||
std::size_t node) const {
|
||||
if (agentTree_[node].end - agentTree_[node].begin <= RVO3D_MAX_LEAF_SIZE) {
|
||||
for (std::size_t i = agentTree_[node].begin; i < agentTree_[node].end;
|
||||
++i) {
|
||||
agent->insertAgentNeighbor(agents_[i], rangeSq);
|
||||
}
|
||||
} else {
|
||||
const float distSqLeftMinX =
|
||||
std::max(0.0F, agentTree_[agentTree_[node].left].minCoord[0] -
|
||||
agent->position_.x());
|
||||
const float distSqLeftMaxX =
|
||||
std::max(0.0F, agent->position_.x() -
|
||||
agentTree_[agentTree_[node].left].maxCoord[0]);
|
||||
const float distSqLeftMinY =
|
||||
std::max(0.0F, agentTree_[agentTree_[node].left].minCoord[1] -
|
||||
agent->position_.y());
|
||||
const float distSqLeftMaxY =
|
||||
std::max(0.0F, agent->position_.y() -
|
||||
agentTree_[agentTree_[node].left].maxCoord[1]);
|
||||
const float distSqLeftMinZ =
|
||||
std::max(0.0F, agentTree_[agentTree_[node].left].minCoord[2] -
|
||||
agent->position_.z());
|
||||
const float distSqLeftMaxZ =
|
||||
std::max(0.0F, agent->position_.z() -
|
||||
agentTree_[agentTree_[node].left].maxCoord[2]);
|
||||
|
||||
const float distSqLeft =
|
||||
distSqLeftMinX * distSqLeftMinX + distSqLeftMaxX * distSqLeftMaxX +
|
||||
distSqLeftMinY * distSqLeftMinY + distSqLeftMaxY * distSqLeftMaxY +
|
||||
distSqLeftMinZ * distSqLeftMinZ + distSqLeftMaxZ * distSqLeftMaxZ;
|
||||
|
||||
const float distSqRightMinX =
|
||||
std::max(0.0F, agentTree_[agentTree_[node].right].minCoord[0] -
|
||||
agent->position_.x());
|
||||
const float distSqRightMaxX =
|
||||
std::max(0.0F, agent->position_.x() -
|
||||
agentTree_[agentTree_[node].right].maxCoord[0]);
|
||||
const float distSqRightMinY =
|
||||
std::max(0.0F, agentTree_[agentTree_[node].right].minCoord[1] -
|
||||
agent->position_.y());
|
||||
const float distSqRightMaxY =
|
||||
std::max(0.0F, agent->position_.y() -
|
||||
agentTree_[agentTree_[node].right].maxCoord[1]);
|
||||
const float distSqRightMinZ =
|
||||
std::max(0.0F, agentTree_[agentTree_[node].right].minCoord[2] -
|
||||
agent->position_.z());
|
||||
const float distSqRightMaxZ =
|
||||
std::max(0.0F, agent->position_.z() -
|
||||
agentTree_[agentTree_[node].right].maxCoord[2]);
|
||||
|
||||
const float distSqRight =
|
||||
distSqRightMinX * distSqRightMinX + distSqRightMaxX * distSqRightMaxX +
|
||||
distSqRightMinY * distSqRightMinY + distSqRightMaxY * distSqRightMaxY +
|
||||
distSqRightMinZ * distSqRightMinZ + distSqRightMaxZ * distSqRightMaxZ;
|
||||
|
||||
if (distSqLeft < distSqRight) {
|
||||
if (distSqLeft < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].left);
|
||||
|
||||
if (distSqRight < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].right);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if (distSqRight < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].right);
|
||||
|
||||
if (distSqLeft < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].left);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
} /* namespace RVO3D */
|
||||
@ -1,161 +0,0 @@
|
||||
/*
|
||||
* KdTree.cpp
|
||||
* RVO2-3D Library
|
||||
*
|
||||
* Copyright 2008 University of North Carolina at Chapel Hill
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "KdTree3d.h"
|
||||
|
||||
#include <algorithm>
|
||||
|
||||
#include "Agent3d.h"
|
||||
#include "Definitions.h"
|
||||
#include "RVOSimulator3d.h"
|
||||
|
||||
namespace RVO3D {
|
||||
const size_t RVO3D_MAX_LEAF_SIZE = 10;
|
||||
|
||||
KdTree3D::KdTree3D(RVOSimulator3D *sim) : sim_(sim) { }
|
||||
|
||||
void KdTree3D::buildAgentTree(std::vector<Agent3D *> agents)
|
||||
{
|
||||
agents_.swap(agents);
|
||||
|
||||
if (!agents_.empty()) {
|
||||
agentTree_.resize(2 * agents_.size() - 1);
|
||||
buildAgentTreeRecursive(0, agents_.size(), 0);
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree3D::buildAgentTreeRecursive(size_t begin, size_t end, size_t node)
|
||||
{
|
||||
agentTree_[node].begin = begin;
|
||||
agentTree_[node].end = end;
|
||||
agentTree_[node].minCoord = agents_[begin]->position_;
|
||||
agentTree_[node].maxCoord = agents_[begin]->position_;
|
||||
|
||||
for (size_t i = begin + 1; i < end; ++i) {
|
||||
agentTree_[node].maxCoord[0] = std::max(agentTree_[node].maxCoord[0], agents_[i]->position_.x());
|
||||
agentTree_[node].minCoord[0] = std::min(agentTree_[node].minCoord[0], agents_[i]->position_.x());
|
||||
agentTree_[node].maxCoord[1] = std::max(agentTree_[node].maxCoord[1], agents_[i]->position_.y());
|
||||
agentTree_[node].minCoord[1] = std::min(agentTree_[node].minCoord[1], agents_[i]->position_.y());
|
||||
agentTree_[node].maxCoord[2] = std::max(agentTree_[node].maxCoord[2], agents_[i]->position_.z());
|
||||
agentTree_[node].minCoord[2] = std::min(agentTree_[node].minCoord[2], agents_[i]->position_.z());
|
||||
}
|
||||
|
||||
if (end - begin > RVO3D_MAX_LEAF_SIZE) {
|
||||
/* No leaf node. */
|
||||
size_t coord;
|
||||
|
||||
if (agentTree_[node].maxCoord[0] - agentTree_[node].minCoord[0] > agentTree_[node].maxCoord[1] - agentTree_[node].minCoord[1] && agentTree_[node].maxCoord[0] - agentTree_[node].minCoord[0] > agentTree_[node].maxCoord[2] - agentTree_[node].minCoord[2]) {
|
||||
coord = 0;
|
||||
}
|
||||
else if (agentTree_[node].maxCoord[1] - agentTree_[node].minCoord[1] > agentTree_[node].maxCoord[2] - agentTree_[node].minCoord[2]) {
|
||||
coord = 1;
|
||||
}
|
||||
else {
|
||||
coord = 2;
|
||||
}
|
||||
|
||||
const float splitValue = 0.5f * (agentTree_[node].maxCoord[coord] + agentTree_[node].minCoord[coord]);
|
||||
|
||||
size_t left = begin;
|
||||
|
||||
size_t right = end;
|
||||
|
||||
while (left < right) {
|
||||
while (left < right && agents_[left]->position_[coord] < splitValue) {
|
||||
++left;
|
||||
}
|
||||
|
||||
while (right > left && agents_[right - 1]->position_[coord] >= splitValue) {
|
||||
--right;
|
||||
}
|
||||
|
||||
if (left < right) {
|
||||
std::swap(agents_[left], agents_[right - 1]);
|
||||
++left;
|
||||
--right;
|
||||
}
|
||||
}
|
||||
|
||||
size_t leftSize = left - begin;
|
||||
|
||||
if (leftSize == 0) {
|
||||
++leftSize;
|
||||
++left;
|
||||
++right;
|
||||
}
|
||||
|
||||
agentTree_[node].left = node + 1;
|
||||
agentTree_[node].right = node + 2 * leftSize;
|
||||
|
||||
buildAgentTreeRecursive(begin, left, agentTree_[node].left);
|
||||
buildAgentTreeRecursive(left, end, agentTree_[node].right);
|
||||
}
|
||||
}
|
||||
|
||||
void KdTree3D::computeAgentNeighbors(Agent3D *agent, float rangeSq) const
|
||||
{
|
||||
queryAgentTreeRecursive(agent, rangeSq, 0);
|
||||
}
|
||||
|
||||
void KdTree3D::queryAgentTreeRecursive(Agent3D *agent, float &rangeSq, size_t node) const
|
||||
{
|
||||
if (agentTree_[node].end - agentTree_[node].begin <= RVO3D_MAX_LEAF_SIZE) {
|
||||
for (size_t i = agentTree_[node].begin; i < agentTree_[node].end; ++i) {
|
||||
agent->insertAgentNeighbor(agents_[i], rangeSq);
|
||||
}
|
||||
}
|
||||
else {
|
||||
const float distSqLeft = sqr(std::max(0.0f, agentTree_[agentTree_[node].left].minCoord[0] - agent->position_.x())) + sqr(std::max(0.0f, agent->position_.x() - agentTree_[agentTree_[node].left].maxCoord[0])) + sqr(std::max(0.0f, agentTree_[agentTree_[node].left].minCoord[1] - agent->position_.y())) + sqr(std::max(0.0f, agent->position_.y() - agentTree_[agentTree_[node].left].maxCoord[1])) + sqr(std::max(0.0f, agentTree_[agentTree_[node].left].minCoord[2] - agent->position_.z())) + sqr(std::max(0.0f, agent->position_.z() - agentTree_[agentTree_[node].left].maxCoord[2]));
|
||||
|
||||
const float distSqRight = sqr(std::max(0.0f, agentTree_[agentTree_[node].right].minCoord[0] - agent->position_.x())) + sqr(std::max(0.0f, agent->position_.x() - agentTree_[agentTree_[node].right].maxCoord[0])) + sqr(std::max(0.0f, agentTree_[agentTree_[node].right].minCoord[1] - agent->position_.y())) + sqr(std::max(0.0f, agent->position_.y() - agentTree_[agentTree_[node].right].maxCoord[1])) + sqr(std::max(0.0f, agentTree_[agentTree_[node].right].minCoord[2] - agent->position_.z())) + sqr(std::max(0.0f, agent->position_.z() - agentTree_[agentTree_[node].right].maxCoord[2]));
|
||||
|
||||
if (distSqLeft < distSqRight) {
|
||||
if (distSqLeft < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].left);
|
||||
|
||||
if (distSqRight < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].right);
|
||||
}
|
||||
}
|
||||
}
|
||||
else {
|
||||
if (distSqRight < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].right);
|
||||
|
||||
if (distSqLeft < rangeSq) {
|
||||
queryAgentTreeRecursive(agent, rangeSq, agentTree_[node].left);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,67 @@
|
||||
/*
|
||||
* Plane.h
|
||||
* RVO2-3D Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#ifndef RVO3D_PLANE_H_
|
||||
#define RVO3D_PLANE_H_
|
||||
|
||||
/**
|
||||
* @file Plane.h
|
||||
* @brief Contains the Plane class.
|
||||
*/
|
||||
|
||||
#include "Vector3.h"
|
||||
|
||||
namespace RVO3D {
|
||||
/**
|
||||
* @brief Defines a plane.
|
||||
*/
|
||||
class Plane {
|
||||
public:
|
||||
/**
|
||||
* @brief Constructs a plane.
|
||||
*/
|
||||
Plane();
|
||||
|
||||
/**
|
||||
* @brief A point on the plane.
|
||||
*/
|
||||
Vector3 point;
|
||||
|
||||
/**
|
||||
* @brief The normal to the plane.
|
||||
*/
|
||||
Vector3 normal;
|
||||
};
|
||||
} /* namespace RVO3D */
|
||||
|
||||
#endif /* RVO3D_PLANE_H_ */
|
||||
@ -0,0 +1,250 @@
|
||||
/*
|
||||
* RVOSimulator3d.cc
|
||||
* RVO2-3D Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "RVOSimulator3d.h"
|
||||
|
||||
#include <utility>
|
||||
|
||||
#ifdef _OPENMP
|
||||
#include <omp.h>
|
||||
#endif /* _OPENMP */
|
||||
|
||||
#include "Agent3d.h"
|
||||
#include "KdTree3d.h"
|
||||
#include "Plane.h"
|
||||
|
||||
namespace RVO3D {
|
||||
RVOSimulator3D::RVOSimulator3D()
|
||||
: defaultAgent_(NULL),
|
||||
kdTree_(new KdTree3D(this)),
|
||||
globalTime_(0.0F),
|
||||
timeStep_(0.0F) {}
|
||||
|
||||
RVOSimulator3D::RVOSimulator3D(float timeStep, float neighborDist,
|
||||
std::size_t maxNeighbors, float timeHorizon,
|
||||
float radius, float maxSpeed,
|
||||
const Vector3 &velocity)
|
||||
: defaultAgent_(new Agent3D()),
|
||||
kdTree_(new KdTree3D(this)),
|
||||
globalTime_(0.0F),
|
||||
timeStep_(timeStep) {
|
||||
defaultAgent_->maxNeighbors_ = maxNeighbors;
|
||||
defaultAgent_->maxSpeed_ = maxSpeed;
|
||||
defaultAgent_->neighborDist_ = neighborDist;
|
||||
defaultAgent_->radius_ = radius;
|
||||
defaultAgent_->timeHorizon_ = timeHorizon;
|
||||
defaultAgent_->velocity_ = velocity;
|
||||
}
|
||||
|
||||
RVOSimulator3D::~RVOSimulator3D() {
|
||||
delete defaultAgent_;
|
||||
delete kdTree_;
|
||||
|
||||
for (std::size_t i = 0U; i < agents_.size(); ++i) {
|
||||
delete agents_[i];
|
||||
}
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator3D::getAgentNumAgentNeighbors(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->agentNeighbors_.size();
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator3D::getAgentAgentNeighbor(std::size_t agentNo,
|
||||
std::size_t neighborNo) const {
|
||||
return agents_[agentNo]->agentNeighbors_[neighborNo].second->id_;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator3D::getAgentNumORCAPlanes(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->orcaPlanes_.size();
|
||||
}
|
||||
|
||||
const Plane &RVOSimulator3D::getAgentORCAPlane(std::size_t agentNo,
|
||||
std::size_t planeNo) const {
|
||||
return agents_[agentNo]->orcaPlanes_[planeNo];
|
||||
}
|
||||
|
||||
void RVOSimulator3D::removeAgent(std::size_t agentNo) {
|
||||
delete agents_[agentNo];
|
||||
agents_[agentNo] = agents_.back();
|
||||
agents_.pop_back();
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator3D::addAgent(const Vector3 &position) {
|
||||
if (defaultAgent_ == NULL) {
|
||||
return RVO3D_ERROR;
|
||||
}
|
||||
|
||||
Agent3D *agent = new Agent3D();
|
||||
|
||||
agent->position_ = position;
|
||||
agent->maxNeighbors_ = defaultAgent_->maxNeighbors_;
|
||||
agent->maxSpeed_ = defaultAgent_->maxSpeed_;
|
||||
agent->neighborDist_ = defaultAgent_->neighborDist_;
|
||||
agent->radius_ = defaultAgent_->radius_;
|
||||
agent->timeHorizon_ = defaultAgent_->timeHorizon_;
|
||||
agent->velocity_ = defaultAgent_->velocity_;
|
||||
|
||||
agent->id_ = agents_.size();
|
||||
|
||||
agents_.push_back(agent);
|
||||
|
||||
return agents_.size() - 1U;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator3D::addAgent(const Vector3 &position, float neighborDist,
|
||||
std::size_t maxNeighbors, float timeHorizon,
|
||||
float radius, float maxSpeed,
|
||||
const Vector3 &velocity) {
|
||||
Agent3D *agent = new Agent3D();
|
||||
|
||||
agent->position_ = position;
|
||||
agent->maxNeighbors_ = maxNeighbors;
|
||||
agent->maxSpeed_ = maxSpeed;
|
||||
agent->neighborDist_ = neighborDist;
|
||||
agent->radius_ = radius;
|
||||
agent->timeHorizon_ = timeHorizon;
|
||||
agent->velocity_ = velocity;
|
||||
|
||||
agent->id_ = agents_.size();
|
||||
|
||||
agents_.push_back(agent);
|
||||
|
||||
return agents_.size() - 1U;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::doStep() {
|
||||
kdTree_->buildAgentTree(agents_);
|
||||
|
||||
#ifdef _OPENMP
|
||||
#pragma omp parallel for
|
||||
#endif /* _OPENMP */
|
||||
for (int i = 0; i < static_cast<int>(agents_.size()); ++i) {
|
||||
agents_[i]->computeNeighbors(this);
|
||||
agents_[i]->computeNewVelocity(this);
|
||||
}
|
||||
|
||||
#ifdef _OPENMP
|
||||
#pragma omp parallel for
|
||||
#endif /* _OPENMP */
|
||||
for (int i = 0; i < static_cast<int>(agents_.size()); ++i) {
|
||||
agents_[i]->update(this);
|
||||
}
|
||||
|
||||
globalTime_ += timeStep_;
|
||||
}
|
||||
|
||||
std::size_t RVOSimulator3D::getAgentMaxNeighbors(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->maxNeighbors_;
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getAgentMaxSpeed(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->maxSpeed_;
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getAgentNeighborDist(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->neighborDist_;
|
||||
}
|
||||
|
||||
const Vector3 &RVOSimulator3D::getAgentPosition(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->position_;
|
||||
}
|
||||
|
||||
const Vector3 &RVOSimulator3D::getAgentPrefVelocity(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->prefVelocity_;
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getAgentRadius(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->radius_;
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getAgentTimeHorizon(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->timeHorizon_;
|
||||
}
|
||||
|
||||
const Vector3 &RVOSimulator3D::getAgentVelocity(std::size_t agentNo) const {
|
||||
return agents_[agentNo]->velocity_;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentDefaults(float neighborDist,
|
||||
std::size_t maxNeighbors, float timeHorizon,
|
||||
float radius, float maxSpeed,
|
||||
const Vector3 &velocity) {
|
||||
if (defaultAgent_ == NULL) {
|
||||
defaultAgent_ = new Agent3D();
|
||||
}
|
||||
|
||||
defaultAgent_->maxNeighbors_ = maxNeighbors;
|
||||
defaultAgent_->maxSpeed_ = maxSpeed;
|
||||
defaultAgent_->neighborDist_ = neighborDist;
|
||||
defaultAgent_->radius_ = radius;
|
||||
defaultAgent_->timeHorizon_ = timeHorizon;
|
||||
defaultAgent_->velocity_ = velocity;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentMaxNeighbors(std::size_t agentNo,
|
||||
std::size_t maxNeighbors) {
|
||||
agents_[agentNo]->maxNeighbors_ = maxNeighbors;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentMaxSpeed(std::size_t agentNo, float maxSpeed) {
|
||||
agents_[agentNo]->maxSpeed_ = maxSpeed;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentNeighborDist(std::size_t agentNo,
|
||||
float neighborDist) {
|
||||
agents_[agentNo]->neighborDist_ = neighborDist;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentPosition(std::size_t agentNo,
|
||||
const Vector3 &position) {
|
||||
agents_[agentNo]->position_ = position;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentPrefVelocity(std::size_t agentNo,
|
||||
const Vector3 &prefVelocity) {
|
||||
agents_[agentNo]->prefVelocity_ = prefVelocity;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentRadius(std::size_t agentNo, float radius) {
|
||||
agents_[agentNo]->radius_ = radius;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentTimeHorizon(std::size_t agentNo, float timeHorizon) {
|
||||
agents_[agentNo]->timeHorizon_ = timeHorizon;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentVelocity(std::size_t agentNo,
|
||||
const Vector3 &velocity) {
|
||||
agents_[agentNo]->velocity_ = velocity;
|
||||
}
|
||||
} /* namespace RVO3D */
|
||||
@ -1,274 +0,0 @@
|
||||
/*
|
||||
* RVOSimulator.cpp
|
||||
* RVO2-3D Library
|
||||
*
|
||||
* Copyright 2008 University of North Carolina at Chapel Hill
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <http://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "RVOSimulator3d.h"
|
||||
|
||||
#ifdef _OPENMP
|
||||
#include <omp.h>
|
||||
#endif
|
||||
|
||||
#include "Agent3d.h"
|
||||
#include "KdTree3d.h"
|
||||
|
||||
namespace RVO3D {
|
||||
RVOSimulator3D::RVOSimulator3D() : defaultAgent_(NULL), kdTree_(NULL), globalTime_(0.0f), timeStep_(0.0f)
|
||||
{
|
||||
kdTree_ = new KdTree3D(this);
|
||||
}
|
||||
|
||||
RVOSimulator3D::RVOSimulator3D(float timeStep, float neighborDist, size_t maxNeighbors, float timeHorizon, float radius, float maxSpeed, const Vector3 &velocity) : defaultAgent_(NULL), kdTree_(NULL), globalTime_(0.0f), timeStep_(timeStep)
|
||||
{
|
||||
kdTree_ = new KdTree3D(this);
|
||||
defaultAgent_ = new Agent3D();
|
||||
|
||||
defaultAgent_->maxNeighbors_ = maxNeighbors;
|
||||
defaultAgent_->maxSpeed_ = maxSpeed;
|
||||
defaultAgent_->neighborDist_ = neighborDist;
|
||||
defaultAgent_->radius_ = radius;
|
||||
defaultAgent_->timeHorizon_ = timeHorizon;
|
||||
defaultAgent_->velocity_ = velocity;
|
||||
}
|
||||
|
||||
RVOSimulator3D::~RVOSimulator3D()
|
||||
{
|
||||
if (defaultAgent_ != NULL) {
|
||||
delete defaultAgent_;
|
||||
}
|
||||
|
||||
for (size_t i = 0; i < agents_.size(); ++i) {
|
||||
delete agents_[i];
|
||||
}
|
||||
|
||||
if (kdTree_ != NULL) {
|
||||
delete kdTree_;
|
||||
}
|
||||
}
|
||||
|
||||
size_t RVOSimulator3D::getAgentNumAgentNeighbors(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->agentNeighbors_.size();
|
||||
}
|
||||
|
||||
size_t RVOSimulator3D::getAgentAgentNeighbor(size_t agentNo, size_t neighborNo) const
|
||||
{
|
||||
return agents_[agentNo]->agentNeighbors_[neighborNo].second->id_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator3D::getAgentNumORCAPlanes(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->orcaPlanes_.size();
|
||||
}
|
||||
|
||||
const Plane &RVOSimulator3D::getAgentORCAPlane(size_t agentNo, size_t planeNo) const
|
||||
{
|
||||
return agents_[agentNo]->orcaPlanes_[planeNo];
|
||||
}
|
||||
|
||||
void RVOSimulator3D::removeAgent(size_t agentNo)
|
||||
{
|
||||
delete agents_[agentNo];
|
||||
agents_[agentNo] = agents_.back();
|
||||
agents_.pop_back();
|
||||
}
|
||||
|
||||
size_t RVOSimulator3D::addAgent(const Vector3 &position)
|
||||
{
|
||||
if (defaultAgent_ == NULL) {
|
||||
return RVO3D_ERROR;
|
||||
}
|
||||
|
||||
Agent3D *agent = new Agent3D();
|
||||
|
||||
agent->position_ = position;
|
||||
agent->maxNeighbors_ = defaultAgent_->maxNeighbors_;
|
||||
agent->maxSpeed_ = defaultAgent_->maxSpeed_;
|
||||
agent->neighborDist_ = defaultAgent_->neighborDist_;
|
||||
agent->radius_ = defaultAgent_->radius_;
|
||||
agent->timeHorizon_ = defaultAgent_->timeHorizon_;
|
||||
agent->velocity_ = defaultAgent_->velocity_;
|
||||
|
||||
agent->id_ = agents_.size();
|
||||
|
||||
agents_.push_back(agent);
|
||||
|
||||
return agents_.size() - 1;
|
||||
}
|
||||
|
||||
size_t RVOSimulator3D::addAgent(const Vector3 &position, float neighborDist, size_t maxNeighbors, float timeHorizon, float radius, float maxSpeed, const Vector3 &velocity)
|
||||
{
|
||||
Agent3D *agent = new Agent3D();
|
||||
|
||||
agent->position_ = position;
|
||||
agent->maxNeighbors_ = maxNeighbors;
|
||||
agent->maxSpeed_ = maxSpeed;
|
||||
agent->neighborDist_ = neighborDist;
|
||||
agent->radius_ = radius;
|
||||
agent->timeHorizon_ = timeHorizon;
|
||||
agent->velocity_ = velocity;
|
||||
|
||||
agent->id_ = agents_.size();
|
||||
|
||||
agents_.push_back(agent);
|
||||
|
||||
return agents_.size() - 1;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::doStep()
|
||||
{
|
||||
kdTree_->buildAgentTree(agents_);
|
||||
|
||||
for (int i = 0; i < static_cast<int>(agents_.size()); ++i) {
|
||||
agents_[i]->computeNeighbors(this);
|
||||
agents_[i]->computeNewVelocity(this);
|
||||
}
|
||||
|
||||
for (int i = 0; i < static_cast<int>(agents_.size()); ++i) {
|
||||
agents_[i]->update(this);
|
||||
}
|
||||
|
||||
globalTime_ += timeStep_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator3D::getAgentMaxNeighbors(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->maxNeighbors_;
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getAgentMaxSpeed(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->maxSpeed_;
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getAgentNeighborDist(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->neighborDist_;
|
||||
}
|
||||
|
||||
const Vector3 &RVOSimulator3D::getAgentPosition(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->position_;
|
||||
}
|
||||
|
||||
const Vector3 &RVOSimulator3D::getAgentPrefVelocity(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->prefVelocity_;
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getAgentRadius(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->radius_;
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getAgentTimeHorizon(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->timeHorizon_;
|
||||
}
|
||||
|
||||
const Vector3 &RVOSimulator3D::getAgentVelocity(size_t agentNo) const
|
||||
{
|
||||
return agents_[agentNo]->velocity_;
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getGlobalTime() const
|
||||
{
|
||||
return globalTime_;
|
||||
}
|
||||
|
||||
size_t RVOSimulator3D::getNumAgents() const
|
||||
{
|
||||
return agents_.size();
|
||||
}
|
||||
|
||||
float RVOSimulator3D::getTimeStep() const
|
||||
{
|
||||
return timeStep_;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentDefaults(float neighborDist, size_t maxNeighbors, float timeHorizon, float radius, float maxSpeed, const Vector3 &velocity)
|
||||
{
|
||||
if (defaultAgent_ == NULL) {
|
||||
defaultAgent_ = new Agent3D();
|
||||
}
|
||||
|
||||
defaultAgent_->maxNeighbors_ = maxNeighbors;
|
||||
defaultAgent_->maxSpeed_ = maxSpeed;
|
||||
defaultAgent_->neighborDist_ = neighborDist;
|
||||
defaultAgent_->radius_ = radius;
|
||||
defaultAgent_->timeHorizon_ = timeHorizon;
|
||||
defaultAgent_->velocity_ = velocity;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentMaxNeighbors(size_t agentNo, size_t maxNeighbors)
|
||||
{
|
||||
agents_[agentNo]->maxNeighbors_ = maxNeighbors;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentMaxSpeed(size_t agentNo, float maxSpeed)
|
||||
{
|
||||
agents_[agentNo]->maxSpeed_ = maxSpeed;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentNeighborDist(size_t agentNo, float neighborDist)
|
||||
{
|
||||
agents_[agentNo]->neighborDist_ = neighborDist;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentPosition(size_t agentNo, const Vector3 &position)
|
||||
{
|
||||
agents_[agentNo]->position_ = position;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentPrefVelocity(size_t agentNo, const Vector3 &prefVelocity)
|
||||
{
|
||||
agents_[agentNo]->prefVelocity_ = prefVelocity;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentRadius(size_t agentNo, float radius)
|
||||
{
|
||||
agents_[agentNo]->radius_ = radius;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentTimeHorizon(size_t agentNo, float timeHorizon)
|
||||
{
|
||||
agents_[agentNo]->timeHorizon_ = timeHorizon;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setAgentVelocity(size_t agentNo, const Vector3 &velocity)
|
||||
{
|
||||
agents_[agentNo]->velocity_ = velocity;
|
||||
}
|
||||
|
||||
void RVOSimulator3D::setTimeStep(float timeStep)
|
||||
{
|
||||
timeStep_ = timeStep;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,170 @@
|
||||
/*
|
||||
* Vector3.cc
|
||||
* RVO2-3D Library
|
||||
*
|
||||
* SPDX-FileCopyrightText: 2008 University of North Carolina at Chapel Hill
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Please send all bug reports to <geom@cs.unc.edu>.
|
||||
*
|
||||
* The authors may be contacted via:
|
||||
*
|
||||
* Jur van den Berg, Stephen J. Guy, Jamie Snape, Ming C. Lin, Dinesh Manocha
|
||||
* Dept. of Computer Science
|
||||
* 201 S. Columbia St.
|
||||
* Frederick P. Brooks, Jr. Computer Science Bldg.
|
||||
* Chapel Hill, N.C. 27599-3175
|
||||
* United States of America
|
||||
*
|
||||
* <https://gamma.cs.unc.edu/RVO2/>
|
||||
*/
|
||||
|
||||
#include "Vector3.h"
|
||||
|
||||
#include <cmath>
|
||||
#include <ostream>
|
||||
|
||||
namespace RVO3D {
|
||||
Vector3::Vector3() : val_() {
|
||||
val_[0] = 0.0F;
|
||||
val_[1] = 0.0F;
|
||||
val_[2] = 0.0F;
|
||||
}
|
||||
|
||||
Vector3::Vector3(const Vector3 &vector) : val_() {
|
||||
val_[0] = vector[0];
|
||||
val_[1] = vector[1];
|
||||
val_[2] = vector[2];
|
||||
}
|
||||
|
||||
Vector3::Vector3(const float val[3]) : val_() {
|
||||
val_[0] = val[0];
|
||||
val_[1] = val[1];
|
||||
val_[2] = val[2];
|
||||
}
|
||||
|
||||
Vector3::Vector3(float x, float y, float z) : val_() {
|
||||
val_[0] = x;
|
||||
val_[1] = y;
|
||||
val_[2] = z;
|
||||
}
|
||||
|
||||
Vector3::~Vector3() {}
|
||||
|
||||
Vector3 &Vector3::operator=(const Vector3 &vector) {
|
||||
if (this != &vector) {
|
||||
val_[0] = vector[0];
|
||||
val_[1] = vector[1];
|
||||
val_[2] = vector[2];
|
||||
}
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
float Vector3::operator[](std::size_t i) const { return val_[i]; }
|
||||
|
||||
float &Vector3::operator[](std::size_t i) { return val_[i]; }
|
||||
|
||||
Vector3 Vector3::operator-() const {
|
||||
return Vector3(-val_[0], -val_[1], -val_[2]);
|
||||
}
|
||||
|
||||
float Vector3::operator*(const Vector3 &vector) const {
|
||||
return val_[0] * vector[0] + val_[1] * vector[1] + val_[2] * vector[2];
|
||||
}
|
||||
|
||||
Vector3 Vector3::operator*(float scalar) const {
|
||||
return Vector3(val_[0] * scalar, val_[1] * scalar, val_[2] * scalar);
|
||||
}
|
||||
|
||||
Vector3 Vector3::operator/(float scalar) const {
|
||||
const float invScalar = 1.0F / scalar;
|
||||
|
||||
return Vector3(val_[0] * invScalar, val_[1] * invScalar, val_[2] * invScalar);
|
||||
}
|
||||
|
||||
Vector3 Vector3::operator+(const Vector3 &vector) const {
|
||||
return Vector3(val_[0] + vector[0], val_[1] + vector[1], val_[2] + vector[2]);
|
||||
}
|
||||
|
||||
Vector3 Vector3::operator-(const Vector3 &vector) const {
|
||||
return Vector3(val_[0] - vector[0], val_[1] - vector[1], val_[2] - vector[2]);
|
||||
}
|
||||
|
||||
bool Vector3::operator==(const Vector3 &vector) const {
|
||||
return val_[0] == vector[0] && val_[1] == vector[1] && val_[2] == vector[2];
|
||||
}
|
||||
|
||||
bool Vector3::operator!=(const Vector3 &vector) const {
|
||||
return val_[0] != vector[0] || val_[1] != vector[1] || val_[2] != vector[2];
|
||||
}
|
||||
|
||||
Vector3 &Vector3::operator*=(float scalar) {
|
||||
val_[0] *= scalar;
|
||||
val_[1] *= scalar;
|
||||
val_[2] *= scalar;
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
Vector3 &Vector3::operator/=(float scalar) {
|
||||
const float invScalar = 1.0F / scalar;
|
||||
|
||||
val_[0] *= invScalar;
|
||||
val_[1] *= invScalar;
|
||||
val_[2] *= invScalar;
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
Vector3 &Vector3::operator+=(const Vector3 &vector) {
|
||||
val_[0] += vector[0];
|
||||
val_[1] += vector[1];
|
||||
val_[2] += vector[2];
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
Vector3 &Vector3::operator-=(const Vector3 &vector) {
|
||||
val_[0] -= vector[0];
|
||||
val_[1] -= vector[1];
|
||||
val_[2] -= vector[2];
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
Vector3 operator*(float scalar, const Vector3 &vector) {
|
||||
return Vector3(scalar * vector[0], scalar * vector[1], scalar * vector[2]);
|
||||
}
|
||||
|
||||
std::ostream &operator<<(std::ostream &stream, const Vector3 &vector) {
|
||||
stream << "(" << vector[0] << "," << vector[1] << "," << vector[2] << ")";
|
||||
|
||||
return stream;
|
||||
}
|
||||
|
||||
float abs(const Vector3 &vector) { return std::sqrt(vector * vector); }
|
||||
|
||||
float absSq(const Vector3 &vector) { return vector * vector; }
|
||||
|
||||
Vector3 cross(const Vector3 &vector1, const Vector3 &vector2) {
|
||||
return Vector3(vector1[1] * vector2[2] - vector1[2] * vector2[1],
|
||||
vector1[2] * vector2[0] - vector1[0] * vector2[2],
|
||||
vector1[0] * vector2[1] - vector1[1] * vector2[0]);
|
||||
}
|
||||
|
||||
Vector3 normalize(const Vector3 &vector) { return vector / abs(vector); }
|
||||
|
||||
} /* namespace RVO3D */
|
||||
Loading…
Reference in New Issue