Wang Tiles

src/libtiled/libtiled-src.pri

@@ -29,7 +29,8 @@ SOURCES += $$PWD/compression.cpp \
     $$PWD/tileset.cpp \
     $$PWD/tilesetformat.cpp \
     $$PWD/tilesetmanager.cpp \
-    $$PWD/varianttomapconverter.cpp
+    $$PWD/varianttomapconverter.cpp \
+    $$PWD/wangset.cpp
 HEADERS += $$PWD/compression.h \
     $$PWD/filesystemwatcher.h \
     $$PWD/gidmapper.h \
@@ -65,4 +66,5 @@ HEADERS += $$PWD/compression.h \
     $$PWD/tileset.h \
     $$PWD/tilesetformat.h \
     $$PWD/tilesetmanager.h \
-    $$PWD/varianttomapconverter.h
+    $$PWD/varianttomapconverter.h \
+    $$PWD/wangset.h

src/libtiled/libtiled.qbs

@@ -99,6 +99,8 @@ DynamicLibrary {
         "tilesetmanager.h",
         "varianttomapconverter.cpp",
         "varianttomapconverter.h",
+        "wangset.cpp",
+        "wangset.h",
     ]
 
     Group {

src/libtiled/wangset.cpp

@@ -0,0 +1,148 @@
+/*
+ * wangset.cpp
+ * Copyright 2017, Benjamin Trotter <bdtrotte@ucsc.edu>
+ * This file is part of libtiled.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ *    1. Redistributions of source code must retain the above copyright notice,
+ *       this list of conditions and the following disclaimer.
+ *
+ *    2. Redistributions in binary form must reproduce the above copyright
+ *       notice, this list of conditions and the following disclaimer in the
+ *       documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
+ * EVENT SHALL THE CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
+ * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+ * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
+ * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+ * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "wangset.h"
+
+#include <QStack>
+#include <QPoint>
+
+using namespace Tiled;
+
+WangSet::WangSet(Tileset *tileset,
+                 int edgeColors,
+                 int cornerColors,
+                 QString name,
+                 int imageTileId):
+    Object(Object::TerrainType), //for now, will add unique type soon
+    mTileSet(tileset),
+    mEdgeColors(edgeColors),
+    mCornerColors(cornerColors),
+    mName(std::move(name)),
+    mImageTileId(imageTileId),
+    mWangIdToTile(QMultiMap<WangId, Tile*>()),
+    mTileIdToWangId(QMap<int, WangId>())
+{
+}
+
+void WangSet::addTile(Tile *tile, WangId wangId)
+{
+    Q_ASSERT(tile->tileset() == mTileSet);
+
+    for (int i = 0; i < 4; ++i) {
+        Q_ASSERT(wangId.getColor(i,true) <= mEdgeColors);
+    }
+
+    for (int i = 0; i < 4; ++i) {
+        Q_ASSERT(wangId.getColor(i,false) <= mCornerColors);
+    }
+
+    mWangIdToTile.insert(wangId, tile);
+    mTileIdToWangId.insert(tile->id(), wangId);
+}
+
+Tile *WangSet::getMatchingTile(WangId wangId) const
+{
+    auto potentials = getAllTiles(wangId);
+
+    if (potentials.length() > 0)
+        return potentials[qrand() % potentials.length()];
+    else
+        return NULL;
+}
+
+QList<Tile*> WangSet::getAllTiles(WangId wangId) const
+{
+    QList<Tile*> list;
+
+    //Stores the space of a wild card, followed by how many colors that space can have.
+    QVector<QPoint> wildCards;
+
+    if (mEdgeColors > 0) {
+        for (int i = 0; i < 4; ++i) {
+            if (!wangId.getColor(i,true)) {
+                wildCards.append(QPoint(i * 8, mEdgeColors));
+            }
+        }
+    }
+
+    if (mCornerColors > 0) {
+        for (int i = 0; i < 4; ++i) {
+            if (!wangId.getColor(i,false)) {
+                wildCards.append(QPoint(i * 8 + 4, mCornerColors));
+            }
+        }
+    }
+
+    if (wildCards.isEmpty()) {
+        list.append(mWangIdToTile.values(wangId));
+    } else {
+        QStack<QPoint> stack;
+
+        for (int i = 0; i < wildCards.size(); ++i) {
+            stack.push(wildCards[i]);
+        }
+
+        int max = wildCards.size();
+
+        while (!stack.isEmpty()) {
+            if (stack.size() == max) {
+                int idVariation = 0;
+
+                for (int i = 0; i < max; ++i) {
+                    idVariation |= stack[i].y() << stack[i].x();
+                }
+
+                list.append(mWangIdToTile.values(idVariation | wangId.id()));
+
+                QPoint top = stack.pop();
+                top.setY(top.y() - 1);
+                if (top.y() > 0)
+                    stack.push(top);
+            } else {
+                QPoint top = stack.pop();
+                top.setY(top.y() - 1);
+                if (top.y() > 0) {
+                    stack.push(top);
+
+                    for (int i = stack.size(); i < max; ++i) {
+                        stack.push(wildCards[i]);
+                    }
+                }
+            }
+        }
+    }
+
+    return list;
+}
+
+WangId WangSet::getWangIdOfTile(Tile *tile) const
+{
+    if (tile->tileset() == mTileSet && mTileIdToWangId.contains(tile->id()))
+        return mTileIdToWangId.value(tile->id());
+    else
+        return WangId(0);
+}

src/libtiled/wangset.h

@@ -0,0 +1,162 @@
+/*
+ * wangset.h
+ * Copyright 2017, Benjamin Trotter <bdtrotte@ucsc.edu>
+ * This file is part of libtiled.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ *    1. Redistributions of source code must retain the above copyright notice,
+ *       this list of conditions and the following disclaimer.
+ *
+ *    2. Redistributions in binary form must reproduce the above copyright
+ *       notice, this list of conditions and the following disclaimer in the
+ *       documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
+ * EVENT SHALL THE CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
+ * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+ * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
+ * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+ * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#pragma once
+
+#include "tile.h"
+#include "tileset.h"
+
+#include <QMap>
+#include <QMultiMap>
+#include <QString>
+#include <QList>
+
+namespace Tiled {
+
+class WangId
+{
+public:
+    WangId(unsigned id){ mId = id; }
+    WangId() { mId = 0; }
+
+    inline unsigned id() const { return mId; }
+    inline void setId(unsigned id) { mId = id; }
+
+    /**
+     * @brief getColor returns the color of a desired edge/corner of a given wang id
+     * @param index index 0-3 with zero being the top right, and 3 left top
+     * @param edges requesting edge color (corners if false)
+     * @return
+     */
+    int getColor(int index, bool edges) const;
+
+    /**
+     * @brief rotateWangId rotates the wang id 90 * rotations degrees ccw
+     * @param rotations 1-3
+     * @return
+     */
+    void rotate(int rotations);
+
+    //used in maping
+    bool operator <(const WangId w) const
+    {
+        return mId < w.id();
+    }
+
+private:
+    unsigned mId;
+};
+
+inline int WangId::getColor(int index, bool edges) const
+{
+    int shift = (index * 8) + ((!edges) * 4);
+
+    int color = (mId >> shift) & 0xf;
+
+    return color;
+}
+
+void WangId::rotate(int rotations)
+{
+    if (rotations < 0)
+        rotations = 4 + (rotations % 4);
+    else
+        rotations %= 4;
+
+    unsigned rotated = mId << rotations*8;
+    rotated = rotated | (mId >> ((4 - rotations) * 8));
+
+    mId = rotated;
+}
+
+/**
+ * Represents a wang set.
+ */
+class TILEDSHARED_EXPORT WangSet : public Object
+{
+public:
+    WangSet(Tileset *tileset,
+            int edgeColors,
+            int cornerColors,
+            QString name,
+            int imageTileId);
+
+    QString name() const { return mName; }
+    void setName(const QString &name) { mName = name; }
+
+    int imageTileId() const { return mImageTileId; }
+    void setImageTileId(int imageTileId) { mImageTileId = imageTileId; }
+
+    Tile *imageTile() const { return mTileSet->findTile(mImageTileId); }
+
+    int edgeColors() const { return mEdgeColors; }
+    int cornerColors() const { return mCornerColors; }
+    void setEdgeColors(int n) { mEdgeColors = n; }
+    void setCornerColors(int n) { mCornerColors = n; }
+
+    /**
+     * @brief addTile Adds a tile to the wang set
+     * @param tile
+     * @param wangId
+     */
+    void addTile(Tile *tile, WangId wangId);
+
+    /**
+     * @brief getMatchingTile Returns a tile matching the given wangId.
+     * 0s in the id are wild cards, and can be filled with any color.
+     * If there are multiple possible options, one will be choosen at random.
+     * @param wangId
+     * @return
+     */
+    Tile *getMatchingTile(WangId wangId) const;
+
+    /**
+     * @brief getAllTiles Returns a list of all tiles which match a wangId
+     * 0s in the id are wild cards, and can be filled with any color.
+     * @param wangId
+     * @return
+     */
+    QList<Tile*> getAllTiles(WangId wangId) const;
+
+
+    /**
+     * @brief wangIdOfTile returns the wangId of a given tileId
+     * @param tileId
+     */
+    WangId getWangIdOfTile(Tile *tile) const;
+
+private:
+    Tileset *mTileSet;
+    QString mName;
+    int mImageTileId;
+    QMultiMap<WangId, Tile*> mWangIdToTile;
+    QMap<int, WangId> mTileIdToWangId; //This could be stored in the tile object.
+    int mEdgeColors;
+    int mCornerColors;
+};
+
+} // namespace Tiled