/*
* ARX: Powerful Data Anonymization
* Copyright 2012 - 2017 Fabian Prasser, Florian Kohlmayer and contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.deidentifier.arx.gui.view.impl.risk;
import org.deidentifier.arx.gui.Controller;
import org.deidentifier.arx.gui.model.ModelEvent;
import org.deidentifier.arx.gui.model.ModelEvent.ModelPart;
import org.deidentifier.arx.gui.model.ModelRisk.ViewRiskType;
import org.deidentifier.arx.gui.resources.Resources;
import org.deidentifier.arx.gui.view.SWTUtil;
import org.deidentifier.arx.gui.view.impl.common.ClipboardHandlerTable;
import org.deidentifier.arx.gui.view.impl.common.ComponentStatusLabelProgressProvider;
import org.deidentifier.arx.gui.view.impl.common.async.Analysis;
import org.deidentifier.arx.gui.view.impl.common.async.AnalysisContext;
import org.deidentifier.arx.gui.view.impl.common.async.AnalysisManager;
import org.deidentifier.arx.risk.RiskEstimateBuilderInterruptible;
import org.deidentifier.arx.risk.RiskModelSampleRiskDistribution;
import org.eclipse.swt.SWT;
import org.eclipse.swt.layout.FillLayout;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.TableColumn;
import org.eclipse.swt.widgets.TableItem;
import de.linearbits.swt.table.DynamicTable;
import de.linearbits.swt.table.DynamicTableColumn;
/**
* This view displays basic risk estimates.
*
* @author Fabian Prasser
*/
public class ViewRisksRiskDistributionTable extends ViewRisks<AnalysisContextRisk> {
/** View */
private Composite root;
/** View */
private DynamicTable table;
/** Internal stuff. */
private AnalysisManager manager;
/**
* Creates a new instance.
*
* @param parent
* @param controller
* @param target
* @param reset
*/
public ViewRisksRiskDistributionTable(final Composite parent,
final Controller controller,
final ModelPart target,
final ModelPart reset) {
super(parent, controller, target, reset);
this.manager = new AnalysisManager(parent.getDisplay());
controller.addListener(ModelPart.ATTRIBUTE_TYPE, this);
}
@Override
public void update(ModelEvent event) {
super.update(event);
if (event.part == ModelPart.ATTRIBUTE_TYPE) {
triggerUpdate();
}
}
@Override
protected Control createControl(Composite parent) {
this.root = new Composite(parent, SWT.NONE);
this.root.setLayout(new FillLayout());
table = SWTUtil.createTableDynamic(root, SWT.BORDER | SWT.V_SCROLL | SWT.H_SCROLL | SWT.FULL_SELECTION);
table.setHeaderVisible(true);
table.setLinesVisible(true);
table.setMenu(new ClipboardHandlerTable(table).getMenu());
DynamicTableColumn c = new DynamicTableColumn(table, SWT.LEFT);
c.setWidth("33%", "100px"); //$NON-NLS-1$ //$NON-NLS-2$
c.setText(Resources.getMessage("RiskAnalysis.1")); //$NON-NLS-1$
c = new DynamicTableColumn(table, SWT.LEFT);
SWTUtil.createColumnWithBarCharts(table, c);
c.setWidth("33%", "100px"); //$NON-NLS-1$ //$NON-NLS-2$
c.setText(Resources.getMessage("RiskAnalysis.2")); //$NON-NLS-1$
c = new DynamicTableColumn(table, SWT.LEFT);
SWTUtil.createColumnWithBarCharts(table, c);
c.setWidth("33%", "100px"); //$NON-NLS-1$ //$NON-NLS-2$
c.setText(Resources.getMessage("RiskAnalysis.3")); //$NON-NLS-1$
for (final TableColumn col : table.getColumns()) {
col.pack();
}
SWTUtil.createGenericTooltip(table);
return this.root;
}
@Override
protected AnalysisContextRisk createViewConfig(AnalysisContext context) {
return new AnalysisContextRisk(context);
}
@Override
protected void doReset() {
if (this.manager != null) {
this.manager.stop();
}
table.setRedraw(false);
for (final TableItem i : table.getItems()) {
i.dispose();
}
table.setRedraw(true);
setStatusEmpty();
}
@Override
protected void doUpdate(final AnalysisContextRisk context) {
// Enable/disable
final RiskEstimateBuilderInterruptible builder = getBuilder(context);
if (!this.isEnabled() || builder == null) {
if (manager != null) {
manager.stop();
}
this.setStatusEmpty();
return;
}
// Create an analysis
Analysis analysis = new Analysis() {
private boolean stopped = false;
private double[] frequencies;
private double[] cumulative;
private String[] labels;
@Override
public int getProgress() {
return 0;
}
@Override
public void onError() {
setStatusEmpty();
}
@Override
public void onFinish() {
if (stopped || !isEnabled()) {
return;
}
// Disable drawing
table.setRedraw(false);
// Update chart
for (final TableItem i : table.getItems()) {
i.dispose();
}
// Create entries
for (int i = labels.length-1; i >=0 ; i--) {
TableItem item = new TableItem(table, SWT.NONE);
item.setText(0, labels[i]);
item.setData("1", frequencies[i]);
item.setData("2", cumulative[i]);
}
// Enable drawing and redraw
table.setRedraw(true);
table.redraw();
root.layout();
// Set status
setStatusDone();
}
@Override
public void onInterrupt() {
if (!isEnabled() || !isValid()) {
setStatusEmpty();
} else {
setStatusWorking();
}
}
@Override
public void run() throws InterruptedException {
// Timestamp
long time = System.currentTimeMillis();
// Perform work
RiskModelSampleRiskDistribution model = builder.getSampleBasedRiskDistribution();
// Create array
frequencies = model.getFractionOfRecordsForRiskThresholds();
cumulative = model.getFractionOfRecordsForCumulativeRiskThresholds();
labels = new String[frequencies.length];
for (int i = 0; i < frequencies.length; i++) {
labels[i] = "]" + String.valueOf(SWTUtil.getPrettyString(model.getAvailableLowerRiskThresholds()[i] * 100d)) + //$NON-NLS-1$
", " + String.valueOf(SWTUtil.getPrettyString(model.getAvailableUpperRiskThresholds()[i] * 100d)) + "]"; //$NON-NLS-1$ $NON-NLS-2$
}
// Our users are patient
while (System.currentTimeMillis() - time < MINIMAL_WORKING_TIME && !stopped) {
Thread.sleep(10);
}
}
@Override
public void stop() {
if (builder != null) builder.interrupt();
this.stopped = true;
}
};
this.manager.start(analysis);
}
@Override
protected ComponentStatusLabelProgressProvider getProgressProvider() {
return null;
}
@Override
protected ViewRiskType getViewType() {
return ViewRiskType.CLASSES_TABLE;
}
/**
* Is an analysis running
*/
protected boolean isRunning() {
return manager != null && manager.isRunning();
}
}