com.google.gxp.compiler.dot..svn.text-base.DotWriter.svn-base Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of google-gxp Show documentation

Google XML Pages (GXP) is a templating system used to generate XML/SGML markup (most often HTML).

The newest version!

/*
 * Copyright (C) 2008 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.gxp.compiler.dot;

import com.google.common.base.CharEscaper;
import com.google.common.base.CharEscaperBuilder;
import com.google.common.base.Joiner;
import com.google.common.collect.Lists;
import com.google.gxp.compiler.io.CIndenter;

import java.io.*;
import java.util.*;

/**
 * A {@link GraphSink} that writes graphviz "dot" format to an {@code
 * Appendable}.
 */
public class DotWriter implements GraphSink {
  private final CIndenter out;

  public DotWriter(Appendable out) {
    this.out = new CIndenter(out);
  }

  public void digraphStart(String name) {
    out.appendLine(String.format("digraph %s {", name));
  }

  public void digraphEnd() {
    out.appendLine("}");
  }

  public void simpleNode(String nodeId, NodeShape shape, String label) {
    out.appendLine(String.format(
        "node%s [label=\"%s\",shape=%s];",
        nodeId, dotEscape(label), shape));
  }

  public void recordNode(String nodeId, Map properties) {
    List names = Lists.newArrayListWithExpectedSize(properties.size());
    List values = Lists.newArrayListWithExpectedSize(properties.size());
    for (Map.Entry entry : properties.entrySet()) {
      names.add(dotEscape(entry.getKey()));
      values.add(dotEscape(entry.getValue()));
    }
    out.appendLine(String.format(
        "node%s [label=\"{%s} | {%s}\",shape=Mrecord];",
        nodeId,
        Joiner.on(" | ").join(names),
        Joiner.on(" | ").join(values)));
  }

  public void edge(String fromNodeId, String label, String toNodeId) {
    String suffix = (label == null)
        ? ""
        : String.format(" [label=\"%s\"]", dotEscape(label));
    out.appendLine(String.format("node%s -> node%s%s;\n",
                                 fromNodeId, toNodeId, suffix));
  }

  private static CharEscaper DOT_ESCAPER =
      new CharEscaperBuilder()
      .addEscape('\n', "\\n")
      .addEscape('\r', "\\r")
      .addEscape('\t', "\\t")
      .addEscape('\\', "\\\\")
      .addEscape('\"', "\\\"")
      .addEscape('{', "\\{")
      .addEscape('}', "\\}")
      .addEscape('<', "\\<")
      .addEscape('>', "\\>")
      .addEscape('|', "\\|")
      .addEscape(' ', "\\ ")
      .toEscaper();

  private static String dotEscape(String s) {
    return DOT_ESCAPER.escape(s);
  }
}