Implement TmfTrace changes - introduce TmfTraceException

[deliverable/tracecompass.git] / org.eclipse.linuxtools.tmf.ui / src / org / eclipse / linuxtools / internal / tmf / ui / parsers / wizards / CustomXmlParserOutputWizardPage.java

package org.eclipse.linuxtools.internal.tmf.ui.parsers.wizards;\r
\r
import java.io.File;\r
import java.io.FileWriter;\r
import java.io.IOException;\r
import java.util.ArrayList;\r
import java.util.Iterator;\r
import java.util.List;\r
\r
import org.eclipse.jface.wizard.WizardPage;\r
import org.eclipse.linuxtools.internal.tmf.ui.Messages;\r
import org.eclipse.linuxtools.internal.tmf.ui.TmfUiPlugin;\r
import org.eclipse.linuxtools.internal.tmf.ui.parsers.custom.CustomEventsTable;\r
import org.eclipse.linuxtools.internal.tmf.ui.parsers.custom.CustomTraceDefinition;\r
import org.eclipse.linuxtools.internal.tmf.ui.parsers.custom.CustomTraceDefinition.OutputColumn;\r
import org.eclipse.linuxtools.internal.tmf.ui.parsers.custom.CustomXmlTrace;\r
import org.eclipse.linuxtools.internal.tmf.ui.parsers.custom.CustomXmlTraceDefinition;\r
import org.eclipse.linuxtools.tmf.core.exceptions.TmfTraceException;\r
import org.eclipse.linuxtools.tmf.core.trace.ITmfTrace;\r
import org.eclipse.swt.SWT;\r
import org.eclipse.swt.custom.SashForm;\r
import org.eclipse.swt.custom.ScrolledComposite;\r
import org.eclipse.swt.events.SelectionAdapter;\r
import org.eclipse.swt.events.SelectionEvent;\r
import org.eclipse.swt.graphics.Image;\r
import org.eclipse.swt.layout.GridData;\r
import org.eclipse.swt.layout.GridLayout;\r
import org.eclipse.swt.widgets.Button;\r
import org.eclipse.swt.widgets.Composite;\r
import org.eclipse.swt.widgets.Text;\r
\r
public class CustomXmlParserOutputWizardPage extends WizardPage {\r
\r
    private static final Image upImage = TmfUiPlugin.getDefault().getImageFromPath("/icons/elcl16/up_button.gif"); //$NON-NLS-1$\r
    private static final Image downImage = TmfUiPlugin.getDefault().getImageFromPath("/icons/elcl16/down_button.gif"); //$NON-NLS-1$\r
    private final CustomXmlParserWizard wizard;\r
    private CustomXmlTraceDefinition definition;\r
    ArrayList<Output> outputs = new ArrayList<Output>();\r
    //    Output messageOutput;\r
    Composite container;\r
    SashForm sash;\r
    //    Text timestampFormatText;\r
    //    Text timestampPreviewText;\r
    ScrolledComposite outputsScrolledComposite;\r
    Composite outputsContainer;\r
    //    ScrolledComposite inputScrolledComposite;\r
    Composite tableContainer;\r
    CustomEventsTable previewTable;\r
    File tmpFile;\r
\r
    protected CustomXmlParserOutputWizardPage(final CustomXmlParserWizard wizard) {\r
        super("CustomParserOutputWizardPage"); //$NON-NLS-1$\r
        setTitle(wizard.inputPage.getTitle());\r
        setDescription(Messages.CustomXmlParserOutputWizardPage_description);\r
        this.wizard = wizard;\r
        setPageComplete(false);\r
    }\r
\r
    @Override\r
    public void createControl(final Composite parent) {\r
        container = new Composite(parent, SWT.NULL);\r
        container.setLayout(new GridLayout());\r
\r
        sash = new SashForm(container, SWT.VERTICAL);\r
        sash.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));\r
        sash.setBackground(sash.getDisplay().getSystemColor(SWT.COLOR_GRAY));\r
\r
        outputsScrolledComposite = new ScrolledComposite(sash, SWT.V_SCROLL);\r
        outputsScrolledComposite.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));\r
        outputsContainer = new Composite(outputsScrolledComposite, SWT.NONE);\r
        final GridLayout outputsLayout = new GridLayout(4, false);\r
        outputsLayout.marginHeight = 10;\r
        outputsLayout.marginWidth = 0;\r
        outputsContainer.setLayout(outputsLayout);\r
        outputsScrolledComposite.setContent(outputsContainer);\r
        outputsScrolledComposite.setExpandHorizontal(true);\r
        outputsScrolledComposite.setExpandVertical(true);\r
\r
        outputsContainer.layout();\r
\r
        outputsScrolledComposite.setMinSize(outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).x, outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y-5);\r
\r
        tableContainer = new Composite(sash, SWT.NONE);\r
        final GridLayout tableLayout = new GridLayout();\r
        tableLayout.marginHeight = 0;\r
        tableLayout.marginWidth = 0;\r
        tableContainer.setLayout(tableLayout);\r
        previewTable = new CustomEventsTable(new CustomXmlTraceDefinition(), tableContainer, 0);\r
        previewTable.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));\r
\r
        if (wizard.definition != null)\r
            loadDefinition(wizard.definition);\r
        setControl(container);\r
\r
    }\r
\r
    @Override\r
    public void dispose() {\r
        previewTable.dispose();\r
        super.dispose();\r
    }\r
\r
    private void loadDefinition(final CustomTraceDefinition definition) {\r
        for (final OutputColumn outputColumn : definition.outputs) {\r
            final Output output = new Output(outputsContainer, outputColumn.name);\r
            outputs.add(output);\r
        }\r
    }\r
\r
    /* (non-Javadoc)\r
     * @see org.eclipse.jface.dialogs.DialogPage#setVisible(boolean)\r
     */\r
    @Override\r
    public void setVisible(final boolean visible) {\r
        if (visible) {\r
            this.definition = wizard.inputPage.getDefinition();\r
            final List<String> outputNames = wizard.inputPage.getInputNames();\r
\r
            // dispose outputs that have been removed in the input page\r
            final Iterator<Output> iter = outputs.iterator();\r
            while (iter.hasNext()) {\r
                final Output output = iter.next();\r
                boolean found = false;\r
                for (final String name : outputNames)\r
                    if (output.name.equals(name)) {\r
                        found = true;\r
                        break;\r
                    }\r
                if (!found) {\r
                    output.dispose();\r
                    iter.remove();\r
                }\r
            }\r
\r
            // create outputs that have been added in the input page\r
            for (final String name : outputNames) {\r
                boolean found = false;\r
                for (final Output output : outputs)\r
                    if (output.name.equals(name)) {\r
                        found = true;\r
                        break;\r
                    }\r
                if (!found)\r
                    outputs.add(new Output(outputsContainer, name));\r
            }\r
\r
            outputsContainer.layout();\r
            outputsScrolledComposite.setMinSize(outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).x, outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y-5);\r
            updatePreviewTable();\r
            if (sash.getSize().y > outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y + previewTable.getTable().getItemHeight())\r
                sash.setWeights(new int[] {outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y, sash.getSize().y - outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y});\r
            else\r
                sash.setWeights(new int[] {outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y, previewTable.getTable().getItemHeight()});\r
            setPageComplete(true);\r
        } else\r
            setPageComplete(false);\r
        super.setVisible(visible);\r
    }\r
\r
    private void moveBefore(final Output moved) {\r
        final int i = outputs.indexOf(moved);\r
        if (i > 0) {\r
            final Output previous = outputs.get(i-1);\r
            moved.enabledButton.moveAbove(previous.enabledButton);\r
            moved.nameLabel.moveBelow(moved.enabledButton);\r
            moved.upButton.moveBelow(moved.nameLabel);\r
            moved.downButton.moveBelow(moved.upButton);\r
            outputs.add(i-1, outputs.remove(i));\r
            outputsContainer.layout();\r
            outputsScrolledComposite.setMinSize(outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).x, outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y-5);\r
            container.layout();\r
            updatePreviewTable();\r
        }\r
    }\r
\r
    private void moveAfter(final Output moved) {\r
        final int i = outputs.indexOf(moved);\r
        if (i+1 < outputs.size()) {\r
            final Output next = outputs.get(i+1);\r
            moved.enabledButton.moveBelow(next.downButton);\r
            moved.nameLabel.moveBelow(moved.enabledButton);\r
            moved.upButton.moveBelow(moved.nameLabel);\r
            moved.downButton.moveBelow(moved.upButton);\r
            outputs.add(i+1, outputs.remove(i));\r
            outputsContainer.layout();\r
            outputsScrolledComposite.setMinSize(outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).x, outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y-5);\r
            container.layout();\r
            updatePreviewTable();\r
        }\r
    }\r
\r
    private void updatePreviewTable() {\r
        final int CACHE_SIZE = 50;\r
        definition.outputs = extractOutputs();\r
\r
        try {\r
            tmpFile = TmfUiPlugin.getDefault().getStateLocation().addTrailingSeparator().append("customwizard.tmp").toFile(); //$NON-NLS-1$\r
            final FileWriter writer = new FileWriter(tmpFile);\r
            writer.write(wizard.inputPage.getInputText());\r
            writer.close();\r
\r
            final ITmfTrace<?> trace = new CustomXmlTrace(null, definition, tmpFile.getAbsolutePath(), CACHE_SIZE);\r
            previewTable.dispose();\r
            previewTable = new CustomEventsTable(definition, tableContainer, CACHE_SIZE);\r
            previewTable.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));\r
            previewTable.setTrace(trace, true);\r
        } catch (final TmfTraceException e) {\r
            e.printStackTrace();\r
        } catch (final IOException e) {\r
            e.printStackTrace();\r
        }\r
\r
        tableContainer.layout();\r
        container.layout();\r
    }\r
\r
    public List<OutputColumn> extractOutputs() {\r
        int numColumns = 0;\r
        for (int i = 0; i < outputs.size(); i++)\r
            if (outputs.get(i).enabledButton.getSelection())\r
                numColumns++;\r
        final List<OutputColumn> outputColumns = new ArrayList<OutputColumn>(numColumns);\r
        numColumns = 0;\r
        for (int i = 0; i < outputs.size(); i++) {\r
            final Output output = outputs.get(i);\r
            if (output.enabledButton.getSelection()) {\r
                final OutputColumn column = new OutputColumn();\r
                column.name = output.nameLabel.getText();\r
                outputColumns.add(column);\r
            }\r
        }\r
        return outputColumns;\r
    }\r
\r
    private class Output {\r
        String name;\r
        Button enabledButton;\r
        Text nameLabel;\r
        Button upButton;\r
        Button downButton;\r
\r
        public Output(final Composite parent, final String name) {\r
            this.name = name;\r
\r
            enabledButton = new Button(parent, SWT.CHECK);\r
            enabledButton.setToolTipText(Messages.CustomXmlParserOutputWizardPage_visible);\r
            enabledButton.setSelection(true);\r
            enabledButton.addSelectionListener(new SelectionAdapter() {\r
                @Override\r
                public void widgetSelected(final SelectionEvent e) {\r
                    updatePreviewTable();\r
                }\r
            });\r
            //            if (messageOutput != null) {\r
            //                enabledButton.moveAbove(messageOutput.enabledButton);\r
            //            }\r
\r
            nameLabel = new Text(parent, SWT.BORDER | SWT.READ_ONLY | SWT.SINGLE);\r
            nameLabel.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, false, false));\r
            nameLabel.setText(name);\r
            nameLabel.moveBelow(enabledButton);\r
\r
            upButton = new Button(parent, SWT.PUSH);\r
            upButton.setImage(upImage);\r
            upButton.setToolTipText(Messages.CustomXmlParserOutputWizardPage_moveBefore);\r
            upButton.addSelectionListener(new SelectionAdapter() {\r
                @Override\r
                public void widgetSelected(final SelectionEvent e) {\r
                    moveBefore(Output.this);\r
                }\r
            });\r
            upButton.moveBelow(nameLabel);\r
\r
            downButton = new Button(parent, SWT.PUSH);\r
            downButton.setImage(downImage);\r
            downButton.setToolTipText(Messages.CustomXmlParserOutputWizardPage_moveAfter);\r
            downButton.addSelectionListener(new SelectionAdapter() {\r
                @Override\r
                public void widgetSelected(final SelectionEvent e) {\r
                    moveAfter(Output.this);\r
                }\r
            });\r
            downButton.moveBelow(upButton);\r
        }\r
\r
        private void dispose() {\r
            enabledButton.dispose();\r
            nameLabel.dispose();\r
            upButton.dispose();\r
            downButton.dispose();\r
        }\r
    }\r
\r
    public CustomXmlTraceDefinition getDefinition() {\r
        return definition;\r
    }\r
\r
}\r