package de.lmu.ifi.dbs.elki.visualization.visualizers.scatterplot.outlier;
/*
This file is part of ELKI:
Environment for Developing KDD-Applications Supported by Index-Structures
Copyright (C) 2012
Ludwig-Maximilians-Universität München
Lehr- und Forschungseinheit für Datenbanksysteme
ELKI Development Team
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
import java.util.List;
import org.apache.batik.util.SVGConstants;
import org.w3c.dom.Element;
import de.lmu.ifi.dbs.elki.data.NumberVector;
import de.lmu.ifi.dbs.elki.database.datastore.DataStoreListener;
import de.lmu.ifi.dbs.elki.database.ids.DBID;
import de.lmu.ifi.dbs.elki.result.HierarchicalResult;
import de.lmu.ifi.dbs.elki.result.Result;
import de.lmu.ifi.dbs.elki.result.ResultUtil;
import de.lmu.ifi.dbs.elki.result.outlier.OutlierResult;
import de.lmu.ifi.dbs.elki.utilities.documentation.Reference;
import de.lmu.ifi.dbs.elki.utilities.iterator.IterableIterator;
import de.lmu.ifi.dbs.elki.utilities.optionhandling.AbstractParameterizer;
import de.lmu.ifi.dbs.elki.utilities.optionhandling.OptionID;
import de.lmu.ifi.dbs.elki.utilities.optionhandling.parameterization.Parameterization;
import de.lmu.ifi.dbs.elki.utilities.optionhandling.parameters.Flag;
import de.lmu.ifi.dbs.elki.utilities.optionhandling.parameters.ObjectParameter;
import de.lmu.ifi.dbs.elki.utilities.scaling.ScalingFunction;
import de.lmu.ifi.dbs.elki.utilities.scaling.outlier.OutlierScalingFunction;
import de.lmu.ifi.dbs.elki.visualization.VisualizationTask;
import de.lmu.ifi.dbs.elki.visualization.colors.ColorLibrary;
import de.lmu.ifi.dbs.elki.visualization.css.CSSClass;
import de.lmu.ifi.dbs.elki.visualization.projector.ScatterPlotProjector;
import de.lmu.ifi.dbs.elki.visualization.style.ClassStylingPolicy;
import de.lmu.ifi.dbs.elki.visualization.style.StyleLibrary;
import de.lmu.ifi.dbs.elki.visualization.style.StylingPolicy;
import de.lmu.ifi.dbs.elki.visualization.svg.SVGPlot;
import de.lmu.ifi.dbs.elki.visualization.svg.SVGUtil;
import de.lmu.ifi.dbs.elki.visualization.visualizers.AbstractVisFactory;
import de.lmu.ifi.dbs.elki.visualization.visualizers.Visualization;
import de.lmu.ifi.dbs.elki.visualization.visualizers.scatterplot.AbstractScatterplotVisualization;
/**
* Generates a SVG-Element containing bubbles. A Bubble is a circle visualizing
* an outlierness-score, with its center at the position of the visualized
* object and its radius depending on the objects score.
*
* @author Remigius Wojdanowski
* @author Erich Schubert
*
* @apiviz.has OutlierResult oneway - - visualizes
*/
@Reference(authors = "E. Achtert, H.-P. Kriegel, L. Reichert, E. Schubert, R. Wojdanowski, A. Zimek", title = "Visual Evaluation of Outlier Detection Models", booktitle = "Proceedings of the 15th International Conference on Database Systems for Advanced Applications (DASFAA), Tsukuba, Japan, 2010", url = "http://dx.doi.org/10.1007/978-3-642-12098-5_34")
public class BubbleVisualization extends AbstractScatterplotVisualization implements DataStoreListener {
/**
* Generic tag to indicate the type of element. Used in IDs, CSS-Classes etc.
*/
public static final String BUBBLE = "bubble";
/**
* A short name characterizing this Visualizer.
*/
public static final String NAME = "Outlier Bubbles";
/**
* Fill parameter.
*/
protected boolean fill;
/**
* Scaling function to use for Bubbles
*/
protected ScalingFunction scaling;
/**
* The outlier result to visualize
*/
protected OutlierResult result;
/**
* Constructor.
*
* @param task Visualization task
* @param scaling Scaling function
*/
public BubbleVisualization(VisualizationTask task, ScalingFunction scaling) {
super(task);
this.result = task.getResult();
this.scaling = scaling;
context.addDataStoreListener(this);
incrementalRedraw();
}
@Override
public void destroy() {
super.destroy();
context.removeDataStoreListener(this);
}
@Override
public void redraw() {
StylingPolicy stylepolicy = context.getStyleResult().getStylingPolicy();
// bubble size
final double bubble_size = context.getStyleLibrary().getSize(StyleLibrary.BUBBLEPLOT);
if(stylepolicy instanceof ClassStylingPolicy) {
ClassStylingPolicy colors = (ClassStylingPolicy) stylepolicy;
setupCSS(svgp, colors);
// draw data
for(DBID objId : sample.getSample()) {
final Double radius = getScaledForId(objId);
if(radius > 0.01 && !Double.isInfinite(radius)) {
final NumberVector<?, ?> vec = rel.get(objId);
if(vec != null) {
double[] v = proj.fastProjectDataToRenderSpace(vec);
Element circle = svgp.svgCircle(v[0], v[1], radius * bubble_size);
SVGUtil.addCSSClass(circle, BUBBLE + colors.getStyleForDBID(objId));
layer.appendChild(circle);
}
}
}
}
else {
// draw data
for(DBID objId : sample.getSample()) {
final Double radius = getScaledForId(objId);
if(radius > 0.01 && !Double.isInfinite(radius)) {
final NumberVector<?, ?> vec = rel.get(objId);
if(vec != null) {
double[] v = proj.fastProjectDataToRenderSpace(vec);
Element circle = svgp.svgCircle(v[0], v[1], radius * bubble_size);
int color = stylepolicy.getColorForDBID(objId);
final StringBuffer style = new StringBuffer();
if(fill) {
style.append(SVGConstants.CSS_FILL_PROPERTY + ":").append(SVGUtil.colorToString(color));
style.append(SVGConstants.CSS_FILL_OPACITY_PROPERTY + ":0.5");
}
else {
style.append(SVGConstants.CSS_STROKE_VALUE + ":").append(SVGUtil.colorToString(color));
style.append(SVGConstants.CSS_FILL_PROPERTY + ":" + SVGConstants.CSS_NONE_VALUE);
}
SVGUtil.setAtt(circle, SVGConstants.SVG_STYLE_ATTRIBUTE, style.toString());
layer.appendChild(circle);
}
}
}
}
}
@Override
public void resultChanged(Result current) {
if(sample == current) {
synchronizedRedraw();
}
}
/**
* Registers the Bubble-CSS-Class at a SVGPlot.
*
* @param svgp the SVGPlot to register the Tooltip-CSS-Class.
* @param policy Clustering to use
*/
private void setupCSS(SVGPlot svgp, ClassStylingPolicy policy) {
ColorLibrary colors = context.getStyleLibrary().getColorSet(StyleLibrary.PLOT);
// creating IDs manually because cluster often return a null-ID.
for(int clusterID = policy.getMinStyle(); clusterID < policy.getMaxStyle(); clusterID++) {
CSSClass bubble = new CSSClass(svgp, BUBBLE + clusterID);
bubble.setStatement(SVGConstants.CSS_STROKE_WIDTH_PROPERTY, context.getStyleLibrary().getLineWidth(StyleLibrary.PLOT));
String color = colors.getColor(clusterID);
if(fill) {
bubble.setStatement(SVGConstants.CSS_FILL_PROPERTY, color);
bubble.setStatement(SVGConstants.CSS_FILL_OPACITY_PROPERTY, 0.5);
}
else {
// for diamond-shaped strokes, see bugs.sun.com, bug ID 6294396
bubble.setStatement(SVGConstants.CSS_STROKE_VALUE, color);
bubble.setStatement(SVGConstants.CSS_FILL_PROPERTY, SVGConstants.CSS_NONE_VALUE);
}
svgp.addCSSClassOrLogError(bubble);
}
}
/**
* Convenience method to apply scalings in the right order.
*
* @param id object ID to get scaled score for
* @return a Double representing a outlierness-score, after it has modified by
* the given scales.
*/
protected double getScaledForId(DBID id) {
double d = result.getScores().get(id).doubleValue();
if(Double.isNaN(d) || Double.isInfinite(d)) {
return 0.0;
}
if(scaling == null) {
return result.getOutlierMeta().normalizeScore(d);
}
else {
return scaling.getScaled(d);
}
}
/**
* Factory for producing bubble visualizations
*
* @author Erich Schubert
*
* @apiviz.stereotype factory
* @apiviz.uses BubbleVisualization oneway - - «create»
*/
public static class Factory extends AbstractVisFactory {
/**
* Flag for half-transparent filling of bubbles.
*
* <p>
* Key: {@code -bubble.fill}
* </p>
*/
public static final OptionID FILL_ID = OptionID.getOrCreateOptionID("bubble.fill", "Half-transparent filling of bubbles.");
/**
* Parameter for scaling functions
*
* <p>
* Key: {@code -bubble.scaling}
* </p>
*/
public static final OptionID SCALING_ID = OptionID.getOrCreateOptionID("bubble.scaling", "Additional scaling function for bubbles.");
/**
* Fill parameter.
*/
protected boolean fill;
/**
* Scaling function to use for Bubbles
*/
protected ScalingFunction scaling;
/**
* Constructor.
*
* @param fill
* @param scaling
*/
public Factory(boolean fill, ScalingFunction scaling) {
super();
this.fill = fill;
this.scaling = scaling;
}
@Override
public Visualization makeVisualization(VisualizationTask task) {
if(this.scaling != null && this.scaling instanceof OutlierScalingFunction) {
final OutlierResult outlierResult = task.getResult();
((OutlierScalingFunction) this.scaling).prepare(outlierResult);
}
return new BubbleVisualization(task, scaling);
}
@Override
public void processNewResult(HierarchicalResult baseResult, Result result) {
List<OutlierResult> ors = ResultUtil.filterResults(result, OutlierResult.class);
for(OutlierResult o : ors) {
IterableIterator<ScatterPlotProjector<?>> ps = ResultUtil.filteredResults(baseResult, ScatterPlotProjector.class);
boolean vis = true;
// Quick and dirty hack: hide if parent result is also an outlier result
// Since that probably is already visible and we're redundant.
for(Result r : o.getHierarchy().getParents(o)) {
if(r instanceof OutlierResult) {
vis = false;
break;
}
}
for(ScatterPlotProjector<?> p : ps) {
final VisualizationTask task = new VisualizationTask(NAME, o, p.getRelation(), this);
task.put(VisualizationTask.META_LEVEL, VisualizationTask.LEVEL_DATA);
if(!vis) {
task.put(VisualizationTask.META_VISIBLE_DEFAULT, false);
}
baseResult.getHierarchy().add(o, task);
baseResult.getHierarchy().add(p, task);
}
}
}
/**
* Parameterization class.
*
* @author Erich Schubert
*
* @apiviz.exclude
*/
public static class Parameterizer extends AbstractParameterizer {
/**
* Fill parameter.
*/
protected boolean fill = false;
/**
* Scaling function to use for Bubbles
*/
protected ScalingFunction scaling = null;
@Override
protected void makeOptions(Parameterization config) {
super.makeOptions(config);
Flag fillF = new Flag(FILL_ID);
if(config.grab(fillF)) {
fill = fillF.getValue();
}
ObjectParameter<ScalingFunction> scalingP = new ObjectParameter<ScalingFunction>(SCALING_ID, OutlierScalingFunction.class, true);
if(config.grab(scalingP)) {
scaling = scalingP.instantiateClass(config);
}
}
@Override
protected Factory makeInstance() {
return new Factory(fill, scaling);
}
}
}
}