src/platform/qt/FrameView.cpp (view raw)
1/* Copyright (c) 2013-2019 Jeffrey Pfau
2 *
3 * This Source Code Form is subject to the terms of the Mozilla Public
4 * License, v. 2.0. If a copy of the MPL was not distributed with this
5 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
6#include "FrameView.h"
7
8#include <QMouseEvent>
9#include <QPalette>
10
11#include <array>
12#include <cmath>
13
14#include "CoreController.h"
15#include "GBAApp.h"
16
17#include <mgba/core/core.h>
18#include <mgba/feature/video-logger.h>
19#ifdef M_CORE_GBA
20#include <mgba/internal/gba/gba.h>
21#include <mgba/internal/gba/io.h>
22#include <mgba/internal/gba/memory.h>
23#include <mgba/internal/gba/video.h>
24#endif
25#ifdef M_CORE_GB
26#include <mgba/internal/gb/gb.h>
27#include <mgba/internal/gb/memory.h>
28#endif
29
30using namespace QGBA;
31
32FrameView::FrameView(std::shared_ptr<CoreController> controller, QWidget* parent)
33 : AssetView(controller, parent)
34{
35 m_ui.setupUi(this);
36
37 m_glowTimer.setInterval(33);
38 connect(&m_glowTimer, &QTimer::timeout, this, [this]() {
39 ++m_glowFrame;
40 invalidateQueue();
41 });
42
43 m_ui.compositedView->installEventFilter(this);
44
45 connect(m_ui.queue, &QListWidget::itemChanged, this, [this](QListWidgetItem* item) {
46 Layer& layer = m_queue[item->data(Qt::UserRole).toInt()];
47 layer.enabled = item->checkState() == Qt::Checked;
48 if (layer.enabled) {
49 m_disabled.remove(layer.id);
50 } else {
51 m_disabled.insert(layer.id);
52 }
53 invalidateQueue();
54 });
55 connect(m_ui.queue, &QListWidget::currentItemChanged, this, [this](QListWidgetItem* item) {
56 if (item) {
57 m_active = m_queue[item->data(Qt::UserRole).toInt()].id;
58 } else {
59 m_active = {};
60 }
61 invalidateQueue();
62 });
63 connect(m_ui.magnification, static_cast<void (QSpinBox::*)(int)>(&QSpinBox::valueChanged), this, [this]() {
64 invalidateQueue();
65 });
66 connect(m_ui.exportButton, &QAbstractButton::pressed, this, &FrameView::exportFrame);
67 connect(m_ui.reset, &QAbstractButton::pressed, this, &FrameView::reset);
68
69 m_backdropPicker = ColorPicker(m_ui.backdrop, QColor(0, 0, 0, 0));
70 connect(&m_backdropPicker, &ColorPicker::colorChanged, this, [this](const QColor& color) {
71 m_overrideBackdrop = color;
72 });
73 m_controller->addFrameAction(std::bind(&FrameView::frameCallback, this, m_callbackLocker));
74
75 {
76 CoreController::Interrupter interrupter(m_controller);
77 refreshVl();
78 }
79 m_controller->frameAdvance();
80}
81
82FrameView::~FrameView() {
83 QMutexLocker locker(&m_mutex);
84 *m_callbackLocker = false;
85 if (m_vl) {
86 m_vl->deinit(m_vl);
87 }
88}
89
90bool FrameView::lookupLayer(const QPointF& coord, Layer*& out) {
91 for (Layer& layer : m_queue) {
92 if (!layer.enabled || m_disabled.contains(layer.id)) {
93 continue;
94 }
95 QPointF location = layer.location;
96 QSizeF layerDims(layer.image.width(), layer.image.height());
97 QRegion region;
98 if (layer.repeats) {
99 if (location.x() + layerDims.width() < 0) {
100 location.setX(std::fmod(location.x(), layerDims.width()));
101 }
102 if (location.y() + layerDims.height() < 0) {
103 location.setY(std::fmod(location.y(), layerDims.height()));
104 }
105
106 region += layer.mask.translated(location.x(), location.y());
107 region += layer.mask.translated(location.x() + layerDims.width(), location.y());
108 region += layer.mask.translated(location.x(), location.y() + layerDims.height());
109 region += layer.mask.translated(location.x() + layerDims.width(), location.y() + layerDims.height());
110 } else {
111 region = layer.mask.translated(location.x(), location.y());
112 }
113
114 if (region.contains(QPoint(coord.x(), coord.y()))) {
115 out = &layer;
116 return true;
117 }
118 }
119 return false;
120}
121
122void FrameView::selectLayer(const QPointF& coord) {
123 Layer* layer;
124 if (!lookupLayer(coord, layer)) {
125 return;
126 }
127 if (layer->id == m_active) {
128 m_active = {};
129 } else {
130 m_active = layer->id;
131 }
132 m_glowFrame = 0;
133}
134
135void FrameView::disableLayer(const QPointF& coord) {
136 Layer* layer;
137 if (!lookupLayer(coord, layer)) {
138 return;
139 }
140 layer->enabled = false;
141 m_disabled.insert(layer->id);
142}
143
144#ifdef M_CORE_GBA
145void FrameView::updateTilesGBA(bool) {
146 if (m_ui.freeze->checkState() == Qt::Checked) {
147 return;
148 }
149 QMutexLocker locker(&m_mutex);
150 m_queue.clear();
151 {
152 CoreController::Interrupter interrupter(m_controller);
153
154 uint16_t* io = static_cast<GBA*>(m_controller->thread()->core->board)->memory.io;
155 QRgb backdrop = M_RGB5_TO_RGB8(static_cast<GBA*>(m_controller->thread()->core->board)->video.palette[0]);
156 m_gbaDispcnt = io[REG_DISPCNT >> 1];
157 int mode = GBARegisterDISPCNTGetMode(m_gbaDispcnt);
158
159 std::array<bool, 4> enabled{
160 bool(GBARegisterDISPCNTIsBg0Enable(m_gbaDispcnt)),
161 bool(GBARegisterDISPCNTIsBg1Enable(m_gbaDispcnt)),
162 bool(GBARegisterDISPCNTIsBg2Enable(m_gbaDispcnt)),
163 bool(GBARegisterDISPCNTIsBg3Enable(m_gbaDispcnt)),
164 };
165
166 for (int priority = 0; priority < 4; ++priority) {
167 for (int sprite = 0; sprite < 128; ++sprite) {
168 ObjInfo info;
169 lookupObj(sprite, &info);
170
171 if (!info.enabled || info.priority != priority) {
172 continue;
173 }
174
175 QPointF offset(info.x, info.y);
176 QImage obj(compositeObj(info));
177 if (info.hflip || info.vflip) {
178 obj = obj.mirrored(info.hflip, info.vflip);
179 }
180 if (!info.xform.isIdentity()) {
181 offset += QPointF(obj.width(), obj.height()) / 2;
182 obj = obj.transformed(info.xform);
183 offset -= QPointF(obj.width() / 2, obj.height() / 2);
184 }
185 m_queue.append({
186 { LayerId::SPRITE, sprite },
187 !m_disabled.contains({ LayerId::SPRITE, sprite }),
188 QPixmap::fromImage(obj),
189 {}, offset, false, false
190 });
191 if (m_queue.back().image.hasAlpha()) {
192 m_queue.back().mask = QRegion(m_queue.back().image.mask());
193 } else {
194 m_queue.back().mask = QRegion(0, 0, m_queue.back().image.width(), m_queue.back().image.height());
195 }
196 }
197
198 for (int bg = 0; bg < 4; ++bg) {
199 if (!enabled[bg]) {
200 continue;
201 }
202 if (GBARegisterBGCNTGetPriority(io[(REG_BG0CNT >> 1) + bg]) != priority) {
203 continue;
204 }
205
206 QPointF offset;
207 if (mode == 0) {
208 offset.setX(-(io[(REG_BG0HOFS >> 1) + (bg << 1)] & 0x1FF));
209 offset.setY(-(io[(REG_BG0VOFS >> 1) + (bg << 1)] & 0x1FF));
210 };
211 m_queue.append({
212 { LayerId::BACKGROUND, bg },
213 !m_disabled.contains({ LayerId::BACKGROUND, bg }),
214 QPixmap::fromImage(compositeMap(bg, m_mapStatus[bg])),
215 {}, offset, true, false
216 });
217 if (m_queue.back().image.hasAlpha()) {
218 m_queue.back().mask = QRegion(m_queue.back().image.mask());
219 } else {
220 m_queue.back().mask = QRegion(0, 0, m_queue.back().image.width(), m_queue.back().image.height());
221 }
222 }
223 }
224 QImage backdropImage(QSize(GBA_VIDEO_HORIZONTAL_PIXELS, GBA_VIDEO_VERTICAL_PIXELS), QImage::Format_Mono);
225 backdropImage.fill(1);
226 backdropImage.setColorTable({backdrop, backdrop | 0xFF000000 });
227 m_queue.append({
228 { LayerId::BACKDROP },
229 !m_disabled.contains({ LayerId::BACKDROP }),
230 QPixmap::fromImage(backdropImage),
231 {}, {0, 0}, false, true
232 });
233 updateRendered();
234 }
235 invalidateQueue(QSize(GBA_VIDEO_HORIZONTAL_PIXELS, GBA_VIDEO_VERTICAL_PIXELS));
236}
237
238void FrameView::injectGBA() {
239 mVideoLogger* logger = m_vl->videoLogger;
240 GBA* gba = static_cast<GBA*>(m_vl->board);
241 gba->video.renderer->highlightBG[0] = false;
242 gba->video.renderer->highlightBG[1] = false;
243 gba->video.renderer->highlightBG[2] = false;
244 gba->video.renderer->highlightBG[3] = false;
245 for (int i = 0; i < 128; ++i) {
246 gba->video.renderer->highlightOBJ[i] = false;
247 }
248 QPalette palette;
249 gba->video.renderer->highlightColor = palette.color(QPalette::HighlightedText).rgb();
250 gba->video.renderer->highlightAmount = sin(m_glowFrame * M_PI / 30) * 48 + 64;
251 if (!m_overrideBackdrop.isValid()) {
252 QRgb backdrop = M_RGB5_TO_RGB8(gba->video.palette[0]) | 0xFF000000;
253 m_backdropPicker.setColor(backdrop);
254 }
255
256 m_vl->reset(m_vl);
257 for (const Layer& layer : m_queue) {
258 switch (layer.id.type) {
259 case LayerId::SPRITE:
260 if (!layer.enabled) {
261 mVideoLoggerInjectOAM(logger, layer.id.index << 2, 0x200);
262 }
263 if (layer.id == m_active) {
264 gba->video.renderer->highlightOBJ[layer.id.index] = true;
265 }
266 break;
267 case LayerId::BACKGROUND:
268 m_vl->enableVideoLayer(m_vl, layer.id.index, layer.enabled);
269 if (layer.id == m_active) {
270 gba->video.renderer->highlightBG[layer.id.index] = true;
271 }
272 break;
273 }
274 }
275 if (m_overrideBackdrop.isValid()) {
276 mVideoLoggerInjectPalette(logger, 0, M_RGB8_TO_RGB5(m_overrideBackdrop.rgb()));
277 }
278}
279#endif
280
281#ifdef M_CORE_GB
282void FrameView::updateTilesGB(bool) {
283 if (m_ui.freeze->checkState() == Qt::Checked) {
284 return;
285 }
286 m_queue.clear();
287 {
288 CoreController::Interrupter interrupter(m_controller);
289 for (int sprite = 0; sprite < 40; ++sprite) {
290 ObjInfo info;
291 lookupObj(sprite, &info);
292
293 if (!info.enabled) {
294 continue;
295 }
296
297 QPointF offset(info.x, info.y);
298 QImage obj(compositeObj(info));
299 if (info.hflip || info.vflip) {
300 obj = obj.mirrored(info.hflip, info.vflip);
301 }
302 m_queue.append({
303 { LayerId::SPRITE, sprite },
304 !m_disabled.contains({ LayerId::SPRITE, sprite }),
305 QPixmap::fromImage(obj),
306 {}, offset, false, false
307 });
308 if (m_queue.back().image.hasAlpha()) {
309 m_queue.back().mask = QRegion(m_queue.back().image.mask());
310 } else {
311 m_queue.back().mask = QRegion(0, 0, m_queue.back().image.width(), m_queue.back().image.height());
312 }
313 }
314
315 updateRendered();
316 }
317 invalidateQueue(m_controller->screenDimensions());
318}
319
320void FrameView::injectGB() {
321 mVideoLogger* logger = m_vl->videoLogger;
322
323 m_vl->reset(m_vl);
324 for (const Layer& layer : m_queue) {
325 switch (layer.id.type) {
326 case LayerId::SPRITE:
327 if (!layer.enabled) {
328 mVideoLoggerInjectOAM(logger, layer.id.index << 2, 0);
329 }
330 break;
331 }
332 }
333}
334#endif
335
336void FrameView::invalidateQueue(const QSize& dims) {
337 if (dims.isValid()) {
338 m_dims = dims;
339 }
340 bool blockSignals = m_ui.queue->blockSignals(true);
341 QMutexLocker locker(&m_mutex);
342 if (m_vl) {
343 mVideoLogger* logger = m_vl->videoLogger;
344 mVideoLoggerInjectionPoint(logger, LOGGER_INJECTION_FIRST_SCANLINE);
345 switch (m_controller->platform()) {
346#ifdef M_CORE_GBA
347 case PLATFORM_GBA:
348 injectGBA();
349 break;
350#endif
351#ifdef M_CORE_GB
352 case PLATFORM_GB:
353 injectGB();
354 break;
355#endif
356 }
357 if (m_ui.disableScanline->checkState() == Qt::Checked) {
358 mVideoLoggerIgnoreAfterInjection(logger, (1 << DIRTY_PALETTE) | (1 << DIRTY_OAM) | (1 << DIRTY_REGISTER));
359 } else {
360 mVideoLoggerIgnoreAfterInjection(logger, 0);
361 }
362 m_vl->runFrame(m_vl);
363 }
364
365 for (int i = 0; i < m_queue.count(); ++i) {
366 const Layer& layer = m_queue[i];
367 QListWidgetItem* item;
368 if (i >= m_ui.queue->count()) {
369 item = new QListWidgetItem;
370 m_ui.queue->addItem(item);
371 } else {
372 item = m_ui.queue->item(i);
373 }
374 item->setText(layer.id.readable());
375 item->setFlags(Qt::ItemIsSelectable | (layer.fixed ? Qt::NoItemFlags : Qt::ItemIsUserCheckable) | Qt::ItemIsEnabled);
376 item->setCheckState(layer.enabled ? Qt::Checked : Qt::Unchecked);
377 item->setData(Qt::UserRole, i);
378 item->setSelected(layer.id == m_active);
379 }
380
381 while (m_ui.queue->count() > m_queue.count()) {
382 delete m_ui.queue->takeItem(m_queue.count());
383 }
384 m_ui.queue->blockSignals(blockSignals);
385
386 QPixmap composited;
387 if (m_framebuffer.isNull()) {
388 updateRendered();
389 composited = m_rendered;
390 } else {
391 QImage framebuffer(m_framebuffer);
392 m_ui.exportButton->setEnabled(true);
393 if (framebuffer.size() != m_dims) {
394 framebuffer = framebuffer.copy({QPoint(), m_dims});
395 }
396 composited.convertFromImage(framebuffer);
397 }
398 m_composited = composited.scaled(m_dims * m_ui.magnification->value());
399 m_ui.compositedView->setPixmap(m_composited);
400}
401
402void FrameView::updateRendered() {
403 if (m_ui.freeze->checkState() == Qt::Checked) {
404 return;
405 }
406 m_rendered.convertFromImage(m_controller->getPixels());
407}
408
409bool FrameView::eventFilter(QObject* obj, QEvent* event) {
410 QPointF pos;
411 switch (event->type()) {
412 case QEvent::MouseButtonPress:
413 pos = static_cast<QMouseEvent*>(event)->localPos();
414 pos /= m_ui.magnification->value();
415 selectLayer(pos);
416 return true;
417 case QEvent::MouseButtonDblClick:
418 pos = static_cast<QMouseEvent*>(event)->localPos();
419 pos /= m_ui.magnification->value();
420 disableLayer(pos);
421 return true;
422 default:
423 break;
424 }
425 return false;
426}
427
428void FrameView::refreshVl() {
429 QMutexLocker locker(&m_mutex);
430 m_currentFrame = m_nextFrame;
431 m_nextFrame = VFileMemChunk(nullptr, 0);
432 if (m_currentFrame) {
433 m_controller->endVideoLog(false);
434 QMetaObject::invokeMethod(this, "newVl");
435 }
436 m_controller->endVideoLog();
437 m_controller->startVideoLog(m_nextFrame, false);
438}
439
440void FrameView::newVl() {
441 if (!m_glowTimer.isActive()) {
442 m_glowTimer.start();
443 }
444 QMutexLocker locker(&m_mutex);
445 if (!m_currentFrame) {
446 return;
447 }
448 if (m_vl) {
449 m_vl->deinit(m_vl);
450 }
451 m_vl = mCoreFindVF(m_currentFrame);
452 m_vl->init(m_vl);
453 m_vl->loadROM(m_vl, m_currentFrame);
454 m_currentFrame = nullptr;
455 mCoreInitConfig(m_vl, nullptr);
456 unsigned width, height;
457 m_vl->desiredVideoDimensions(m_vl, &width, &height);
458 m_framebuffer = QImage(width, height, QImage::Format_RGBX8888);
459 m_vl->setVideoBuffer(m_vl, reinterpret_cast<color_t*>(m_framebuffer.bits()), width);
460 m_vl->reset(m_vl);
461}
462
463void FrameView::frameCallback(FrameView* viewer, std::shared_ptr<bool> lock) {
464 if (!*lock) {
465 return;
466 }
467 CoreController::Interrupter interrupter(viewer->m_controller, true);
468 viewer->refreshVl();
469 viewer->m_controller->addFrameAction(std::bind(&FrameView::frameCallback, viewer, lock));
470}
471
472void FrameView::exportFrame() {
473 QString filename = GBAApp::app()->getSaveFileName(this, tr("Export frame"),
474 tr("Portable Network Graphics (*.png)"));
475 CoreController::Interrupter interrupter(m_controller);
476 m_framebuffer.save(filename, "PNG");
477}
478
479void FrameView::reset() {
480 m_disabled.clear();
481 for (Layer& layer : m_queue) {
482 layer.enabled = true;
483 }
484 m_overrideBackdrop = QColor();
485 invalidateQueue();
486}
487
488QString FrameView::LayerId::readable() const {
489 QString typeStr;
490 switch (type) {
491 case NONE:
492 return tr("None");
493 case BACKGROUND:
494 typeStr = tr("Background");
495 break;
496 case WINDOW:
497 typeStr = tr("Window");
498 break;
499 case SPRITE:
500 typeStr = tr("Sprite");
501 break;
502 case BACKDROP:
503 typeStr = tr("Backdrop");
504 break;
505 case FRAME:
506 typeStr = tr("Frame");
507 break;
508 }
509 if (index < 0) {
510 return typeStr;
511 }
512 return tr("%1 %2").arg(typeStr).arg(index);
513}