diagram-definition/plugins/org.eclipse.papyrus.dd.editor/src/org/eclipse/papyrus/dd/dg/presentation/DGSVGSourcePage.java

/**
 * Copyright (c) 2014 CEA LIST.
 *  
 * 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
 * http://www.eclipse.org/legal/epl-v10.html
 *  
 * Contributors:
 *  CEA LIST - Initial API and implementation
 */
package org.eclipse.papyrus.dd.dg.presentation;

import java.io.StringWriter;

import org.eclipse.jface.text.Document;
import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.source.SourceViewer;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.papyrus.dd.dg.RootCanvas;
import org.eclipse.papyrus.dd.presentation.DDEditorPage;
import org.eclipse.papyrus.dd.presentation.DDEditorPlugin;
import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.Composite;
import org.w3c.dom.DOMImplementation;

/**
 * This is a page that shows the source of the corresponding SVG for a DG model
 */
public class DGSVGSourcePage extends DDEditorPage {

	/**
	 * The page id
	 */
	public static final String ID = DGSVGSourcePage.class.getName();

	/**
	 * The source viewer of the page
	 */
	protected SourceViewer viewer;

	/**
	 * Constructs a new SVG source page for a given DG editor
	 * 
	 * @param editor
	 *        The DG editor
	 */
	public DGSVGSourcePage(DGEditor editor) {
		super(editor, ID, getString("DGSVGSourcePage.title"));
	}

	@Override
	protected Viewer createViewer(Composite formBody) {
		viewer = new SourceViewer(formBody, null, SWT.H_SCROLL | SWT.V_SCROLL);
		viewer.getTextWidget().setAlwaysShowScrollBars(false);
		viewer.setEditable(false);
		return viewer;
	}

	@Override
	public void refresh() {
		RootCanvas canvas = (RootCanvas)getDDEditor().getSelectedEObject();
		if(canvas == null) {
			viewer.setDocument(null);
			return;
		}
		// (re)generate the SVG document
		org.w3c.dom.Document svgDocument = getConverter().convert(canvas);
		String contents;
		try {
			StringWriter writer = new StringWriter();
			// DOMUtilities.writeDocument(svgDocument, writer);
			StringWriter formatted = new StringWriter();
//			PrettyPrinter printer = new PrettyPrinter();
//			pSrinter.setTabulationWidth(4);
//			printer.setDocumentWidth(2000);
//			printer.print(new StringReader(writer.toString()), formatted);
			contents = formatted.toString().substring(1);
			writer.close();
		} catch (Exception e) {
			DDEditorPlugin.INSTANCE.log(e);
			return;
		}
		// update the input of the viewer
		IDocument document = new Document();
		document.set(contents);
		viewer.setDocument(document);
	}

	/**
	 * Gets an instance of <code>DGToSVGConverter</code>
	 * 
	 * @return DGToSVGConverter
	 */
	protected DGToSVGConverter getConverter() {
		return new DGToSVGConverter() {

			@Override
			protected DOMImplementation getDOMImplementation() {
				return null;
				//return GenericDOMImplementation.getDOMImplementation();
			}
		};
	}
}