Refactor TmfTrace and dependencies - remove indexTrace()

[deliverable/tracecompass.git] / org.eclipse.linuxtools.tmf.ui / src / org / eclipse / linuxtools / internal / tmf / ui / parsers / custom / CustomXmlTrace.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.internal.tmf.ui.parsers.custom;\r
\r
import java.io.ByteArrayInputStream;\r
import java.io.File;\r
import java.io.FileNotFoundException;\r
import java.io.IOException;\r
import java.io.RandomAccessFile;\r
\r
import javax.xml.parsers.DocumentBuilder;\r
import javax.xml.parsers.DocumentBuilderFactory;\r
import javax.xml.parsers.ParserConfigurationException;\r
\r
import org.eclipse.core.resources.IResource;\r
import org.eclipse.linuxtools.internal.tmf.ui.parsers.custom.CustomXmlTraceDefinition.InputAttribute;\r
import org.eclipse.linuxtools.internal.tmf.ui.parsers.custom.CustomXmlTraceDefinition.InputElement;\r
import org.eclipse.linuxtools.tmf.core.event.TmfEvent;\r
import org.eclipse.linuxtools.tmf.core.event.TmfTimestamp;\r
import org.eclipse.linuxtools.tmf.core.io.BufferedRandomAccessFile;\r
import org.eclipse.linuxtools.tmf.core.trace.ITmfContext;\r
import org.eclipse.linuxtools.tmf.core.trace.ITmfLocation;\r
import org.eclipse.linuxtools.tmf.core.trace.TmfContext;\r
import org.eclipse.linuxtools.tmf.core.trace.TmfLocation;\r
import org.eclipse.linuxtools.tmf.core.trace.TmfTrace;\r
import org.w3c.dom.Document;\r
import org.w3c.dom.Element;\r
import org.w3c.dom.Node;\r
import org.w3c.dom.NodeList;\r
import org.xml.sax.EntityResolver;\r
import org.xml.sax.ErrorHandler;\r
import org.xml.sax.InputSource;\r
import org.xml.sax.SAXException;\r
import org.xml.sax.SAXParseException;\r
\r
public class CustomXmlTrace extends TmfTrace<CustomXmlEvent> {\r
\r
    private static final TmfLocation<Long> NULL_LOCATION = new TmfLocation<Long>((Long) null);\r
    private static final int DEFAULT_CACHE_SIZE = 100;\r
\r
    private final CustomXmlTraceDefinition fDefinition;\r
    private final CustomXmlEventType fEventType;\r
    private final InputElement fRecordInputElement;\r
\r
    public CustomXmlTrace(final CustomXmlTraceDefinition definition) {\r
        fDefinition = definition;\r
        fEventType = new CustomXmlEventType(fDefinition);\r
        fRecordInputElement = getRecordInputElement(fDefinition.rootInputElement);\r
    }\r
\r
    public CustomXmlTrace(final IResource resource, final CustomXmlTraceDefinition definition, final String path, final int pageSize) throws FileNotFoundException {\r
        super(null, CustomXmlEvent.class, path, (pageSize > 0) ? pageSize : DEFAULT_CACHE_SIZE);\r
        fDefinition = definition;\r
        fEventType = new CustomXmlEventType(fDefinition);\r
        fRecordInputElement = getRecordInputElement(fDefinition.rootInputElement);\r
    }\r
\r
    @Override\r
    public void initTrace(final IResource resource, final String path, final Class<CustomXmlEvent> eventType) throws FileNotFoundException {\r
        super.initTrace(resource, path, eventType);\r
    }\r
\r
    @Override\r
    public TmfContext seekLocation(final ITmfLocation<?> location) {\r
        final CustomXmlTraceContext context = new CustomXmlTraceContext(NULL_LOCATION, ITmfContext.INITIAL_RANK);\r
        if (NULL_LOCATION.equals(location) || !new File(getPath()).isFile())\r
            return context;\r
        try {\r
            context.raFile = new BufferedRandomAccessFile(getPath(), "r"); //$NON-NLS-1$\r
            if (location != null && location.getLocation() instanceof Long) {\r
                context.raFile.seek((Long)location.getLocation());\r
            }\r
\r
            String line;\r
            final String recordElementStart = "<" + fRecordInputElement.elementName; //$NON-NLS-1$\r
            long rawPos = context.raFile.getFilePointer();\r
\r
            while ((line = context.raFile.getNextLine()) != null) {\r
                final int idx = line.indexOf(recordElementStart);\r
                if (idx != -1) {\r
                    context.setLocation(new TmfLocation<Long>(rawPos + idx));\r
                    return context;\r
                }\r
                rawPos = context.raFile.getFilePointer();\r
            }\r
            return context;\r
        } catch (final FileNotFoundException e) {\r
            e.printStackTrace();\r
            return context;\r
        } catch (final IOException e) {\r
            e.printStackTrace();\r
            return context;\r
        }\r
\r
    }\r
\r
    @Override\r
    public TmfContext seekLocation(final double ratio) {\r
        BufferedRandomAccessFile raFile = null;\r
        try {\r
            raFile = new BufferedRandomAccessFile(getPath(), "r"); //$NON-NLS-1$\r
            long pos = (long) (ratio * raFile.length());\r
            while (pos > 0) {\r
                raFile.seek(pos - 1);\r
                if (raFile.read() == '\n') {\r
                    break;\r
                }\r
                pos--;\r
            }\r
            final ITmfLocation<?> location = new TmfLocation<Long>(pos);\r
            final TmfContext context = seekLocation(location);\r
            context.setRank(ITmfContext.UNKNOWN_RANK);\r
            return context;\r
        } catch (final FileNotFoundException e) {\r
            e.printStackTrace();\r
            return new CustomXmlTraceContext(NULL_LOCATION, ITmfContext.INITIAL_RANK);\r
        } catch (final IOException e) {\r
            e.printStackTrace();\r
            return new CustomXmlTraceContext(NULL_LOCATION, ITmfContext.INITIAL_RANK);\r
        } finally {\r
            if (raFile != null) {\r
                try {\r
                    raFile.close();\r
                } catch (final IOException e) {\r
                }\r
            }\r
        }\r
    }\r
\r
    @Override\r
    public double getLocationRatio(final ITmfLocation<?> location) {\r
        RandomAccessFile raFile = null;\r
        try {\r
            if (location.getLocation() instanceof Long) {\r
                raFile = new RandomAccessFile(getPath(), "r"); //$NON-NLS-1$\r
                return (double) ((Long) location.getLocation()) / raFile.length();\r
            }\r
        } catch (final FileNotFoundException e) {\r
            e.printStackTrace();\r
        } catch (final IOException e) {\r
            e.printStackTrace();\r
        } finally {\r
            if (raFile != null) {\r
                try {\r
                    raFile.close();\r
                } catch (final IOException e) {\r
                }\r
            }\r
        }\r
        return 0;\r
    }\r
\r
    @Override\r
    public ITmfLocation<?> getCurrentLocation() {\r
        // TODO Auto-generated method stub\r
        return null;\r
    }\r
\r
    @Override\r
    public synchronized TmfEvent getNextEvent(final ITmfContext context) {\r
        final ITmfContext savedContext = context.clone();\r
        final TmfEvent event = parseEvent(context);\r
        if (event != null) {\r
            updateIndex(savedContext, savedContext.getRank(), event.getTimestamp());\r
            context.increaseRank();\r
        }\r
        return event;\r
    }\r
\r
    @Override\r
    public TmfEvent parseEvent(final ITmfContext tmfContext) {\r
        if (!(tmfContext instanceof CustomXmlTraceContext))\r
            return null;\r
\r
        final CustomXmlTraceContext context = (CustomXmlTraceContext) tmfContext;\r
        if (!(context.getLocation().getLocation() instanceof Long) || NULL_LOCATION.equals(context.getLocation()))\r
            return null;\r
\r
        synchronized (context.raFile) {\r
            CustomXmlEvent event = null;\r
            try {\r
                if (context.raFile.getFilePointer() != (Long)context.getLocation().getLocation() + 1)\r
                {\r
                    context.raFile.seek((Long)context.getLocation().getLocation() + 1); // +1 is for the <\r
                }\r
                final StringBuffer elementBuffer = new StringBuffer("<"); //$NON-NLS-1$\r
                readElement(elementBuffer, context.raFile);\r
                final Element element = parseElementBuffer(elementBuffer);\r
\r
                event = extractEvent(element, fRecordInputElement);\r
                ((StringBuffer) event.getContent().getValue()).append(elementBuffer);\r
\r
                String line;\r
                final String recordElementStart = "<" + fRecordInputElement.elementName; //$NON-NLS-1$\r
                long rawPos = context.raFile.getFilePointer();\r
\r
                while ((line = context.raFile.getNextLine()) != null) {\r
                    final int idx = line.indexOf(recordElementStart);\r
                    if (idx != -1) {\r
                        context.setLocation(new TmfLocation<Long>(rawPos + idx));\r
                        return event;\r
                    }\r
                    rawPos = context.raFile.getFilePointer();\r
                }\r
            } catch (final IOException e) {\r
                e.printStackTrace();\r
            }\r
            context.setLocation(NULL_LOCATION);\r
            return event;\r
        }\r
    }\r
\r
    private Element parseElementBuffer(final StringBuffer elementBuffer) {\r
        try {\r
            final DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();\r
            final DocumentBuilder db = dbf.newDocumentBuilder();\r
\r
            // The following allows xml parsing without access to the dtd\r
            final EntityResolver resolver = new EntityResolver () {\r
                @Override\r
                public InputSource resolveEntity (final String publicId, final String systemId) {\r
                    final String empty = ""; //$NON-NLS-1$\r
                    final ByteArrayInputStream bais = new ByteArrayInputStream(empty.getBytes());\r
                    return new InputSource(bais);\r
                }\r
            };\r
            db.setEntityResolver(resolver);\r
\r
            // The following catches xml parsing exceptions\r
            db.setErrorHandler(new ErrorHandler(){\r
                @Override\r
                public void error(final SAXParseException saxparseexception) throws SAXException {}\r
                @Override\r
                public void warning(final SAXParseException saxparseexception) throws SAXException {}\r
                @Override\r
                public void fatalError(final SAXParseException saxparseexception) throws SAXException {\r
                    throw saxparseexception;\r
                }});\r
\r
            final Document doc = db.parse(new ByteArrayInputStream(elementBuffer.toString().getBytes()));\r
            return doc.getDocumentElement();\r
        } catch (final ParserConfigurationException e) {\r
            e.printStackTrace();\r
        } catch (final SAXException e) {\r
            e.printStackTrace();\r
        } catch (final IOException e) {\r
            e.printStackTrace();\r
        }\r
        return null;\r
    }\r
\r
    private void readElement(final StringBuffer buffer, final RandomAccessFile raFile) {\r
        try {\r
            int numRead = 0;\r
            boolean startTagClosed = false;\r
            int i;\r
            while ((i = raFile.read()) != -1) {\r
                numRead++;\r
                final char c = (char)i;\r
                buffer.append(c);\r
                if (c == '"') {\r
                    readQuote(buffer, raFile, '"');\r
                } else if (c == '\'') {\r
                    readQuote(buffer, raFile, '\'');\r
                } else if (c == '<') {\r
                    readElement(buffer, raFile);\r
                } else if (c == '/' && numRead == 1) {\r
                    break; // found "</"\r
                } else if (c == '-' && numRead == 3 && buffer.substring(buffer.length() - 3, buffer.length() - 1).equals("!-")) {\r
                    readComment(buffer, raFile); // found "<!--"\r
                } else if (i == '>')\r
                    if (buffer.charAt(buffer.length() - 2) == '/') {\r
                        break; // found "/>"\r
                    } else if (startTagClosed) {\r
                        break; // found "<...>...</...>"\r
                    }\r
                    else {\r
                        startTagClosed = true; // found "<...>"\r
                    }\r
            }\r
            return;\r
        } catch (final IOException e) {\r
            return;\r
        }\r
    }\r
\r
    private void readQuote(final StringBuffer buffer, final RandomAccessFile raFile, final char eq) {\r
        try {\r
            int i;\r
            while ((i = raFile.read()) != -1) {\r
                final char c = (char)i;\r
                buffer.append(c);\r
                if (c == eq)\r
                {\r
                    break; // found matching end-quote\r
                }\r
            }\r
            return;\r
        } catch (final IOException e) {\r
            return;\r
        }\r
    }\r
\r
    private void readComment(final StringBuffer buffer, final RandomAccessFile raFile) {\r
        try {\r
            int numRead = 0;\r
            int i;\r
            while ((i = raFile.read()) != -1) {\r
                numRead++;\r
                final char c = (char)i;\r
                buffer.append(c);\r
                if (c == '>' && numRead >= 2 && buffer.substring(buffer.length() - 3, buffer.length() - 1).equals("--"))\r
                {\r
                    break; // found "-->"\r
                }\r
            }\r
            return;\r
        } catch (final IOException e) {\r
            return;\r
        }\r
    }\r
\r
    public static StringBuffer parseElement(final Element parentElement, final StringBuffer buffer) {\r
        final NodeList nodeList = parentElement.getChildNodes();\r
        String separator = null;\r
        for (int i = 0; i < nodeList.getLength(); i++) {\r
            final Node node = nodeList.item(i);\r
            if (node.getNodeType() == Node.ELEMENT_NODE) {\r
                if (separator == null) {\r
                    separator = " | "; //$NON-NLS-1$\r
                } else {\r
                    buffer.append(separator);\r
                }\r
                final Element element = (Element) node;\r
                if (element.hasChildNodes() == false) {\r
                    buffer.append(element.getNodeName());\r
                } else if (element.getChildNodes().getLength() == 1 && element.getFirstChild().getNodeType() == Node.TEXT_NODE) {\r
                    buffer.append(element.getNodeName() + ":" + element.getFirstChild().getNodeValue().trim()); //$NON-NLS-1$\r
                } else {\r
                    buffer.append(element.getNodeName());\r
                    buffer.append(" [ "); //$NON-NLS-1$\r
                    parseElement(element, buffer);\r
                    buffer.append(" ]"); //$NON-NLS-1$\r
                }\r
            } else if (node.getNodeType() == Node.TEXT_NODE)\r
                if (node.getNodeValue().trim().length() != 0) {\r
                    buffer.append(node.getNodeValue().trim());\r
                }\r
        }\r
        return buffer;\r
    }\r
\r
    public InputElement getRecordInputElement(final InputElement inputElement) {\r
        if (inputElement.logEntry)\r
            return inputElement;\r
        else if (inputElement.childElements != null) {\r
            for (final InputElement childInputElement : inputElement.childElements) {\r
                final InputElement recordInputElement = getRecordInputElement(childInputElement);\r
                if (recordInputElement != null)\r
                    return recordInputElement;\r
            }\r
        }\r
        return null;\r
    }\r
\r
    public CustomXmlEvent extractEvent(final Element element, final InputElement inputElement) {\r
        final CustomXmlEvent event = new CustomXmlEvent(fDefinition, this, TmfTimestamp.ZERO, "", fEventType,""); //$NON-NLS-1$ //$NON-NLS-2$\r
        event.setContent(new CustomEventContent(event, "")); //$NON-NLS-1$\r
        parseElement(element, event, inputElement);\r
        return event;\r
    }\r
\r
    private void parseElement(final Element element, final CustomXmlEvent event, final InputElement inputElement) {\r
        if (inputElement.inputName != null && !inputElement.inputName.equals(CustomXmlTraceDefinition.TAG_IGNORE)) {\r
            event.parseInput(parseElement(element, new StringBuffer()).toString(), inputElement.inputName, inputElement.inputAction, inputElement.inputFormat);\r
        }\r
        if (inputElement.attributes != null) {\r
            for (final InputAttribute attribute : inputElement.attributes) {\r
                event.parseInput(element.getAttribute(attribute.attributeName), attribute.inputName, attribute.inputAction, attribute.inputFormat);\r
            }\r
        }\r
        final NodeList childNodes = element.getChildNodes();\r
        if (inputElement.childElements != null) {\r
            for (int i = 0; i < childNodes.getLength(); i++) {\r
                final Node node = childNodes.item(i);\r
                if (node instanceof Element) {\r
                    for (final InputElement child : inputElement.childElements)\r
                        if (node.getNodeName().equals(child.elementName)) {\r
                            parseElement((Element) node, event, child);\r
                            break;\r
                        }\r
                }\r
            }\r
        }\r
        return;\r
    }\r
\r
    public CustomTraceDefinition getDefinition() {\r
        return fDefinition;\r
    }\r
}\r