franta-hg@76: /**
franta-hg@76:  * XML Web generátor – program na generování webových stránek
franta-hg@76:  * Copyright © 2012 František Kučera (frantovo.cz)
franta-hg@76:  * 
franta-hg@76:  * This program is free software: you can redistribute it and/or modify
franta-hg@76:  * it under the terms of the GNU General Public License as published by
franta-hg@76:  * the Free Software Foundation, either version 3 of the License, or
franta-hg@76:  * (at your option) any later version.
franta-hg@76:  * 
franta-hg@76:  * This program is distributed in the hope that it will be useful,
franta-hg@76:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
franta-hg@76:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
franta-hg@76:  * GNU General Public License for more details.
franta-hg@76:  * 
franta-hg@76:  * You should have received a copy of the GNU General Public License
franta-hg@76:  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
franta-hg@76:  */
franta-hg@76: package cz.frantovo.xmlWebGenerator.makra;
franta-hg@76: 
franta-hg@76: import java.io.File;
franta-hg@76: import java.io.IOException;
franta-hg@76: import java.io.PrintStream;
franta-hg@76: import java.net.URLDecoder;
franta-hg@76: import java.nio.charset.Charset;
franta-hg@76: import static cz.frantovo.xmlWebGenerator.NástrojeCLI.*;
franta-hg@76: 
franta-hg@76: /**
franta-hg@76:  * Diagramy
franta-hg@76:  * 
franta-hg@76:  * @author František Kučera (frantovo.cz)
franta-hg@76:  */
franta-hg@76: public class Diagram {
franta-hg@76: 
franta-hg@76: 	private static final String ADRESÁŘ_VÝSTUPNÍ = "výstup";
franta-hg@76: 	private static final String PŘÍKAZ_DOT = "dot";
franta-hg@76: 	private static int počítadloDiagramů = 0;
franta-hg@76: 	private static String počítadloDiagramůKontext = ""; // aktuálně zpracovávaná stránka, při změně vynulujeme počítadlo
franta-hg@76: 
franta-hg@76: 	/**
franta-hg@76: 	 * Vytvoří obrázek s diagramem.
franta-hg@76: 	 * @param zadání definice diagramu ve formátu dot
franta-hg@76: 	 * @param vodorovně zda má být graf orientovaný vodorovně (funguje jen při <code>kompletní = false</code>)
franta-hg@76: 	 * @param kompletní false, pokud k zadání chceme doplnit <code>digraph d {…}</code>
franta-hg@76: 	 * @param kontext kam diagram patří – typicky název stránky, do které je vložen
franta-hg@76: 	 * diagramy se pak budou číslovat v rámci tohoto kontextu 
franta-hg@76: 	 * → nebude docházet k přepisování diagramů jiných stránek při částečném přegenerování webu.
franta-hg@76: 	 * @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: 	 * @return název souboru bez přípony, který byl vytvořen, nebo null, pokud došlo k chybě.
franta-hg@76: 	 */
franta-hg@76: 	public static String vytvořDiagram(String zadání, boolean vodorovně, boolean kompletní, String kontext, String souborZadání) throws IOException, InterruptedException {
franta-hg@76: 		if (isPříkazDostupný(PŘÍKAZ_DOT)) {
franta-hg@76: 
franta-hg@76: 			String soubor;
franta-hg@76: 			if (souborZadání == null) {
franta-hg@76: 				if (kontext == null) {
franta-hg@76: 					počítadloDiagramů++;
franta-hg@76: 					soubor = "diagram-" + počítadloDiagramů;
franta-hg@76: 				} else {
franta-hg@76: 					// TODO: tohle by se mělo udělat v XSLT
franta-hg@76: 					kontext = URLDecoder.decode(kontext, Charset.defaultCharset().name());
franta-hg@76: 
franta-hg@76: 					// Každá stránka bude mít svoje diagramy číslované od 1
franta-hg@76: 					if (!počítadloDiagramůKontext.equals(kontext)) {
franta-hg@76: 						počítadloDiagramůKontext = kontext;
franta-hg@76: 						počítadloDiagramů = 0;
franta-hg@76: 					}
franta-hg@76: 
franta-hg@76: 					počítadloDiagramů++;
franta-hg@76: 					soubor = "diagram-" + kontext + "-" + počítadloDiagramů;
franta-hg@76: 				}
franta-hg@76: 			} else {
franta-hg@76: 				soubor = souborZadání;
franta-hg@76: 			}
franta-hg@76: 			String souborSložka = ADRESÁŘ_VÝSTUPNÍ + File.separator + soubor;
franta-hg@76: 
franta-hg@76: 			String zdroják;
franta-hg@76: 			if (kompletní) {
franta-hg@76: 				zdroják = zadání;
franta-hg@76: 			} else {
franta-hg@76: 				StringBuilder b = new StringBuilder(zadání.length() + 200);
franta-hg@76: 				b.append("digraph d {\n");
franta-hg@76: 				b.append("\tbgcolor=\"transparent\";\n");
franta-hg@76: 				if (vodorovně) {
franta-hg@76: 					b.append("\trankdir=LR;");
franta-hg@76: 				}
franta-hg@76: 				b.append(zadání);
franta-hg@76: 				b.append("}\n");
franta-hg@76: 				zdroják = b.toString();
franta-hg@76: 			}
franta-hg@76: 
franta-hg@76: 			Runtime r = Runtime.getRuntime();
franta-hg@76: 			Process p = r.exec(new String[]{PŘÍKAZ_DOT, "-T", "svg", "-o", souborSložka + ".svg"});
franta-hg@76: 
franta-hg@76: 			/**
franta-hg@76: 			 * TODO: generovat i PNG bitmapu
franta-hg@76: 			 */
franta-hg@76: 			PrintStream vstupProcesu = new PrintStream(p.getOutputStream());
franta-hg@76: 			vstupProcesu.print(zdroják.toString());
franta-hg@76: 			vstupProcesu.close();
franta-hg@76: 
franta-hg@76: 			String chyby = načtiProud(p.getErrorStream());
franta-hg@76: 
franta-hg@76: 			p.waitFor();
franta-hg@76: 
franta-hg@76: 			if (chyby.length() == 0) {
franta-hg@76: 				return soubor;
franta-hg@76: 			} else {
franta-hg@76: 				System.err.print("Při vytváření diagramu došlo k chybě: " + chyby);
franta-hg@76: 				return null;
franta-hg@76: 			}
franta-hg@76: 		} else {
franta-hg@76: 			System.err.println("Příkaz " + PŘÍKAZ_DOT + " není na vašem systému dostupný → diagramy nelze vygreslit.");
franta-hg@76: 			System.err.println("Můžete ho nainstalovat pomocí:");
franta-hg@76: 			System.err.println("\t$ aptitude install graphviz   # (Debian/Ubuntu)");
franta-hg@76: 			System.err.println("\t$ yum install graphviz        # (Fedora/RedHat)");
franta-hg@76: 			return null;
franta-hg@76: 		}
franta-hg@76: 	}
franta-hg@76: }