Extent.h

Go to the documentation of this file.

// -*- lsst-c++ -*-

/* 
 * LSST Data Management System
 * Copyright 2008, 2009, 2010 LSST Corporation.
 * 
 * This product includes software developed by the
 * LSST Project (http://www.lsst.org/).
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the LSST License Statement and 
 * the GNU General Public License along with this program.  If not, 
 * see <http://www.lsstcorp.org/LegalNotices/>.
 */
 
#ifndef LSST_AFW_GEOM_EXTENT_H
#define LSST_AFW_GEOM_EXTENT_H

#include <boost/type_traits/is_integral.hpp>
#include <boost/static_assert.hpp>
#include <boost/type_traits/is_same.hpp>

#include "lsst/pex/exceptions.h"
#include "lsst/afw/geom/CoordinateExpr.h"

namespace lsst { namespace afw { namespace geom {

// These are present to avoid a static assertion for instantiating computeNorm() on integer types.
namespace detail {

template <int N>
double computeExtentNorm(Extent<double,N> const & s) {
    return s.asEigen().norm();
}

template <int N>
int computeExtentNorm(Extent<int,N> const & s) {
    throw LSST_EXCEPT(
        pex::exceptions::LogicError,
        "Cannot compute norm of integer extent"
    );
#if 1                                   // make compilers happy in non-void function
    return -1;
#endif
}

} // namespace detail


template<typename T, int N>
class ExtentBase : public CoordinateBase<Extent<T,N>,T,N> {
    typedef CoordinateBase<Extent<T,N>,T,N> Super;
public:

    T computeSquaredNorm() const { return this->asEigen().squaredNorm(); }

    T computeNorm() const { return detail::computeExtentNorm(static_cast<Extent<T,N> const &>(*this)); }

    bool operator==(Extent<T,N> const & other) const { return all(this->eq(other)); }

    bool operator!=(Extent<T,N> const & other) const { return any(this->ne(other)); }

    CoordinateExpr<N> eq(Extent<T,N> const & other) const;
    CoordinateExpr<N> ne(Extent<T,N> const & other) const;
    CoordinateExpr<N> lt(Extent<T,N> const & other) const;
    CoordinateExpr<N> le(Extent<T,N> const & other) const;
    CoordinateExpr<N> gt(Extent<T,N> const & other) const;
    CoordinateExpr<N> ge(Extent<T,N> const & other) const;
    CoordinateExpr<N> eq(T scalar) const { return this->eq(Extent<T,N>(scalar)); }
    CoordinateExpr<N> ne(T scalar) const { return this->ne(Extent<T,N>(scalar)); }
    CoordinateExpr<N> lt(T scalar) const { return this->lt(Extent<T,N>(scalar)); }
    CoordinateExpr<N> le(T scalar) const { return this->le(Extent<T,N>(scalar)); }
    CoordinateExpr<N> gt(T scalar) const { return this->gt(Extent<T,N>(scalar)); }
    CoordinateExpr<N> ge(T scalar) const { return this->ge(Extent<T,N>(scalar)); }

    Point<T,N> operator+(Point<T,N> const & other) const;
    Extent<T,N> operator+(Extent<T,N> const & other) const {
        return Extent<T,N>(this->_vector + other._vector);
    }
    Extent<T,N> operator-(Extent<T,N> const & other) const {
        return Extent<T,N>(this->_vector - other._vector);
    }
    Extent<T,N> & operator+=(Extent<T,N> const & other) {
        this->_vector += other._vector;
        return static_cast<Extent<T,N>&>(*this);
    }
    Extent<T,N> & operator-=(Extent<T,N> const & other) {
        this->_vector -= other._vector;
        return static_cast<Extent<T,N>&>(*this);
    }
    Extent<T,N> operator+() const { return static_cast<Extent<T,N> const &>(*this); }
    Extent<T,N> operator-() const { return Extent<T,N>(-this->_vector); }

    Extent<T,N> operator*(T scalar) const { return Extent<T,N>(this->_vector * scalar); }
    Extent<T,N> & operator*=(T scalar) { this->_vector *= scalar; return static_cast<Extent<T,N>&>(*this); }
    Extent<T,N> operator/(T scalar) const { return Extent<T,N>(this->_vector / scalar); }
    Extent<T,N> & operator/=(T scalar) { this->_vector /= scalar; return static_cast<Extent<T,N>&>(*this); }

    Point<T,N> asPoint() const;

    std::string toString() const {
        std::stringstream out;
        out << "Extent(";
        for (size_t i = 0; i < N; ++i) {
            if (i != 0) {
                out << ",";
            }
            out << (*this)[i];
        }
        out << ")";
        return out.str();
    }

protected:

    explicit ExtentBase(T val=static_cast<T>(0)) : Super(val) {}

    template <typename Vector>
    explicit ExtentBase(Eigen::MatrixBase<Vector> const & vector) : Super(vector) {}

};

template<typename T, int N>
class Extent : public ExtentBase<T,N> {
    typedef ExtentBase<T,N> Super;
public:
    typedef typename Super::EigenVector EigenVector;

    explicit Extent(T val=static_cast<T>(0)) : Super(val) {}

    explicit Extent(EigenVector const & vector) : Super(vector) {}

    explicit Extent(Point<T,N> const & other);

    template<typename U>
    explicit Extent(Extent<U,N> const & other);
    template<typename U>
    explicit Extent(Point<U,N> const & other);

    T computeSquaredNorm() const { return this->asEigen().squaredNorm(); }

    T computeNorm() const { return this->asEigen().norm(); }
    
    void swap(Extent & other) { this->_swap(other); }
};

template<typename T>
class Extent<T,2> : public ExtentBase<T,2> {
    typedef ExtentBase<T,2> Super;
public:
    typedef typename Super::EigenVector EigenVector;

    explicit Extent(T val=static_cast<T>(0)) : Super(val) {}

    explicit Extent(EigenVector const & vector) : Super(vector) {}

    explicit Extent(Point<T,2> const & other);

    template<typename U>
    explicit Extent(Extent<U,2> const & other);
    template<typename U>
    explicit Extent(Point<U,2> const & other);

    explicit Extent(T x, T y) : Super(EigenVector(x, y)) {}

    explicit Extent(T const xy[2]) : Super(EigenVector(xy[0], xy[1])) {}

    explicit Extent(std::pair<T,T> const & xy) : Super(EigenVector(xy.first, xy.second)) {}

    explicit Extent(boost::tuple<T,T> const & xy) : 
        Super(EigenVector(xy.template get<0>(), xy.template get<1>())) {}

#ifdef SWIG
    T getX() const;
    T getY() const;
    void setX(T x);
    void setY(T y);
#endif

    void swap(Extent & other) { this->_swap(other); }
};

template<typename T>
class Extent<T,3> : public ExtentBase<T,3> {
    typedef ExtentBase<T,3> Super;
public:
    typedef typename Super::EigenVector EigenVector;

    explicit Extent(T val=static_cast<T>(0)) : Super(val) {}

    explicit Extent(EigenVector const & vector) : Super(vector) {}

    explicit Extent(Point<T,3> const & other);

    template<typename U>
    explicit Extent(Extent<U,3> const & other);
    template<typename U>
    explicit Extent(Point<U,3> const & other);

    explicit Extent(T x, T y, T z) : Super(EigenVector(x, y, z)) {}

    explicit Extent(T const xyz[3]) : Super(EigenVector(xyz[0], xyz[1], xyz[2])) {}

    explicit Extent(boost::tuple<T,T,T> const & xyz) : 
        Super(EigenVector(xyz.template get<0>(), xyz.template get<1>(), xyz.template get<2>())) {}

#ifdef SWIG
    T getX() const;
    T getY() const;
    T getZ() const;
    void setX(T x);
    void setY(T y);
    void setZ(T z);
#endif

    void swap(Extent & other) { this->_swap(other); }
};

// Constructor for any 2D type from 2I type
template<typename T>
template<typename U>
Extent<T, 2>::Extent(Extent<U, 2> const & other) 
{
    BOOST_STATIC_ASSERT( (!boost::is_same<T,U>::value && boost::is_integral<U>::value) );
    this->setX(static_cast<T>(other.getX()));
    this->setY(static_cast<T>(other.getY()));
};

template<typename T>
template<typename U>
Extent<T, 2>::Extent(Point<U, 2> const & other) 
{
    BOOST_STATIC_ASSERT( (!boost::is_same<T,U>::value && boost::is_integral<U>::value) );
    this->setX(static_cast<T>(other.getX()));
    this->setY(static_cast<T>(other.getY()));
};

// Constructor for any 3D type from 3I type
template<typename T>
template<typename U>
Extent<T, 3>::Extent(Extent<U, 3> const & other) 
{
    BOOST_STATIC_ASSERT( (!boost::is_same<T,U>::value && boost::is_integral<U>::value) );
    this->setX(static_cast<T>(other.getX()));
    this->setY(static_cast<T>(other.getY()));
    this->setZ(static_cast<T>(other.getZ()));
};

// Constructor for any 3D type from 3I type
template<typename T>
template<typename U>
Extent<T, 3>::Extent(Point<U, 3> const & other) 
{
    BOOST_STATIC_ASSERT( (!boost::is_same<T,U>::value && boost::is_integral<U>::value) );
    this->setX(static_cast<T>(other.getX()));
    this->setY(static_cast<T>(other.getY()));
    this->setZ(static_cast<T>(other.getZ()));
};

typedef Extent<int,2> ExtentI;
typedef Extent<int,2> Extent2I;
typedef Extent<int,3> Extent3I;
typedef Extent<double,2> ExtentD;
typedef Extent<double,2> Extent2D;
typedef Extent<double,3> Extent3D;

template <int N>
Extent<int,N> truncate(Extent<double,N> const & input);

template <int N>
Extent<int,N> floor(Extent<double,N> const & input);

template <int N>
Extent<int,N> ceil(Extent<double,N> const & input);

#ifndef SWIG

// Some operators below need to take ExtentBase arguments rather than Extent to
// avoid ambiguous overloads (since some competing operators are defined as member
// functions on ExtentBase).

template <typename T, int N>
Extent<T,N> operator*(T scalar, ExtentBase<T,N> const & rhs) {
    return rhs * scalar;
}

template <int N>
Extent<double,N> operator*(ExtentBase<int,N> const & lhs, double rhs) {
    return Extent<double,N>(static_cast<Extent<int,N> const &>(lhs)) * rhs;
}

template <int N>
void operator*=(ExtentBase<int,N> & lhs, double rhs) {
    // use "N < 0" so assertion is dependent on template instantiation, instead of triggering all the time
    static_assert(N < 0, "In-place multiplication of Extent<int,N> by double would truncate.");
}

template <int N>
Extent<double,N> operator/(ExtentBase<int,N> const & lhs, double rhs) {
    return Extent<double,N>(static_cast<Extent<int,N> const &>(lhs)) / rhs;
}

template <int N>
void operator/=(ExtentBase<int,N> & lhs, double rhs) {
    // use "N < 0" so assertion is dependent on template instantiation, instead of triggering all the time
    static_assert(N < 0, "In-place division of Extent<int,N> by double would truncate.");
}

template <int N>
Extent<double,N> operator*(double lhs, ExtentBase<int,N> const & rhs) {
    return lhs * Extent<double,N>(static_cast<Extent<int,N> const &>(rhs));
}

template <int N>
Extent<double,N> operator+(Extent<double,N> const & lhs, Extent<int,N> const & rhs) {
    return lhs + Extent<double,N>(rhs);
}

template <int N>
Extent<double,N> & operator+=(Extent<double,N> & lhs, Extent<int,N> const & rhs) {
    return lhs += Extent<double,N>(rhs);
}

template <int N>
Extent<double,N> operator-(Extent<double,N> const & lhs, Extent<int,N> const & rhs) {
    return lhs - Extent<double,N>(rhs);
}

template <int N>
Extent<double,N> & operator-=(Extent<double,N> & lhs, Extent<int,N> const & rhs) {
    return lhs -= Extent<double,N>(rhs);
}

template <int N>
Extent<double,N> operator+(Extent<int,N> const & lhs, Extent<double,N> const & rhs) {
    return Extent<double,N>(lhs) + rhs;
}

template <int N>
Extent<double,N> operator-(Extent<int,N> const & lhs, Extent<double,N> const & rhs) {
    return Extent<double,N>(lhs) - rhs;
}

#endif // !SWIG

}}}

#endif