cocos-engine/native/cocos/scene/AABB.cpp

/****************************************************************************
 Copyright (c) 2021-2022 Xiamen Yaji Software Co., Ltd.

 http://www.cocos.com

 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated engine source code (the "Software"), a limited,
 worldwide, royalty-free, non-assignable, revocable and non-exclusive license
 to use Cocos Creator solely to develop games on your target platforms. You shall
 not use Cocos Creator software for developing other software or tools that's
 used for developing games. You are not granted to publish, distribute,
 sublicense, and/or sell copies of Cocos Creator.

 The software or tools in this License Agreement are licensed, not sold.
 Xiamen Yaji Software Co., Ltd. reserves all rights not expressly granted to you.

 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 THE SOFTWARE.
****************************************************************************/

#include "AABB.h"

#include "base/TypeDef.h"

namespace cc {
namespace scene {
bool AABB::aabbAabb(AABB *aabb) const {
    Vec3 aMin;
    Vec3 aMax;
    Vec3 bMin;
    Vec3 bMax;
    Vec3::subtract(getCenter(), getHalfExtents(), &aMin);
    Vec3::add(getCenter(), getHalfExtents(), &aMax);
    Vec3::subtract(aabb->getCenter(), aabb->getHalfExtents(), &bMin);
    Vec3::add(aabb->getCenter(), aabb->getHalfExtents(), &bMax);
    return (aMin.x <= bMax.x && aMax.x >= bMin.x) &&
           (aMin.y <= bMax.y && aMax.y >= bMin.y) &&
           (aMin.z <= bMax.z && aMax.z >= bMin.z);
}

int AABB::aabbPlane(const Plane &plane) const {
    auto r = getHalfExtents().x * std::abs(plane.n.x) +
             getHalfExtents().y * std::abs(plane.n.y) +
             getHalfExtents().z * std::abs(plane.n.z);
    auto dot = Vec3::dot(plane.n, getCenter());
    if (dot + r < plane.d) {
        return -1;
    }
    if (dot - r > plane.d) {
        return 0;
    }
    return 1;
}

void AABB::initWithData(uint8_t *data) {
    _aabbLayout = reinterpret_cast<AABBLayout *>(data);
}

bool AABB::aabbFrustum(const Frustum &frustum) const {
    const auto &planes = frustum.planes;
    const auto *self   = this;
    return std::all_of(planes.begin(),
                       planes.end(),
                       // frustum plane normal points to the inside
                       [self](const Plane &plane) { return self->aabbPlane(plane) != -1; });
}

void AABB::getBoundary(cc::Vec3 *minPos, cc::Vec3 *maxPos) const {
    Vec3::subtract(getCenter(), getHalfExtents(), minPos);
    Vec3::add(getCenter(), getHalfExtents(), maxPos);
}

void AABB::merge(const AABB &aabb) {
    const cc::Vec3 minA = getCenter() - getHalfExtents();
    const cc::Vec3 minB = aabb.getCenter() - aabb.getHalfExtents();
    const cc::Vec3 maxA = getCenter() + getHalfExtents();
    const cc::Vec3 maxB = aabb.getCenter() + aabb.getHalfExtents();
    cc::Vec3       maxP;
    cc::Vec3       minP;
    cc::Vec3::max(maxA, maxB, &maxP);
    cc::Vec3::min(minA, minB, &minP);

    const cc::Vec3 addP = maxP + minP;
    const cc::Vec3 subP = maxP - minP;
    setCenter(addP * 0.5F);
    setHalfExtents(subP * 0.5F);
}

void AABB::merge(const cc::Vec3 &point) {
    cc::Vec3 minPos = getCenter() - getHalfExtents();
    cc::Vec3 maxPos = getCenter() + getHalfExtents();
    if (point.x < minPos.x) {
        minPos.x = point.x;
    }
    if (point.y < minPos.y) {
        minPos.y = point.y;
    }
    if (point.z < minPos.z) {
        minPos.z = point.z;
    }
    if (point.x > maxPos.x) {
        maxPos.x = point.x;
    }
    if (point.y > maxPos.y) {
        maxPos.y = point.y;
    }
    if (point.z > maxPos.z) {
        maxPos.z = point.z;
    }

    const Vec3 center = (minPos + maxPos) * 0.5F;
    setCenter(center);
    setHalfExtents(maxPos.x - center.x, maxPos.y - center.y, maxPos.z - center.z);
}

void AABB::merge(const Frustum &frustum) {
    const std::array<Vec3, 8> &vertices = frustum.vertices;
    for (uint i = 0; i < vertices.max_size(); ++i) {
        merge(vertices[i]);
    }
}

void AABB::set(const cc::Vec3 &centerVal, const cc::Vec3 &halfExtentVal) {
    setCenter(centerVal);
    setHalfExtents(halfExtentVal);
}

void AABB::transform(const Mat4 &m, AABB *out) const {
    AABBLayout *layout = out->_aabbLayout;
    layout->center.transformMat4(getCenter(), m);
    transformExtentM4(&layout->halfExtents, getHalfExtents(), m);
}

bool AABB::contain(const cc::Vec3 &point) const {
    cc::Vec3 minPos = getCenter() - getHalfExtents();
    cc::Vec3 maxPos = getCenter() + getHalfExtents();

    return !(point.x > maxPos.x || point.x < minPos.x ||
             point.y > maxPos.y || point.y < minPos.y ||
             point.z > maxPos.z || point.z < minPos.z);
}

void AABB::transformExtentM4(Vec3 *out, const Vec3 &extent, const Mat4 &m4) {
    Mat3 m3Tmp;
    m3Tmp.m[0] = std::abs(m4.m[0]);
    m3Tmp.m[1] = std::abs(m4.m[1]);
    m3Tmp.m[2] = std::abs(m4.m[2]);
    m3Tmp.m[3] = std::abs(m4.m[4]);
    m3Tmp.m[4] = std::abs(m4.m[5]);
    m3Tmp.m[5] = std::abs(m4.m[6]);
    m3Tmp.m[6] = std::abs(m4.m[8]);
    m3Tmp.m[7] = std::abs(m4.m[9]);
    m3Tmp.m[8] = std::abs(m4.m[10]);
    out->transformMat3(extent, m3Tmp);
}

AABB::AABB() {
    _aabbLayout = &_embedLayout;
}

void AABB::fromPoints(const Vec3 &minPos, const Vec3 &maxPos, AABB *dst) {
    Vec3 v3Tmp;
    Vec3::add(maxPos, minPos, &v3Tmp);
    dst->setCenter(v3Tmp * 0.5);
    Vec3::subtract(maxPos, minPos, &v3Tmp);
    dst->setHalfExtents(v3Tmp * 0.5);
}

} // namespace scene
} // namespace cc