libcruft-util/geom/aabb.cpp

/*
 * 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.
 *
 * Copyright 2015 Danny Robson <danny@nerdcruft.net>
 */


#include "./aabb.hpp"

#include "./iostream.hpp"
#include "../coord/iostream.hpp"
#include "../debug.hpp"

using util::geom::aabb;

//-----------------------------------------------------------------------------
template <size_t S, typename T>
aabb<S,T>::aabb (point<S,T> _p0, point<S,T> _p1):
    p0 (_p0),
    p1 (_p1)
{
    debug::sanity (*this);
}


//-----------------------------------------------------------------------------
template <size_t S, typename T>
T
aabb<S,T>::diameter (void) const
{
    return magnitude ().diameter ();
}


//-----------------------------------------------------------------------------
template <size_t S, typename T>
util::extent<S,T>
aabb<S,T>::magnitude (void) const
{
    extent<S,T> out;
    for (size_t i = 0; i < S; ++i)
        out[i] = p1[i] - p0[i];
    return out;
}


//-----------------------------------------------------------------------------
template <size_t S, typename T>
bool
aabb<S,T>::overlaps (point<S,T> p) const
{
    for (size_t i = 0; i < S; ++i)
        if (p0[i] > p[i] || p1[i] < p[i])
            return false;

    return true;
}


//-----------------------------------------------------------------------------
template <size_t S, typename T>
util::point<S,T>
aabb<S,T>::closest (point<S,T> q) const
{
    point<S,T> res;

    for (size_t i = 0; i < S; ++i)
        res[i] = q[i] < p0[i] ? p0[i] :
                 q[i] > p1[i] ? p1[i] :
                 q[i];

    return res;
}


///////////////////////////////////////////////////////////////////////////////
template <size_t S, typename T>
aabb<S,T>
aabb<S,T>::expand (vector<S,T> mag) const noexcept
{
    CHECK (all (mag >= T{0}));
    CHECK (all (mag <  p1 - p0));

    return {
        p0 - mag / T{2},
        p1 + mag / T{2}
    };
}


//-----------------------------------------------------------------------------
template <size_t S, typename T>
aabb<S,T>
aabb<S,T>::expand (T t) const noexcept
{
    CHECK_GE (t, T{0});
    CHECK (all (t < p1 - p0));

    return {
        p0 - t / T{2},
        p1 + t / T{2}
    };
}


///////////////////////////////////////////////////////////////////////////////
template <size_t S, typename T>
aabb<S,T>
aabb<S,T>::contract (util::vector<S,T> mag) const noexcept
{
    CHECK (all (mag > T{0}));
    CHECK (all (mag <= p1 - p0));

    return {
        p0 + mag / T{2},
        p1 - mag / T{2}
    };
}


//-----------------------------------------------------------------------------
template <size_t S, typename T>
aabb<S,T>
aabb<S,T>::contract (T mag) const noexcept
{
    CHECK_GE (mag, T{0});
    CHECK (all (mag <= p1 - p0));

    return {
        p0 + mag / T{2},
        p1 - mag / T{2}
    };
}


///////////////////////////////////////////////////////////////////////////////
template <size_t S, typename T>
void
aabb<S,T>::cover (point<S,T> p)
{
    p0 = min (p, p0);
    p1 = max (p, p1);
}


///////////////////////////////////////////////////////////////////////////////
template <size_t S, typename T>
aabb<S,T>
aabb<S,T>::operator+ (vector<S,T> v) const
{
    return { p0 + v, p1 + v };
}


//-----------------------------------------------------------------------------
template <size_t S, typename T>
aabb<S,T>
aabb<S,T>::operator- (vector<S,T> v) const
{
    return { p0 - v, p1 - v };
}


///////////////////////////////////////////////////////////////////////////////
template <size_t S, typename T>
bool
aabb<S,T>::operator== (const aabb<S,T> rhs) const
{
    return rhs.p0 == p0 && rhs.p1 == p1;
}



//-----------------------------------------------------------------------------
namespace util::debug {
    template <size_t S, typename T>
    struct validator<aabb<S,T>> {
        static bool is_valid (const aabb<S,T> &b)
        {
            for (size_t i = 0; i < S; ++i)
                if (b.p1[i] < b.p0[i])
                    return false;

            return true;
        }
    };
}


//-----------------------------------------------------------------------------
template <size_t S, typename T>
std::ostream&
util::geom::operator<< (std::ostream &os, util::geom::aabb<S,T> b)
{
    os << "aabb(" << b.p0 << ", " << b.p1 << ")";
    return os;
}


//-----------------------------------------------------------------------------
#define INSTANTIATE_S_T(S,T)                             \
namespace util::geom { template struct aabb<S,T>; }      \
template bool util::debug::is_valid (const aabb<S,T>&);  \
template std::ostream& util::geom::operator<< (std::ostream&, aabb<S,T>);

#define INSTANTIATE(T)  \
INSTANTIATE_S_T(2,T)    \
INSTANTIATE_S_T(3,T)

INSTANTIATE( int32_t)
INSTANTIATE( int64_t)
INSTANTIATE(uint32_t)
INSTANTIATE(uint64_t)
INSTANTIATE(float)
INSTANTIATE(double)