sym-text.c

   1 /*
   2  *      Hic Est Leo -- Text Symbolizer
   3  *
   4  *      (c) 2014 Martin Mares <mj@ucw.cz>
   5  */
   6
   7 #include <ucw/lib.h>
   8 #include <ucw/stkstring.h>
   9
  10 #include <stdio.h>
  11 #include <ft2build.h>
  12 #include FT_FREETYPE_H
  13 #include <pango/pangoft2.h>
  14
  15 #include "leo.h"
  16 #include "osm.h"
  17 #include "sym.h"
  18 #include "map.h"
  19
  20 #include "labeller.h"
  21
  22 /*** Fonts ***/
  23
  24 struct text_font {
  25   const char *family;
  26   double size;
  27   osm_val_t weight;
  28   osm_val_t style;
  29   PangoFontDescription *pango_font_desc;
  30   char key[1];
  31 };
  32
  33 #define HASH_NODE struct text_font
  34 #define HASH_PREFIX(x) text_font_##x
  35 #define HASH_KEY_ENDSTRING key
  36 #define HASH_WANT_LOOKUP
  37 #define HASH_LOOKUP_DETECT_NEW
  38 #define HASH_USE_POOL sym_mp
  39 #include <ucw/hashtable.h>
  40
  41 static PangoFontMap *pango_font_map;
  42 static PangoContext *pango_context;
  43
  44 static double pt_to_mm(double pt)
  45 {
  46   return pt / 72 * 25.4;
  47 }
  48
  49 static double mm_to_pt(double mm)
  50 {
  51   return mm / 25.4 * 72;
  52 }
  53
  54 static double pango_to_mm(double pango)
  55 {
  56   return pt_to_mm(pango / PANGO_SCALE);
  57 }
  58
  59 static double mm_to_pango(double mm)
  60 {
  61   return mm_to_pt(mm) * PANGO_SCALE;
  62 }
  63
  64 /*
  65  * Pango tries to do pixel-based optimizations in low resolutions.
  66  * Work around it by scaling everything up.
  67  */
  68 #define FONT_HACK_FACTOR 30
  69
  70 static struct text_font *font_get(struct text_font *req)
  71 {
  72   char *key = stk_printf("%s:%.6g:%u:%u", req->family, req->size, req->weight, req->style);
  73   int is_new = 0;
  74   struct text_font *font = text_font_lookup(key, &is_new);
  75   if (is_new)
  76     {
  77       msg(L_DEBUG, "Loading font %s (size %.6g, weight %s, style %s)", req->family, req->size, osm_val_decode(req->weight), osm_val_decode(req->style));
  78       font->family = mp_strdup(sym_mp, req->family);
  79       font->size = req->size;
  80       font->weight = req->weight;
  81       font->style = req->style;
  82
  83       PangoFontDescription *desc;
  84       desc = pango_font_description_new();
  85       ASSERT(desc);
  86       pango_font_description_set_family(desc, font->family);
  87       if (font->weight != VALUE_NORMAL)
  88         pango_font_description_set_weight(desc, PANGO_WEIGHT_BOLD);
  89       if (font->style != VALUE_NORMAL)
  90         pango_font_description_set_style(desc, PANGO_STYLE_ITALIC);
  91       pango_font_description_set_size(desc, mm_to_pango(font->size));
  92       font->pango_font_desc = desc;
  93
  94 #if 1
  95       // FIXME
  96       PangoFont *pfont = pango_font_map_load_font(pango_font_map, pango_context, desc);
  97       ASSERT(pfont);
  98       PangoFontDescription *d2 = pango_font_describe(pfont);
  99       ASSERT(d2);
 100       msg(L_DEBUG, "Font desc: %s", pango_font_description_to_string(d2));
 101
 102       PangoFontMetrics *fm = pango_font_get_metrics(pfont, NULL);
 103       ASSERT(fm);
 104       msg(L_DEBUG, "Font metrics: asc=%.6g desc=%.6g", pango_to_mm(pango_font_metrics_get_ascent(fm)), pango_to_mm(pango_font_metrics_get_descent(fm)));
 105 #endif
 106     }
 107   return font;
 108 }
 109
 110 static void font_init(void)
 111 {
 112   text_font_init();
 113
 114   pango_font_map = pango_ft2_font_map_new();
 115   ASSERT(pango_font_map);
 116   pango_ft2_font_map_set_resolution((PangoFT2FontMap *) pango_font_map, 72 * FONT_HACK_FACTOR, 72 * FONT_HACK_FACTOR);
 117   pango_context = pango_font_map_create_context(PANGO_FONT_MAP(pango_font_map));
 118   ASSERT(pango_context);
 119 }
 120
 121 static void text_size(struct sym_text *st)
 122 {
 123   PangoLayout *layout = pango_layout_new(pango_context);
 124   pango_layout_set_font_description(layout, st->font->pango_font_desc);
 125   pango_layout_set_text(layout, osm_val_decode(st->text), -1);
 126   pango_layout_context_changed(layout);
 127
 128   PangoRectangle ext;
 129   pango_layout_get_extents(layout, NULL, &ext);
 130   // st->tx = pango_to_mm(ext.x) / FONT_HACK_FACTOR;
 131   // st->ty = pango_to_mm(ext.y) / FONT_HACK_FACTOR;
 132   st->tw = pango_to_mm(ext.width) / FONT_HACK_FACTOR;
 133   st->th = pango_to_mm(pango_layout_get_baseline(layout)) / FONT_HACK_FACTOR;
 134   st->td = pango_to_mm(ext.height) / FONT_HACK_FACTOR - st->th;
 135
 136   g_object_unref(layout);
 137 }
 138
 139 /*** Elimination of duplicate texts ***/
 140
 141 // FIXME: Get rid of globals
 142 #define DUP_TILE_SIZE 10
 143 static int dup_tiles_w, dup_tiles_h;
 144 static struct sym_text **dup_tiles;
 145
 146 static void text_dup_init(void)
 147 {
 148   dup_tiles_w = 1 + page_map_width / DUP_TILE_SIZE;
 149   dup_tiles_h = 1 + page_map_height / DUP_TILE_SIZE;
 150   dup_tiles = xmalloc_zero(sizeof(struct sym_text *) * dup_tiles_w * dup_tiles_h);
 151   msg(L_DEBUG, "Allocated text tiles: %u x %u", dup_tiles_w, dup_tiles_h);
 152 }
 153
 154 static int text_dup_coord(int x, int y)
 155 {
 156   ASSERT(x >= 0 && x < dup_tiles_w && y >= 0 && y < dup_tiles_h);
 157   return x + y * dup_tiles_w;
 158 }
 159
 160 static double text_quad_dist(struct sym_text *a, struct sym_text *b)
 161 {
 162   double dx = a->x - b->x;
 163   double dy = a->y - b->y;
 164   return dx*dx + dy*dy;
 165 }
 166
 167 static bool text_dup_detect(struct sym_text *t, struct style_info *si)
 168 {
 169   // Out-of-frame texts are dropped immediately
 170   double x = t->x - page_offset_x;
 171   double y = t->y - page_offset_y;
 172   if (x < 0 || x >= page_map_width ||
 173       y < 0 || y >= page_map_height)
 174         return 0;
 175
 176   int tile_x = x / DUP_TILE_SIZE;
 177   int tile_y = y / DUP_TILE_SIZE;
 178
 179   double radius = 0;
 180   style_get_number(si, PROP_TEXT_DUP_THRESHOLD, &radius);
 181   if (radius)
 182     {
 183       // A rather simple-minded algorithm, but believed to be efficient enough.
 184       int r_tiles = 1 + radius / DUP_TILE_SIZE;
 185       int x_start = MAX(0, tile_x - r_tiles);
 186       int x_stop = MIN(dup_tiles_w - 1, tile_x + r_tiles);
 187       int y_start = MAX(0, tile_y - r_tiles);
 188       int y_stop = MIN(dup_tiles_h - 1, tile_y + r_tiles);
 189       for (int x = x_start; x <= x_stop; x++)
 190         for (int y = y_start; y <= y_stop; y++)
 191           {
 192             for (struct sym_text *d = dup_tiles[text_dup_coord(x, y)]; d; d = d->next_in_tile)
 193               if (d->text == t->text &&
 194                   text_quad_dist(d, t) <= radius*radius &&
 195                   d->text_color == t->text_color &&
 196                   d->font == t->font)
 197                 {
 198                   t->next_duplicate = d->next_duplicate;
 199                   d->next_duplicate = t;
 200                   return 0;
 201                 }
 202           }
 203     }
 204
 205   int tile_i = text_dup_coord(tile_x, tile_y);
 206   t->next_in_tile = dup_tiles[tile_i];
 207   dup_tiles[tile_i] = t;
 208   return 1;
 209 }
 210
 211 /*** Core of the symbolizer ***/
 212
 213 static void prepare_text_element(struct sym_text *t, struct svg *svg)
 214 {
 215   struct text_font *font = t->font;
 216   svg_push_element(svg, "text");
 217   svg_set_attr_dimen(svg, "x", t->x);
 218   svg_set_attr_dimen(svg, "y", t->y);
 219   svg_set_attr(svg, "font-family", font->family);
 220   svg_set_attr_dimen(svg, "font-size", font->size);
 221   if (font->weight != VALUE_NORMAL)
 222     svg_set_attr(svg, "font-weight", osm_val_decode(font->weight));
 223   if (font->style != VALUE_NORMAL)
 224     svg_set_attr(svg, "font-style", osm_val_decode(font->style));
 225 }
 226
 227 static void sym_text_draw(struct symbol *sym, struct svg *svg)
 228 {
 229   struct sym_text *t = (struct sym_text *) sym;
 230 printf("Drawing %s at [%.2f; %.2f]\n", osm_val_decode(t->text), t->x, t->y);
 231
 232   if (t->next_duplicate)
 233     {
 234       // If there is a cluster of duplicate texts, average their positions
 235       double sx = 0, sy = 0;
 236       uns nn = 0;
 237       for (struct sym_text *u = t; u; u = u->next_duplicate)
 238         {
 239           sx += u->x;
 240           sy += u->y;
 241           nn++;
 242         }
 243       t->x = sx / nn;
 244       t->y = sy / nn;
 245     }
 246
 247   if (t->opacity != 1)
 248     {
 249       svg_push_element(svg, "g");
 250       svg_set_attr_float(svg, "opacity", t->opacity);
 251     }
 252
 253   if (t->halo_radius)
 254     {
 255       prepare_text_element(t, svg);
 256       svg_set_attr(svg, "fill", "none");
 257       svg_set_attr_color(svg, "stroke", t->halo_color);
 258       svg_set_attr_dimen(svg, "stroke-width", t->halo_radius);
 259       svg_set_attr(svg, "stroke-linecap", "round");
 260       svg_set_attr(svg, "stroke-linejoin", "round");
 261       if (t->halo_opacity != 1)
 262         svg_set_attr_float(svg, "stroke-opacity", t->halo_opacity);
 263       svg_push_chars(svg)->name = osm_val_decode(t->text);
 264       svg_pop(svg);
 265       svg_pop(svg);
 266     }
 267
 268   prepare_text_element(t, svg);
 269   svg_set_attr_color(svg, "fill", t->text_color);
 270   svg_set_attr(svg, "stroke", "none");
 271   svg_push_chars(svg)->name = osm_val_decode(t->text);
 272   svg_pop(svg);
 273   svg_pop(svg);
 274
 275 #if 0
 276   // Draw bounding box for debugging
 277   svg_push_element(svg, "rect");
 278   svg_set_attr(svg, "fill", "none");
 279   svg_set_attr_color(svg, "stroke", 0x0000ff);
 280   svg_set_attr_dimen(svg, "stroke-width", 0.2);
 281   svg_set_attr_dimen(svg, "x", t->x);
 282   svg_set_attr_dimen(svg, "y", t->y - t->th);
 283   svg_set_attr_dimen(svg, "width", t->tw);
 284   svg_set_attr_dimen(svg, "height", t->th + t->td);
 285   svg_pop(svg);
 286
 287   svg_push_element(svg, "line");
 288   svg_set_attr(svg, "fill", "none");
 289   svg_set_attr_color(svg, "stroke", 0x0000ff);
 290   svg_set_attr_dimen(svg, "stroke-width", 0.2);
 291   svg_set_attr_dimen(svg, "x1", t->x);
 292   svg_set_attr_dimen(svg, "y1", t->y);
 293   svg_set_attr_dimen(svg, "x2", t->x + t->tw);
 294   svg_set_attr_dimen(svg, "y2", t->y);
 295   svg_pop(svg);
 296 #endif
 297
 298   if (t->opacity != 1)
 299     svg_pop(svg);
 300 }
 301
 302 static osm_val_t get_text(struct osm_object *o, struct style_info *si)
 303 {
 304   struct style_prop *prop = style_get_and_check(si, PROP_TEXT, (1 << PROP_TYPE_STRING) | (1 << PROP_TYPE_IDENT));
 305   if (!prop)
 306     return 0;
 307
 308   if (prop->type == PROP_TYPE_IDENT && prop->val.id == VALUE_AUTO)
 309     {
 310       static const osm_key_t auto_text_keys[] = {
 311         KEY_NAME_CZ,            // FIXME: This should be configurable
 312         KEY_NAME,
 313         KEY_REF,
 314         KEY_OPERATOR,
 315         KEY_BRAND,
 316         KEY_ADDR_HOUSENUMBER,
 317       };
 318       for (uns i=0; i < ARRAY_SIZE(auto_text_keys); i++)
 319         {
 320           osm_val_t val = osm_obj_find_tag(o, auto_text_keys[i]);
 321           if (val)
 322             return val;
 323         }
 324       return 0;
 325     }
 326
 327   return osm_obj_find_tag(o, osm_key_encode(osm_val_decode(prop->val.id)));
 328 }
 329
 330 static void get_text_attrs(struct sym_text *st, struct style_info *si)
 331 {
 332   struct osm_object *o = st->s.o;
 333   if (o->type == OSM_TYPE_WAY && osm_way_cyclic_p((struct osm_way *) o))
 334     st->text_color = 0xc0c0c0;  // FIXME: This is an ugly hack, do we need it?
 335   else
 336     st->text_color = 0xffffff;
 337   style_get_color(si, PROP_TEXT_COLOR, &st->text_color);
 338
 339   struct text_font f = {
 340     .family = "Helvetica",
 341     .size = pt_to_mm(8),
 342   };
 343   osm_val_t fam = style_get_string(si, PROP_FONT_FAMILY);
 344   if (fam)
 345     f.family = osm_val_decode(fam);
 346   style_get_number(si, PROP_FONT_SIZE, &f.size);
 347   f.weight = style_get_ident(si, PROP_FONT_WEIGHT);
 348   if (!f.weight)
 349     f.weight = VALUE_NORMAL;
 350   if (f.weight != VALUE_NORMAL && f.weight != VALUE_BOLD)
 351     {
 352       osm_obj_warn(o, "Unknown font-weight %s", osm_val_decode(f.weight));
 353       f.weight = VALUE_NORMAL;
 354     }
 355   f.style = style_get_ident(si, PROP_FONT_STYLE);
 356   if (!f.style)
 357     f.style = VALUE_NORMAL;
 358   if (f.style != VALUE_NORMAL && f.style != VALUE_ITALIC)
 359     {
 360       osm_obj_warn(o, "Unknown font-style %s", osm_val_decode(f.weight));
 361       f.style = VALUE_NORMAL;
 362     }
 363   st->font = font_get(&f);
 364
 365   st->opacity = 1;
 366   style_get_number(si, PROP_TEXT_OPACITY, &st->opacity);
 367
 368   st->halo_color = st->text_color ^ 0xffffff;
 369   style_get_color(si, PROP_TEXT_HALO_COLOR, &st->halo_color);
 370   st->halo_radius = 0;
 371   style_get_number(si, PROP_TEXT_HALO_RADIUS, &st->halo_radius);
 372   st->halo_opacity = 1;
 373   style_get_number(si, PROP_TEXT_HALO_OPACITY, &st->halo_opacity);
 374
 375   double dx = 0, dy = 0;
 376   style_get_number(si, PROP_TEXT_OFFSET_X, &dx);
 377   style_get_number(si, PROP_TEXT_OFFSET_Y, &dy);
 378   style_get_number(si, PROP_TEXT_OFFSET, &dy);
 379   st->x += dx;
 380   st->y -= dy;
 381 }
 382
 383 static void text_fix_placement(struct sym_text *st)
 384 {
 385   // Fix texts which do not fit on the paper
 386   st->x = MIN(st->x, page_offset_x + page_map_width - st->tw);
 387   st->x = MAX(st->x, page_offset_x);
 388   st->y = MIN(st->y, page_offset_y + page_map_height - st->th);
 389   st->y = MAX(st->y, page_offset_y + st->td);
 390 }
 391
 392 static void sym_text_node(struct osm_object *o, struct style_info *si, osm_val_t text)
 393 {
 394   struct osm_node *n = (struct osm_node *) o;
 395
 396   struct sym_text *st = sym_text_new(o);
 397   st->text = text;
 398   st->x = n->x;
 399   st->y = n->y;
 400
 401   get_text_attrs(st, si);
 402   text_size(st);
 403
 404   osm_val_t ah = style_get_ident(si, PROP_TEXT_ANCHOR_HORIZONTAL);
 405   switch (ah)
 406     {
 407     case VALUE_LEFT:
 408       st->x -= st->tw;
 409       break;
 410     case VALUE_CENTER:
 411       st->x -= st->tw / 2;
 412       break;
 413     case 0:
 414     case VALUE_RIGHT:
 415       break;
 416     default:
 417       osm_obj_warn(o, "Unknown text-anchor-horizontal: %s", osm_val_decode(ah));
 418     }
 419
 420   osm_val_t av = style_get_ident(si, PROP_TEXT_ANCHOR_VERTICAL);
 421   switch (av)
 422     {
 423     case VALUE_ABOVE:           // FIXME: What's the difference between above and top?
 424     case VALUE_TOP:
 425       st->y -= st->td;
 426       break;
 427     case VALUE_CENTER:
 428       st->y -= (st->th + st->td) / 2;
 429       // Fall thru
 430     case 0:
 431     case VALUE_BOTTOM:
 432     case VALUE_BELOW:
 433       st->y += st->th;
 434       break;
 435     default:
 436       osm_obj_warn(o, "Unknown text-anchor-vertical: %s", osm_val_decode(av));
 437     }
 438
 439   text_fix_placement(st);
 440   if (!text_dup_detect(st, si))
 441     {
 442       msg(L_DEBUG, "Text <%s> dropped as duplicate", osm_val_decode(text));
 443       return;
 444     }
 445
 446   //sym_plan(&st->s, sym_zindex(o, si, 5));
 447 }
 448
 449 static void sym_text_center(struct osm_object *o, struct style_info *si, osm_val_t text, double x, double y)
 450 {
 451   struct sym_text *st = sym_text_new(o);
 452   st->text = text;
 453   st->x = x;
 454   st->y = y;
 455
 456   get_text_attrs(st, si);
 457   text_size(st);
 458   st->x -= st->tw / 2;
 459   st->y += st->th - (st->th + st->td) / 2;
 460   text_fix_placement(st);
 461   if (o->type == OSM_TYPE_WAY && !osm_way_cyclic_p((struct osm_way *) o))
 462   {
 463     //sym_plan(&st->s, sym_zindex(o, si, 4.9));
 464     printf("[Sym] Labelling way %ju with %s\n", o->id, osm_val_decode(st->text));
 465     labeller_add_linelabel(&st->s, o, sym_zindex(o, si, 4.9));
 466   }
 467   else
 468   {
 469     //sym_plan(&st->s, sym_zindex(o, si, 4.9));
 470     printf("[Sym] Labelling area %ju with %s\n", o->id, osm_val_decode(st->text));
 471     labeller_add_arealabel(&st->s, o, sym_zindex(o, si, 4.9));
 472   }
 473 }
 474
 475 static void sym_text_way(struct osm_object *o, struct style_info *si, osm_val_t text)
 476 {
 477   double x, y;
 478   osm_val_t tp = style_get_ident(si, PROP_TEXT_POSITION);
 479
 480   switch (tp)
 481     {
 482     case VALUE_CENTER:
 483       if (osm_obj_center(o, &x, &y))
 484         sym_text_center(o, si, text, x, y);
 485       break;
 486     case VALUE_LINE:
 487       // FIXME
 488     default:
 489       osm_obj_warn(o, "Unknown text-position: %s", osm_val_decode(tp));
 490     }
 491 }
 492
 493 static void sym_text_mpg(struct osm_object *o, struct style_info *si, osm_val_t text)
 494 {
 495   double x, y;
 496   osm_val_t tp = style_get_ident(si, PROP_TEXT_POSITION);
 497
 498   switch (tp)
 499     {
 500     case VALUE_CENTER:
 501       if (osm_obj_center(o, &x, &y))
 502         sym_text_center(o, si, text, x, y);
 503       break;
 504     case VALUE_LINE:
 505       // FIXME
 506     default:
 507       osm_obj_warn(o, "Unknown text-position: %s", osm_val_decode(tp));
 508     }
 509 }
 510
 511 static void sym_text_gen(struct osm_object *o, struct style_info *si, struct svg *svg UNUSED)
 512 {
 513   osm_val_t text = get_text(o, si);
 514   if (!text)
 515     return;
 516
 517   switch (o->type)
 518     {
 519     case OSM_TYPE_NODE:
 520       sym_text_node(o, si, text);
 521       break;
 522     case OSM_TYPE_WAY:
 523       sym_text_way(o, si, text);
 524       break;
 525     case OSM_TYPE_MULTIPOLYGON:
 526       sym_text_mpg(o, si, text);
 527       break;
 528     default:
 529       osm_obj_warn(o, "Text symbolizer does not support this object type");
 530       return;
 531     }
 532 }
 533
 534 static void sym_text_init(void)
 535 {
 536   font_init();
 537   text_dup_init();
 538 }
 539
 540 struct symbolizer symbolizer_text = {
 541   .name = "text",
 542   .draw = sym_text_draw,
 543   .gen = sym_text_gen,
 544   .init = sym_text_init,
 545 };
 546
 547 struct sym_text *sym_text_new(struct osm_object *o)
 548 {
 549   return sym_new(SYMBOLIZER_TEXT, o, sizeof(struct sym_text));
 550 }
 551
 552 // FIXME: Hack
 553 void scale_text(struct svg *svg, double x, double y, osm_val_t text)
 554 {
 555   struct sym_text *st = sym_text_new(NULL);
 556
 557   struct text_font f = {
 558     .family = "Times",
 559     .weight = VALUE_NORMAL,
 560     .style = VALUE_NORMAL,
 561     .size = pt_to_mm(10),
 562   };
 563
 564   st->text = text;
 565   st->text_color = 0;
 566   st->x = x;
 567   st->y = y;
 568   st->font = font_get(&f);
 569   st->opacity = 1;
 570   st->halo_color = 0xffffff;
 571   st->halo_radius = 0.8;
 572   st->halo_opacity = 1;
 573   text_size(st);
 574   st->x -= st->tw / 2;
 575   sym_text_draw(&st->s, svg);
 576 }