txtwin.c

   1 #include <ctype.h>
   2 #include <errno.h>
   3 #include <fcntl.h>
   4 #include <locale.h>
   5 #include <poll.h>
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <sys/wait.h>
  10 #include <pty.h>
  11
  12 #include <glib.h>
  13 #include <pango/pango.h>
  14 #include <pango/pangoft2.h>
  15
  16 #include "draw.h"
  17 #include "util.h"
  18
  19 #define MAXTEXT                 (1 << 20)
  20 #define DPI                     196
  21 #define PAGESTEPS               8
  22 #define CTRLKEY(x)              ((x) - 96)
  23
  24 struct txtwin {
  25         int n;
  26 };
  27
  28 static struct {
  29         char name[1 << 4];
  30         char desc[1 << 8];
  31 } tags[1 << 4] = {
  32         {"default", "<span size=\"xx-small\">"}
  33 };
  34 static int ntags = 1;
  35
  36 static char name[128];
  37 static PangoContext *context;
  38 static int width = 1;
  39 static int height = 1;
  40 static int head;
  41 static PangoFontMap *fontmap;
  42 static FT_Bitmap *bitmap;
  43 static char text[MAXTEXT];
  44 static char *pos = text;
  45 static int count;
  46
  47 static FT_Bitmap *create_bitmap(int width, int height)
  48 {
  49         FT_Bitmap *bitmap = g_slice_new(FT_Bitmap);
  50         bitmap->width = width;
  51         bitmap->pitch = (bitmap->width + 3) & ~3;
  52         bitmap->rows = height;
  53         bitmap->buffer = g_malloc(bitmap->pitch * bitmap->rows);
  54         bitmap->num_grays = 256;
  55         bitmap->pixel_mode = ft_pixel_mode_grays;
  56         memset(bitmap->buffer, 0, bitmap->pitch * bitmap->rows);
  57         return bitmap;
  58 }
  59
  60 static void destroy_bitmap(FT_Bitmap *bitmap)
  61 {
  62         g_free(bitmap->buffer);
  63         g_slice_free(FT_Bitmap, bitmap);
  64 }
  65
  66 #define COLORMERGE(f, b, c)             ((b) + (((f) - (b)) * (c) >> 8u))
  67
  68 static void view_write(int sr)
  69 {
  70         int r, c;
  71         fbval_t buf[1 << 14];
  72         int nr = MIN(height - sr, fb_rows());
  73         int nc = MIN(width, fb_cols());
  74         for (r = 0; r < nr; r++) {
  75                 for (c = 0; c < nc; c++) {
  76                         int v = bitmap->buffer[(sr + r) * bitmap->pitch + c];
  77                         int r = COLORMERGE(0xd5, 0x00, 255 - v);
  78                         int g = COLORMERGE(0xdd, 0x00, 255 - v);
  79                         int b = COLORMERGE(0xff, 0x00, 255 - v);
  80                         buf[c] = fb_color(r, g, b);
  81                 }
  82                 fb_set(r, 0, buf, width);
  83         }
  84 }
  85
  86 static PangoLayout *make_layout(PangoContext *context, char *text, double size)
  87 {
  88         static PangoFontDescription *font_description;
  89         PangoLayout *layout = pango_layout_new(context);
  90         pango_layout_set_markup(layout, text, -1);
  91         pango_layout_set_wrap(layout, PANGO_WRAP_WORD_CHAR);
  92         pango_layout_set_width(layout, fb_cols() << 10);
  93         pango_layout_set_height(layout, -1);
  94         pango_layout_set_alignment(layout, PANGO_ALIGN_RIGHT);
  95         pango_layout_set_font_description(layout, font_description);
  96         pango_font_description_free(font_description);
  97         return layout;
  98 }
  99
 100 static void do_output(PangoContext *context, char *text)
 101 {
 102         PangoLayout *layout;
 103         PangoRectangle logical_rect;
 104         pango_context_set_language(context, pango_language_get_default());
 105         pango_context_set_base_dir(context, PANGO_DIRECTION_RTL);
 106         layout = make_layout(context, text, -1);
 107         pango_layout_get_pixel_extents(layout, NULL, &logical_rect);
 108         pango_ft2_render_layout(bitmap, layout, 0, 0);
 109         width = MAX(width, MAX(logical_rect.x + logical_rect.width,
 110                         PANGO_PIXELS(pango_layout_get_width(layout))));
 111         height = MAX(logical_rect.y + logical_rect.height,
 112                 PANGO_PIXELS(pango_layout_get_height(layout)));
 113         g_object_unref(layout);
 114 }
 115
 116 static int readkey(void)
 117 {
 118         char b;
 119         if (read(STDIN_FILENO, &b, 1) <= 0)
 120                 return -1;
 121         return b;
 122 }
 123
 124 void txtwin_append(char *s, char *tag)
 125 {
 126         int i;
 127         int t = 0;
 128         for (i = 0; i < ntags; i++)
 129                 if (tag && !strcmp(tag, tags[i].name))
 130                         t = i;
 131         snprintf(pos, MAXTEXT - (pos - text),
 132                  "%s%s</span>", tags[t].desc, s);
 133         pos = strchr(pos, '\0');
 134 }
 135
 136 void txtwin_tag(char *name, char *fg, char *bg, char *font)
 137 {
 138         strcpy(tags[ntags].name, name);
 139         snprintf(tags[ntags].desc, sizeof(tags[ntags].desc),
 140                 "<span font=\"%s\">", font);
 141         ntags++;
 142 }
 143
 144 static int getcount(int def)
 145 {
 146         int result = count ? count : def;
 147         count = 0;
 148         return result;
 149 }
 150
 151 static void printinfo(void)
 152 {
 153         printf("\x1b[H");
 154         printf("CNOOR \t\t\t %d%% \t\t\t %s \x1b[K",
 155                 head * 100 / height, name);
 156         fflush(stdout);
 157 }
 158
 159 void txtwin_loop(void)
 160 {
 161         int step = fb_rows() / PAGESTEPS;
 162         int c;
 163         struct termios oldtermios, termios;
 164         tcgetattr(STDIN_FILENO, &termios);
 165         oldtermios = termios;
 166         cfmakeraw(&termios);
 167         tcsetattr(STDIN_FILENO, TCSAFLUSH, &termios);
 168         bitmap = create_bitmap(width, height);
 169         do_output(context, text);
 170         destroy_bitmap(bitmap);
 171         bitmap = create_bitmap(width, height);
 172         do_output(context, text);
 173         view_write(head);
 174         while ((c = readkey()) != -1) {
 175                 switch (c) {
 176                 case 27:
 177                         count = 0;
 178                         break;
 179                 case 'q':
 180                         tcsetattr(STDIN_FILENO, 0, &oldtermios);
 181                         return;
 182                 default:
 183                         if (isdigit(c))
 184                                 count = count * 10 + c - '0';
 185                 }
 186                 switch (c) {
 187                 case 'j':
 188                         head += step * getcount(1);
 189                         break;
 190                 case 'k':
 191                         head -= step * getcount(1);
 192                         break;
 193                 case CTRLKEY('f'):
 194                 case ' ':
 195                         head += fb_rows() * getcount(1) - step;
 196                         break;
 197                 case CTRLKEY('b'):
 198                 case 127:
 199                         head -= fb_rows() * getcount(1) - step;
 200                         break;
 201                 case CTRLKEY('l'):
 202                         break;
 203                 case 'G':
 204                         if (getcount(0))
 205                                 head = 0;
 206                         else
 207                                 head = height - fb_rows();
 208                         break;
 209                 case '%':
 210                         if (count)
 211                                 head = height * getcount(0) / 100;
 212                         break;
 213                 case 'i':
 214                         printinfo();
 215                         continue;
 216                 default:
 217                         /* no need to redraw */
 218                         continue;
 219                 }
 220                 head = MAX(0, MIN(head, height - fb_rows()));
 221                 view_write(head);
 222         }
 223 }
 224
 225 void txtwin_gtk_init(int argc, char **argv)
 226 {
 227         g_type_init();
 228         g_set_prgname("cpan");
 229         setlocale(LC_ALL, "");
 230 }
 231
 232 void txtwin_init(char *sura)
 233 {
 234         char *clear = "\x1b[2J\x1b[H";
 235         char *hide = "\x1b[?25l";
 236         fontmap = pango_ft2_font_map_new();
 237         pango_ft2_font_map_set_resolution(PANGO_FT2_FONT_MAP(fontmap), DPI, DPI);
 238         context = pango_font_map_create_context(PANGO_FONT_MAP(fontmap));
 239         strcpy(name, sura);
 240         fb_init();
 241         write(STDIN_FILENO, clear, strlen(clear));
 242         write(STDIN_FILENO, hide, strlen(hide));
 243         printinfo();
 244 }
 245
 246 void txtwin_free(void)
 247 {
 248         char *show = "\x1b[?25h\n";
 249         fb_free();
 250         destroy_bitmap(bitmap);
 251         g_object_unref(context);
 252         g_object_unref(fontmap);
 253         write(STDIN_FILENO, show, strlen(show));
 254 }