style.c

   1 /*
   2  *      Hic Est Leo -- Styling
   3  *
   4  *      (c) 2014 Martin Mares <mj@ucw.cz>
   5  */
   6
   7 #include "leo.h"
   8 #include "osm.h"
   9 #include "style.h"
  10
  11 #include <ucw/mempool.h>
  12
  13 #include <stdio.h>
  14
  15 struct dict style_prop_dict, style_layer_dict;
  16
  17 static const char * const style_wk_props[] = {
  18   NULL,
  19 #define P(x,y) [PROP_##x] = y,
  20 #include "dict-props.h"
  21 #undef P
  22   NULL
  23 };
  24
  25 static const char * const style_wk_layers[] = {
  26   NULL,
  27   "default",
  28   NULL
  29 };
  30
  31 void styles_init(void)
  32 {
  33   dict_init(&style_prop_dict, style_wk_props);
  34   dict_init(&style_layer_dict, style_wk_layers);
  35 }
  36
  37 #define HASH_NODE struct style_prop
  38 #define HASH_PREFIX(x) style_prop_##x
  39 #define HASH_KEY_ATOMIC key
  40 #define HASH_WANT_FIND
  41 #define HASH_WANT_LOOKUP
  42 #define HASH_GIVE_ALLOC
  43 // FIXME: Make <ucw/hashtable.h> accept our pool
  44 #define HASH_ZERO_FILL
  45 #define HASH_TABLE_DYNAMIC
  46 #define HASH_TABLE_ALLOC
  47 #define HASH_TABLE_VARS struct mempool *pool;
  48 static void *style_prop_alloc(struct style_prop_table *table, uns size);
  49 static inline void style_prop_free(struct style_prop_table *table UNUSED, void *x UNUSED) { }
  50 #include <ucw/hashtable.h>
  51
  52 static void *style_prop_alloc(struct style_prop_table *table, uns size)
  53 {
  54   return mp_alloc_fast(table->pool, size);
  55 }
  56
  57 void style_init(struct style_results *r)
  58 {
  59   r->pool = mp_new(4096);
  60   r->num_layers = dict_size(&style_layer_dict);
  61   r->layers = mp_alloc_zero(r->pool, r->num_layers * sizeof(struct style_info *));
  62   r->num_active_layers = 0;
  63   r->active_layers = mp_alloc_zero(r->pool, r->num_layers * sizeof(layer_t));
  64 }
  65
  66 void style_cleanup(struct style_results *r)
  67 {
  68   mp_delete(r->pool);
  69 }
  70
  71 void style_begin(struct style_results *r, struct osm_object *o)
  72 {
  73   ASSERT(!r->num_active_layers);
  74   mp_push(r->pool);
  75   r->obj = o;
  76 }
  77
  78 void style_end(struct style_results *r)
  79 {
  80   for (uns i=0; i<r->num_active_layers; i++)
  81     r->layers[r->active_layers[i]] = NULL;
  82   r->layers[0] = NULL;
  83   r->num_active_layers = 0;
  84   mp_pop(r->pool);
  85 }
  86
  87 static inline void style_assign(struct style_prop *dest, struct style_prop *src)
  88 {
  89   dest->type = src->type;
  90   dest->val = src->val;
  91 }
  92
  93 static struct style_info *style_get_info(struct style_results *r, layer_t layer)
  94 {
  95   ASSERT(layer < r->num_layers);
  96   if (!r->layers[layer])
  97     {
  98       struct style_info *si = mp_alloc_zero(r->pool, sizeof(*si));
  99       r->layers[layer] = si;
 100       si->hash = mp_alloc_zero(r->pool, sizeof(struct style_prop_table));
 101       si->hash->pool = r->pool;
 102       style_prop_init(si->hash);
 103
 104       if (layer != STYLE_LAYER_ALL)
 105         {
 106           r->active_layers[r->num_active_layers++] = layer;
 107
 108           // Copy all properties which have been set for all layers
 109           struct style_info *si_all = r->layers[STYLE_LAYER_ALL];
 110           if (si_all)
 111             {
 112               /*
 113                * CAVEAT: This is probably wrong. When no properties are set explicitly
 114                * set for a layer, all-layer properties are not propagated. Hopefully harmless.
 115                */
 116               HASH_FOR_ALL_DYNAMIC(style_prop, si_all->hash, s)
 117                 {
 118                   style_assign(style_prop_lookup(si->hash, s->key), s);
 119                 }
 120               HASH_END_FOR;
 121             }
 122         }
 123     }
 124   return r->layers[layer];
 125 }
 126
 127 void style_enable_default_layer(struct style_results *r)
 128 {
 129   style_get_info(r, STYLE_LAYER_DEFAULT);
 130 }
 131
 132 void style_set_by_layer(struct style_results *r, layer_t layer, struct style_prop *p)
 133 {
 134   if (layer == STYLE_LAYER_ALL)
 135     {
 136       // Set in all existing layers
 137       for (uns i=0; i < r->num_active_layers; i++)
 138         style_assign(style_prop_lookup(r->layers[r->active_layers[i]]->hash, p->key), p);
 139       // ... and let it propagate to STYLE_LAYER_ALL
 140     }
 141
 142   struct style_info *si = style_get_info(r, layer);
 143   style_assign(style_prop_lookup(si->hash, p->key), p);
 144 }
 145
 146 void style_set(struct style_info *si, struct style_prop *p)
 147 {
 148   style_assign(style_prop_lookup(si->hash, p->key), p);
 149 }
 150
 151 struct style_prop *style_get(struct style_info *si, prop_t key)
 152 {
 153   return style_prop_find(si->hash, key);
 154 }
 155
 156 struct style_prop *style_get_and_check(struct style_info *si, prop_t key, uns allowed_types)
 157 {
 158   struct style_prop *p = style_prop_find(si->hash, key);
 159   if (!p)
 160     return NULL;
 161   if (!(allowed_types & (1 << p->type)))
 162     {
 163       // XXX: Better diagnostics?
 164       msg(L_WARN, "Style property %s set to invalid type #%u", style_prop_decode(p->key), p->type);
 165       return NULL;
 166     }
 167   return p;
 168 }
 169
 170 osm_val_t style_get_ident(struct style_info *si, prop_t key)
 171 {
 172   struct style_prop *p = style_get_and_check(si, key, 1 << PROP_TYPE_IDENT);
 173   return p ? p->val.id : 0;
 174 }
 175
 176 osm_val_t style_get_string(struct style_info *si, prop_t key)
 177 {
 178   struct style_prop *p = style_get_and_check(si, key, (1 << PROP_TYPE_STRING) | (1 << PROP_TYPE_IDENT));
 179   return p ? p->val.id : 0;
 180 }
 181
 182 bool style_get_number(struct style_info *si, prop_t key, double *dp)
 183 {
 184   struct style_prop *p = style_get_and_check(si, key, 1 << PROP_TYPE_NUMBER);
 185   if (!p)
 186     return 0;
 187   *dp = p->val.number;
 188   return 1;
 189 }
 190
 191 bool style_get_color(struct style_info *si, prop_t key, color_t *colorp)
 192 {
 193   struct style_prop *p = style_get_and_check(si, key, 1 << PROP_TYPE_COLOR);
 194   if (!p)
 195     return 0;
 196   *colorp = p->val.color;
 197   return 1;
 198 }
 199
 200 void style_dump(struct style_results *r)
 201 {
 202   for (uns i=0; i < r->num_active_layers; i++)
 203     {
 204       layer_t layer = r->active_layers[i];
 205       printf("Layer %s (%u)\n", style_layer_decode(layer), layer);
 206       struct style_info *si = r->layers[layer];
 207       HASH_FOR_ALL_DYNAMIC(style_prop, si->hash, s)
 208         {
 209           printf("\t");
 210           style_dump_prop(s);
 211         }
 212       HASH_END_FOR;
 213     }
 214 }
 215
 216 static void style_dump_val(struct style_prop *s)
 217 {
 218   uns cnt = 0;
 219
 220   switch (s->type)
 221     {
 222     case PROP_TYPE_STRING:
 223       printf("%s [string]", osm_val_decode(s->val.id));
 224       break;
 225     case PROP_TYPE_IDENT:
 226       printf("%s [ident]", osm_val_decode(s->val.id));
 227       break;
 228     case PROP_TYPE_NUMBER:
 229       printf("%.6g [number]", s->val.number);
 230       break;
 231     case PROP_TYPE_COLOR:
 232       printf("%06x [color]", s->val.color);
 233       break;
 234     case PROP_TYPE_LIST:
 235       putchar('(');
 236       CLIST_FOR_EACH(struct style_val_list_entry *, e, *s->val.list)
 237         {
 238           if (cnt++)
 239             printf(", ");
 240           style_dump_val(&e->val);
 241         }
 242       printf(") [list]");
 243       break;
 244     default:
 245       printf("[unknown type %u]", s->type);
 246     }
 247 }
 248
 249 void style_dump_prop(struct style_prop *s)
 250 {
 251   printf("%s = ", style_prop_decode(s->key));
 252   style_dump_val(s);
 253   putchar('\n');
 254 }
 255
 256 void style_scale(struct style_info *si, double *wp, double *hp, prop_t width_prop, prop_t height_prop)
 257 {
 258   double w, h;
 259   bool got_width = style_get_number(si, width_prop, &w);
 260   bool got_height = style_get_number(si, height_prop, &h);
 261
 262   if (got_width + got_height == 2)
 263     {
 264       *wp = w;
 265       *hp = h;
 266     }
 267   else if (got_width + got_height == 1)
 268     {
 269       if (got_width)
 270         {
 271           *hp *= w / *wp;
 272           *wp = w;
 273         }
 274       else
 275         {
 276           *wp *= h / *hp;
 277           *hp = h;
 278         }
 279     }
 280 }