Fix for Bug338155

[deliverable/tracecompass.git] / org.eclipse.linuxtools.tmf.ui / src / org / eclipse / linuxtools / tmf / ui / views / colors / ColorSettingsXML.java

/*******************************************************************************\r
 * Copyright (c) 2010 Ericsson\r
 * \r
 * All rights reserved. This program and the accompanying materials are\r
 * made available under the terms of the Eclipse Public License v1.0 which\r
 * accompanies this distribution, and is available at\r
 * http://www.eclipse.org/legal/epl-v10.html\r
 * \r
 * Contributors:\r
 *   Patrick Tasse - Initial API and implementation\r
 *******************************************************************************/\r
\r
package org.eclipse.linuxtools.tmf.ui.views.colors;\r
\r
import java.io.File;\r
import java.io.IOException;\r
import java.util.ArrayList;\r
\r
import javax.xml.parsers.DocumentBuilder;\r
import javax.xml.parsers.DocumentBuilderFactory;\r
import javax.xml.parsers.ParserConfigurationException;\r
import javax.xml.parsers.SAXParserFactory;\r
import javax.xml.transform.Transformer;\r
import javax.xml.transform.TransformerConfigurationException;\r
import javax.xml.transform.TransformerException;\r
import javax.xml.transform.TransformerFactory;\r
import javax.xml.transform.dom.DOMSource;\r
import javax.xml.transform.stream.StreamResult;\r
\r
import org.eclipse.linuxtools.tmf.filter.model.ITmfFilterTreeNode;\r
import org.eclipse.linuxtools.tmf.filter.xml.TmfFilterContentHandler;\r
import org.eclipse.linuxtools.tmf.filter.xml.TmfFilterXMLWriter;\r
import org.eclipse.swt.graphics.RGB;\r
import org.w3c.dom.Document;\r
import org.w3c.dom.Element;\r
import org.xml.sax.Attributes;\r
import org.xml.sax.SAXException;\r
import org.xml.sax.XMLReader;\r
import org.xml.sax.helpers.DefaultHandler;\r
\r
public class ColorSettingsXML {\r
\r
	private static final String COLOR_SETTINGS_TAG = "COLOR_SETTINGS"; //$NON-NLS-1$\r
	private static final String COLOR_SETTING_TAG = "COLOR_SETTING"; //$NON-NLS-1$\r
	private static final String FG_TAG = "FG"; //$NON-NLS-1$\r
	private static final String BG_TAG = "BG"; //$NON-NLS-1$\r
	private static final String R_ATTR = "R"; //$NON-NLS-1$\r
	private static final String G_ATTR = "G"; //$NON-NLS-1$\r
	private static final String B_ATTR = "B"; //$NON-NLS-1$\r
	private static final String TICK_COLOR_TAG = "TICK_COLOR"; //$NON-NLS-1$\r
	private static final String INDEX_ATTR = "INDEX"; //$NON-NLS-1$\r
	private static final String FILTER_TAG = "FILTER"; //$NON-NLS-1$\r
\r
	public static void save(String pathName, ColorSetting[] colorSettings) {\r
		try {\r
			DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();\r
			DocumentBuilder documentBuilder = documentBuilderFactory.newDocumentBuilder();\r
			Document document = documentBuilder.newDocument();\r
			\r
			Element rootElement = document.createElement(COLOR_SETTINGS_TAG);\r
			document.appendChild(rootElement);\r
	\r
			for (ColorSetting colorSetting : colorSettings) {\r
				Element colorSettingElement = document.createElement(COLOR_SETTING_TAG);\r
				rootElement.appendChild(colorSettingElement);\r
				\r
				Element fgElement = document.createElement(FG_TAG);\r
				colorSettingElement.appendChild(fgElement);\r
				RGB foreground = colorSetting.getForegroundRGB();\r
				fgElement.setAttribute(R_ATTR, Integer.toString(foreground.red));\r
				fgElement.setAttribute(G_ATTR, Integer.toString(foreground.green));\r
				fgElement.setAttribute(B_ATTR, Integer.toString(foreground.blue));\r
				\r
				Element bgElement = document.createElement(BG_TAG);\r
				colorSettingElement.appendChild(bgElement);\r
				RGB background = colorSetting.getBackgroundRGB();\r
				bgElement.setAttribute(R_ATTR, Integer.toString(background.red));\r
				bgElement.setAttribute(G_ATTR, Integer.toString(background.green));\r
				bgElement.setAttribute(B_ATTR, Integer.toString(background.blue));\r
				\r
				Element tickColorElement = document.createElement(TICK_COLOR_TAG);\r
				colorSettingElement.appendChild(tickColorElement);\r
				int index = colorSetting.getTickColorIndex();\r
				tickColorElement.setAttribute(INDEX_ATTR, Integer.toString(index));\r
				\r
				if (colorSetting.getFilter() != null) {\r
					Element filterElement = document.createElement(FILTER_TAG);\r
					colorSettingElement.appendChild(filterElement);\r
					TmfFilterXMLWriter.buildXMLTree(document, colorSetting.getFilter(), filterElement);\r
				}\r
			}\r
	\r
			TransformerFactory transformerFactory = TransformerFactory.newInstance();\r
		\r
			Transformer transformer = transformerFactory.newTransformer();\r
	        DOMSource source = new DOMSource(document);\r
	        StreamResult result =  new StreamResult(new File(pathName));\r
			transformer.transform(source, result);\r
		} catch (ParserConfigurationException e) {\r
			e.printStackTrace();\r
		} catch (TransformerConfigurationException e) {\r
			e.printStackTrace();\r
		} catch (TransformerException e) {\r
			e.printStackTrace();\r
		}\r
	}\r
	\r
	public static ColorSetting[] load(String pathName) {\r
		if (! new File(pathName).canRead()) {\r
			return new ColorSetting[0];\r
		}\r
		SAXParserFactory parserFactory = SAXParserFactory.newInstance(); \r
        parserFactory.setNamespaceAware(true); \r
\r
		try {\r
			XMLReader saxReader = parserFactory.newSAXParser().getXMLReader();\r
			ColorSettingsContentHandler handler = new ColorSettingsContentHandler();\r
	        saxReader.setContentHandler(handler);\r
	        saxReader.parse(pathName);\r
	        return handler.colorSettings.toArray(new ColorSetting[0]);\r
		} catch (ParserConfigurationException e) {\r
			e.printStackTrace();\r
		} catch (SAXException e) {\r
			e.printStackTrace();\r
		} catch (IOException e) {\r
			e.printStackTrace();\r
		}\r
		return new ColorSetting[0];\r
	}\r
	\r
	private static class ColorSettingsContentHandler extends DefaultHandler {\r
\r
		private ArrayList<ColorSetting> colorSettings = new ArrayList<ColorSetting>(0);\r
		private RGB fg;\r
		private RGB bg;\r
		private int tickColorIndex;\r
		private ITmfFilterTreeNode filter;\r
		private TmfFilterContentHandler filterContentHandler;\r
		\r
		/* (non-Javadoc)\r
		 * @see org.xml.sax.helpers.DefaultHandler#startElement(java.lang.String, java.lang.String, java.lang.String, org.xml.sax.Attributes)\r
		 */\r
		@Override\r
		public void startElement(String uri, String localName, String qName, Attributes attributes)\r
				throws SAXException {\r
			if (localName.equals(COLOR_SETTINGS_TAG)) {\r
				colorSettings = new ArrayList<ColorSetting>();\r
			} else if (localName.equals(COLOR_SETTING_TAG)) {\r
				fg = null;\r
				bg = null;\r
				filter = null;\r
			} else if (localName.equals(FG_TAG)) {\r
				int r = Integer.valueOf(attributes.getValue(R_ATTR));\r
				int g = Integer.valueOf(attributes.getValue(G_ATTR));\r
				int b = Integer.valueOf(attributes.getValue(B_ATTR));\r
				fg = new RGB(r, g, b);\r
			} else if (localName.equals(BG_TAG)) {\r
				int r = Integer.valueOf(attributes.getValue(R_ATTR));\r
				int g = Integer.valueOf(attributes.getValue(G_ATTR));\r
				int b = Integer.valueOf(attributes.getValue(B_ATTR));\r
				bg = new RGB(r, g, b);\r
			} else if (localName.equals(TICK_COLOR_TAG)) {\r
				int index = Integer.valueOf(attributes.getValue(INDEX_ATTR));\r
				tickColorIndex = index;\r
			} else if (localName.equals(FILTER_TAG)) {\r
				filterContentHandler = new TmfFilterContentHandler();\r
			} else if (filterContentHandler != null) {\r
				filterContentHandler.startElement(uri, localName, qName, attributes);\r
			}\r
		}\r
		\r
		/* (non-Javadoc)\r
		 * @see org.xml.sax.helpers.DefaultHandler#endElement(java.lang.String, java.lang.String, java.lang.String)\r
		 */\r
		@Override\r
		public void endElement(String uri, String localName, String qName)\r
				throws SAXException {\r
			if (localName.equals(COLOR_SETTINGS_TAG)) {\r
			} else if (localName.equals(COLOR_SETTING_TAG)) {\r
				ColorSetting colorSetting = new ColorSetting(fg, bg, tickColorIndex, filter);\r
				colorSettings.add(colorSetting);\r
			} else if (localName.equals(FILTER_TAG)) {\r
				filter = filterContentHandler.getTree();\r
				filterContentHandler = null;\r
			} else if (filterContentHandler != null) {\r
				filterContentHandler.endElement(uri, localName, qName);\r
			}\r
		}\r
\r
	}\r
}\r