gvcolor.c

Go to the documentation of this file.

 
/*************************************************************************
 * Copyright (c) 2011 AT&T Intellectual Property 
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors: Details at https://graphviz.org
 *************************************************************************/
 
 
/*
 * Written by Stephen North
 * Updated by Emden Gansner
 */
 
/* if NC changes, a bunch of scanf calls below are in trouble */
#define NC      3               /* size of HSB color vector */
 
#include <assert.h>
#include <cgraph/cgraph.h>
#include <cgraph/ingraphs.h>
#include "colorxlate.h"
#include <math.h>
#include <stdbool.h>
#include <stdlib.h>
#include <util/agxbuf.h>
#include <util/alloc.h>
#include <util/exit.h>
 
typedef struct {
    Agrec_t h;
    double relrank;     /* coordinate of its rank, smaller means lower rank */
    double x[NC];       /* color vector */
} Agnodeinfo_t;
 
#define ND_relrank(n) (((Agnodeinfo_t*)((n)->base.data))->relrank)
#define ND_x(n) (((Agnodeinfo_t*)((n)->base.data))->x)
 
#include <stdio.h>
 
#include <getopt.h>
 
double Defcolor[NC] = { 0.0, 0.0, 1.0 };        /* white */
int Forward = 1;                /* how to propagate colors w.r.t. ranks */
int LR = 0;                     /* rank orientation */
 
int AdjustSaturation;
double MinRankSaturation;
double MaxRankSaturation;
 
static int cmpf(const void *x, const void *y) {
// Suppress Clang/GCC -Wcast-qual warning. Casting away const here is acceptable
// as the later usage is const. We need the cast because the macros use
// non-const pointers for genericity.
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wcast-qual"
#endif
  Agnode_t **n0 = (Agnode_t **)x;
  Agnode_t **n1 = (Agnode_t **)y;
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
    double relrank0 = ND_relrank(*n0);
    double relrank1 = ND_relrank(*n1);
    if (relrank0 < relrank1)
        return -1;
    if (relrank0 > relrank1)
        return 1;
    return 0;
}
 
static void setcolor(char *p, double *v)
{
    agxbuf buf = {0};
    if (sscanf(p, "%lf %lf %lf", v, v + 1, v + 2) != 3 && p[0]) {
        colorxlate(p, &buf);
        sscanf(agxbuse(&buf), "%lf %lf %lf", v, v + 1, v + 2);
    }
    agxbfree(&buf);
}
 
static char **Files;
 
static char *useString = "Usage: gvcolor [-?] <files>\n\
  -? - print usage\n\
If no files are specified, stdin is used\n";
 
static void usage(int v)
{
    printf("%s",useString);
    graphviz_exit(v);
}
 
static void init(int argc, char *argv[])
{
    int c;
 
    opterr = 0;
    while ((c = getopt(argc, argv, ":?")) != -1) {
        switch (c) {
        case '?':
            if (optopt == '\0' || optopt == '?')
                usage(0);
            else {
                fprintf(stderr, "gvcolor: option -%c unrecognized\n",
                        optopt);
                usage(1);
            }
            break;
        default:
            fprintf(stderr, "gvcolor: unexpected error\n");
            graphviz_exit(EXIT_FAILURE);
        }
    }
    argv += optind;
    argc -= optind;
 
    if (argc)
        Files = argv;
}
 
static void color(Agraph_t * g)
{
    int nn, j, cnt;
    Agnode_t *n, *v, **nlist;
    Agedge_t *e;
    char *p;
    double x, y, maxrank = 0.0;
    double sum[NC], d, lowsat, highsat;
 
    if (agattr(g, AGNODE, "pos", 0) == NULL) {
        fprintf(stderr,
                "graph must be run through 'dot' before 'gvcolor'\n");
        graphviz_exit(1);
    }
    aginit(g, AGNODE, "nodeinfo", sizeof(Agnodeinfo_t), true);
    if (agattr(g, AGNODE, "style", 0) == NULL)
        agattr(g, AGNODE, "style", "filled");
    if ((p = agget(g, "Defcolor")))
        setcolor(p, Defcolor);
 
    if ((p = agget(g, "rankdir")) && p[0] == 'L')
        LR = 1;
    if ((p = agget(g, "flow")) && p[0] == 'b')
        Forward = 0;
    if ((p = agget(g, "saturation"))) {
        if (sscanf(p, "%lf,%lf", &lowsat, &highsat) == 2) {
            MinRankSaturation = lowsat;
            MaxRankSaturation = highsat;
            AdjustSaturation = 1;
        }
    }
 
    /* assemble the sorted list of nodes and store the initial colors */
    nn = agnnodes(g);
    assert(nn >= 0);
    size_t nnodes = (size_t)nn;
    nlist = gv_calloc(nnodes, sizeof(Agnode_t *));
    size_t i = 0;
    for (n = agfstnode(g); n; n = agnxtnode(g, n)) {
        nlist[i++] = n;
        if ((p = agget(n, "color")))
            setcolor(p, ND_x(n));
        p = agget(n, "pos");
        sscanf(p, "%lf,%lf", &x, &y);
        ND_relrank(n) = (LR ? x : y);
        maxrank = fmax(maxrank, ND_relrank(n));
    }
    if (LR != Forward)
        for (i = 0; i < nnodes; i++) {
            n = nlist[i];
            ND_relrank(n) = maxrank - ND_relrank(n);
        }
    qsort(nlist, nnodes, sizeof(Agnode_t *), cmpf);
 
    /* this is the pass that pushes the colors through the edges */
    for (i = 0; i < nnodes; i++) {
        n = nlist[i];
 
        /* skip nodes that were manually colored */
        cnt = 0;
        for (j = 0; j < NC; j++)
            if (ND_x(n)[j] != 0.0)
                cnt++;
        if (cnt > 0)
            continue;
 
        for (j = 0; j < NC; j++)
            sum[j] = 0.0;
        cnt = 0;
        for (e = agfstedge(g, n); e; e = agnxtedge(g, e, n)) {
            v = aghead(e);
            if (v == n)
                v = agtail(e);
            d = ND_relrank(v) - ND_relrank(n) - 0.01;
            if (d < 0) {
                double t = 0.0;
                for (j = 0; j < NC; j++) {
                    t += ND_x(v)[j];
                    sum[j] += ND_x(v)[j];
                }
                if (t > 0.0)
                    cnt++;
            }
        }
        if (cnt)
            for (j = 0; j < NC; j++)
                ND_x(n)[j] = sum[j] / cnt;
    }
 
    /* apply saturation adjustment and convert color to string */
    for (i = 0; i < nnodes; i++) {
        double h, s, b, t;
        char buf[64];
 
        n = nlist[i];
 
        t = 0.0;
        for (j = 0; j < NC; j++)
            t += ND_x(n)[j];
        if (t > 0.0) {
            h = ND_x(n)[0];
            if (AdjustSaturation) {
                s = ND_relrank(n) / maxrank;
                if (!Forward)
                    s = 1.0 - s;
                s = MinRankSaturation
                    + s * (MaxRankSaturation - MinRankSaturation);
            } else
                s = 1.0;
            s = s * ND_x(n)[1];
            b = ND_x(n)[2];
        } else {
            h = Defcolor[0];
            s = Defcolor[1];
            b = Defcolor[2];
        }
        snprintf(buf, sizeof(buf), "%f %f %f", h, s, b);
        agset(n, "color", buf);
    }
    free (nlist);
}
 
int main(int argc, char **argv)
{
    Agraph_t *g;
    ingraph_state ig;
 
    init(argc, argv);
    newIngraph(&ig, Files);
 
    while ((g = nextGraph(&ig)) != 0) {
        color(g);
        agwrite(g, stdout);
        fflush(stdout);
        agclose(g);
    }
 
    graphviz_exit(0);
}