Don't leak palette if freeing screen object
[lsnes.git] / src / core / render.cpp
blob76ce28836c6268c3f2b5834459e8a4d03fecd504
1 #include "lsnes.hpp"
2 #include <snes/snes.hpp>
4 #include "core/misc.hpp"
5 #include "core/png.hpp"
6 #include "core/render.hpp"
8 #include <sstream>
9 #include <list>
10 #include <iomanip>
11 #include <cstdint>
12 #include <string>
13 #include <map>
14 #include <vector>
16 #define TAG_ZEROWIDTH 0
17 #define TAG_NARROW 1
18 #define TAG_WIDE 2
19 #define TAG_TABULATION 3
20 #define TAG_WIDTH_MASK 3
21 #define TAG_LINECHANGE 4
23 extern const char* font_hex_data;
25 namespace
27 std::vector<uint32_t> font_glyph_data;
28 std::map<uint32_t, uint32_t> font_glyph_offsets;
30 uint32_t parse_word(const char* x)
32 char buf[9] = {0};
33 char* end;
34 memcpy(buf, x, 8);
35 unsigned long v = strtoul(buf, &end, 16);
36 if(end != buf + 8)
37 v = 0xFFFFFFFFUL;
38 //std::cerr << "Parse word " << buf << std::endl;
39 return v;
42 void init_font()
44 static bool iflag = false;
45 if(iflag)
46 return;
47 //Special glyph data.
48 font_glyph_data.resize(7);
49 //Space & Unknown.
50 font_glyph_data[0] = TAG_NARROW;
51 font_glyph_data[1] = 0;
52 font_glyph_data[2] = 0;
53 font_glyph_data[3] = 0;
54 font_glyph_data[4] = 0;
55 //Tabulation.
56 font_glyph_data[5] = TAG_TABULATION;
57 //Linefeed.
58 font_glyph_data[6] = TAG_ZEROWIDTH | TAG_LINECHANGE;
60 size_t lsptr = 0;
61 uint32_t lc = 1;
62 for(size_t i = 0;; i++) {
63 //Skip spaces.
64 switch(font_hex_data[i]) {
65 case ' ':
66 case '\t':
67 //Skip spaces at start of line.
68 if(lsptr == i)
69 lsptr++;
70 case '\r':
71 case '\n':
72 case '\0': {
73 char* end;
74 uint32_t cp;
75 size_t fdatastart;
76 //Is this a comment?
77 if(lsptr == i || font_hex_data[lsptr] == '#')
78 goto skip_line;
79 cp = strtoul(font_hex_data + lsptr, &end, 16);
80 if(*end != ':') {
81 messages << "Malformed line " << lc << " in font data" << std::endl;
82 goto skip_line;
84 fdatastart = end - font_hex_data + 1;
85 if(i - fdatastart == 32) {
86 //Narrow glyph.
87 font_glyph_offsets[cp] = font_glyph_data.size();
88 font_glyph_data.push_back(TAG_NARROW);
89 for(uint32_t k = 0; k < 4; k++)
90 font_glyph_data.push_back(parse_word(end + 1 + 8 * k));
91 } else if(i - fdatastart == 64) {
92 //Wide glyph.
93 font_glyph_offsets[cp] = font_glyph_data.size();
94 font_glyph_data.push_back(TAG_WIDE);
95 for(uint32_t k = 0; k < 8; k++)
96 font_glyph_data.push_back(parse_word(end + 1 + 8 * k));
97 } else {
98 messages << "Malformed line " << lc << " in font data" << std::endl;
99 goto skip_line;
101 skip_line:
102 if(font_hex_data[i] != '\r' || font_hex_data[i + 1] != '\n')
103 lc++;
104 lsptr = i + 1;
107 if(!font_hex_data[i])
108 break;
111 //Special characters.
112 font_glyph_offsets[9] = 5;
113 font_glyph_offsets[10] = 6;
114 font_glyph_offsets[32] = 0;
116 uint32_t glyphs = 0;
117 uint32_t glyphs_narrow = 0;
118 uint32_t glyphs_wide = 0;
119 uint32_t glyphs_special = 0;
120 for(auto i : font_glyph_offsets) {
121 if(font_glyph_data[i.second] == TAG_NARROW)
122 glyphs_narrow++;
123 else if(font_glyph_data[i.second] == TAG_WIDE)
124 glyphs_wide++;
125 else
126 glyphs_special++;
127 glyphs++;
129 messages << "Loaded font data: " << glyphs << " glyphs (" << glyphs_narrow << " narrow, " <<
130 glyphs_wide << " wide, " << glyphs_special << " special)." << std::endl;
131 iflag = true;
134 inline uint32_t find_font_glyph_offset(uint32_t cp)
136 return font_glyph_offsets.count(cp) ? font_glyph_offsets[cp] : 0;
139 inline uint32_t process_tag(uint32_t tag, int32_t& x, int32_t& y, int32_t orig_x, bool hdbl, bool vdbl)
141 uint32_t dwidth;
142 switch(tag & TAG_WIDTH_MASK) {
143 case TAG_ZEROWIDTH:
144 dwidth = 0;
145 break;
146 case TAG_NARROW:
147 dwidth = 8;
148 break;
149 case TAG_WIDE:
150 dwidth = 16;
151 break;
152 case TAG_TABULATION:
153 dwidth = 0x40 - (x & 0x3F);
154 break;
156 x += dwidth * (hdbl ? 2 : 1);
157 if(tag & TAG_LINECHANGE) {
158 y += 16 * (vdbl ? 2 : 1);
159 x = orig_x;
161 return dwidth;
164 inline bool is_visible(uint32_t tag)
166 return ((tag & TAG_WIDTH_MASK) == TAG_NARROW || (tag & TAG_WIDTH_MASK) == TAG_WIDE);
171 void do_init_font()
173 init_font();
176 std::pair<uint32_t, const uint32_t*> find_glyph(uint32_t codepoint, int32_t x, int32_t y, int32_t orig_x,
177 int32_t& next_x, int32_t& next_y, bool hdbl, bool vdbl) throw()
179 init_font();
180 next_x = x;
181 next_y = y;
182 uint32_t offset = find_font_glyph_offset(codepoint);
183 uint32_t tag = font_glyph_data[offset];
184 uint32_t dwidth = process_tag(tag, next_x, next_y, orig_x, hdbl, vdbl);
185 bool visible = is_visible(tag);
186 return std::pair<uint32_t, const uint32_t*>(dwidth, visible ? &font_glyph_data[offset + 1] : NULL);
189 render_object::~render_object() throw()
193 void render_text(struct screen& scr, int32_t x, int32_t y, const std::string& text, premultiplied_color fg,
194 premultiplied_color bg, bool hdbl, bool vdbl) throw(std::bad_alloc)
196 int32_t orig_x = x;
197 uint32_t unicode_code = 0;
198 uint8_t unicode_left = 0;
199 for(size_t i = 0; i < text.length(); i++) {
200 uint8_t ch = text[i];
201 if(ch < 128)
202 unicode_code = text[i];
203 else if(ch < 192) {
204 if(!unicode_left)
205 continue;
206 unicode_code = 64 * unicode_code + ch - 128;
207 if(--unicode_left)
208 continue;
209 } else if(ch < 224) {
210 unicode_code = ch - 192;
211 unicode_left = 1;
212 continue;
213 } else if(ch < 240) {
214 unicode_code = ch - 224;
215 unicode_left = 2;
216 continue;
217 } else if(ch < 248) {
218 unicode_code = ch - 240;
219 unicode_left = 3;
220 continue;
221 } else
222 continue;
223 int32_t next_x, next_y;
224 auto p = find_glyph(unicode_code, x, y, orig_x, next_x, next_y, hdbl, vdbl);
225 uint32_t dx = 0;
226 uint32_t dw = p.first * (hdbl ? 2 : 1);
227 uint32_t dy = 0;
228 uint32_t dh = 16 * (vdbl ? 2 : 1);
229 uint32_t cx = static_cast<uint32_t>(static_cast<int32_t>(scr.originx) + x);
230 uint32_t cy = static_cast<uint32_t>(static_cast<int32_t>(scr.originy) + y);
231 while(cx > scr.width && dw > 0) {
232 dx++;
233 dw--;
234 cx++;
236 while(cy > scr.height && dh > 0) {
237 dy++;
238 dh--;
239 cy++;
241 while(cx + dw > scr.width && dw > 0)
242 dw--;
243 while(cy + dh > scr.height && dh > 0)
244 dh--;
245 if(!dw || !dh)
246 continue; //Outside screen.
248 uint32_t rshift = (p.first == 16) ? (vdbl ? 2 : 1) : (vdbl ? 3 : 2);
249 uint32_t rishift = (p.first == 16) ? 4 : 3;
250 uint32_t xshift = hdbl ? 1 : 0;
251 uint32_t yshift = vdbl ? 1 : 0;
252 uint32_t b = dx & 1;
254 if(p.second == NULL) {
255 //Blank glyph.
256 for(uint32_t j = 0; j < dh; j++) {
257 uint32_t* base = scr.rowptr(cy + j) + cx;
258 for(uint32_t i = 0; i < dw; i++)
259 bg.apply(base[i]);
261 } else {
262 for(uint32_t j = 0; j < dh; j++) {
263 uint32_t dataword = p.second[(dy + j) >> rshift];
264 uint32_t* base = scr.rowptr(cy + j) + cx;
265 uint32_t rbit = (~((dy + j) >> yshift << rishift) & 31) - (dx >> xshift);
266 if(hdbl) {
267 for(uint32_t i = 0; i < dw; i++)
268 if((dataword >> (rbit - ((i + b) >> 1))) & 1)
269 fg.apply(base[i]);
270 else
271 bg.apply(base[i]);
272 } else {
273 for(uint32_t i = 0; i < dw; i++)
274 if((dataword >> (rbit - i)) & 1)
275 fg.apply(base[i]);
276 else
277 bg.apply(base[i]);
281 x = next_x;
282 y = next_y;
286 void render_queue::add(struct render_object& obj) throw(std::bad_alloc)
288 q.push_back(&obj);
291 void render_queue::run(struct screen& scr) throw()
293 for(auto i : q) {
294 try {
295 (*i)(scr);
296 } catch(...) {
301 void render_queue::clear() throw()
303 for(auto i : q)
304 delete i;
305 q.clear();
308 render_queue::~render_queue() throw()
310 clear();
313 uint32_t screen::make_color(uint8_t r, uint8_t g, uint8_t b) throw()
315 uint32_t _r = r;
316 uint32_t _g = g;
317 uint32_t _b = b;
318 return (_r << 16) + (_g << 8) + _b;
321 lcscreen::lcscreen(const uint32_t* mem, bool hires, bool interlace, bool overscan, bool region) throw()
323 uint32_t dataoffset = 0;
324 width = hires ? 512 : 256;
325 height = 0;
326 if(region) {
327 //PAL.
328 height = 239;
329 dataoffset = overscan ? 9 : 1;
330 } else {
331 //presumably NTSC.
332 height = 224;
333 dataoffset = overscan ? 16 : 9;
335 if(interlace)
336 height <<= 1;
337 memory = mem + dataoffset * 1024;
338 pitch = interlace ? 512 : 1024;
339 user_memory = false;
342 lcscreen::lcscreen(const uint32_t* mem, uint32_t _width, uint32_t _height) throw()
344 width = _width;
345 height = _height;
346 memory = mem;
347 pitch = width;
348 user_memory = false;
351 lcscreen::lcscreen() throw()
353 width = 0;
354 height = 0;
355 memory = NULL;
356 user_memory = true;
357 pitch = 0;
358 allocated = 0;
361 lcscreen::lcscreen(const lcscreen& ls) throw(std::bad_alloc)
363 width = ls.width;
364 height = ls.height;
365 pitch = width;
366 user_memory = true;
367 allocated = static_cast<size_t>(width) * height;
368 memory = new uint32_t[allocated];
369 for(size_t l = 0; l < height; l++)
370 memcpy(const_cast<uint32_t*>(memory + l * width), ls.memory + l * ls.pitch, 4 * width);
373 lcscreen& lcscreen::operator=(const lcscreen& ls) throw(std::bad_alloc, std::runtime_error)
375 if(!user_memory)
376 throw std::runtime_error("Can't copy to non-user memory");
377 if(this == &ls)
378 return *this;
379 if(allocated < static_cast<size_t>(ls.width) * ls.height) {
380 size_t p_allocated = static_cast<size_t>(ls.width) * ls.height;
381 memory = new uint32_t[p_allocated];
382 allocated = p_allocated;
384 width = ls.width;
385 height = ls.height;
386 pitch = width;
387 for(size_t l = 0; l < height; l++)
388 memcpy(const_cast<uint32_t*>(memory + l * width), ls.memory + l * ls.pitch, 4 * width);
389 return *this;
392 lcscreen::~lcscreen()
394 if(user_memory)
395 delete[] const_cast<uint32_t*>(memory);
398 void lcscreen::load(const std::vector<char>& data) throw(std::bad_alloc, std::runtime_error)
400 if(!user_memory)
401 throw std::runtime_error("Can't load to non-user memory");
402 const uint8_t* data2 = reinterpret_cast<const uint8_t*>(&data[0]);
403 if(data.size() < 2)
404 throw std::runtime_error("Corrupt saved screenshot data");
405 uint32_t _width = static_cast<uint32_t>(data2[0]) * 256 + static_cast<uint32_t>(data2[1]);
406 if(_width > 1 && data.size() % (3 * _width) != 2)
407 throw std::runtime_error("Corrupt saved screenshot data");
408 uint32_t _height = (data.size() - 2) / (3 * _width);
409 if(allocated < static_cast<size_t>(_width) * _height) {
410 size_t p_allocated = static_cast<size_t>(_width) * _height;
411 memory = new uint32_t[p_allocated];
412 allocated = p_allocated;
414 uint32_t* mem = const_cast<uint32_t*>(memory);
415 width = _width;
416 height = _height;
417 pitch = width;
418 for(size_t i = 0; i < (data.size() - 2) / 3; i++)
419 mem[i] = static_cast<uint32_t>(data2[2 + 3 * i]) * 65536 +
420 static_cast<uint32_t>(data2[2 + 3 * i + 1]) * 256 +
421 static_cast<uint32_t>(data2[2 + 3 * i + 2]);
424 void lcscreen::save(std::vector<char>& data) throw(std::bad_alloc)
426 data.resize(2 + 3 * static_cast<size_t>(width) * height);
427 uint8_t* data2 = reinterpret_cast<uint8_t*>(&data[0]);
428 data2[0] = (width >> 8);
429 data2[1] = width;
430 for(size_t i = 0; i < (data.size() - 2) / 3; i++) {
431 data[2 + 3 * i] = memory[(i / width) * pitch + (i % width)] >> 16;
432 data[2 + 3 * i + 1] = memory[(i / width) * pitch + (i % width)] >> 8;
433 data[2 + 3 * i + 2] = memory[(i / width) * pitch + (i % width)];
437 void lcscreen::save_png(const std::string& file) throw(std::bad_alloc, std::runtime_error)
439 uint8_t* buffer = new uint8_t[3 * static_cast<size_t>(width) * height];
440 for(uint32_t j = 0; j < height; j++)
441 for(uint32_t i = 0; i < width; i++) {
442 uint32_t word = memory[pitch * j + i];
443 uint32_t l = 1 + ((word >> 15) & 0xF);
444 uint32_t r = l * ((word >> 0) & 0x1F);
445 uint32_t g = l * ((word >> 5) & 0x1F);
446 uint32_t b = l * ((word >> 10) & 0x1F);
447 buffer[3 * static_cast<size_t>(width) * j + 3 * i + 0] = r * 255 / 496;
448 buffer[3 * static_cast<size_t>(width) * j + 3 * i + 1] = g * 255 / 496;
449 buffer[3 * static_cast<size_t>(width) * j + 3 * i + 2] = b * 255 / 496;
451 try {
452 save_png_data(file, buffer, width, height);
453 delete[] buffer;
454 } catch(...) {
455 delete[] buffer;
456 throw;
460 void screen::copy_from(lcscreen& scr, uint32_t hscale, uint32_t vscale) throw()
462 if(width < originx || height < originy) {
463 //Just clear the screen.
464 for(uint32_t y = 0; y < height; y++)
465 memset(rowptr(y), 0, 4 * width);
466 return;
468 uint32_t copyable_width = 0, copyable_height = 0;
469 if(hscale)
470 copyable_width = (width - originx) / hscale;
471 if(vscale)
472 copyable_height = (height - originy) / vscale;
473 copyable_width = (copyable_width > scr.width) ? scr.width : copyable_width;
474 copyable_height = (copyable_height > scr.height) ? scr.height : copyable_height;
475 for(uint32_t y = 0; y < height; y++)
476 memset(rowptr(y), 0, 4 * width);
477 for(uint32_t y = 0; y < copyable_height; y++) {
478 uint32_t line = y * vscale + originy;
479 uint32_t* ptr = rowptr(line) + originx;
480 const uint32_t* sbase = scr.memory + y * scr.pitch;
481 for(uint32_t x = 0; x < copyable_width; x++) {
482 uint32_t c = palette[sbase[x] & 0x7FFFF];
483 for(uint32_t i = 0; i < hscale; i++)
484 *(ptr++) = c;
486 for(uint32_t j = 1; j < vscale; j++)
487 memcpy(rowptr(line + j) + originx, rowptr(line) + originx, 4 * hscale * copyable_width);
491 void screen::reallocate(uint32_t _width, uint32_t _height, bool upside_down) throw(std::bad_alloc)
493 if(_width == width && _height == height)
494 return;
495 if(!_width || !_height) {
496 width = height = originx = originy = pitch = 0;
497 if(memory && !user_memory)
498 delete[] memory;
499 memory = NULL;
500 user_memory = false;
501 flipped = upside_down;
502 return;
504 uint32_t* newmem = new uint32_t[_width * _height];
505 width = _width;
506 height = _height;
507 pitch = 4 * _width;
508 if(memory && !user_memory)
509 delete[] memory;
510 memory = newmem;
511 user_memory = false;
512 flipped = upside_down;
515 void screen::set(uint32_t* _memory, uint32_t _width, uint32_t _height, uint32_t _pitch) throw()
517 if(memory && !user_memory)
518 delete[] memory;
519 width = _width;
520 height = _height;
521 pitch = _pitch;
522 user_memory = true;
523 memory = _memory;
524 flipped = false;
527 void screen::set_origin(uint32_t _originx, uint32_t _originy) throw()
529 originx = _originx;
530 originy = _originy;
534 uint32_t* screen::rowptr(uint32_t row) throw()
536 if(flipped)
537 row = height - row - 1;
538 return reinterpret_cast<uint32_t*>(reinterpret_cast<uint8_t*>(memory) + row * pitch);
541 screen::screen() throw()
543 memory = NULL;
544 width = height = originx = originy = pitch = 0;
545 user_memory = false;
546 flipped = false;
547 palette = NULL;
548 set_palette(16, 8, 0);
551 screen::~screen() throw()
553 if(memory && !user_memory)
554 delete[] memory;
555 delete[] palette;
558 void clip_range(uint32_t origin, uint32_t size, int32_t base, int32_t& minc, int32_t& maxc) throw()
560 int64_t _origin = origin;
561 int64_t _size = size;
562 int64_t _base = base;
563 int64_t _minc = minc;
564 int64_t _maxc = maxc;
565 int64_t mincoordinate = _base + _origin + _minc;
566 int64_t maxcoordinate = _base + _origin + _maxc;
567 if(mincoordinate < 0)
568 _minc = _minc - mincoordinate;
569 if(maxcoordinate > _size)
570 _maxc = _maxc - (maxcoordinate - _size);
571 if(_minc >= maxc) {
572 minc = 0;
573 maxc = 0;
574 } else {
575 minc = _minc;
576 maxc = _maxc;
580 void screen::set_palette(uint32_t r, uint32_t g, uint32_t b)
582 if(!palette)
583 palette = new uint32_t[0x80000];
584 else if(r == palette_r && g == palette_g && b == palette_b)
585 return;
586 for(size_t i = 0; i < static_cast<size_t>(width) * height; i++) {
587 uint32_t word = memory[i];
588 uint32_t R = (word >> palette_r) & 0xFF;
589 uint32_t G = (word >> palette_g) & 0xFF;
590 uint32_t B = (word >> palette_b) & 0xFF;
591 memory[i] = (R << r) | (G << g) | (B << b);
593 for(unsigned i = 0; i < 0x80000; i++) {
594 unsigned l = 1 + ((i >> 15) & 0xF);
595 unsigned R = (i >> 0) & 0x1F;
596 unsigned G = (i >> 5) & 0x1F;
597 unsigned B = (i >> 10) & 0x1F;
598 double _l = static_cast<double>(l);
599 double m = 255.0 / 496.0;
600 R = floor(m * R * _l + 0.5);
601 G = floor(m * G * _l + 0.5);
602 B = floor(m * B * _l + 0.5);
603 palette[i] = (R << r) | (G << g) | (B << b);
605 palette_r = r;
606 palette_g = g;
607 palette_b = b;
610 void premultiplied_color::set_palette(unsigned rshift, unsigned gshift, unsigned bshift) throw()
612 uint32_t r = (orig >> 16) & 0xFF;
613 uint32_t g = (orig >> 8) & 0xFF;
614 uint32_t b = orig & 0xFF;
615 uint32_t color = (r << rshift) | (g << gshift) | (b << bshift);
616 hi = color & 0xFF00FF;
617 lo = (color & 0xFF00FF00) >> 8;
618 hi *= origa;
619 lo *= origa;