org.eclipse.linuxtools.tmf.ui/src/org/eclipse/linuxtools/tmf/ui/wizards/CustomXmlParserOutputWizardPage.java

   1 package org.eclipse.linuxtools.tmf.ui.wizards;
   2
   3 import java.io.File;
   4 import java.io.FileNotFoundException;
   5 import java.io.FileWriter;
   6 import java.io.IOException;
   7 import java.util.ArrayList;
   8 import java.util.Iterator;
   9 import java.util.List;
  10
  11 import org.eclipse.jface.wizard.WizardPage;
  12 import org.eclipse.linuxtools.tmf.trace.ITmfTrace;
  13 import org.eclipse.linuxtools.tmf.ui.TmfUiPlugin;
  14 import org.eclipse.linuxtools.tmf.ui.parsers.custom.CustomEventsTable;
  15 import org.eclipse.linuxtools.tmf.ui.parsers.custom.CustomTraceDefinition;
  16 import org.eclipse.linuxtools.tmf.ui.parsers.custom.CustomXmlTrace;
  17 import org.eclipse.linuxtools.tmf.ui.parsers.custom.CustomXmlTraceDefinition;
  18 import org.eclipse.linuxtools.tmf.ui.parsers.custom.CustomTraceDefinition.OutputColumn;
  19 import org.eclipse.swt.SWT;
  20 import org.eclipse.swt.custom.SashForm;
  21 import org.eclipse.swt.custom.ScrolledComposite;
  22 import org.eclipse.swt.events.SelectionAdapter;
  23 import org.eclipse.swt.events.SelectionEvent;
  24 import org.eclipse.swt.graphics.Image;
  25 import org.eclipse.swt.layout.GridData;
  26 import org.eclipse.swt.layout.GridLayout;
  27 import org.eclipse.swt.widgets.Button;
  28 import org.eclipse.swt.widgets.Composite;
  29 import org.eclipse.swt.widgets.Text;
  30
  31 public class CustomXmlParserOutputWizardPage extends WizardPage {
  32
  33     private static final Image upImage = TmfUiPlugin.getDefault().getImageFromPath("/icons/up_button.gif");
  34     private static final Image downImage = TmfUiPlugin.getDefault().getImageFromPath("/icons/down_button.gif");
  35     private CustomXmlParserWizard wizard;
  36     private CustomXmlTraceDefinition definition;
  37     ArrayList<Output> outputs = new ArrayList<Output>();
  38     Output messageOutput;
  39     Composite container;
  40     SashForm sash;
  41     Text timestampFormatText;
  42     Text timestampPreviewText;
  43     ScrolledComposite outputsScrolledComposite;
  44     Composite outputsContainer;
  45     ScrolledComposite inputScrolledComposite;
  46     Composite tableContainer;
  47     CustomEventsTable previewTable;
  48     File tmpFile;
  49
  50     protected CustomXmlParserOutputWizardPage(CustomXmlParserWizard wizard) {
  51         super("CustomParserOutputWizardPage");
  52         setTitle(wizard.inputPage.getTitle());
  53         setDescription("Customize the output of the parser");
  54         this.wizard = wizard;
  55         setPageComplete(false);
  56     }
  57
  58     public void createControl(Composite parent) {
  59         container = new Composite(parent, SWT.NULL);
  60         container.setLayout(new GridLayout());
  61
  62         sash = new SashForm(container, SWT.VERTICAL);
  63         sash.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));
  64         sash.setBackground(sash.getDisplay().getSystemColor(SWT.COLOR_GRAY));
  65
  66         outputsScrolledComposite = new ScrolledComposite(sash, SWT.V_SCROLL);
  67         outputsScrolledComposite.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));
  68         outputsContainer = new Composite(outputsScrolledComposite, SWT.NONE);
  69         GridLayout outputsLayout = new GridLayout(4, false);
  70         outputsLayout.marginHeight = 10;
  71         outputsLayout.marginWidth = 0;
  72         outputsContainer.setLayout(outputsLayout);
  73         outputsScrolledComposite.setContent(outputsContainer);
  74         outputsScrolledComposite.setExpandHorizontal(true);
  75         outputsScrolledComposite.setExpandVertical(true);
  76
  77         outputsContainer.layout();
  78
  79         outputsScrolledComposite.setMinSize(outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).x, outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y-5);
  80
  81         tableContainer = new Composite(sash, SWT.NONE);
  82         GridLayout tableLayout = new GridLayout();
  83         tableLayout.marginHeight = 0;
  84         tableLayout.marginWidth = 0;
  85         tableContainer.setLayout(tableLayout);
  86         previewTable = new CustomEventsTable(new CustomXmlTraceDefinition(), tableContainer, 0);
  87
  88         if (wizard.definition != null) {
  89             loadDefinition(wizard.definition);
  90         }
  91         setControl(container);
  92
  93     }
  94
  95     @Override
  96     public void dispose() {
  97         previewTable.dispose();
  98         super.dispose();
  99     }
 100
 101     private void loadDefinition(CustomTraceDefinition definition) {
 102         for (OutputColumn outputColumn : definition.outputs) {
 103             Output output = new Output(outputsContainer, outputColumn.name);
 104             outputs.add(output);
 105         }
 106     }
 107
 108     /* (non-Javadoc)
 109      * @see org.eclipse.jface.dialogs.DialogPage#setVisible(boolean)
 110      */
 111     @Override
 112     public void setVisible(boolean visible) {
 113         if (visible) {
 114             this.definition = wizard.inputPage.getDefinition();
 115             List<String> outputNames = wizard.inputPage.getInputNames();
 116
 117             // dispose outputs that have been removed in the input page
 118             Iterator<Output> iter = outputs.iterator();
 119             while (iter.hasNext()) {
 120                 Output output = iter.next();
 121                 boolean found = false;
 122                 for (String name : outputNames) {
 123                     if (output.name.equals(name)) {
 124                         found = true;
 125                         break;
 126                     }
 127                 }
 128                 if (!found) {
 129                     output.dispose();
 130                     iter.remove();
 131                 }
 132             }
 133
 134             // create outputs that have been added in the input page
 135             for (String name : outputNames) {
 136                 boolean found = false;
 137                 for (Output output : outputs) {
 138                     if (output.name.equals(name)) {
 139                         found = true;
 140                         break;
 141                     }
 142                 }
 143                 if (!found) {
 144                     outputs.add(new Output(outputsContainer, name));
 145                 }
 146             }
 147
 148             outputsContainer.layout();
 149             outputsScrolledComposite.setMinSize(outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).x, outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y-5);
 150             updatePreviewTable();
 151             if (sash.getSize().y > outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y + previewTable.getTable().getItemHeight()) {
 152                 sash.setWeights(new int[] {outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y, sash.getSize().y - outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y});
 153             } else {
 154                 sash.setWeights(new int[] {outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y, previewTable.getTable().getItemHeight()});
 155             }
 156             setPageComplete(true);
 157         } else {
 158             setPageComplete(false);
 159         }
 160         super.setVisible(visible);
 161     }
 162
 163     private void moveBefore(Output moved) {
 164         int i = outputs.indexOf(moved);
 165         if (i > 0) {
 166             Output previous = outputs.get(i-1);
 167             moved.enabledButton.moveAbove(previous.enabledButton);
 168             moved.nameLabel.moveBelow(moved.enabledButton);
 169             moved.upButton.moveBelow(moved.nameLabel);
 170             moved.downButton.moveBelow(moved.upButton);
 171             outputs.add(i-1, outputs.remove(i));
 172             outputsContainer.layout();
 173             outputsScrolledComposite.setMinSize(outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).x, outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y-5);
 174             container.layout();
 175             updatePreviewTable();
 176         }
 177     }
 178
 179     private void moveAfter(Output moved) {
 180         int i = outputs.indexOf(moved);
 181         if (i+1 < outputs.size()) {
 182             Output next = outputs.get(i+1);
 183             moved.enabledButton.moveBelow(next.downButton);
 184             moved.nameLabel.moveBelow(moved.enabledButton);
 185             moved.upButton.moveBelow(moved.nameLabel);
 186             moved.downButton.moveBelow(moved.upButton);
 187             outputs.add(i+1, outputs.remove(i));
 188             outputsContainer.layout();
 189             outputsScrolledComposite.setMinSize(outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).x, outputsContainer.computeSize(SWT.DEFAULT, SWT.DEFAULT).y-5);
 190             container.layout();
 191             updatePreviewTable();
 192         }
 193     }
 194
 195     private void updatePreviewTable() {
 196         final int MAX_NUM_ENTRIES = 50;
 197         definition.outputs = extractOutputs();
 198
 199         try {
 200             tmpFile = TmfUiPlugin.getDefault().getStateLocation().addTrailingSeparator().append("customwizard.tmp").toFile();
 201             FileWriter writer = new FileWriter(tmpFile);
 202             writer.write(wizard.inputPage.getInputText());
 203             writer.close();
 204
 205             ITmfTrace trace = new CustomXmlTrace(tmpFile.getName(), definition, tmpFile.getAbsolutePath(), MAX_NUM_ENTRIES);
 206             previewTable.dispose();
 207             previewTable = new CustomEventsTable(definition, tableContainer, MAX_NUM_ENTRIES);
 208             previewTable.setTrace(trace);
 209             previewTable.getTable().setItemCount(MAX_NUM_ENTRIES); // Trigger parsing
 210         } catch (FileNotFoundException e) {
 211             e.printStackTrace();
 212         } catch (IOException e) {
 213             e.printStackTrace();
 214         }
 215
 216         tableContainer.layout();
 217         container.layout();
 218     }
 219
 220     public List<OutputColumn> extractOutputs() {
 221         int numColumns = 0;
 222         for (int i = 0; i < outputs.size(); i++) {
 223             if (outputs.get(i).enabledButton.getSelection()) {
 224                 numColumns++;
 225             }
 226         }
 227         List<OutputColumn> outputColumns = new ArrayList<OutputColumn>(numColumns);
 228         numColumns = 0;
 229         for (int i = 0; i < outputs.size(); i++) {
 230             Output output = outputs.get(i);
 231             if (output.enabledButton.getSelection()) {
 232                 OutputColumn column = new OutputColumn();
 233                 column.name = output.nameLabel.getText();
 234                 outputColumns.add(column);
 235             }
 236         }
 237         return outputColumns;
 238     }
 239
 240     private class Output {
 241         String name;
 242         Button enabledButton;
 243         Text nameLabel;
 244         Button upButton;
 245         Button downButton;
 246
 247         public Output(Composite parent, String name) {
 248             this.name = name;
 249
 250             enabledButton = new Button(parent, SWT.CHECK);
 251             enabledButton.setToolTipText("Visible");
 252             enabledButton.setSelection(true);
 253             enabledButton.addSelectionListener(new SelectionAdapter() {
 254                 public void widgetSelected(SelectionEvent e) {
 255                     updatePreviewTable();
 256                 }
 257             });
 258             if (messageOutput != null) {
 259                 enabledButton.moveAbove(messageOutput.enabledButton);
 260             }
 261
 262             nameLabel = new Text(parent, SWT.BORDER | SWT.READ_ONLY | SWT.SINGLE);
 263             nameLabel.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, false, false));
 264             nameLabel.setText(name);
 265             nameLabel.moveBelow(enabledButton);
 266
 267             upButton = new Button(parent, SWT.PUSH);
 268             upButton.setImage(upImage);
 269             upButton.setToolTipText("Move Before");
 270             upButton.addSelectionListener(new SelectionAdapter() {
 271                 public void widgetSelected(SelectionEvent e) {
 272                     moveBefore(Output.this);
 273                 }
 274             });
 275             upButton.moveBelow(nameLabel);
 276
 277             downButton = new Button(parent, SWT.PUSH);
 278             downButton.setImage(downImage);
 279             downButton.setToolTipText("Move After");
 280             downButton.addSelectionListener(new SelectionAdapter() {
 281                 public void widgetSelected(SelectionEvent e) {
 282                     moveAfter(Output.this);
 283                 }
 284             });
 285             downButton.moveBelow(upButton);
 286         }
 287
 288         private void dispose() {
 289             enabledButton.dispose();
 290             nameLabel.dispose();
 291             upButton.dispose();
 292             downButton.dispose();
 293         }
 294     }
 295
 296     public CustomXmlTraceDefinition getDefinition() {
 297         return definition;
 298     }
 299
 300 }