Java tutorial
/* * Copyright (c) 2012 Diamond Light Source Ltd. * * All rights reserved. This program and the accompanying materials * are made available under the terms of the Eclipse Public License v1.0 * which accompanies this distribution, and is available at * http://www.eclipse.org/legal/epl-v10.html */ package org.dawnsci.plotting.tools.masking; import java.io.Serializable; import java.util.ArrayList; import java.util.Collection; import java.util.HashSet; import java.util.List; import org.dawb.common.ui.image.CursorUtils; import org.dawb.common.ui.image.IconUtils; import org.dawb.common.ui.image.ShapeType; import org.dawb.common.ui.menu.CheckableActionGroup; import org.dawb.common.ui.menu.MenuAction; import org.dawb.common.ui.util.EclipseUtils; import org.dawb.common.ui.util.GridUtils; import org.dawb.common.ui.wizard.persistence.PersistenceExportWizard; import org.dawb.common.ui.wizard.persistence.PersistenceImportWizard; import org.dawnsci.common.widgets.spinner.FloatSpinner; import org.dawnsci.plotting.AbstractPlottingViewer; import org.dawnsci.plotting.tools.Activator; import org.dawnsci.plotting.util.ColorUtility; import org.eclipse.core.commands.operations.IOperationHistoryListener; import org.eclipse.core.commands.operations.OperationHistoryEvent; import org.eclipse.core.runtime.IProgressMonitor; import org.eclipse.core.runtime.IStatus; import org.eclipse.core.runtime.NullProgressMonitor; import org.eclipse.core.runtime.Status; import org.eclipse.core.runtime.jobs.Job; import org.eclipse.dawnsci.analysis.dataset.impl.BooleanDataset; import org.eclipse.dawnsci.analysis.dataset.impl.Dataset; import org.eclipse.dawnsci.plotting.api.IPlottingSystem; import org.eclipse.dawnsci.plotting.api.histogram.HistogramBound; import org.eclipse.dawnsci.plotting.api.preferences.BasePlottingConstants; import org.eclipse.dawnsci.plotting.api.preferences.PlottingConstants; import org.eclipse.dawnsci.plotting.api.region.IROIListener; import org.eclipse.dawnsci.plotting.api.region.IRegion; import org.eclipse.dawnsci.plotting.api.region.IRegion.RegionType; import org.eclipse.dawnsci.plotting.api.region.IRegionAction; import org.eclipse.dawnsci.plotting.api.region.IRegionListener; import org.eclipse.dawnsci.plotting.api.region.ROIEvent; import org.eclipse.dawnsci.plotting.api.region.RegionEvent; import org.eclipse.dawnsci.plotting.api.region.RegionUtils; import org.eclipse.dawnsci.plotting.api.tool.AbstractToolPage; import org.eclipse.dawnsci.plotting.api.tool.IToolPage; import org.eclipse.dawnsci.plotting.api.trace.IImageTrace; import org.eclipse.dawnsci.plotting.api.trace.IPaletteListener; import org.eclipse.dawnsci.plotting.api.trace.ITraceListener; import org.eclipse.dawnsci.plotting.api.trace.PaletteEvent; import org.eclipse.dawnsci.plotting.api.trace.TraceEvent; import org.eclipse.draw2d.MouseMotionListener; import org.eclipse.draw2d.geometry.PrecisionPoint; import org.eclipse.jface.action.Action; import org.eclipse.jface.action.ActionContributionItem; import org.eclipse.jface.action.IAction; import org.eclipse.jface.action.Separator; import org.eclipse.jface.action.ToolBarManager; import org.eclipse.jface.dialogs.ErrorDialog; import org.eclipse.jface.dialogs.MessageDialog; import org.eclipse.jface.preference.ColorSelector; import org.eclipse.jface.resource.ImageDescriptor; import org.eclipse.jface.util.IPropertyChangeListener; import org.eclipse.jface.util.PropertyChangeEvent; import org.eclipse.jface.viewers.CellEditor; import org.eclipse.jface.viewers.ColumnLabelProvider; import org.eclipse.jface.viewers.ColumnViewer; import org.eclipse.jface.viewers.ColumnViewerToolTipSupport; import org.eclipse.jface.viewers.EditingSupport; import org.eclipse.jface.viewers.IStructuredContentProvider; import org.eclipse.jface.viewers.TableViewer; import org.eclipse.jface.viewers.TableViewerColumn; import org.eclipse.jface.viewers.TextCellEditor; import org.eclipse.jface.viewers.Viewer; import org.eclipse.jface.viewers.ViewerCell; import org.eclipse.jface.window.ToolTip; import org.eclipse.jface.wizard.IWizard; import org.eclipse.jface.wizard.WizardDialog; import org.eclipse.swt.SWT; import org.eclipse.swt.custom.CLabel; import org.eclipse.swt.custom.ScrolledComposite; import org.eclipse.swt.custom.StackLayout; import org.eclipse.swt.events.ControlAdapter; import org.eclipse.swt.events.ControlEvent; import org.eclipse.swt.events.KeyAdapter; import org.eclipse.swt.events.KeyEvent; import org.eclipse.swt.events.MouseEvent; import org.eclipse.swt.events.MouseListener; import org.eclipse.swt.events.SelectionAdapter; import org.eclipse.swt.events.SelectionEvent; import org.eclipse.swt.graphics.Image; import org.eclipse.swt.graphics.Point; import org.eclipse.swt.graphics.RGB; import org.eclipse.swt.graphics.Rectangle; import org.eclipse.swt.layout.GridData; import org.eclipse.swt.layout.GridLayout; import org.eclipse.swt.widgets.Button; import org.eclipse.swt.widgets.Composite; import org.eclipse.swt.widgets.Control; import org.eclipse.swt.widgets.Display; import org.eclipse.swt.widgets.Group; import org.eclipse.swt.widgets.Label; import org.eclipse.swt.widgets.TableItem; import org.eclipse.ui.IActionBars; import org.eclipse.ui.IViewPart; import org.slf4j.Logger; import org.slf4j.LoggerFactory; public class MaskingTool extends AbstractToolPage implements MouseListener { private static final Logger logger = LoggerFactory.getLogger(MaskingTool.class); private ScrolledComposite scrollComp; private FloatSpinner minimum, maximum; private Button autoApply; private MaskObject maskObject; private MaskJob maskJob; private TableViewer regionTable; private ToolBarManager directToolbar; private Button apply; private IPaletteListener paletteListener; private ITraceListener traceListener; private IRegionListener regionListener; private IROIListener regionBoundsListener; private MaskMouseListener clickListener; private ColorSelector colorSelector; private Collection<Control> enableControls; public MaskingTool() { enableControls = new HashSet<Control>(7); this.traceListener = new ITraceListener.Stub() { @Override public void traceAdded(TraceEvent evt) { updateMask(evt); } @Override public void traceUpdated(TraceEvent evt) { if (!autoApply.getSelection()) return; processMask(true, false, null); } @Override public void traceRemoved(TraceEvent evt) { if (evt.getSource() instanceof IImageTrace) { ((IImageTrace) evt.getSource()).removePaletteListener(paletteListener); } } }; this.paletteListener = new IPaletteListener.Stub() { @Override public void nanBoundsChanged(PaletteEvent evt) { updateIcons(evt.getTrace().getNanBound().getColor()); } }; this.clickListener = new MaskMouseListener(); this.regionListener = new IRegionListener.Stub() { @Override public void regionCreated(RegionEvent evt) { // Those created while the tool is active are mask regions evt.getRegion().setMaskRegion(true); if (MaskMarker.MASK_REGION == evt.getRegion().getUserObject()) { int wid = Activator.getPlottingPreferenceStore().getInt(PlottingConstants.FREE_DRAW_WIDTH); evt.getRegion().setLineWidth(wid); evt.getRegion().setUserObject(MaskObject.MaskRegionType.REGION_FROM_MASKING); } } @Override public void regionAdded(final RegionEvent evt) { if (MaskMarker.MASK_REGION == evt.getRegion().getUserObject()) { int wid = Activator.getPlottingPreferenceStore().getInt(PlottingConstants.FREE_DRAW_WIDTH); evt.getRegion().setLineWidth(wid); evt.getRegion().setUserObject(MaskObject.MaskRegionType.REGION_FROM_MASKING); } setLastActionRange(false); evt.getRegion().addROIListener(regionBoundsListener); processMask(evt.getRegion()); regionTable.refresh(); if (Activator.getPlottingPreferenceStore().getBoolean(PlottingConstants.MASK_DRAW_MULTIPLE)) { Display.getDefault().asyncExec(new Runnable() { public void run() { try { getPlottingSystem().createRegion( RegionUtils.getUniqueName(evt.getRegion().getRegionType().getName(), getPlottingSystem()), evt.getRegion().getRegionType()); } catch (Exception e) { logger.error("Cannot add multple regions.", e); } } }); } } @Override public void regionRemoved(RegionEvent evt) { evt.getRegion().removeROIListener(regionBoundsListener); processMask(true, false, null); regionTable.refresh(); } @Override public void regionsRemoved(RegionEvent evt) { processMask(true, false, null); regionTable.refresh(); } }; this.regionBoundsListener = new IROIListener.Stub() { @Override public void roiChanged(ROIEvent evt) { processMask((IRegion) evt.getSource()); } }; this.maskJob = new MaskJob(); maskJob.setPriority(Job.INTERACTIVE); maskJob.setUser(false); } private void updateMask(TraceEvent evt) { try { setEnabled(getImageTrace() != null); if (evt.getSource() instanceof IImageTrace) { ((IImageTrace) evt.getSource()).setMask(maskObject.getMaskDataset()); ((IImageTrace) evt.getSource()).addPaletteListener(paletteListener); int[] ia = ((IImageTrace) evt.getSource()).getImageServiceBean().getNanBound().getColor(); updateIcons(ia); colorSelector.setColorValue(ColorUtility.getRGB(ia)); if (autoApplySavedMask && savedMask != null) { Display.getDefault().asyncExec(new Runnable() { public void run() { try { mergeSavedMask(); } catch (Throwable ne) { logger.error("Problem loading saved mask!", ne); } } }); } } else { saveMaskBuffer(); } } catch (Exception ne) { logger.error("Cannot update trace!", ne); } } private AbstractPlottingViewer viewer; protected final class MaskMouseListener extends MouseMotionListener.Stub implements org.eclipse.draw2d.MouseListener { @Override public void mousePressed(org.eclipse.draw2d.MouseEvent me) { if (me.button != 1) return; if (viewer.getSelectedCursor() == null) { ActionContributionItem item = (ActionContributionItem) directToolbar.find(ShapeType.NONE.getId()); if (item != null) item.getAction().setChecked(true); return; } setLastActionRange(false); maskJob.schedule(false, null, me.getLocation()); } @Override public void mouseDragged(org.eclipse.draw2d.MouseEvent me) { if (me.button != 0) return; if (viewer.getSelectedCursor() == null) { ActionContributionItem item = (ActionContributionItem) directToolbar.find(ShapeType.NONE.getId()); if (item != null) item.getAction().setChecked(true); return; } setLastActionRange(false); maskJob.schedule(false, null, me.getLocation()); } @Override public void mouseReleased(org.eclipse.draw2d.MouseEvent me) { // record shift point viewer.setShiftPoint(me.getLocation()); } @Override public void mouseDoubleClicked(org.eclipse.draw2d.MouseEvent me) { } } public void setPlottingSystem(IPlottingSystem system) { super.setPlottingSystem(system); this.maskObject = new MaskObject(); //TODO maybe make maskCreator by only processing visible regions. } @Override public ToolPageRole getToolPageRole() { return ToolPageRole.ROLE_2D; } @Override public void createControl(Composite parent) { scrollComp = new ScrolledComposite(parent, SWT.V_SCROLL | SWT.H_SCROLL); final Group composite = new Group(scrollComp, SWT.NONE); composite.setLayout(new GridLayout(1, false)); final IImageTrace image = getImageTrace(); if (image != null) { composite.setText("Masking '" + image.getName() + "'"); } else { composite.setText("Masking "); } final Composite minMaxComp = new Composite(composite, SWT.NONE); minMaxComp.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, false)); minMaxComp.setLayout(new GridLayout(2, false)); Label label = new Label(minMaxComp, SWT.WRAP); label.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, false, 2, 1)); label.setText("Create a mask, the mask can be saved and available in other tools."); final CLabel warningMessage = new CLabel(minMaxComp, SWT.WRAP); warningMessage.setText("Changing lower / upper can reset the mask and is not undoable."); warningMessage.setToolTipText( "The reset can occur because the algorithm which processes intensity values,\ndoes not know if the mask pixel should be unmasked or not.\nIt can only take into account intensity.\nTherefore it is best to define intensity masks first,\nbefore the custom masks using pen or region tools."); warningMessage.setImage(Activator.getImage("icons/error.png")); warningMessage.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, false, 2, 1)); GridUtils.setVisible(warningMessage, false); // Max and min final Button minEnabled = new Button(minMaxComp, SWT.CHECK); minEnabled.setLayoutData(new GridData(SWT.LEFT, SWT.CENTER, false, false, 1, 1)); minEnabled.setText("Enable lower mask "); minEnabled.setToolTipText("Enable the lower bound mask, removing pixels with lower intensity."); enableControls.add(minEnabled); this.minimum = new FloatSpinner(minMaxComp, SWT.NONE); minimum.setIncrement(1d); minimum.setEnabled(false); minimum.setMinimum(Integer.MIN_VALUE); minimum.setMaximum(Integer.MAX_VALUE); minimum.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false, 1, 1)); if (image != null) minimum.setDouble(getValue(image.getMin(), image.getMinCut(), 0)); minimum.setToolTipText("Press enter to apply a full update of the mask."); minimum.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { processMask(true, false, null); setLastActionRange(true); } }); minimum.addKeyListener(new KeyAdapter() { public void keyPressed(KeyEvent e) { if (e.character == '\n' || e.character == '\r') { processMask(isLastActionRange(), true, null); } } }); final Button maxEnabled = new Button(minMaxComp, SWT.CHECK); maxEnabled.setLayoutData(new GridData(SWT.LEFT, SWT.CENTER, false, false, 1, 1)); maxEnabled.setText("Enable upper mask "); maxEnabled.setToolTipText("Enable the upper bound mask, removing pixels with higher intensity."); enableControls.add(maxEnabled); minEnabled.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { minimum.setEnabled(minEnabled.getSelection()); GridUtils.setVisible(warningMessage, minEnabled.getSelection() || maxEnabled.getSelection()); warningMessage.getParent().getParent().layout(); if (!minEnabled.getSelection()) { warningMessage.getParent().getParent().layout(); } else { processMask(false, true, null); } } }); maxEnabled.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { maximum.setEnabled(maxEnabled.getSelection()); GridUtils.setVisible(warningMessage, minEnabled.getSelection() || maxEnabled.getSelection()); warningMessage.getParent().getParent().layout(); if (!maxEnabled.getSelection()) { processMask(true, true, null); } else { processMask(false, true, null); } } }); this.maximum = new FloatSpinner(minMaxComp, SWT.NONE); maximum.setIncrement(1d); maximum.setEnabled(false); maximum.setMinimum(Integer.MIN_VALUE); maximum.setMaximum(Integer.MAX_VALUE); maximum.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false, 1, 1)); if (image != null) maximum.setDouble(getValue(image.getMax(), image.getMaxCut(), Integer.MAX_VALUE)); maximum.setToolTipText("Press enter to apply a full update of the mask."); maximum.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { processMask(true, false, null); setLastActionRange(true); } }); maximum.addKeyListener(new KeyAdapter() { public void keyPressed(KeyEvent e) { if (e.character == '\n' || e.character == '\r') { processMask(isLastActionRange(), true, null); } } }); final Button ignoreAlreadyMasked = new Button(minMaxComp, SWT.CHECK); enableControls.add(ignoreAlreadyMasked); ignoreAlreadyMasked.setLayoutData(new GridData(SWT.LEFT, SWT.CENTER, true, false, 2, 1)); ignoreAlreadyMasked.setText("Keep pixels already masked"); ignoreAlreadyMasked.setToolTipText( "When using bounds, pixels already masked can be ignored and not checked for range.\nThis setting is ignored when removing the bounds mask."); ignoreAlreadyMasked.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { maskObject.setIgnoreAlreadyMasked(ignoreAlreadyMasked.getSelection()); } }); label = new Label(minMaxComp, SWT.NONE); label.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false, 2, 1)); label = new Label(minMaxComp, SWT.NONE); label.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false, 1, 1)); label.setText("Mask Color"); enableControls.add(label); this.colorSelector = new ColorSelector(minMaxComp); enableControls.add(colorSelector.getButton()); colorSelector.getButton().setLayoutData(new GridData(SWT.LEFT, SWT.CENTER, false, false, 1, 1)); if (image != null) colorSelector.setColorValue(ColorUtility.getRGB(image.getNanBound().getColor())); colorSelector.addListener(new IPropertyChangeListener() { @Override public void propertyChange(PropertyChangeEvent event) { getImageTrace().setNanBound( new HistogramBound(Double.NaN, ColorUtility.getIntArray(colorSelector.getColorValue()))); getImageTrace().rehistogram(); } }); final Group drawGroup = new Group(composite, SWT.SHADOW_NONE); drawGroup.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true)); drawGroup.setLayout(new GridLayout(3, false)); label = new Label(drawGroup, SWT.NONE); GridData data = new GridData(SWT.LEFT, SWT.FILL, false, false); label.setLayoutData(data); label.setText("Mask using: "); enableControls.add(label); final Button directDraw = new Button(drawGroup, SWT.RADIO); directDraw.setText("Direct draw"); directDraw.setLayoutData(new GridData(SWT.LEFT, SWT.FILL, false, false)); enableControls.add(directDraw); final Button regionDraw = new Button(drawGroup, SWT.RADIO); regionDraw.setText("Regions"); regionDraw.setLayoutData(new GridData(SWT.LEFT, SWT.FILL, true, false)); enableControls.add(regionDraw); label = new Label(drawGroup, SWT.HORIZONTAL | SWT.SEPARATOR); GridData gdata = new GridData(SWT.FILL, SWT.FILL, true, false, 3, 1); gdata.verticalIndent = 5; label.setLayoutData(gdata); final Composite drawContent = new Composite(drawGroup, SWT.NONE); drawContent.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true, 3, 1)); drawContent.setLayout(new StackLayout()); final Composite directComp = new Composite(drawContent, SWT.NONE); directComp.setLayout(new GridLayout(1, false)); GridUtils.removeMargins(directComp); this.directToolbar = new ToolBarManager(SWT.FLAT | SWT.RIGHT | SWT.WRAP); createDirectToolbarActions(directToolbar); Control tb = directToolbar.createControl(directComp); tb.setLayoutData(new GridData(SWT.RIGHT, SWT.TOP, true, false)); enableControls.add(directToolbar.getControl()); new Label(directComp, SWT.NONE); final Label shiftLabel = new Label(directComp, SWT.WRAP); shiftLabel.setLayoutData(new GridData(SWT.RIGHT, SWT.BOTTOM, true, false)); shiftLabel.setText("(Hold down the 'shift' key to draw lines.)"); new Label(directComp, SWT.NONE); new Label(directComp, SWT.NONE); final Button useThresh = new Button(directComp, SWT.CHECK); useThresh.setLayoutData(new GridData(SWT.FILL, SWT.BOTTOM, true, false)); useThresh.setText("Use threshold on brush."); enableControls.add(useThresh); final Composite threshComp = new Composite(directComp, SWT.NONE); threshComp.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true)); threshComp.setLayout(new GridLayout(2, false)); final Label minThreshLabel = new Label(threshComp, SWT.NONE); threshComp.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false)); minThreshLabel.setText("Minimum Threshold"); enableControls.add(minThreshLabel); final FloatSpinner minThresh = new FloatSpinner(threshComp, SWT.NONE); minThresh.setIncrement(1d); minThresh.setMinimum(Integer.MIN_VALUE); minThresh.setMaximum(Integer.MAX_VALUE); minThresh.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false, 1, 1)); if (image != null) minThresh.setDouble(getValue(image.getMin(), image.getMinCut(), 0)); minThresh.setToolTipText("Press enter to set minimum threshold for brush."); enableControls.add(minThresh); final Label maxThreshLabel = new Label(threshComp, SWT.NONE); maxThreshLabel.setLayoutData(new GridData(SWT.FILL, SWT.FILL, false, false)); maxThreshLabel.setText("Maximum Threshold"); enableControls.add(maxThreshLabel); final FloatSpinner maxThresh = new FloatSpinner(threshComp, SWT.NONE); maxThresh.setIncrement(1d); maxThresh.setMinimum(Integer.MIN_VALUE); maxThresh.setMaximum(Integer.MAX_VALUE); maxThresh.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false, 1, 1)); if (image != null) maxThresh.setDouble(getValue(image.getMax(), image.getMaxCut(), Integer.MAX_VALUE)); maxThresh.setToolTipText("Press enter to set maximum threshold for brush."); enableControls.add(maxThresh); minThresh.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { maskObject.setBrushThreshold(new PrecisionPoint(minThresh.getDouble(), maxThresh.getDouble())); } }); minThresh.addKeyListener(new KeyAdapter() { public void keyPressed(KeyEvent e) { if (e.character == '\n' || e.character == '\r') { maskObject.setBrushThreshold(new PrecisionPoint(minThresh.getDouble(), maxThresh.getDouble())); } } }); maxThresh.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { maskObject.setBrushThreshold(new PrecisionPoint(minThresh.getDouble(), maxThresh.getDouble())); } }); maxThresh.addKeyListener(new KeyAdapter() { public void keyPressed(KeyEvent e) { if (e.character == '\n' || e.character == '\r') { maskObject.setBrushThreshold(new PrecisionPoint(minThresh.getDouble(), maxThresh.getDouble())); } } }); useThresh.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { GridUtils.setVisible(threshComp, useThresh.getSelection()); PrecisionPoint thresh = useThresh.getSelection() ? new PrecisionPoint(minThresh.getDouble(), maxThresh.getDouble()) : null; maskObject.setBrushThreshold(thresh); threshComp.getParent().layout(); } }); GridUtils.setVisible(threshComp, false); // Regions final Composite regionComp = new Composite(drawContent, SWT.NONE); regionComp.setLayout(new GridLayout(1, false)); GridUtils.removeMargins(regionComp); final ToolBarManager regionToolbar = new ToolBarManager(SWT.FLAT | SWT.RIGHT); enableControls.add(regionToolbar.getControl()); createMaskingRegionActions(regionToolbar); tb = regionToolbar.createControl(regionComp); tb.setLayoutData(new GridData(SWT.RIGHT, SWT.TOP, true, false)); this.regionTable = new TableViewer(regionComp, SWT.FULL_SELECTION | SWT.SINGLE | SWT.H_SCROLL | SWT.V_SCROLL | SWT.BORDER); regionTable.getTable().setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true)); createColumns(regionTable); regionTable.getTable().setLinesVisible(true); regionTable.getTable().setHeaderVisible(true); regionTable.getTable().addMouseListener(this); getSite().setSelectionProvider(regionTable); regionTable.setContentProvider(new IStructuredContentProvider() { @Override public void inputChanged(Viewer viewer, Object oldInput, Object newInput) { } @Override public void dispose() { } @Override public Object[] getElements(Object inputElement) { final Collection<IRegion> regions = getPlottingSystem().getRegions(); if (regions == null || regions.isEmpty()) return new Object[] { "-" }; final List<IRegion> supported = new ArrayList<IRegion>(regions.size()); for (IRegion iRegion : regions) if (maskObject.isSupportedRegion(iRegion) && iRegion.isUserRegion()) { supported.add(iRegion); } return supported.toArray(new IRegion[supported.size()]); } }); regionTable.setInput(new Object()); enableControls.add(regionTable.getControl()); final Composite buttons = new Composite(drawGroup, SWT.BORDER); buttons.setLayout(new GridLayout(2, false)); buttons.setLayoutData(new GridData(SWT.FILL, SWT.BOTTOM, true, false, 3, 1)); this.autoApply = new Button(buttons, SWT.CHECK); enableControls.add(autoApply); autoApply.setLayoutData(new GridData(SWT.LEFT, SWT.CENTER, true, true, 2, 1)); autoApply.setText("Automatically apply mask when something changes."); autoApply .setSelection(Activator.getPlottingPreferenceStore().getBoolean(PlottingConstants.MASK_AUTO_APPLY)); this.apply = new Button(buttons, SWT.PUSH); apply.setLayoutData(new GridData(SWT.CENTER, SWT.CENTER, true, true)); apply.setImage(Activator.getImage("icons/apply.gif")); apply.setText("Apply"); apply.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { processMask(isLastActionRange(), true, null); } }); apply.setEnabled(!Activator.getPlottingPreferenceStore().getBoolean(PlottingConstants.MASK_AUTO_APPLY)); enableControls.add(apply); autoApply.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_AUTO_APPLY, autoApply.getSelection()); apply.setEnabled(!autoApply.getSelection()); processMask(); } }); final Button reset = new Button(buttons, SWT.PUSH); reset.setLayoutData(new GridData(SWT.CENTER, SWT.CENTER, true, true)); reset.setImage(Activator.getImage("icons/reset.gif")); reset.setText("Reset"); reset.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { resetMask(); } }); enableControls.add(reset); createActions(getSite().getActionBars()); final StackLayout layout = (StackLayout) drawContent.getLayout(); layout.topControl = directComp; directDraw.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_DRAW_TYPE, "direct"); layout.topControl = directComp; drawContent.layout(); ShapeType penShape = ShapeType.valueOf( Activator.getPlottingPreferenceStore().getString(PlottingConstants.MASK_PEN_SHAPE)); ActionContributionItem item = ((ActionContributionItem) directToolbar.find(penShape.getId())); if (item != null) item.getAction().run(); setRegionsVisible(false); } }); regionDraw.addSelectionListener(new SelectionAdapter() { public void widgetSelected(SelectionEvent e) { Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_DRAW_TYPE, "region"); layout.topControl = regionComp; drawContent.layout(); viewer.setSelectedCursor(null); setRegionsVisible(true); } }); String drawType = Activator.getPlottingPreferenceStore().getString(PlottingConstants.MASK_DRAW_TYPE); if (drawType != null && !"".equals(drawType)) { if ("direct".equals(drawType)) { directDraw.setSelection(true); layout.topControl = directComp; ShapeType penShape = ShapeType.NONE; if (Activator.getPlottingPreferenceStore().contains(PlottingConstants.MASK_PEN_SHAPE)) { penShape = ShapeType.valueOf( Activator.getPlottingPreferenceStore().getString(PlottingConstants.MASK_PEN_SHAPE)); } ActionContributionItem item = ((ActionContributionItem) directToolbar.find(penShape.getId())); if (item != null) item.getAction().run(); } else if ("region".equals(drawType)) { regionDraw.setSelection(true); layout.topControl = regionComp; } } scrollComp.setContent(composite); scrollComp.setExpandVertical(true); scrollComp.setExpandHorizontal(true); scrollComp.addControlListener(new ControlAdapter() { @Override public void controlResized(ControlEvent e) { Rectangle r = scrollComp.getClientArea(); scrollComp.setMinHeight(composite.computeSize(r.width, SWT.DEFAULT).y); scrollComp.setMinWidth(composite.computeSize(SWT.DEFAULT, r.height).x); } }); try { enableControls.add(((ToolBarManager) getViewPart().getViewSite().getActionBars().getToolBarManager()) .getControl()); } catch (Throwable ignored) { } } private boolean lastActionRange = false; protected void setLastActionRange(boolean isRange) { lastActionRange = isRange; } protected boolean isLastActionRange() { return lastActionRange; } protected void setRegionsVisible(boolean isVis) { if (getPlottingSystem() == null) return; final Collection<IRegion> regions = getPlottingSystem().getRegions(); for (IRegion iRegion : regions) { if (iRegion.isMaskRegion()) iRegion.setVisible(isVis); } } /** * Actions for * @param freeToolbar2 */ private void createDirectToolbarActions(ToolBarManager man) { final MenuAction penSize = new MenuAction("Pen Size"); penSize.setToolTipText("Pen size"); man.add(penSize); CheckableActionGroup group = new CheckableActionGroup(); final int[] pensizes = new int[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 14, 16, 18, 20, 24, 32, 64 }; Action currentSize = null; for (final int pensize : pensizes) { final Action action = new Action("Pen size of " + String.valueOf(pensize), IAction.AS_CHECK_BOX) { public void run() { Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_PEN_SIZE, pensize); penSize.setSelectedAction(this); penSize.setToolTipText(getToolTipText()); ShapeType penShape = ShapeType.valueOf( Activator.getPlottingPreferenceStore().getString(PlottingConstants.MASK_PEN_SHAPE)); if (penShape != null) { viewer.setSelectedCursor(CursorUtils.getPenCursor(pensize, penShape)); } } }; action.setImageDescriptor(IconUtils.createPenDescriptor(pensize)); penSize.add(action); group.add(action); if (Activator.getPlottingPreferenceStore().getInt(PlottingConstants.MASK_PEN_SIZE) == pensize) { currentSize = action; } action.setToolTipText("Set pen size to " + pensize); } if (currentSize != null) { currentSize.setChecked(true); penSize.setSelectedAction(currentSize); penSize.setToolTipText(currentSize.getToolTipText()); } man.add(new Separator()); group = new CheckableActionGroup(); Action action = new Action("Square brush", IAction.AS_CHECK_BOX) { public void run() { int pensize = Activator.getPlottingPreferenceStore().getInt(PlottingConstants.MASK_PEN_SIZE); ShapeType penShape = ShapeType.SQUARE; viewer.setSelectedCursor(CursorUtils.getPenCursor(pensize, penShape)); Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_PEN_SHAPE, penShape.name()); } }; action.setId(ShapeType.SQUARE.getId()); group.add(action); man.add(action); action = new Action("Triangle brush", IAction.AS_CHECK_BOX) { public void run() { int pensize = Activator.getPlottingPreferenceStore().getInt(PlottingConstants.MASK_PEN_SIZE); ShapeType penShape = ShapeType.TRIANGLE; viewer.setSelectedCursor(CursorUtils.getPenCursor(pensize, penShape)); Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_PEN_SHAPE, penShape.name()); } }; action.setId(ShapeType.TRIANGLE.getId()); group.add(action); man.add(action); action = new Action("Circular brush", IAction.AS_CHECK_BOX) { public void run() { int pensize = Activator.getPlottingPreferenceStore().getInt(PlottingConstants.MASK_PEN_SIZE); ShapeType penShape = ShapeType.CIRCLE; viewer.setSelectedCursor(CursorUtils.getPenCursor(pensize, penShape)); Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_PEN_SHAPE, penShape.name()); } }; action.setId(ShapeType.CIRCLE.getId()); group.add(action); man.add(action); action = new Action("None", IAction.AS_CHECK_BOX) { public void run() { ShapeType penShape = ShapeType.NONE; Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_PEN_SHAPE, penShape.name()); if (viewer != null) viewer.setSelectedCursor(null); } }; action.setId(ShapeType.NONE.getId()); action.setImageDescriptor(Activator.getImageDescriptor("icons/MouseArrow.png")); group.add(action); man.add(action); man.add(new Separator()); group = new CheckableActionGroup(); final Action mask = new Action("Mask", IAction.AS_CHECK_BOX) { public void run() { Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_PEN_MASKOUT, true); updateIcons(getImageTrace().getNanBound().getColor()); } }; mask.setImageDescriptor(Activator.getImageDescriptor("icons/mask-add.png")); group.add(mask); man.add(mask); final Action unmask = new Action("Unmask", IAction.AS_CHECK_BOX) { public void run() { Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_PEN_MASKOUT, false); updateIcons(null); } }; unmask.setImageDescriptor(Activator.getImageDescriptor("icons/mask-remove.png")); group.add(unmask); man.add(unmask); man.add(new Separator()); boolean maskout = Activator.getPlottingPreferenceStore().getBoolean(PlottingConstants.MASK_PEN_MASKOUT); if (maskout) { if (getImageTrace() != null) { updateIcons(getImageTrace().getNanBound().getColor()); } else { updateIcons(new int[] { 0, 255, 0 }); } mask.setChecked(true); } else { updateIcons(null); unmask.setChecked(true); } Display.getDefault().asyncExec(new Runnable() { public void run() { String savedShape = Activator.getPlottingPreferenceStore() .getString(PlottingConstants.MASK_PEN_SHAPE); if (savedShape != null && !"".equals(savedShape)) { ShapeType type = ShapeType.valueOf(savedShape); ((ActionContributionItem) directToolbar.find(type.getId())).getAction().setChecked(true); } } }); } private void updateIcons(int[] ia) { try { RGB maskColor = ColorUtility.getRGB(ia); ((ActionContributionItem) directToolbar.find(ShapeType.SQUARE.getId())).getAction() .setImageDescriptor(IconUtils.getBrushIcon(12, ShapeType.SQUARE, maskColor)); ((ActionContributionItem) directToolbar.find(ShapeType.TRIANGLE.getId())).getAction() .setImageDescriptor(IconUtils.getBrushIcon(12, ShapeType.TRIANGLE, maskColor)); ((ActionContributionItem) directToolbar.find(ShapeType.CIRCLE.getId())).getAction() .setImageDescriptor(IconUtils.getBrushIcon(12, ShapeType.CIRCLE, maskColor)); } catch (Throwable ne) { logger.error("Cannot set mask color!", ne); } } private static BooleanDataset savedMask = null; private static boolean autoApplySavedMask = false; private static MaskingTool currentMaskingTool = null; private static final String AUTO_SAVE_PROP = "org.dawb.workbench.plotting.tools.auto.save.mask"; private IAction loadMask; private static IAction autoApplyMask, alwaysSave; /** * You can programmatically set the last saved mask by calling this method. * * NOTE: In future a mask saving and retrieving methodology will exist. * This method is used for an NCD workaround and is not intended * as a long term solution for mask saving/loading * * @param sm - static dataset to use as the mask * @param autoApply - if true the mask will be applied when the tool is activated */ public static void setSavedMask(final BooleanDataset sm, final boolean autoApply) { savedMask = sm; autoApplySavedMask = autoApply; if (autoApplyMask != null) { autoApplyMask.setChecked(autoApply); } } public static BooleanDataset getSavedMask() { return savedMask; } private void createActions(IActionBars actionBars) { createToolPageActions(); final Action exportMask = new Action("Export mask to file", Activator.getImageDescriptor("icons/mask-export-wiz.png")) { public void run() { try { IWizard wiz = EclipseUtils.openWizard(PersistenceExportWizard.ID, false); WizardDialog wd = new WizardDialog(Display.getCurrent().getActiveShell(), wiz); wd.setTitle(wiz.getWindowTitle()); wd.open(); } catch (Exception e) { logger.error("Problem opening import!", e); } } }; actionBars.getToolBarManager().add(exportMask); final Action importMask = new Action("Import mask from file", Activator.getImageDescriptor("icons/mask-import-wiz.png")) { public void run() { try { autoApply.setSelection(false); apply.setEnabled(true); IWizard wiz = EclipseUtils.openWizard(PersistenceImportWizard.ID, false); WizardDialog wd = new WizardDialog(Display.getCurrent().getActiveShell(), wiz); wd.setTitle(wiz.getWindowTitle()); wd.open(); } catch (Exception e) { logger.error("Problem opening import!", e); } } }; actionBars.getToolBarManager().add(importMask); actionBars.getToolBarManager().add(new Separator()); final Action invertMask = new Action("Invert mask", Activator.getImageDescriptor("icons/mask-invert.png")) { public void run() { try { maskObject.invert(); final IImageTrace image = getImageTrace(); image.setMask(maskObject.getMaskDataset()); } catch (Exception e) { logger.error("Problem opening import!", e); } } }; actionBars.getToolBarManager().add(invertMask); actionBars.getToolBarManager().add(new Separator()); final Action undo = new Action("Undo mask operation", Activator.getImageDescriptor("icons/mask-undo.png")) { public void run() { maskObject.undo(); getImageTrace().setMask(maskObject.getMaskDataset()); } }; actionBars.getToolBarManager().add(undo); undo.setEnabled(false); final Action redo = new Action("Redo mask operation", Activator.getImageDescriptor("icons/mask-redo.png")) { public void run() { maskObject.redo(); getImageTrace().setMask(maskObject.getMaskDataset()); } }; actionBars.getToolBarManager().add(redo); redo.setEnabled(false); actionBars.getToolBarManager().add(new Separator()); maskObject.getOperationManager().addOperationHistoryListener(new IOperationHistoryListener() { @Override public void historyNotification(OperationHistoryEvent event) { undo.setEnabled(maskObject.getOperationManager().canUndo(MaskOperation.MASK_CONTEXT)); redo.setEnabled(maskObject.getOperationManager().canRedo(MaskOperation.MASK_CONTEXT)); } }); final Action reset = new Action("Reset Mask", Activator.getImageDescriptor("icons/reset.gif")) { public void run() { resetMask(); } }; actionBars.getToolBarManager().add(reset); actionBars.getToolBarManager().add(new Separator()); final Action saveMask = new Action("Export the mask into a temporary buffer", Activator.getImageDescriptor("icons/export_wiz.gif")) { public void run() { saveMaskBuffer(); } }; actionBars.getToolBarManager().add(saveMask); loadMask = new Action("Import the mask from temporary buffer", Activator.getImageDescriptor("icons/import_wiz.gif")) { public void run() { mergeSavedMask(); } }; loadMask.setEnabled(savedMask != null); actionBars.getToolBarManager().add(loadMask); if (autoApplyMask == null) { autoApplyMask = new Action("Automatically apply the mask buffer to any image plotted.", IAction.AS_CHECK_BOX) { public void run() { autoApplySavedMask = isChecked(); if (autoApplySavedMask && currentMaskingTool != null) { if (savedMask == null) currentMaskingTool.saveMaskBuffer(); currentMaskingTool.mergeSavedMask(); if (!currentMaskingTool.isDedicatedView()) { try { currentMaskingTool.createToolInDedicatedView(); } catch (Exception e) { logger.error("Internal error!", e); } } } } }; autoApplyMask.setChecked(autoApplySavedMask); autoApplyMask.setImageDescriptor(Activator.getImageDescriptor("icons/mask-autoapply.png")); } ; actionBars.getToolBarManager().add(autoApplyMask); actionBars.getToolBarManager().add(new Separator()); if (alwaysSave == null) { // Needs to be static else you have to go back to all editors and // uncheck each one. alwaysSave = new Action("Auto-save the mask to the buffer when it changes", IAction.AS_CHECK_BOX) { public void run() { Activator.getPlottingPreferenceStore().setValue(AUTO_SAVE_PROP, isChecked()); } }; alwaysSave.setImageDescriptor(Activator.getImageDescriptor("icons/plot-tool-masking-autosave.png")); if (Activator.getPlottingPreferenceStore().contains(AUTO_SAVE_PROP)) { alwaysSave.setChecked(Activator.getPlottingPreferenceStore().getBoolean(AUTO_SAVE_PROP)); } else { Activator.getPlottingPreferenceStore().setValue(AUTO_SAVE_PROP, true); alwaysSave.setChecked(true); } } ; actionBars.getToolBarManager().add(alwaysSave); final IPlottingSystem system = getPlottingSystem(); system.getPlotActionSystem().fillTraceActions(actionBars.getToolBarManager(), getImageTrace(), system); } protected void saveMaskBuffer() { savedMask = maskObject.getMaskDataset(); if (loadMask != null) loadMask.setEnabled(savedMask != null); } protected void mergeSavedMask() { if (getImageTrace() != null) { if (savedMask == null) return; if (maskObject.getImageDataset() == null) { maskObject.setImageDataset((Dataset) getImageTrace().getData()); } maskObject.process(savedMask); getImageTrace().setMask(maskObject.getMaskDataset()); } } private void createColumns(TableViewer viewer) { ColumnViewerToolTipSupport.enableFor(viewer, ToolTip.NO_RECREATE); viewer.setColumnProperties(new String[] { "Mask", "Name", "Type" }); TableViewerColumn var = new TableViewerColumn(viewer, SWT.LEFT, 0); var.getColumn().setText("Mask"); var.getColumn().setWidth(50); var.setLabelProvider(new MaskingLabelProvider()); var = new TableViewerColumn(viewer, SWT.CENTER, 1); var.getColumn().setText("Name"); var.getColumn().setWidth(200); var.setLabelProvider(new MaskingLabelProvider()); var.setEditingSupport(new NameEditingSupport(viewer)); var = new TableViewerColumn(viewer, SWT.CENTER, 2); var.getColumn().setText("Type"); var.getColumn().setWidth(120); var.setLabelProvider(new MaskingLabelProvider()); } private boolean isInDoubleClick = false; private class NameEditingSupport extends EditingSupport { public NameEditingSupport(ColumnViewer viewer) { super(viewer); } @Override protected CellEditor getCellEditor(Object element) { return new TextCellEditor((Composite) getViewer().getControl()); } @Override protected boolean canEdit(Object element) { return isInDoubleClick; } @Override protected Object getValue(Object element) { return ((IRegion) element).getName(); } @Override protected void setValue(Object element, Object value) { IRegion region = (IRegion) element; try { String name = (String) value; if (name != null) { name = name.trim(); if (name.equals(region.getName())) return; } if (getPlottingSystem().getRegion(name) != null || name == null || "".equals(name)) { throw new Exception("The region name '" + name + "' is not allowed."); } getPlottingSystem().renameRegion(region, name); } catch (Exception e) { final String message = "The name '" + value + "' is not valid."; final Status status = new Status(Status.WARNING, "org.dawnsci.plotting", message, e); ErrorDialog.openError(Display.getDefault().getActiveShell(), "Cannot rename region", message, status); return; } getViewer().refresh(element); } } private class MaskingLabelProvider extends ColumnLabelProvider { private Image checkedIcon; private Image uncheckedIcon; public MaskingLabelProvider() { ImageDescriptor id = Activator.getImageDescriptor("icons/ticked.png"); checkedIcon = id.createImage(); id = Activator.getImageDescriptor("icons/unticked.gif"); uncheckedIcon = id.createImage(); } private int columnIndex; public void update(ViewerCell cell) { columnIndex = cell.getColumnIndex(); super.update(cell); } public Image getImage(Object element) { if (columnIndex != 0) return null; if (!(element instanceof IRegion)) return null; final IRegion region = (IRegion) element; return region.isMaskRegion() && regionTable.getTable().isEnabled() ? checkedIcon : uncheckedIcon; } public String getText(Object element) { if (element instanceof String) return ""; final IRegion region = (IRegion) element; switch (columnIndex) { case 1: return region.getName(); case 2: return region.getRegionType().getName(); } return ""; } public void dispose() { super.dispose(); checkedIcon.dispose(); uncheckedIcon.dispose(); } } private static final Collection<RegionType> maskingTypes; static { maskingTypes = new ArrayList<RegionType>(6); //maskingTypes.add(RegionType.FREE_DRAW); maskingTypes.add(RegionType.RING); maskingTypes.add(RegionType.BOX); maskingTypes.add(RegionType.LINE); //maskingTypes.add(RegionType.POLYLINE); maskingTypes.add(RegionType.POLYGON); maskingTypes.add(RegionType.POINT); maskingTypes.add(RegionType.SECTOR); } private void createMaskingRegionActions(ToolBarManager man) { final Action multipleRegion = new Action("Continuously add the same region", IAction.AS_CHECK_BOX) { public void run() { Activator.getPlottingPreferenceStore().setValue(PlottingConstants.MASK_DRAW_MULTIPLE, isChecked()); } }; multipleRegion.setImageDescriptor(Activator.getImageDescriptor("icons/RegionMultiple.png")); multipleRegion.setChecked( Activator.getPlottingPreferenceStore().getBoolean(PlottingConstants.MASK_DRAW_MULTIPLE)); man.add(multipleRegion); man.add(new Separator()); final MenuAction widthChoice = new MenuAction("Line Width"); widthChoice.setToolTipText("Line width for free draw and line regions"); man.add(widthChoice); // Region actions supported ActionContributionItem menu = (ActionContributionItem) getPlottingSystem().getActionBars().getMenuManager() .find(BasePlottingConstants.ADD_REGION); MenuAction menuAction = (MenuAction) menu.getAction(); IAction ld = null; for (RegionType type : maskingTypes) { final IAction action = menuAction.findAction(type.getId()); if (action == null) continue; man.add(action); if (type == RegionType.LINE) { ld = action; man.add(new Separator()); } if (action instanceof IRegionAction) { ((IRegionAction) action).setUserObject(MaskMarker.MASK_REGION); } } CheckableActionGroup group = new CheckableActionGroup(); final IAction lineDraw = ld; final int maxWidth = 10; for (int iwidth = 1; iwidth <= maxWidth; iwidth++) { final int width = iwidth; final Action action = new Action("Draw width of " + String.valueOf(width), IAction.AS_CHECK_BOX) { public void run() { Activator.getPlottingPreferenceStore().setValue(PlottingConstants.FREE_DRAW_WIDTH, width); widthChoice.setSelectedAction(this); lineDraw.run(); } }; action.setImageDescriptor(IconUtils.createIconDescriptor(String.valueOf(iwidth))); widthChoice.add(action); group.add(action); action.setChecked(false); action.setToolTipText("Set line width to " + iwidth); } int wid = Activator.getPlottingPreferenceStore().getInt(PlottingConstants.FREE_DRAW_WIDTH); widthChoice.setSelectedAction(wid - 1); widthChoice.setCheckedAction(wid - 1, true); man.add(new Separator()); menu = (ActionContributionItem) getPlottingSystem().getActionBars().getToolBarManager() .find(BasePlottingConstants.REMOVE_REGION); if (menu != null) { menuAction = (MenuAction) menu.getAction(); man.add(menuAction); } enableControls.add(man.getControl()); } private int getValue(Number bound, HistogramBound hb, int defaultInt) { if (bound != null) return bound.intValue(); if (hb != null && hb.getBound() != null) { if (!Double.isInfinite(hb.getBound().doubleValue())) { return hb.getBound().intValue(); } } return defaultInt; } private void processMask() { processMask(false, false, null); } private void processMask(final IRegion region) { processMask(false, false, region); } /** * Either adds a new region directly or * @param forceProcess * @param roi * @return true if did some masking */ private void processMask(final boolean resetMask, boolean ignoreAuto, final IRegion region) { if (!ignoreAuto && (autoApply != null && !autoApply.getSelection())) return; final IImageTrace image = getImageTrace(); if (image == null) return; maskJob.schedule(resetMask, region, null); } protected void resetMask() { // Reread the file from disk or cached one if this is a view final IImageTrace image = getImageTrace(); if (image == null) return; maskObject.reset(); image.setMask(null); if (autoApplySavedMask) { // If it is auto-apply then it needs to be reset when you reset. final boolean ok = MessageDialog.openQuestion(Display.getDefault().getActiveShell(), "Confirm Removal of Saved Mask", "You have a saved mask which can be cleared as well.\n\nWould you like to remove the saved mask?"); if (ok) { savedMask = null; autoApplySavedMask = false; } loadMask.setEnabled(savedMask != null); } } @Override public Control getControl() { return scrollComp; } @Override public void setFocus() { if (scrollComp != null) scrollComp.setFocus(); } @Override public void activate() { super.activate(); setEnabled(getImageTrace() != null); currentMaskingTool = this; if (getPlottingSystem() != null) { getPlottingSystem().addTraceListener(traceListener); // If it changes get reference to image. getPlottingSystem().addRegionListener(regionListener); // Listen to new regions // For all supported regions, add listener for rois final Collection<IRegion> regions = getPlottingSystem().getRegions(); if (regions != null) for (IRegion region : regions) { if (!maskObject.isSupportedRegion(region)) continue; region.addROIListener(this.regionBoundsListener); } if (getImageTrace() != null) { getImageTrace().addPaletteListener(paletteListener); } this.viewer = (AbstractPlottingViewer) getPlottingSystem().getAdapter(AbstractPlottingViewer.class); if (viewer != null) { viewer.addMouseClickListener(clickListener); viewer.addMouseMotionListener(clickListener); } } if (this.regionTable != null && !regionTable.getControl().isDisposed()) { regionTable.refresh(); } if (loadMask != null) loadMask.setEnabled(savedMask != null); if (autoApplyMask != null) { autoApplyMask.setEnabled(!(getPlottingSystem().getPart() instanceof IViewPart)); } } @Override public void deactivate() { super.deactivate(); if (savedMask == null) saveMaskBuffer(); currentMaskingTool = null; if (getPlottingSystem() != null) { getPlottingSystem().removeTraceListener(traceListener); getPlottingSystem().removeRegionListener(regionListener);// Stop listen to new regions // For all supported regions, add listener final Collection<IRegion> regions = getPlottingSystem().getRegions(); if (regions != null) for (IRegion region : regions) region.removeROIListener(this.regionBoundsListener); if (getImageTrace() != null) { getImageTrace().removePaletteListener(paletteListener); } AbstractPlottingViewer viewer = (AbstractPlottingViewer) getPlottingSystem() .getAdapter(AbstractPlottingViewer.class); viewer.removeMouseClickListener(clickListener); viewer.removeMouseMotionListener(clickListener); viewer.setSelectedCursor(null); viewer.setShiftPoint(null); } } @Override public void dispose() { super.dispose(); enableControls.clear(); if (scrollComp != null) scrollComp.dispose(); if (maskObject != null) maskObject.dispose(); scrollComp = null; traceListener = null; regionListener = null; regionBoundsListener = null; if (this.regionTable != null && !regionTable.getControl().isDisposed()) { regionTable.getTable().removeMouseListener(this); } } private void setEnabled(boolean enabled) { for (Control control : enableControls) { if (control != null) control.setEnabled(enabled); } if (autoApply.getSelection()) apply.setEnabled(false); } @Override public void sync(IToolPage with) { if (with instanceof MaskingTool) { MaskingTool sync = (MaskingTool) with; maskObject.sync(sync.maskObject); } } public class MaskJob extends Job { public MaskJob() { super("Masking image"); } private boolean resetMask = false; private boolean isRegionsEnabled = false; private IRegion region = null; private Double min = null, max = null; private org.eclipse.draw2d.geometry.Point location; @Override protected IStatus run(final IProgressMonitor monitor) { try { if (isDisposed() || !isActive()) { return Status.CANCEL_STATUS; } monitor.beginTask("Mask Process", 100); final IImageTrace image = getImageTrace(); if (image == null) return Status.CANCEL_STATUS; if (monitor.isCanceled()) return Status.CANCEL_STATUS; if (region != null && !isRegionsEnabled) return Status.CANCEL_STATUS; if (resetMask && !maskObject.isIgnoreAlreadyMasked()) { maskObject.setMaskDataset(null, false); } if (maskObject.getMaskDataset() == null) { // The mask must be maintained as a BooleanDataset so that there is the option // of applying the same mask to many images. final Dataset unmasked = (Dataset) image.getData(); maskObject.setMaskDataset(new BooleanDataset(unmasked.getShape()), true); maskObject.setImageDataset(unmasked); } if (maskObject.getImageDataset() == null) { final Dataset unmasked = (Dataset) image.getData(); maskObject.setImageDataset(unmasked); } // Keep the saved mask if (autoApplySavedMask && savedMask != null) maskObject.process(savedMask); monitor.worked(1); // Just process a changing region if (location != null) { maskObject.process(location, getPlottingSystem(), monitor); } else if (region != null) { if (!maskObject.isSupportedRegion(region)) return Status.CANCEL_STATUS; maskObject.process(region, monitor); } else { // process everything maskObject.process(min, max, isRegionsEnabled ? getPlottingSystem().getRegions() : null, monitor); } if (Activator.getPlottingPreferenceStore().getBoolean(AUTO_SAVE_PROP)) { saveMaskBuffer(); } Display.getDefault().syncExec(new Runnable() { public void run() { // NOTE the mask will have a reference kept and // will downsample with the data. image.setMask(maskObject.getMaskDataset()); } }); return Status.OK_STATUS; } catch (Throwable ne) { logger.error("Cannot mask properly at the edges as yet!", ne); return Status.CANCEL_STATUS; } finally { monitor.done(); } } /** * Uses job thread * @param resetMask * @param region * @param loc */ public void schedule(boolean resetMask, IRegion region, org.eclipse.draw2d.geometry.Point loc) { assign(resetMask, region, loc); schedule(); } private void assign(boolean resetMask, IRegion region, org.eclipse.draw2d.geometry.Point loc) { if (isDisposed()) { return; } this.isRegionsEnabled = regionTable != null ? regionTable.getTable().isEnabled() : false; this.resetMask = resetMask; this.region = region; this.location = loc; min = (minimum != null && minimum.isEnabled()) ? minimum.getDouble() : null; max = (maximum != null && maximum.isEnabled()) ? maximum.getDouble() : null; } /** * Done in calling thread. * @param resetMask * @param region * @param loc */ public void run(boolean resetMask, IRegion region, org.eclipse.draw2d.geometry.Point loc) { assign(resetMask, region, loc); try { run(new NullProgressMonitor()); } catch (Throwable ne) { logger.error("Cannot run masking job!", ne); } } } @Override public void mouseDoubleClick(MouseEvent e) { final TableItem item = this.regionTable.getTable().getItem(new Point(e.x, e.y)); if (e.button == 1 && item != null) { Rectangle rect = item.getBounds(1); // Name if (!rect.contains(e.x, e.y)) return; try { IRegion region = (IRegion) item.getData(); isInDoubleClick = true; regionTable.editElement(region, 1); } finally { isInDoubleClick = false; } } } @Override public void mouseDown(MouseEvent e) { final TableItem item = this.regionTable.getTable().getItem(new Point(e.x, e.y)); if (item != null) { Rectangle rect = item.getBounds(0); // Tick if (!rect.contains(e.x, e.y)) return; IRegion region = (IRegion) item.getData(); region.setMaskRegion(!region.isMaskRegion()); region.setUserObject(MaskObject.MaskRegionType.REGION_FROM_MASKING); regionTable.refresh(region); processMask(true, false, null); } } @Override public void mouseUp(MouseEvent e) { } private enum MaskMarker { MASK_REGION; } public void setToolData(Serializable toolData) { if (toolData instanceof BooleanDataset) { maskObject.setMaskDataset((BooleanDataset) toolData, false); processMask(true, false, null); } } }