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