šablona/funkce/src/cz/frantovo/xmlWebGenerator/makra/Diagram.java
author František Kučera <franta-hg@frantovo.cz>
Sun, 08 Jan 2012 17:27:08 +0100
changeset 80 386c7e4ac34b
parent 76 c7746d95283d
child 87 25dec6931f18
permissions -rw-r--r--
Added signature for changeset 7e478bfa5694
franta-hg@76
     1
/**
franta-hg@76
     2
 * XML Web generátor – program na generování webových stránek
franta-hg@76
     3
 * Copyright © 2012 František Kučera (frantovo.cz)
franta-hg@76
     4
 * 
franta-hg@76
     5
 * This program is free software: you can redistribute it and/or modify
franta-hg@76
     6
 * it under the terms of the GNU General Public License as published by
franta-hg@76
     7
 * the Free Software Foundation, either version 3 of the License, or
franta-hg@76
     8
 * (at your option) any later version.
franta-hg@76
     9
 * 
franta-hg@76
    10
 * This program is distributed in the hope that it will be useful,
franta-hg@76
    11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
franta-hg@76
    12
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
franta-hg@76
    13
 * GNU General Public License for more details.
franta-hg@76
    14
 * 
franta-hg@76
    15
 * You should have received a copy of the GNU General Public License
franta-hg@76
    16
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
franta-hg@76
    17
 */
franta-hg@76
    18
package cz.frantovo.xmlWebGenerator.makra;
franta-hg@76
    19
franta-hg@76
    20
import java.io.File;
franta-hg@76
    21
import java.io.IOException;
franta-hg@76
    22
import java.io.PrintStream;
franta-hg@76
    23
import java.net.URLDecoder;
franta-hg@76
    24
import java.nio.charset.Charset;
franta-hg@76
    25
import static cz.frantovo.xmlWebGenerator.NástrojeCLI.*;
franta-hg@76
    26
franta-hg@76
    27
/**
franta-hg@76
    28
 * Diagramy
franta-hg@76
    29
 * 
franta-hg@76
    30
 * @author František Kučera (frantovo.cz)
franta-hg@76
    31
 */
franta-hg@76
    32
public class Diagram {
franta-hg@76
    33
franta-hg@76
    34
	private static final String ADRESÁŘ_VÝSTUPNÍ = "výstup";
franta-hg@76
    35
	private static final String PŘÍKAZ_DOT = "dot";
franta-hg@76
    36
	private static int počítadloDiagramů = 0;
franta-hg@76
    37
	private static String počítadloDiagramůKontext = ""; // aktuálně zpracovávaná stránka, při změně vynulujeme počítadlo
franta-hg@76
    38
franta-hg@76
    39
	/**
franta-hg@76
    40
	 * Vytvoří obrázek s diagramem.
franta-hg@76
    41
	 * @param zadání definice diagramu ve formátu dot
franta-hg@76
    42
	 * @param vodorovně zda má být graf orientovaný vodorovně (funguje jen při <code>kompletní = false</code>)
franta-hg@76
    43
	 * @param kompletní false, pokud k zadání chceme doplnit <code>digraph d {…}</code>
franta-hg@76
    44
	 * @param kontext kam diagram patří – typicky název stránky, do které je vložen
franta-hg@76
    45
	 * diagramy se pak budou číslovat v rámci tohoto kontextu 
franta-hg@76
    46
	 * → nebude docházet k přepisování diagramů jiných stránek při částečném přegenerování webu.
franta-hg@76
    47
	 * @param souborZadání null pokud chceme automatické číslování | nebo zadáme název souboru se zadáním diagramu – vygenerovaný diagram se pak bude jmenovat stejně
franta-hg@76
    48
	 * @return název souboru bez přípony, který byl vytvořen, nebo null, pokud došlo k chybě.
franta-hg@76
    49
	 */
franta-hg@76
    50
	public static String vytvořDiagram(String zadání, boolean vodorovně, boolean kompletní, String kontext, String souborZadání) throws IOException, InterruptedException {
franta-hg@76
    51
		if (isPříkazDostupný(PŘÍKAZ_DOT)) {
franta-hg@76
    52
franta-hg@76
    53
			String soubor;
franta-hg@76
    54
			if (souborZadání == null) {
franta-hg@76
    55
				if (kontext == null) {
franta-hg@76
    56
					počítadloDiagramů++;
franta-hg@76
    57
					soubor = "diagram-" + počítadloDiagramů;
franta-hg@76
    58
				} else {
franta-hg@76
    59
					// TODO: tohle by se mělo udělat v XSLT
franta-hg@76
    60
					kontext = URLDecoder.decode(kontext, Charset.defaultCharset().name());
franta-hg@76
    61
franta-hg@76
    62
					// Každá stránka bude mít svoje diagramy číslované od 1
franta-hg@76
    63
					if (!počítadloDiagramůKontext.equals(kontext)) {
franta-hg@76
    64
						počítadloDiagramůKontext = kontext;
franta-hg@76
    65
						počítadloDiagramů = 0;
franta-hg@76
    66
					}
franta-hg@76
    67
franta-hg@76
    68
					počítadloDiagramů++;
franta-hg@76
    69
					soubor = "diagram-" + kontext + "-" + počítadloDiagramů;
franta-hg@76
    70
				}
franta-hg@76
    71
			} else {
franta-hg@76
    72
				soubor = souborZadání;
franta-hg@76
    73
			}
franta-hg@76
    74
			String souborSložka = ADRESÁŘ_VÝSTUPNÍ + File.separator + soubor;
franta-hg@76
    75
franta-hg@76
    76
			String zdroják;
franta-hg@76
    77
			if (kompletní) {
franta-hg@76
    78
				zdroják = zadání;
franta-hg@76
    79
			} else {
franta-hg@76
    80
				StringBuilder b = new StringBuilder(zadání.length() + 200);
franta-hg@76
    81
				b.append("digraph d {\n");
franta-hg@76
    82
				b.append("\tbgcolor=\"transparent\";\n");
franta-hg@76
    83
				if (vodorovně) {
franta-hg@76
    84
					b.append("\trankdir=LR;");
franta-hg@76
    85
				}
franta-hg@76
    86
				b.append(zadání);
franta-hg@76
    87
				b.append("}\n");
franta-hg@76
    88
				zdroják = b.toString();
franta-hg@76
    89
			}
franta-hg@76
    90
franta-hg@76
    91
			Runtime r = Runtime.getRuntime();
franta-hg@76
    92
			Process p = r.exec(new String[]{PŘÍKAZ_DOT, "-T", "svg", "-o", souborSložka + ".svg"});
franta-hg@76
    93
franta-hg@76
    94
			/**
franta-hg@76
    95
			 * TODO: generovat i PNG bitmapu
franta-hg@76
    96
			 */
franta-hg@76
    97
			PrintStream vstupProcesu = new PrintStream(p.getOutputStream());
franta-hg@76
    98
			vstupProcesu.print(zdroják.toString());
franta-hg@76
    99
			vstupProcesu.close();
franta-hg@76
   100
franta-hg@76
   101
			String chyby = načtiProud(p.getErrorStream());
franta-hg@76
   102
franta-hg@76
   103
			p.waitFor();
franta-hg@76
   104
franta-hg@76
   105
			if (chyby.length() == 0) {
franta-hg@76
   106
				return soubor;
franta-hg@76
   107
			} else {
franta-hg@76
   108
				System.err.print("Při vytváření diagramu došlo k chybě: " + chyby);
franta-hg@76
   109
				return null;
franta-hg@76
   110
			}
franta-hg@76
   111
		} else {
franta-hg@76
   112
			System.err.println("Příkaz " + PŘÍKAZ_DOT + " není na vašem systému dostupný → diagramy nelze vygreslit.");
franta-hg@76
   113
			System.err.println("Můžete ho nainstalovat pomocí:");
franta-hg@76
   114
			System.err.println("\t$ aptitude install graphviz   # (Debian/Ubuntu)");
franta-hg@76
   115
			System.err.println("\t$ yum install graphviz        # (Fedora/RedHat)");
franta-hg@76
   116
			return null;
franta-hg@76
   117
		}
franta-hg@76
   118
	}
franta-hg@76
   119
}