2010-10-26 Francois Chouinard <fchouinard@gmail.com> Contribution for Bug309042

[deliverable/tracecompass.git] / org.eclipse.linuxtools.tmf.ui / src / org / eclipse / linuxtools / tmf / ui / wizards / CustomTxtParserOutputWizardPage.java

package org.eclipse.linuxtools.tmf.ui.wizards;\r
\r
import java.io.File;\r
import java.io.FileNotFoundException;\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.tmf.trace.ITmfTrace;\r
import org.eclipse.linuxtools.tmf.ui.TmfUiPlugin;\r
import org.eclipse.linuxtools.tmf.ui.parsers.custom.CustomEventsTable;\r
import org.eclipse.linuxtools.tmf.ui.parsers.custom.CustomTraceDefinition.OutputColumn;\r
import org.eclipse.linuxtools.tmf.ui.parsers.custom.CustomTxtTrace;\r
import org.eclipse.linuxtools.tmf.ui.parsers.custom.CustomTxtTraceDefinition;\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 CustomTxtParserOutputWizardPage extends WizardPage {\r
\r
    private static final Image upImage = TmfUiPlugin.getDefault().getImageFromPath("/icons/up_button.gif");\r
    private static final Image downImage = TmfUiPlugin.getDefault().getImageFromPath("/icons/down_button.gif");\r
    private CustomTxtParserWizard wizard;\r
    private CustomTxtTraceDefinition 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 CustomTxtParserOutputWizardPage(CustomTxtParserWizard wizard) {\r
        super("CustomParserOutputWizardPage");\r
        setTitle(wizard.inputPage.getTitle());\r
        setDescription("Customize the output of the parser");\r
        this.wizard = wizard;\r
        setPageComplete(false);\r
    }\r
\r
	@Override\r
    public void createControl(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
        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
        GridLayout tableLayout = new GridLayout();\r
        tableLayout.marginHeight = 0;\r
        tableLayout.marginWidth = 0;\r
        tableContainer.setLayout(tableLayout);\r
        previewTable = new CustomEventsTable(new CustomTxtTraceDefinition(), tableContainer, 0);\r
        \r
        if (wizard.definition != null) {\r
            loadDefinition(wizard.definition);\r
        }\r
        setControl(container);\r
        \r
    }\r
\r
    @Override\r
    public void dispose() {\r
        previewTable.dispose();\r
        super.dispose();\r
    }\r
\r
    private void loadDefinition(CustomTxtTraceDefinition definition) {\r
        for (OutputColumn outputColumn : definition.outputs) {\r
            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(boolean visible) {\r
        if (visible) {\r
            this.definition = wizard.inputPage.getDefinition();\r
            List<String> outputNames = wizard.inputPage.getInputNames();\r
            \r
            // dispose outputs that have been removed in the input page\r
            Iterator<Output> iter = outputs.iterator();\r
            while (iter.hasNext()) {\r
                Output output = iter.next();\r
                boolean found = false;\r
                for (String name : outputNames) {\r
                    if (output.name.equals(name)) {\r
                        found = true;\r
                        break;\r
                    }\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 (String name : outputNames) {\r
                boolean found = false;\r
                for (Output output : outputs) {\r
                    if (output.name.equals(name)) {\r
                        found = true;\r
                        break;\r
                    }\r
                }\r
                if (!found) {\r
                    outputs.add(new Output(outputsContainer, name));\r
                }\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
            }\r
            setPageComplete(true);\r
        } else {\r
            setPageComplete(false);\r
        }\r
        super.setVisible(visible);\r
    }\r
\r
    private void moveBefore(Output moved) {\r
        int i = outputs.indexOf(moved);\r
        if (i > 0) {\r
            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(Output moved) {\r
        int i = outputs.indexOf(moved);\r
        if (i+1 < outputs.size()) {\r
            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 MAX_NUM_ENTRIES = 50;\r
        definition.outputs = extractOutputs();\r
\r
        try {\r
            tmpFile = TmfUiPlugin.getDefault().getStateLocation().addTrailingSeparator().append("customwizard.tmp").toFile();\r
            FileWriter writer = new FileWriter(tmpFile);\r
            writer.write(wizard.inputPage.getInputText());\r
            writer.close();\r
            \r
            ITmfTrace trace = new CustomTxtTrace(tmpFile.getName(), definition, tmpFile.getAbsolutePath(), MAX_NUM_ENTRIES);\r
            previewTable.dispose();\r
            previewTable = new CustomEventsTable(definition, tableContainer, MAX_NUM_ENTRIES);\r
            previewTable.setTrace(trace, true);\r
            previewTable.getTable().setItemCount(MAX_NUM_ENTRIES); // Trigger parsing\r
        } catch (FileNotFoundException e) {\r
            e.printStackTrace();\r
        } catch (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
            }\r
        }\r
        List<OutputColumn> outputColumns = new ArrayList<OutputColumn>(numColumns);\r
        numColumns = 0;\r
        for (int i = 0; i < outputs.size(); i++) {\r
            Output output = outputs.get(i);\r
            if (output.enabledButton.getSelection()) {\r
                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(Composite parent, String name) {\r
            this.name = name;\r
            \r
            enabledButton = new Button(parent, SWT.CHECK);\r
            enabledButton.setToolTipText("Visible");\r
            enabledButton.setSelection(true);\r
            enabledButton.addSelectionListener(new SelectionAdapter() {\r
            	@Override\r
                public void widgetSelected(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("Move Before");\r
            upButton.addSelectionListener(new SelectionAdapter() {\r
            	@Override\r
                public void widgetSelected(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("Move After");\r
            downButton.addSelectionListener(new SelectionAdapter() {\r
            	@Override\r
                public void widgetSelected(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 CustomTxtTraceDefinition getDefinition() {\r
        return definition;\r
    }\r
    \r
}\r