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