Fix zero luma corner case
[lsnes.git] / src / core / render.cpp
blob9a02325dc136d21b483bfb421ffac7b4c02af8ff
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)
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;
157 if(tag & TAG_LINECHANGE) {
158 y += 16;
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) 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);
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) 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);
225 uint32_t dx = 0;
226 uint32_t dw = p.first;
227 uint32_t dy = 0;
228 uint32_t dh = 16;
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 if(p.second == NULL) {
249 //Blank glyph.
250 for(uint32_t j = 0; j < dh; j++) {
251 uint32_t* base = scr.rowptr(cy + j) + cx;
252 for(uint32_t i = 0; i < dw; i++)
253 bg.apply(base[i]);
255 } else if(p.first == 16) {
256 //Wide glyph.
257 for(uint32_t j = 0; j < dh; j++) {
258 uint32_t dataword = p.second[(dy + j) >> 1];
259 uint32_t* base = scr.rowptr(cy + j) + cx;
260 uint32_t rbit = (~((dy + j) << 4) & 0x1F) - dx;
261 for(uint32_t i = 0; i < dw; i++)
262 if((dataword >> (rbit - i)) & 1)
263 fg.apply(base[i]);
264 else
265 bg.apply(base[i]);
267 } else {
268 //narrow glyph.
269 for(uint32_t j = 0; j < dh; j++) {
270 uint32_t dataword = p.second[(dy + j) >> 2];
271 uint32_t* base = scr.rowptr(cy + j) + cx;
272 uint32_t rbit = (~((dy + j) << 3) & 0x1F) - dx;
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]);
280 x = next_x;
281 y = next_y;
285 void render_queue::add(struct render_object& obj) throw(std::bad_alloc)
287 q.push_back(&obj);
290 void render_queue::run(struct screen& scr) throw()
292 for(auto i : q) {
293 try {
294 (*i)(scr);
295 } catch(...) {
297 delete i;
299 q.clear();
302 void render_queue::clear() throw()
304 for(auto i : q)
305 delete i;
306 q.clear();
309 render_queue::~render_queue() throw()
311 clear();
314 uint32_t screen::make_color(uint8_t r, uint8_t g, uint8_t b) throw()
316 uint32_t _r = r;
317 uint32_t _g = g;
318 uint32_t _b = b;
319 return (_r << 16) + (_g << 8) + _b;
322 lcscreen::lcscreen(const uint32_t* mem, bool hires, bool interlace, bool overscan, bool region) throw()
324 uint32_t dataoffset = 0;
325 width = hires ? 512 : 256;
326 height = 0;
327 if(region) {
328 //PAL.
329 height = 239;
330 dataoffset = overscan ? 9 : 1;
331 } else {
332 //presumably NTSC.
333 height = 224;
334 dataoffset = overscan ? 16 : 9;
336 if(interlace)
337 height <<= 1;
338 memory = mem + dataoffset * 1024;
339 pitch = interlace ? 512 : 1024;
340 user_memory = false;
343 lcscreen::lcscreen(const uint32_t* mem, uint32_t _width, uint32_t _height) throw()
345 width = _width;
346 height = _height;
347 memory = mem;
348 pitch = width;
349 user_memory = false;
352 lcscreen::lcscreen() throw()
354 width = 0;
355 height = 0;
356 memory = NULL;
357 user_memory = true;
358 pitch = 0;
359 allocated = 0;
362 lcscreen::lcscreen(const lcscreen& ls) throw(std::bad_alloc)
364 width = ls.width;
365 height = ls.height;
366 pitch = width;
367 user_memory = true;
368 allocated = static_cast<size_t>(width) * height;
369 memory = new uint32_t[allocated];
370 for(size_t l = 0; l < height; l++)
371 memcpy(const_cast<uint32_t*>(memory + l * width), ls.memory + l * ls.pitch, 4 * width);
374 lcscreen& lcscreen::operator=(const lcscreen& ls) throw(std::bad_alloc, std::runtime_error)
376 if(!user_memory)
377 throw std::runtime_error("Can't copy to non-user memory");
378 if(this == &ls)
379 return *this;
380 if(allocated < static_cast<size_t>(ls.width) * ls.height) {
381 size_t p_allocated = static_cast<size_t>(ls.width) * ls.height;
382 memory = new uint32_t[p_allocated];
383 allocated = p_allocated;
385 width = ls.width;
386 height = ls.height;
387 pitch = width;
388 for(size_t l = 0; l < height; l++)
389 memcpy(const_cast<uint32_t*>(memory + l * width), ls.memory + l * ls.pitch, 4 * width);
390 return *this;
393 lcscreen::~lcscreen()
395 if(user_memory)
396 delete[] const_cast<uint32_t*>(memory);
399 void lcscreen::load(const std::vector<char>& data) throw(std::bad_alloc, std::runtime_error)
401 if(!user_memory)
402 throw std::runtime_error("Can't load to non-user memory");
403 const uint8_t* data2 = reinterpret_cast<const uint8_t*>(&data[0]);
404 if(data.size() < 2)
405 throw std::runtime_error("Corrupt saved screenshot data");
406 uint32_t _width = static_cast<uint32_t>(data2[0]) * 256 + static_cast<uint32_t>(data2[1]);
407 if(_width > 1 && data.size() % (3 * _width) != 2)
408 throw std::runtime_error("Corrupt saved screenshot data");
409 uint32_t _height = (data.size() - 2) / (3 * _width);
410 if(allocated < static_cast<size_t>(_width) * _height) {
411 size_t p_allocated = static_cast<size_t>(_width) * _height;
412 memory = new uint32_t[p_allocated];
413 allocated = p_allocated;
415 uint32_t* mem = const_cast<uint32_t*>(memory);
416 width = _width;
417 height = _height;
418 pitch = width;
419 for(size_t i = 0; i < (data.size() - 2) / 2; i++)
420 mem[i] = static_cast<uint32_t>(data2[2 + 3 * i]) * 65536 +
421 static_cast<uint32_t>(data2[2 + 3 * i + 1]) * 256 +
422 static_cast<uint32_t>(data2[2 + 3 * i + 2]);
425 void lcscreen::save(std::vector<char>& data) throw(std::bad_alloc)
427 data.resize(2 + 3 * static_cast<size_t>(width) * height);
428 uint8_t* data2 = reinterpret_cast<uint8_t*>(&data[0]);
429 data2[0] = (width >> 8);
430 data2[1] = width;
431 for(size_t i = 0; i < (data.size() - 2) / 3; i++) {
432 data[2 + 3 * i] = memory[(i / width) * pitch + (i % width)] >> 16;
433 data[2 + 3 * i + 1] = memory[(i / width) * pitch + (i % width)] >> 8;
434 data[2 + 3 * i + 2] = memory[(i / width) * pitch + (i % width)];
438 void lcscreen::save_png(const std::string& file) throw(std::bad_alloc, std::runtime_error)
440 uint8_t* buffer = new uint8_t[3 * static_cast<size_t>(width) * height];
441 for(uint32_t j = 0; j < height; j++)
442 for(uint32_t i = 0; i < width; i++) {
443 uint32_t word = memory[pitch * j + i];
444 uint32_t l = 1 + ((word >> 15) & 0xF);
445 uint32_t r = l * ((word >> 0) & 0x1F);
446 uint32_t g = l * ((word >> 5) & 0x1F);
447 uint32_t b = l * ((word >> 10) & 0x1F);
448 buffer[3 * static_cast<size_t>(width) * j + 3 * i + 0] = r * 255 / 496;
449 buffer[3 * static_cast<size_t>(width) * j + 3 * i + 1] = g * 255 / 496;
450 buffer[3 * static_cast<size_t>(width) * j + 3 * i + 2] = b * 255 / 496;
452 try {
453 save_png_data(file, buffer, width, height);
454 delete[] buffer;
455 } catch(...) {
456 delete[] buffer;
457 throw;
461 void screen::copy_from(lcscreen& scr, uint32_t hscale, uint32_t vscale) throw()
463 uint32_t copyable_width = (width - originx) / hscale;
464 uint32_t copyable_height = (height - originy) / vscale;
465 copyable_width = (copyable_width > scr.width) ? scr.width : copyable_width;
466 copyable_height = (copyable_height > scr.height) ? scr.height : copyable_height;
467 for(uint32_t y = 0; y < height; y++) {
468 memset(rowptr(y), 0, 4 * width);
470 for(uint32_t y = 0; y < copyable_height; y++) {
471 uint32_t line = y * vscale + originy;
472 uint32_t* ptr = rowptr(line) + originx;
473 const uint32_t* sbase = scr.memory + y * scr.pitch;
474 for(uint32_t x = 0; x < copyable_width; x++) {
475 uint32_t c = palette[sbase[x] & 0x7FFFF];
476 for(uint32_t i = 0; i < hscale; i++)
477 *(ptr++) = c;
479 for(uint32_t j = 1; j < vscale; j++)
480 memcpy(rowptr(line + j) + originx, rowptr(line) + originx, 4 * hscale * copyable_width);
484 void screen::reallocate(uint32_t _width, uint32_t _height, uint32_t _originx, uint32_t _originy, bool upside_down)
485 throw(std::bad_alloc)
487 if(_width == width && _height == height) {
488 originx = _originx;
489 originy = _originy;
490 return;
492 if(!_width || !_height) {
493 width = height = originx = originy = pitch = 0;
494 if(memory && !user_memory)
495 delete[] memory;
496 memory = NULL;
497 user_memory = false;
498 flipped = upside_down;
499 return;
501 uint32_t* newmem = new uint32_t[_width * _height];
502 width = _width;
503 height = _height;
504 originx = _originx;
505 originy = _originy;
506 pitch = 4 * _width;
507 if(memory && !user_memory)
508 delete[] memory;
509 memory = newmem;
510 user_memory = false;
511 flipped = upside_down;
514 void screen::set(uint32_t* _memory, uint32_t _width, uint32_t _height, uint32_t _originx, uint32_t _originy,
515 uint32_t _pitch) throw()
517 if(memory && !user_memory)
518 delete[] memory;
519 width = _width;
520 height = _height;
521 originx = _originx;
522 originy = _originy;
523 pitch = _pitch;
524 user_memory = true;
525 memory = _memory;
526 flipped = false;
529 uint32_t* screen::rowptr(uint32_t row) throw()
531 if(flipped)
532 row = height - row - 1;
533 return reinterpret_cast<uint32_t*>(reinterpret_cast<uint8_t*>(memory) + row * pitch);
536 screen::screen() throw()
538 memory = NULL;
539 width = height = originx = originy = pitch = 0;
540 user_memory = false;
541 flipped = false;
542 palette = NULL;
543 set_palette(16, 8, 0);
546 screen::~screen() throw()
548 if(memory && !user_memory)
549 delete[] memory;
552 void clip_range(uint32_t origin, uint32_t size, int32_t base, int32_t& minc, int32_t& maxc) throw()
554 int64_t _origin = origin;
555 int64_t _size = size;
556 int64_t _base = base;
557 int64_t _minc = minc;
558 int64_t _maxc = maxc;
559 int64_t mincoordinate = _base + _origin + _minc;
560 int64_t maxcoordinate = _base + _origin + _maxc;
561 if(mincoordinate < 0)
562 _minc = _minc - mincoordinate;
563 if(maxcoordinate > _size)
564 _maxc = _maxc - (maxcoordinate - _size);
565 if(_minc >= maxc) {
566 minc = 0;
567 maxc = 0;
568 } else {
569 minc = _minc;
570 maxc = _maxc;
574 void screen::set_palette(uint32_t r, uint32_t g, uint32_t b)
576 if(!palette)
577 palette = new uint32_t[0x80000];
578 else if(r == palette_r && g == palette_g && b == palette_b)
579 return;
580 for(size_t i = 0; i < static_cast<size_t>(width) * height; i++) {
581 uint32_t word = memory[i];
582 uint32_t R = (word >> palette_r) & 0xFF;
583 uint32_t G = (word >> palette_g) & 0xFF;
584 uint32_t B = (word >> palette_b) & 0xFF;
585 memory[i] = (R << r) | (G << g) | (B << b);
587 for(unsigned i = 0; i < 0x80000; i++) {
588 unsigned l = 1 + ((i >> 15) & 0xF);
589 unsigned R = (i >> 0) & 0x1F;
590 unsigned G = (i >> 5) & 0x1F;
591 unsigned B = (i >> 10) & 0x1F;
592 double _l = static_cast<double>(l);
593 double m = 255.0 / 496.0;
594 R = floor(m * R * _l + 0.5);
595 G = floor(m * G * _l + 0.5);
596 B = floor(m * B * _l + 0.5);
597 palette[i] = (R << r) | (G << g) | (B << b);
599 palette_r = r;
600 palette_g = g;
601 palette_b = b;