org.jfree.data.category.CategoryDatasetSelectionState Java Examples

/**
 * Clears the selection.
 *
 * @since 1.2.0
 */
public void clearSelection() {
    // cycle through the datasets and clear the selection state
    int datasetCount = this.datasets.size();
    for (int d = 0; d < datasetCount; d++) {
        CategoryDataset dataset = (CategoryDataset) this.datasets.get(d);
        if (dataset instanceof SelectableCategoryDataset) {
            // FIXME:  actually, we need to get the selection state
            // taking into account the rendering source
            SelectableCategoryDataset scd
                    = (SelectableCategoryDataset) dataset;
            if (scd.getSelectionState() != null) {
                CategoryDatasetSelectionState selState
                        = scd.getSelectionState();
                selState.clearSelection();
            }
        }
    }
}
 

/**
 * Perform a select for the item(s) at the specified (x, y) coordinates
 * in Java2D space.
 *
 * @param x  the x-coordinate.
 * @param y  the y-coordinate.
 * @param dataArea  the data area.
 * @param source  the rendering source.
 *
 * @since 1.2.0
 */
public void select(double x, double y, Rectangle2D dataArea,
        RenderingSource source) {

    int datasetCount = this.datasets.size();
    for (int d = 0; d < datasetCount; d++) {
        CategoryDataset dataset = (CategoryDataset) this.datasets.get(d);
        if (dataset == null) {
            continue;
        }
        CategoryDatasetSelectionState state = findSelectionStateForDataset(
                dataset, source);
        if (state == null) {
            continue;
        }
        Graphics2D g2 = source.createGraphics2D();
        CategoryItemRenderer renderer = getRendererForDataset(dataset);
        CategoryItemRendererState rs = renderer.initialise(g2, dataArea,
                this, dataset, null);
        int rowCount = dataset.getRowCount();
        int columnCount = dataset.getColumnCount();
        for (int r = 0; r < rowCount; r++) {
            for (int c = 0; c < columnCount; c++) {
                if (renderer.hitTest(x, y, null, dataArea, this,
                        getDomainAxisForDataset(d),
                        getRangeAxisForDataset(d), dataset, r, c, false,
                        rs)) {
                    state.setSelected(r, c, !state.isSelected(r, c));
                }
            }
        }
    }
}
 

/**
 * Returns the selection state for the specified dataset.  This could be
 * <code>null</code> if the dataset hasn't been set up to support
 * selections.
 *
 * @param dataset  the dataset.
 * @param source  the selection source.
 *
 * @return The selection state (possibly <code>null</code>).
 */
private CategoryDatasetSelectionState findSelectionStateForDataset(
        CategoryDataset dataset, Object source) {
    if (dataset instanceof SelectableCategoryDataset) {
        SelectableCategoryDataset sd = (SelectableCategoryDataset) dataset;
        CategoryDatasetSelectionState s = sd.getSelectionState();
        return s;
    }
    throw new RuntimeException();
    //return null;  // TODO: implement
}
 

/**
 * Returns the selection state.
 *
 * @return The selection state (possibly <code>null</code>).
 *
 * @since 1.2.0
 */
public CategoryDatasetSelectionState getSelectionState() {
    return this.selectionState;
}
 

/**
 * Sets the selection state.
 *
 * @param state  the selection state (<code>null</code> permitted).
 *
 * @since 1.2.0
 */
public void setSelectionState(CategoryDatasetSelectionState state) {
    this.selectionState = state;
}