c for performance1414962500

Signed-off-by: Mattias Andrée <maandree@operamail.com>

1 files changed, 332 insertions, 0 deletions

diff --git a/src/assemble.c b/src/assemble.c
new file mode 100644
index 0000000..e08d2d3
--- /dev/null
+++ b/src/assemble.c
@@ -0,0 +1,332 @@
+/**
+ * splashtool – A simple tool for creating SYSLINUX splashes without fuss
+ * 
+ * Copyright © 2013, 2014  Mattias Andrée (maandree@member.fsf.org)
+ * 
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+#define _GNU_SOURCE
+#include <math.h>
+#include <string.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <unistd.h>
+
+
+#define WIDTH   640
+#define HEIGHT  480
+
+
+static __attribute__((const)) double linear(unsigned long c)
+{
+  if (c <= 10)
+    return (double)c / (255. * 12.92);
+  return pow(((double)c  + 14.025) / 269.025, 2.4);
+}
+
+
+static __attribute__((const)) unsigned long join(unsigned long fg, unsigned long bg, unsigned long alpha)
+{
+  double t = (double)alpha * linear(fg) + (double)(255 - alpha) * linear(bg);
+  t /= 255.;
+  if (t <= 0.00304)
+    t *= 12.92;
+  else
+    t = 1.055 * pow(t, 1.0 / 2.4) - 0.055;
+  return (unsigned long)(255. * t + 0.5);
+}
+
+
+static inline void subpixels(unsigned long p, unsigned long* a,
+			     unsigned long* r, unsigned long* g, unsigned long* b)
+{
+  *a = (p >> 24) & 255;
+  *r = (p >> 16) & 255;
+  *g = (p >>  8) & 255;
+  *b = (p >>  0) & 255;
+}
+
+
+static void buffer_join(unsigned long* restrict restrict* layer, unsigned long* restrict* restrict base,
+			unsigned long width, unsigned long height, unsigned long offx, unsigned long offy)
+{
+  unsigned long x, y, a, r, g, b;
+  unsigned long* restrict layer_column;
+  unsigned long* restrict base_column;
+  base += offx;
+  for (x = 0; x < width; x++)
+    {
+      layer_column = *layer++;
+      base_column = *base++ + offy;
+      for (y = 0; y < height; y++, base_column++)
+	{
+	  subpixels(*layer_column++, &a, &r, &g, &b);
+	  if (a == 255)
+	    *base_column = (r << 16) | (g << 8) | b;
+	}
+    }
+}
+
+
+static void widescreen(unsigned long* restrict* restrict rc, unsigned long* restrict* restrict img)
+{
+  unsigned long x, y, e, rgb;
+  unsigned long a1, r1, g1, b1, a2, r2, g2, b2;
+  for (y = 0; y < HEIGHT; y++)
+    for (x = 0, e = 0; x < WIDTH; x++)
+      if (rc[x + e][y] = img[x][y], x % 3 == 2)
+	{
+	  if (rgb = img[x][y], x < WIDTH - 1)
+	    {
+	      subpixels(rgb, &a1, &r1, &g1, &b1);
+	      subpixels(img[x + 1][y], &a2, &r2, &g2, &b2);
+	      r1 = ((r1 + r2) / 2) << 16;
+	      g1 = ((g1 + g2) / 2) << 8;
+	      b1 = ((b1 + b2) / 2) << 0;
+	      rgb = r1 | g1 | b1;
+	    }
+	  rc[x + ++e][y] = rgb;
+	}
+}
+
+
+static unsigned long* restrict* make_buffer(unsigned long n, unsigned long m)
+{
+  unsigned long i;
+  unsigned long* restrict* restrict rc = malloc(n * sizeof(unsigned long*));
+  for (i = 0; i < n; i++)
+    rc[i] = calloc(m, sizeof(unsigned long));
+  return rc;
+}
+
+
+static void free_buffer(unsigned long* restrict* restrict buffer, unsigned long n)
+{
+  unsigned long i;
+  for (i = 0; i < n; i++)
+    free(buffer[i]);
+# pragma GCC diagnostic push
+# pragma GCC diagnostic ignored "-Wcast-qual"
+  free((unsigned long**)buffer);
+# pragma GCC diagnostic pop
+}
+
+
+static unsigned long* restrict* load_p6(const char* restrict filename)
+{
+  unsigned char* restrict data = malloc((size_t)(WIDTH * HEIGHT * 3) * sizeof(unsigned char));
+  unsigned long* restrict* restrict rc = make_buffer(WIDTH, HEIGHT);
+  unsigned long y, x, p = 0;
+  ssize_t got;
+  int fd;
+  
+  fd = open(filename, O_RDONLY);
+  while (p < (unsigned long)(WIDTH * HEIGHT * 3))
+    {
+      got = read(fd, data + p, ((unsigned long)(WIDTH * HEIGHT * 3) - p) * sizeof(char));
+      if (got <= 0)
+	if ((got == 0) || (errno != EINTR))
+	  abort();
+      if (got > 0)
+	p += (unsigned long)got;
+    }
+  close(fd);
+  
+  for (y = 0; y < HEIGHT; y++)
+    for (x = 0; x < WIDTH; x++)
+      {
+	unsigned long r = (unsigned long)(*data++);
+	unsigned long g = (unsigned long)(*data++);
+	unsigned long b = (unsigned long)(*data++);
+	rc[x][y] = (r << 16) | (g << 8) | b;
+      }
+
+  free(data - WIDTH * HEIGHT * 3);
+  return rc;
+}
+
+
+static char* next_line(void)
+{
+  static char buf[512];
+  unsigned long ptr = 0;
+  char c;
+  while (c = (char)getchar(), c != '\n')
+    buf[ptr++] = c;
+  buf[ptr] = '\0';
+  return buf;
+}
+
+
+int main(int argc, char** argv)
+{
+  unsigned long y, x, i, n, ci, yi, xi, offx, offy, width, height;
+  unsigned long back = 0, fore = 0, chars, chary, charx;
+  unsigned long long foreback = 0;
+  unsigned long* restrict chr;
+  char* restrict line;
+  unsigned long* restrict* restrict charmap;
+  char* restrict splash_filename;
+  unsigned long* restrict* restrict splash;
+  unsigned long* restrict* restrict background;
+  unsigned long* restrict* restrict foreground;
+  unsigned long* restrict* restrict shadow;
+  char** restrict rows;
+  
+  /* Read image data. */
+  splash_filename = strdup(next_line());
+  width = (unsigned long)atol(next_line());
+  height = (unsigned long)atol(next_line());
+  rows = malloc(height * sizeof(char*));
+  for (i = 0; i < height; i++)
+    rows[i] = strdup(next_line());
+  
+  /* Read font data. */
+  chars = (unsigned long)atol(next_line());
+  charx = (unsigned long)atol(next_line());
+  chary = (unsigned long)atol(next_line());
+  charmap = malloc(chars * sizeof(unsigned long*));
+  for (i = 0; i < chars; i++)
+    {
+      charmap[i] = calloc(chary, sizeof(unsigned long));
+      for (y = 0; y < chary; y++)
+	for (line = next_line(), x = 0; x < charx; x++)
+	  charmap[i][y] |= (line[x] == ' ' ? 0UL : 1UL) << x;
+    }
+  
+  /* The overlay should be centered on the background. */
+  offx = (WIDTH - width * charx) / 2;
+  offy = (HEIGHT - height * chary) / 2;
+  
+  /* Buffers for layers. */
+  splash     = load_p6(splash_filename);
+  background = make_buffer(width * charx, height * chary);
+  foreground = make_buffer(width * charx, height * chary);
+  shadow     = make_buffer(width * charx, height * chary);
+  free(splash_filename);
+  
+  /* Fill largers. */
+  for (y = 0; y < height; y++)
+    {
+      char* restrict row = rows[y];
+      char escape = 0;
+      for (i = 0, x = 0, n = strlen(row); i < n; i++)
+	{
+	  char c = row[i];
+	  if (c == '\033')
+	    {
+	      fore = (unsigned long)(foreback >> 32);
+	      back = (unsigned long)(foreback & ((1LL << 32) - 1LL));
+	      escape ^= 1;
+	      foreback = 0;
+	    }
+	  else if (escape)
+	    {
+	      if (c == '#')
+		continue;
+	      foreback <<= 4;
+	      foreback |= (unsigned long long)((c & 15) + (c <= '9' ? 0 : 9));
+	    }
+	  else
+	    {
+	      if (ci = (unsigned long)(unsigned char)c % chars, c & 0x80)
+		{
+		  char* restrict s = row + i;
+		  i += 2;
+		  if      (strstr(s, "┌") == s)  ci = 218;
+		  else if (strstr(s, "─") == s)  ci = 196;
+		  else if (strstr(s, "┐") == s)  ci = 191;
+		  else if (strstr(s, "│") == s)  ci = 179;
+		  else if (strstr(s, "├") == s)  ci = 195;
+		  else if (strstr(s, "┤") == s)  ci = 180;
+		  else if (strstr(s, "└") == s)  ci = 192;
+		  else if (strstr(s, "┘") == s)  ci = 217;
+		  else
+		    i -= 2;
+		}
+	      chr = charmap[ci];
+	      for (yi = 0; yi < chary; yi++)
+		for (xi = 0; xi < charx; xi++)
+		  {
+		    unsigned long xo = (chr[yi] >> xi) & 1;
+		    unsigned long t = offx + x * charx + xi + xo;
+		    unsigned long rgb = splash[t < WIDTH ? t : (WIDTH - 1)][offy + y * chary + yi];
+		    unsigned long _a, r, g, b, ca, cr, cg, cb, c_;
+		    unsigned long* restrict* restrict i_;
+		    subpixels(rgb, &_a, &r, &g, &b);
+		    if (xo)
+		      shadow[x * charx + xi][y * chary + yi] = 128UL << 24;
+		    c_ = xo == 1 ? fore : back;
+		    i_ = xo == 1 ? foreground : background;
+		    subpixels(c_, &ca, &cr, &cg, &cb);
+		    r = join(cr, r, ca);
+		    g = join(cg, g, ca);
+		    b = join(cb, b, ca);
+		    rgb = (255UL << 24) | (r << 16) | (g << 8) | b;
+		    i_[x * charx + xi][y * chary + yi] = rgb;
+		    (void) _a;
+		  }
+	      x++;
+	    }
+	}
+    }
+  
+  /* Apply layers. */
+  width *= charx;
+  buffer_join(background, splash, width, height * chary, offx, offy);
+  buffer_join(shadow,     splash, width, height * chary, offx, offy);
+  buffer_join(foreground, splash, width, height * chary, offx, offy);
+  
+  free_buffer(background, width);
+  free_buffer(foreground, width);
+  free_buffer(shadow, width);
+  
+  /* Make widescreen preview. */
+  if (strchr(argv[2], 'W'))
+    *strchr(argv[2], 'W') = 'w';
+  if ((strstr(argv[2], "-w") == argv[2]) || (strstr(argv[2], "--w") == argv[2]))
+    {
+      unsigned long* restrict* restrict original = splash;
+      width = HEIGHT * 16 / 9;
+      splash = make_buffer(width, HEIGHT);
+      widescreen(splash, original);
+      free_buffer(original, WIDTH);
+    }
+  else
+    width = WIDTH;
+  
+  /* Print image. */
+  printf("P6\n%lu %i\n255\n", width, HEIGHT);
+  for (y = 0; y < HEIGHT; y++)
+    for (x = 0; x < width; x++)
+      {
+	unsigned long _a, r, g, b;
+	subpixels(splash[x][y], &_a, &r, &g, &b);
+	putchar_unlocked((int)r);
+	putchar_unlocked((int)g);
+	putchar_unlocked((int)b);
+      }
+  fflush(stdout);
+  
+  for (i = 0; i < height; i++)
+    free(rows[i]);
+  free(rows);
+  free_buffer(charmap, chars);
+  free_buffer(splash, width);
+  return 0;
+  (void) argc;
+}
+