You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
521 lines
15 KiB
521 lines
15 KiB
/*
|
|
This file is part of the TDE games library
|
|
Copyright (C) 2001-02 Nicolas Hadacek (hadacek@kde.org)
|
|
|
|
This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Library General Public
|
|
License version 2 as published by the Free Software Foundation.
|
|
|
|
This library 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
|
|
Library General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Library General Public License
|
|
along with this library; see the file COPYING.LIB. If not, write to
|
|
the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
|
Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#ifndef __KGRID2D_H_
|
|
#define __KGRID2D_H_
|
|
|
|
#include <math.h>
|
|
|
|
#include <tqpair.h>
|
|
#include <tqvaluelist.h>
|
|
#include <tqvaluevector.h>
|
|
|
|
#include <kglobal.h>
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
namespace KGrid2D
|
|
{
|
|
/**
|
|
* This type represents coordinates on a bidimensionnal grid.
|
|
* @since 3.2
|
|
*/
|
|
typedef TQPair<int, int> Coord;
|
|
|
|
/**
|
|
* This type represents a list of @ref Coord.
|
|
* @since 3.2
|
|
*/
|
|
typedef TQValueList<Coord> CoordList;
|
|
}
|
|
|
|
inline KGrid2D::Coord
|
|
operator +(const KGrid2D::Coord &c1, const KGrid2D::Coord &c2) {
|
|
return KGrid2D::Coord(c1.first + c2.first, c1.second + c2.second);
|
|
}
|
|
|
|
inline KGrid2D::Coord
|
|
operator -(const KGrid2D::Coord &c1, const KGrid2D::Coord &c2) {
|
|
return KGrid2D::Coord(c1.first - c2.first, c1.second - c2.second);
|
|
}
|
|
|
|
/**
|
|
* @return the maximum of both coordinates.
|
|
* @since 3.2
|
|
*/
|
|
inline KGrid2D::Coord
|
|
maximum(const KGrid2D::Coord &c1, const KGrid2D::Coord &c2) {
|
|
return KGrid2D::Coord(kMax(c1.first, c2.first), kMax(c1.second, c2.second));
|
|
}
|
|
/**
|
|
* @return the minimum of both coordinates.
|
|
* @since 3.2
|
|
*/
|
|
inline KGrid2D::Coord
|
|
minimum(const KGrid2D::Coord &c1, const KGrid2D::Coord &c2) {
|
|
return KGrid2D::Coord(kMin(c1.first, c2.first), kMin(c1.second, c2.second));
|
|
}
|
|
|
|
inline TQTextStream &operator <<(TQTextStream &s, const KGrid2D::Coord &c) {
|
|
return s << '(' << c.second << ", " << c.first << ')';
|
|
}
|
|
|
|
inline TQTextStream &operator <<(TQTextStream &s, const KGrid2D::CoordList &list)
|
|
{
|
|
for(KGrid2D::CoordList::const_iterator i=list.begin(); i!=list.end(); ++i)
|
|
s << *i;
|
|
return s;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
namespace KGrid2D
|
|
{
|
|
/**
|
|
* This template class represents a generic bidimensionnal grid. Each node
|
|
* contains an element of the template type.
|
|
*
|
|
* @since 3.2
|
|
*/
|
|
template <class Type>
|
|
class Generic
|
|
{
|
|
public:
|
|
/**
|
|
* Constructor.
|
|
*/
|
|
Generic(uint width = 0, uint height = 0) {
|
|
resize(width, height);
|
|
}
|
|
|
|
virtual ~Generic() {}
|
|
|
|
/**
|
|
* Resize the grid.
|
|
*/
|
|
void resize(uint width, uint height) {
|
|
_width = width;
|
|
_height = height;
|
|
_vector.resize(width*height);
|
|
}
|
|
|
|
/**
|
|
* Fill the nodes with the given value.
|
|
*/
|
|
void fill(const Type &value) {
|
|
for (uint i=0; i<_vector.count(); i++) _vector[i] = value;
|
|
}
|
|
|
|
/**
|
|
* @return the width.
|
|
*/
|
|
uint width() const { return _width; }
|
|
/**
|
|
* @return the height.
|
|
*/
|
|
uint height() const { return _height; }
|
|
/**
|
|
* @return the number of nodes (ie width*height).
|
|
*/
|
|
uint size() const { return _width*_height; }
|
|
|
|
/**
|
|
* @return the linear index for the given coordinate.
|
|
*/
|
|
uint index(const Coord &c) const {
|
|
return c.first + c.second*_width;
|
|
}
|
|
|
|
/**
|
|
* @return the coordinate corresponding to the linear index.
|
|
*/
|
|
Coord coord(uint index) const {
|
|
return Coord(index % _width, index / _width);
|
|
}
|
|
|
|
/**
|
|
* @return the value at the given coordinate.
|
|
*/
|
|
const Type &at(const Coord &c) const { return _vector[index(c)]; }
|
|
/**
|
|
* @return the value at the given coordinate.
|
|
*/
|
|
Type &at(const Coord &c) { return _vector[index(c)]; }
|
|
/**
|
|
* @return the value at the given coordinate.
|
|
*/
|
|
const Type &operator [](const Coord &c) const { return _vector[index(c)]; }
|
|
/**
|
|
* @return the value at the given coordinate.
|
|
*/
|
|
Type &operator [](const Coord &c) { return _vector[index(c)]; }
|
|
|
|
/**
|
|
* @return the value at the given linear index.
|
|
*/
|
|
const Type &at(uint index) const { return _vector[index]; }
|
|
/**
|
|
* @return the value at the given linear index.
|
|
*/
|
|
Type &at(uint index) { return _vector[index]; }
|
|
/**
|
|
* @return the value at the given linear index.
|
|
*/
|
|
const Type &operator [](uint index) const { return _vector[index]; }
|
|
/**
|
|
* @return the value at the given linear index.
|
|
*/
|
|
Type &operator [](uint index) { return _vector[index]; }
|
|
|
|
/**
|
|
* @return if the given coordinate is inside the grid.
|
|
*/
|
|
bool inside(const Coord &c) const {
|
|
return ( c.first>=0 && c.first<(int)_width
|
|
&& c.second>=0 && c.second<(int)_height );
|
|
}
|
|
|
|
/**
|
|
* Bound the given coordinate with the grid dimensions.
|
|
*/
|
|
void bound(Coord &c) const {
|
|
c.first = kMax(kMin(c.first, (int)_width-1), 0);
|
|
c.second = kMax(kMin(c.second, (int)_height-1), 0);
|
|
}
|
|
|
|
protected:
|
|
uint _width, _height;
|
|
TQValueVector<Type> _vector;
|
|
};
|
|
}
|
|
|
|
template <class Type>
|
|
TQDataStream &operator <<(TQDataStream &s, const KGrid2D::Generic<Type> &m) {
|
|
s << (TQ_UINT32)m.width() << (TQ_UINT32)m.height();
|
|
for (uint i=0; i<m.size(); i++) s << m[i];
|
|
return s;
|
|
}
|
|
|
|
template <class Type>
|
|
TQDataStream &operator >>(TQDataStream &s, KGrid2D::Generic<Type> &m) {
|
|
TQ_UINT32 w, h;
|
|
s >> w >> h;
|
|
m.resize(w, h);
|
|
for (uint i=0; i<m.size(); i++) s >> m[i];
|
|
return s;
|
|
}
|
|
|
|
|
|
namespace KGrid2D
|
|
{
|
|
|
|
//-----------------------------------------------------------------------------
|
|
/**
|
|
* This class contains static methods to manipulate coordinates for a
|
|
* square bidimensionnal grid.
|
|
*
|
|
* @since 3.2
|
|
*/
|
|
class SquareBase
|
|
{
|
|
public:
|
|
/**
|
|
* Identify the eight neighbours.
|
|
*/
|
|
enum Neighbour { Left=0, Right, Up, Down, LeftUp, LeftDown,
|
|
RightUp, RightDown, Nb_Neighbour };
|
|
|
|
/**
|
|
* @return the trigonometric angle in radians for the given neighbour.
|
|
*/
|
|
static double angle(Neighbour n) {
|
|
switch (n) {
|
|
case Left: return M_PI;
|
|
case Right: return 0;
|
|
case Up: return M_PI_2;
|
|
case Down: return -M_PI_2;
|
|
case LeftUp: return 3.0*M_PI_4;
|
|
case LeftDown: return -3.0*M_PI_4;
|
|
case RightUp: return M_PI_4;
|
|
case RightDown: return -M_PI_4;
|
|
case Nb_Neighbour: Q_ASSERT(false);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* @return the opposed neighbour.
|
|
*/
|
|
static Neighbour opposed(Neighbour n) {
|
|
switch (n) {
|
|
case Left: return Right;
|
|
case Right: return Left;
|
|
case Up: return Down;
|
|
case Down: return Up;
|
|
case LeftUp: return RightDown;
|
|
case LeftDown: return RightUp;
|
|
case RightUp: return LeftDown;
|
|
case RightDown: return LeftUp;
|
|
case Nb_Neighbour: Q_ASSERT(false);
|
|
}
|
|
return Nb_Neighbour;
|
|
}
|
|
|
|
/**
|
|
* @return true if the neighbour is a direct one (ie is one of the four
|
|
* nearest).
|
|
*/
|
|
static bool isDirect(Neighbour n) { return n<LeftUp; }
|
|
|
|
/**
|
|
* @return the neighbour for the given coordinate.
|
|
*/
|
|
static Coord neighbour(const Coord &c, Neighbour n) {
|
|
switch (n) {
|
|
case Left: return c + Coord(-1, 0);
|
|
case Right: return c + Coord( 1, 0);
|
|
case Up: return c + Coord( 0, -1);
|
|
case Down: return c + Coord( 0, 1);
|
|
case LeftUp: return c + Coord(-1, -1);
|
|
case LeftDown: return c + Coord(-1, 1);
|
|
case RightUp: return c + Coord( 1, -1);
|
|
case RightDown: return c + Coord( 1, 1);
|
|
case Nb_Neighbour: Q_ASSERT(false);
|
|
}
|
|
return c;
|
|
}
|
|
};
|
|
|
|
/**
|
|
* This template is a @ref Generic implementation for a square bidimensionnal
|
|
* grid (@ref SquareBase).
|
|
*
|
|
* @since 3.2
|
|
*/
|
|
template <class T>
|
|
class Square : public Generic<T>, public SquareBase
|
|
{
|
|
public:
|
|
/**
|
|
* Constructor.
|
|
*/
|
|
Square(uint width = 0, uint height = 0)
|
|
: Generic<T>(width, height) {}
|
|
|
|
/**
|
|
* @return the neighbours of coordinate @param c
|
|
* to the given set of coordinates
|
|
* @param c the coordinate to use as the reference point
|
|
* @param insideOnly only add coordinates that are inside the grid.
|
|
* @param directOnly only add the four nearest neighbours.
|
|
*/
|
|
CoordList neighbours(const Coord &c, bool insideOnly = true,
|
|
bool directOnly = false) const {
|
|
CoordList neighbours;
|
|
for (uint i=0; i<(directOnly ? LeftUp : Nb_Neighbour); i++) {
|
|
Coord n = neighbour(c, (Neighbour)i);
|
|
if ( insideOnly && !Generic<T>::inside(n) ) continue;
|
|
neighbours.append(n);
|
|
}
|
|
return neighbours;
|
|
}
|
|
|
|
/**
|
|
* @return the "projection" of the given coordinate on the grid edges.
|
|
*
|
|
* @param c the coordinate to use as the reference point
|
|
* @param n the direction of projection.
|
|
*/
|
|
Coord toEdge(const Coord &c, Neighbour n) const {
|
|
switch (n) {
|
|
case Left: return Coord(0, c.second);
|
|
case Right: return Coord(Generic<T>::width()-1, c.second);
|
|
case Up: return Coord(c.first, 0);
|
|
case Down: return Coord(c.first, Generic<T>::height()-1);
|
|
case LeftUp: return Coord(0, 0);
|
|
case LeftDown: return Coord(0, Generic<T>::height()-1);
|
|
case RightUp: return Coord(Generic<T>::width()-1, 0);
|
|
case RightDown: return Coord(Generic<T>::width()-1, Generic<T>::height()-1);
|
|
case Nb_Neighbour: Q_ASSERT(false);
|
|
}
|
|
return c;
|
|
}
|
|
};
|
|
|
|
//-----------------------------------------------------------------------------
|
|
/**
|
|
* This class contains static methods to manipulate coordinates on an
|
|
* hexagonal grid where hexagons form horizontal lines:
|
|
* <pre>
|
|
* (0,0) (0,1) (0,2)
|
|
* (1,0) (1,1) (1,2)
|
|
* (2,0) (2,1) (2,2)
|
|
* </pre>
|
|
*
|
|
* @since 3.2
|
|
*/
|
|
class HexagonalBase
|
|
{
|
|
public:
|
|
/**
|
|
* Identify the six neighbours.
|
|
*/
|
|
enum Neighbour { Left = 0, Right, LeftUp, LeftDown,
|
|
RightUp, RightDown, Nb_Neighbour };
|
|
|
|
/**
|
|
* @return the trigonometric angle in radians for the given neighbour.
|
|
*/
|
|
static double angle(Neighbour n) {
|
|
switch (n) {
|
|
case Left: return M_PI;
|
|
case Right: return 0;
|
|
case LeftUp: return 2.0*M_PI/3;
|
|
case LeftDown: return -2.0*M_PI/3;
|
|
case RightUp: return M_PI/3;
|
|
case RightDown: return -M_PI/3;
|
|
case Nb_Neighbour: Q_ASSERT(false);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* @return the opposed neighbour.
|
|
*/
|
|
static Neighbour opposed(Neighbour n) {
|
|
switch (n) {
|
|
case Left: return Right;
|
|
case Right: return Left;
|
|
case LeftUp: return RightDown;
|
|
case LeftDown: return RightUp;
|
|
case RightUp: return LeftDown;
|
|
case RightDown: return LeftUp;
|
|
case Nb_Neighbour: Q_ASSERT(false);
|
|
}
|
|
return Nb_Neighbour;
|
|
}
|
|
|
|
/**
|
|
* @return the neighbour of the given coordinate.
|
|
*/
|
|
static Coord neighbour(const Coord &c, Neighbour n) {
|
|
bool oddRow = c.second%2;
|
|
switch (n) {
|
|
case Left: return c + Coord(-1, 0);
|
|
case Right: return c + Coord( 1, 0);
|
|
case LeftUp: return c + (oddRow ? Coord( 0, -1) : Coord(-1, -1));
|
|
case LeftDown: return c + (oddRow ? Coord( 0, 1) : Coord(-1, 1));
|
|
case RightUp: return c + (oddRow ? Coord( 1, -1) : Coord( 0, -1));
|
|
case RightDown: return c + (oddRow ? Coord( 1, 1) : Coord( 0, 1));
|
|
case Nb_Neighbour: Q_ASSERT(false);
|
|
}
|
|
return c;
|
|
}
|
|
|
|
/**
|
|
* @return the distance between the two coordinates in term of hexagons.
|
|
*/
|
|
static uint distance(const Coord &c1, const Coord &c2) {
|
|
return kAbs(c1.first - c2.first) + kAbs(c1.second - c2.second)
|
|
+ (c1.first==c2.first || c1.second==c2.second ? 0 : -1);
|
|
}
|
|
};
|
|
|
|
/**
|
|
* This template implements a hexagonal grid
|
|
* where hexagons form horizontal lines:
|
|
* <pre>
|
|
* (0,0) (0,1) (0,2)
|
|
* (1,0) (1,1) (1,2)
|
|
* (2,0) (2,1) (2,2)
|
|
* </pre>
|
|
*
|
|
* @ since 3.2
|
|
*/
|
|
template <class Type>
|
|
class Hexagonal : public Generic<Type>, public HexagonalBase
|
|
{
|
|
public:
|
|
/**
|
|
* Constructor.
|
|
*/
|
|
Hexagonal(uint width = 0, uint height = 0)
|
|
: Generic<Type>(width, height) {}
|
|
|
|
/**
|
|
* @return the neighbours of coordinate @param c
|
|
* to the given set of coordinates
|
|
* @param c the coordiante to use as the reference point
|
|
* @param insideOnly only add coordinates that are inside the grid.
|
|
*/
|
|
CoordList neighbours(const Coord &c, bool insideOnly = true) const {
|
|
CoordList neighbours;
|
|
for (uint i=0; i<Nb_Neighbour; i++) {
|
|
Coord n = neighbour(c, (Neighbour)i);
|
|
if ( insideOnly && !Generic<Type>::inside(n) ) continue;
|
|
neighbours.append(n);
|
|
}
|
|
return neighbours;
|
|
}
|
|
|
|
|
|
/**
|
|
* @return the neighbours at distance @param distance of coordinate
|
|
* @param c the coordinate to use as the reference point
|
|
* @param distance distance to the neighbour (1 means at contact).
|
|
* @param insideOnly only add coordinates that are inside the grid.
|
|
* @param all returns all neighbours at distance equal and less than
|
|
* @param distance (the original coordinate is not included).
|
|
*/
|
|
CoordList neighbours(const Coord &c, uint distance, bool all,
|
|
bool insideOnly = true) const {
|
|
// brute force algorithm -- you're welcome to make it more efficient :)
|
|
CoordList ring;
|
|
if ( distance==0 ) return ring;
|
|
ring = neighbours(c, insideOnly);
|
|
if ( distance==1 ) return ring;
|
|
CoordList center;
|
|
center.append(c);
|
|
for (uint i=1; i<distance; i++) {
|
|
CoordList newRing;
|
|
CoordList::const_iterator it;
|
|
for (it=ring.begin(); it!=ring.end(); ++it) {
|
|
CoordList n = neighbours(*it, insideOnly);
|
|
CoordList::const_iterator it2;
|
|
for (it2=n.begin(); it2!=n.end(); ++it2)
|
|
if ( center.find(*it2)==center.end()
|
|
&& ring.find(*it2)==ring.end()
|
|
&& newRing.find(*it2)==newRing.end() )
|
|
newRing.append(*it2);
|
|
center.append(*it);
|
|
}
|
|
ring = newRing;
|
|
}
|
|
if ( !all ) return ring;
|
|
CoordList::const_iterator it;
|
|
for (it=ring.begin(); it!=ring.end(); ++it)
|
|
center.append(*it);
|
|
center.remove(c);
|
|
return center;
|
|
}
|
|
};
|
|
|
|
} // namespace
|
|
|
|
#endif
|