[nostalgia/core/studio] Add ability to export tile sheet to png

src/nostalgia/core/consts.hpp

@@ -4,6 +4,8 @@
 namespace nostalgia::core {
 
 constexpr auto PixelsPerTile = 64;
+constexpr auto TileWidth = 8;
+constexpr auto TileHeight = 8;
 
 constexpr auto FileExt_ng = ".ng";
 constexpr auto FileExt_npal = ".npal";

src/nostalgia/core/studio/tilesheeteditor.cpp

@@ -8,6 +8,7 @@
 
 #include <array>
 
+#include <QBuffer>
 #include <QHBoxLayout>
 #include <QHeaderView>
 #include <QPointer>
@@ -25,6 +26,7 @@
 #include <memory>
 
 #include <nostalgia/core/consts.hpp>
+#include <nostalgia/common/point.hpp>
 
 #include "consts.hpp"
 #include "tilesheeteditor.hpp"
@@ -346,6 +348,32 @@ void SheetData::setSelectedColor(int index) {
 	m_selectedColor = index;
 }
 
+std::unique_ptr<NostalgiaGraphic> SheetData::toNostalgiaGraphic() const {
+	auto ng = std::make_unique<NostalgiaGraphic>();
+	const auto highestColorIdx = static_cast<uint8_t>(*std::max_element(m_pixels.begin(), m_pixels.end()));
+	ng->defaultPalette = m_currentPalettePath.toUtf8().data();
+	ng->bpp = highestColorIdx > 15 ? 8 : 4;
+	ng->columns = m_columns;
+	ng->rows = m_rows;
+	auto pixelCount = ng->rows * ng->columns * PixelsPerTile;
+	if (ng->bpp == 4) {
+		ng->tiles.resize(pixelCount / 2);
+		for (int i = 0; i < m_pixels.size(); ++i) {
+			if (i & 1) {
+				ng->tiles[i / 2] |= static_cast<uint8_t>(m_pixels[i]) << 4;
+			} else {
+				ng->tiles[i / 2] = static_cast<uint8_t>(m_pixels[i]);
+			}
+		}
+	} else {
+		ng->tiles.resize(pixelCount);
+		for (std::size_t i = 0; i < ng->tiles.size(); ++i) {
+			ng->tiles[i] = static_cast<uint8_t>(m_pixels[i]);
+		}
+	}
+	return ng;
+}
+
 void SheetData::setColumns(int columns) {
 	m_columns = columns;
 	emit columnsChanged(columns);
@@ -385,31 +413,6 @@ void SheetData::updatePixels(const NostalgiaGraphic *ng) {
 	emit pixelsChanged();
 }
 
-std::unique_ptr<NostalgiaGraphic> SheetData::toNostalgiaGraphic() const {
-	auto ng = std::make_unique<NostalgiaGraphic>();
-	const auto highestColorIdx = static_cast<uint8_t>(*std::max_element(m_pixels.begin(), m_pixels.end()));
-	ng->defaultPalette = m_currentPalettePath.toUtf8().data();
-	ng->bpp = highestColorIdx > 15 ? 8 : 4;
-	ng->columns = m_columns;
-	ng->rows = m_rows;
-	if (ng->bpp == 4) {
-		ng->tiles.resize(m_pixels.size() / 2);
-		for (int i = 0; i < m_pixels.size(); ++i) {
-			if (i & 1) {
-				ng->tiles[i / 2] |= static_cast<uint8_t>(m_pixels[i]) << 4;
-			} else {
-				ng->tiles[i / 2] = static_cast<uint8_t>(m_pixels[i]);
-			}
-		}
-	} else {
-		ng->tiles.resize(m_pixels.size());
-		for (int i = 0; i < m_pixels.size(); ++i) {
-			ng->tiles[i] = static_cast<uint8_t>(m_pixels[i]);
-		}
-	}
-	return ng;
-}
-
 TileSheetEditor::TileSheetEditor(QString path, const studio::Context *ctx, QWidget *parent): studio::Editor(parent), m_sheetData(undoStack()) {
 	m_ctx = ctx;
 	m_itemPath = path;
@@ -451,6 +454,7 @@ TileSheetEditor::TileSheetEditor(QString path, const studio::Context *ctx, QWidg
 	restoreState();
 	connect(&m_sheetData, &SheetData::changeOccurred, [this] { setUnsavedChanges(true); });
 	connect(&m_sheetData, &SheetData::paletteChanged, this, &TileSheetEditor::setColorTable);
+	setExportable(true);
 }
 
 TileSheetEditor::~TileSheetEditor() {
@@ -461,6 +465,24 @@ QString TileSheetEditor::itemName() {
 	return m_itemName;
 }
 
+void TileSheetEditor::exportFile() {
+	auto path = QFileDialog::getSaveFileName(this, tr("Export to Image"), "",
+	                                         tr("PNG (*.png)"));
+	auto ng = m_sheetData.toNostalgiaGraphic();
+	QString palPath;
+	if (palPath == "" && ng->defaultPalette.type() == ox::FileAddressType::Path) {
+		palPath = ng->defaultPalette.getPath().value;
+	}
+	std::unique_ptr<NostalgiaPalette> npal;
+	try {
+		npal = m_ctx->project->loadObj<NostalgiaPalette>(palPath);
+	} catch (ox::Error) {
+		qWarning() << "Could not open palette" << palPath;
+		// TODO: message box to notify of failure
+	}
+	toQImage(ng.get(), npal.get()).save(path, "PNG");
+}
+
 void TileSheetEditor::saveItem() {
 	m_sheetData.save(m_ctx, m_itemPath);
 }
@@ -531,6 +553,54 @@ QString TileSheetEditor::palettePath(QString paletteName) const {
 	return PaletteDir + paletteName + FileExt_npal;
 }
 
+constexpr common::Point idxToPt(int i, int c, int) noexcept {
+	common::Point p;
+	const auto t = i / PixelsPerTile; // tile number
+	const auto iti = i % PixelsPerTile; // in tile index
+	const auto tc = t % c; // tile column
+	const auto tr = t / c; // tile row
+	const auto itx = iti % TileWidth; // in tile x
+	const auto ity = iti / TileHeight; // in tile y
+	return {
+		itx + tc * TileWidth,
+		ity + tr * TileHeight,
+	};
+}
+
+static_assert(idxToPt(4, 1, 1) == common::Point{4, 0});
+static_assert(idxToPt(8, 1, 1) == common::Point{0, 1});
+static_assert(idxToPt(8, 2, 2) == common::Point{0, 1});
+static_assert(idxToPt(64, 2, 2) == common::Point{8, 0});
+static_assert(idxToPt(128, 2, 2) == common::Point{0, 8});
+static_assert(idxToPt(129, 2, 2) == common::Point{1, 8});
+static_assert(idxToPt(192, 2, 2) == common::Point{8, 8});
+static_assert(idxToPt(384, 8, 6) == common::Point{48, 0});
+
+QImage TileSheetEditor::toQImage(NostalgiaGraphic *ng, NostalgiaPalette *npal) const {
+	const auto w = ng->columns * TileWidth;
+	const auto h = ng->rows * TileHeight;
+	QImage dst(w, h, QImage::Format_RGB32);
+	auto setPixel = [&dst, npal, ng](std::size_t i, uint8_t p) {
+		const auto color = toColor32(npal->colors[p]) >> 8;
+		const auto pt = idxToPt(i, ng->columns, ng->rows);
+		dst.setPixel(pt.x, pt.y, color);
+	};
+	if (ng->bpp == 4) {
+		for (std::size_t i = 0; i < ng->tiles.size(); ++i) {
+			auto p1 = ng->tiles[i] & 0xF;
+			auto p2 = ng->tiles[i] >> 4;
+			setPixel(i * 2 + 0, p1);
+			setPixel(i * 2 + 1, p2);
+		}
+	} else {
+		for (std::size_t i = 0; i < ng->tiles.size(); i++) {
+			const auto p = ng->tiles[i];
+			setPixel(i, p);
+		}
+	}
+	return dst;
+}
+
 void TileSheetEditor::colorSelected() {
 	m_sheetData.setSelectedColor(m_colorPicker.colorTable->currentRow());
 }

src/nostalgia/core/studio/tilesheeteditor.hpp

@@ -53,7 +53,7 @@ class SheetData: public QObject {
 
 		[[nodiscard]] int rows() const;
 
-		const QVector<int> &pixels() const;
+		[[nodiscard]] const QVector<int> &pixels() const;
 
 		[[nodiscard]] QStringList palette() const;
 
@@ -73,7 +73,7 @@ class SheetData: public QObject {
 
 		void setSelectedColor(int index);
 
-		QUndoStack *undoStack();
+		[[nodiscard]] std::unique_ptr<NostalgiaGraphic> toNostalgiaGraphic() const;
 
 	public slots:
 		void setColumns(int columns);
@@ -93,8 +93,6 @@ class SheetData: public QObject {
 	private:
 		void updatePixels(const NostalgiaGraphic *ng);
 
-		[[nodiscard]] std::unique_ptr<NostalgiaGraphic> toNostalgiaGraphic() const;
-
 	signals:
 		void changeOccurred();
 
@@ -133,6 +131,8 @@ class TileSheetEditor: public studio::Editor {
 
 		QString itemName() override;
 
+		void exportFile() override;
+
 	protected:
 		void saveItem() override;
 
@@ -149,6 +149,8 @@ class TileSheetEditor: public studio::Editor {
 
 		[[nodiscard]] QString palettePath(QString palettePath) const;
 
+		[[nodiscard]] QImage toQImage(NostalgiaGraphic *ng, NostalgiaPalette *npal) const;
+
 	public slots:
 		void colorSelected();