Fix tilemap layer copy/paste

src/app/ui/editor/editor.cpp

@@ -2730,7 +2730,10 @@ void Editor::setZoomAndCenterInMouse(const Zoom& zoom,
   }
 }
 
-void Editor::pasteImage(const Image* image, const Mask* mask, const gfx::Point* position)
+void Editor::pasteImage(const Image* image,
+                        const Mask* mask,
+                        const gfx::Point* position,
+                        const Tileset* srcTileset)
 {
   ASSERT(image);
 
@@ -2822,6 +2825,10 @@ void Editor::pasteImage(const Image* image, const Mask* mask, const gfx::Point*
   PixelsMovementPtr pixelsMovement(
     new PixelsMovement(UIContext::instance(), site, image, &mask2, "Paste"));
 
+  // Adjust image when copying between tilemap layers
+  if (site.tilemapMode() == TilemapMode::Tiles && srcTileset != nullptr)
+    pixelsMovement->remapTilesForPaste(srcTileset);
+
   setState(EditorStatePtr(new MovingPixelsState(this, NULL, pixelsMovement, NoHandle)));
 }
 

src/app/ui/editor/editor.h

@@ -244,7 +244,8 @@ public:
 
   void pasteImage(const Image* image,
                   const Mask* mask = nullptr,
-                  const gfx::Point* position = nullptr);
+                  const gfx::Point* position = nullptr,
+                  const Tileset* srcTileset = nullptr);
 
   void startSelectionTransformation(const gfx::Point& move, double angle);
   void startFlipTransformation(doc::algorithm::FlipType flipType);

src/app/ui/editor/pixels_movement.cpp

@@ -12,6 +12,7 @@
 #include "app/ui/editor/pixels_movement.h"
 
 #include "app/app.h"
+#include "app/cmd/add_tile.h"
 #include "app/cmd/clear_mask.h"
 #include "app/cmd/deselect_mask.h"
 #include "app/cmd/set_mask.h"
@@ -1356,6 +1357,39 @@ void PixelsMovement::drawTransformedTilemap(const Transformation& transformation
   draw_row(dst->height() - 1, src->height() - 1, 1);
 }
 
+void PixelsMovement::remapTilesForPaste(const Tileset* srcTileset)
+{
+  // Check that tile size matches before doing anything
+  LayerTilemap* dstLayer = static_cast<LayerTilemap*>(m_site.layer());
+  Tileset* dstTileset = dstLayer->tileset();
+  const doc::Grid& srcGrid = srcTileset->grid();
+  const doc::Grid& dstGrid = dstTileset->grid();
+  if (srcGrid.tileSize() != dstGrid.tileSize())
+    throw base::Exception("Tile size does not match.");
+
+  // Map source to destination
+  tile_index dstSz = dstTileset->size();
+  for (int y = 0; y < m_originalImage->height(); ++y) {
+    for (int x = 0; x < m_originalImage->width(); ++x) {
+      // Blank tile can be skipped
+      const color_t srcTi = m_originalImage->getPixel(x, y);
+      if (!srcTi)
+        continue;
+
+      // Add tile to destination if missing
+      tile_index dstTi;
+      const ImageRef t = srcTileset->get(srcTi);
+      if (!dstTileset->findTileIndex(t, dstTi)) {
+        m_tx(new cmd::AddTile(dstTileset, t, srcTileset->getTileData(srcTi)));
+        dstTi = dstSz++;
+      }
+
+      // Update tile index in image
+      m_originalImage->putPixel(x, y, dstTi);
+    }
+  }
+}
+
 void PixelsMovement::onPivotChange()
 {
   set_pivot_from_preferences(m_currentData);

src/app/ui/editor/pixels_movement.h

@@ -130,6 +130,9 @@ public:
   const Transformation& getTransformation() const { return m_currentData; }
   void setTransformation(const Transformation& t);
 
+  // For copy/paste between tilemap layers
+  void remapTilesForPaste(const Tileset* srcTileset);
+
 private:
   void setTransformationBase(const Transformation& t);
   void adjustPivot();

src/app/util/clipboard.cpp

@@ -10,6 +10,7 @@
 #endif
 
 #include "app/app.h"
+#include "app/cmd/add_tileset.h"
 #include "app/cmd/clear_mask.h"
 #include "app/cmd/deselect_mask.h"
 #include "app/cmd/set_mask.h"
@@ -584,7 +585,10 @@ void Clipboard::paste(Context* ctx, const bool interactive, const gfx::Point* po
         // TODO add post-command parameters (issue #2324)
 
         // Change to MovingTilemapState
-        editor->pasteImage(m_data->tilemap.get(), m_data->mask.get(), position);
+        editor->pasteImage(m_data->tilemap.get(),
+                           m_data->mask.get(),
+                           position,
+                           m_data->tileset.get());
       }
       else {
         // TODO non-interactive version (for scripts)
@@ -763,7 +767,14 @@ void Clipboard::paste(Context* ctx, const bool interactive, const gfx::Point* po
               afterThis = dstSpr->root()->lastLayer();
 
             Layer* newLayer = nullptr;
-            if (srcLayer->isImage())
+            if (srcLayer->isTilemap()) {
+              Tileset* srcTileset = static_cast<LayerTilemap*>(srcLayer)->tileset();
+              Tileset* tilesetCopy = Tileset::MakeCopyCopyingImagesForSprite(srcTileset, dstSpr);
+              const tileset_index tsi = dstSpr->tilesets()->size();
+              tx(new cmd::AddTileset(dstSpr, tilesetCopy));
+              newLayer = new LayerTilemap(dstSpr, tsi);
+            }
+            else if (srcLayer->isImage())
               newLayer = new LayerImage(dstSpr);
             else if (srcLayer->isGroup())
               newLayer = new LayerGroup(dstSpr);