    001    /* ===========================================================
    002     * JFreeChart : a free chart library for the Java(tm) platform
    003     * ===========================================================
    004     *
    005     * (C) Copyright 2000-2008, by Object Refinery Limited and Contributors.
    006     *
    007     * Project Info:  http://www.jfree.org/jfreechart/index.html
    008     *
    009     * This library is free software; you can redistribute it and/or modify it
    010     * under the terms of the GNU Lesser General Public License as published by
    011     * the Free Software Foundation; either version 2.1 of the License, or
    012     * (at your option) any later version.
    013     *
    014     * This library is distributed in the hope that it will be useful, but
    015     * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
    016     * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
    017     * License for more details.
    018     *
    019     * You should have received a copy of the GNU Lesser General Public
    020     * License along with this library; if not, write to the Free Software
    021     * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301,
    022     * USA.
    023     *
    024     * [Java is a trademark or registered trademark of Sun Microsystems, Inc.
    025     * in the United States and other countries.]
    026     *
    027     * --------------------------------
    028     * SlidingGanttCategoryDataset.java
    029     * --------------------------------
    030     * (C) Copyright 2008, by Object Refinery Limited.
    031     *
    032     * Original Author:  David Gilbert (for Object Refinery Limited);
    033     * Contributor(s):   -;
    034     *
    035     * Changes
    036     * -------
    037     * 09-May-2008 : Version 1 (DG);
    038     *
    039     */
    040
    041    package org.jfree.data.gantt;
    042
    043    import java.util.Collections;
    044    import java.util.List;
    045
    046    import org.jfree.data.UnknownKeyException;
    047    import org.jfree.data.general.AbstractDataset;
    048    import org.jfree.data.general.DatasetChangeEvent;
    049    import org.jfree.util.PublicCloneable;
    050
    051    /**
    052     * A {@link GanttCategoryDataset} implementation that presents a subset of the
    053     * categories in an underlying dataset.  The index of the first "visible"
    054     * category can be modified, which provides a means of "sliding" through
    055     * the categories in the underlying dataset.
    056     *
    057     * @since 1.0.10
    058     */
    059    public class SlidingGanttCategoryDataset extends AbstractDataset
    060            implements GanttCategoryDataset {
    061
    062            /** The underlying dataset. */
    063            private GanttCategoryDataset underlying;
    064
    065            /** The index of the first category to present. */
    066            private int firstCategoryIndex;
    067
    068            /** The maximum number of categories to present. */
    069            private int maximumCategoryCount;
    070
    071            /**
    072             * Creates a new instance.
    073             *
    074             * @param underlying  the underlying dataset (<code>null</code> not
    075             *     permitted).
    076             * @param firstColumn  the index of the first visible column from the
    077             *     underlying dataset.
    078             * @param maxColumns  the maximumColumnCount.
    079             */
    080            public SlidingGanttCategoryDataset(GanttCategoryDataset underlying,
    081                            int firstColumn, int maxColumns) {
    082            this.underlying = underlying;
    083            this.firstCategoryIndex = firstColumn;
    084            this.maximumCategoryCount = maxColumns;
    085            }
    086
    087            /**
    088             * Returns the underlying dataset that was supplied to the constructor.
    089             *
    090             * @return The underlying dataset (never <code>null</code>).
    091             */
    092            public GanttCategoryDataset getUnderlyingDataset() {
    093                    return this.underlying;
    094            }
    095
    096            /**
    097             * Returns the index of the first visible category.
    098             *
    099             * @return The index.
    100             *
    101             * @see #setFirstCategoryIndex(int)
    102             */
    103            public int getFirstCategoryIndex() {
    104                    return this.firstCategoryIndex;
    105            }
    106
    107            /**
    108             * Sets the index of the first category that should be used from the
    109             * underlying dataset, and sends a {@link DatasetChangeEvent} to all
    110             * registered listeners.
    111             *
    112             * @param first  the index.
    113             *
    114             * @see #getFirstCategoryIndex()
    115             */
    116            public void setFirstCategoryIndex(int first) {
    117                    if (first < 0 || first >= this.underlying.getColumnCount()) {
    118                            throw new IllegalArgumentException("Invalid index.");
    119                    }
    120                    this.firstCategoryIndex = first;
    121                    fireDatasetChanged();
    122            }
    123
    124            /**
    125             * Returns the maximum category count.
    126             *
    127             * @return The maximum category count.
    128             *
    129             * @see #setMaximumCategoryCount(int)
    130             */
    131            public int getMaximumCategoryCount() {
    132                    return this.maximumCategoryCount;
    133            }
    134
    135            /**
    136             * Sets the maximum category count and sends a {@link DatasetChangeEvent}
    137             * to all registered listeners.
    138             *
    139             * @param max  the maximum.
    140             *
    141             * @see #getMaximumCategoryCount()
    142             */
    143            public void setMaximumCategoryCount(int max) {
    144                    if (max < 0) {
    145                            throw new IllegalArgumentException("Requires 'max' >= 0.");
    146                    }
    147                    this.maximumCategoryCount = max;
    148                    fireDatasetChanged();
    149            }
    150
    151            /**
    152             * Returns the index of the last column for this dataset, or -1.
    153             *
    154             * @return The index.
    155             */
    156            private int lastCategoryIndex() {
    157                    if (this.maximumCategoryCount == 0) {
    158                            return -1;
    159                    }
    160                    return Math.min(this.firstCategoryIndex + this.maximumCategoryCount,
    161                                    this.underlying.getColumnCount()) - 1;
    162            }
    163
    164            /**
    165             * Returns the index for the specified column key.
    166             *
    167             * @param key  the key.
    168             *
    169             * @return The column index, or -1 if the key is not recognised.
    170             */
    171            public int getColumnIndex(Comparable key) {
    172                    int index = this.underlying.getColumnIndex(key);
    173                    if (index >= this.firstCategoryIndex && index <= lastCategoryIndex()) {
    174                            return index - this.firstCategoryIndex;
    175                    }
    176                    return -1;  // we didn't find the key
    177            }
    178
    179        /**
    180         * Returns the column key for a given index.
    181         *
    182         * @param column  the column index (zero-based).
    183         *
    184         * @return The column key.
    185         *
    186         * @throws IndexOutOfBoundsException if <code>row</code> is out of bounds.
    187         */
    188            public Comparable getColumnKey(int column) {
    189                    return this.underlying.getColumnKey(column + this.firstCategoryIndex);
    190            }
    191
    192        /**
    193         * Returns the column keys.
    194         *
    195         * @return The keys.
    196         *
    197         * @see #getColumnKey(int)
    198         */
    199            public List getColumnKeys() {
    200                    List result = new java.util.ArrayList();
    201                    int last = lastCategoryIndex();
    202                    for (int i = this.firstCategoryIndex; i < last; i++) {
    203                            result.add(this.underlying.getColumnKey(i));
    204                    }
    205                    return Collections.unmodifiableList(result);
    206            }
    207
    208        /**
    209         * Returns the row index for a given key.
    210         *
    211         * @param key  the row key.
    212         *
    213         * @return The row index, or <code>-1</code> if the key is unrecognised.
    214         */
    215            public int getRowIndex(Comparable key) {
    216                    return this.underlying.getRowIndex(key);
    217            }
    218
    219        /**
    220         * Returns the row key for a given index.
    221         *
    222         * @param row  the row index (zero-based).
    223         *
    224         * @return The row key.
    225         *
    226         * @throws IndexOutOfBoundsException if <code>row</code> is out of bounds.
    227         */
    228            public Comparable getRowKey(int row) {
    229                    return this.underlying.getRowKey(row);
    230            }
    231
    232        /**
    233         * Returns the row keys.
    234         *
    235         * @return The keys.
    236         */
    237            public List getRowKeys() {
    238                    return this.underlying.getRowKeys();
    239            }
    240
    241        /**
    242         * Returns the value for a pair of keys.
    243         *
    244         * @param rowKey  the row key (<code>null</code> not permitted).
    245         * @param columnKey  the column key (<code>null</code> not permitted).
    246         *
    247         * @return The value (possibly <code>null</code>).
    248         *
    249         * @throws UnknownKeyException if either key is not defined in the dataset.
    250         */
    251            public Number getValue(Comparable rowKey, Comparable columnKey) {
    252                    int r = getRowIndex(rowKey);
    253                    int c = getColumnIndex(columnKey);
    254                    if (c != -1) {
    255                return this.underlying.getValue(r, c + this.firstCategoryIndex);
    256                    }
    257                    else {
    258                            throw new UnknownKeyException("Unknown columnKey: " + columnKey);
    259                    }
    260            }
    261
    262        /**
    263         * Returns the number of columns in the table.
    264         *
    265         * @return The column count.
    266         */
    267            public int getColumnCount() {
    268                    int last = lastCategoryIndex();
    269                    if (last == -1) {
    270                            return 0;
    271                    }
    272                    else {
    273                return Math.max(last - this.firstCategoryIndex + 1, 0);
    274                    }
    275            }
    276
    277        /**
    278         * Returns the number of rows in the table.
    279         *
    280         * @return The row count.
    281         */
    282            public int getRowCount() {
    283                    return this.underlying.getRowCount();
    284            }
    285
    286        /**
    287         * Returns a value from the table.
    288         *
    289         * @param row  the row index (zero-based).
    290         * @param column  the column index (zero-based).
    291         *
    292         * @return The value (possibly <code>null</code>).
    293         */
    294            public Number getValue(int row, int column) {
    295                    return this.underlying.getValue(row, column + this.firstCategoryIndex);
    296            }
    297
    298        /**
    299         * Returns the percent complete for a given item.
    300         *
    301         * @param rowKey  the row key.
    302         * @param columnKey  the column key.
    303         *
    304         * @return The percent complete.
    305         */
    306            public Number getPercentComplete(Comparable rowKey, Comparable columnKey) {
    307                    int r = getRowIndex(rowKey);
    308                    int c = getColumnIndex(columnKey);
    309                    if (c != -1) {
    310                return this.underlying.getPercentComplete(r,
    311                            c + this.firstCategoryIndex);
    312                    }
    313                    else {
    314                            throw new UnknownKeyException("Unknown columnKey: " + columnKey);
    315                    }
    316            }
    317
    318        /**
    319         * Returns the percentage complete value of a sub-interval for a given item.
    320         *
    321         * @param rowKey  the row key.
    322         * @param columnKey  the column key.
    323         * @param subinterval  the sub-interval.
    324         *
    325         * @return The percent complete value (possibly <code>null</code>).
    326         *
    327         * @see #getPercentComplete(int, int, int)
    328         */
    329            public Number getPercentComplete(Comparable rowKey, Comparable columnKey,
    330                            int subinterval) {
    331                    int r = getRowIndex(rowKey);
    332                    int c = getColumnIndex(columnKey);
    333                    if (c != -1) {
    334                return this.underlying.getPercentComplete(r,
    335                            c + this.firstCategoryIndex, subinterval);
    336                    }
    337                    else {
    338                            throw new UnknownKeyException("Unknown columnKey: " + columnKey);
    339                    }
    340            }
    341
    342        /**
    343         * Returns the end value of a sub-interval for a given item.
    344         *
    345         * @param rowKey  the row key.
    346         * @param columnKey  the column key.
    347         * @param subinterval  the sub-interval.
    348         *
    349         * @return The end value (possibly <code>null</code>).
    350         *
    351         * @see #getStartValue(Comparable, Comparable, int)
    352         */
    353            public Number getEndValue(Comparable rowKey, Comparable columnKey,
    354                            int subinterval) {
    355                    int r = getRowIndex(rowKey);
    356                    int c = getColumnIndex(columnKey);
    357                    if (c != -1) {
    358                return this.underlying.getEndValue(r,
    359                            c + this.firstCategoryIndex, subinterval);
    360                    }
    361                    else {
    362                            throw new UnknownKeyException("Unknown columnKey: " + columnKey);
    363                    }
    364            }
    365
    366        /**
    367         * Returns the end value of a sub-interval for a given item.
    368         *
    369         * @param row  the row index (zero-based).
    370         * @param column  the column index (zero-based).
    371         * @param subinterval  the sub-interval.
    372         *
    373         * @return The end value (possibly <code>null</code>).
    374         *
    375         * @see #getStartValue(int, int, int)
    376         */
    377            public Number getEndValue(int row, int column, int subinterval) {
    378                    return this.underlying.getEndValue(row,
    379                                    column + this.firstCategoryIndex, subinterval);
    380            }
    381
    382        /**
    383         * Returns the percent complete for a given item.
    384         *
    385         * @param series  the row index (zero-based).
    386         * @param category  the column index (zero-based).
    387         *
    388         * @return The percent complete.
    389         */
    390            public Number getPercentComplete(int series, int category) {
    391                    return this.underlying.getPercentComplete(series,
    392                                    category + this.firstCategoryIndex);
    393            }
    394
    395        /**
    396         * Returns the percentage complete value of a sub-interval for a given item.
    397         *
    398         * @param row  the row index (zero-based).
    399         * @param column  the column index (zero-based).
    400         * @param subinterval  the sub-interval.
    401         *
    402         * @return The percent complete value (possibly <code>null</code>).
    403         *
    404         * @see #getPercentComplete(Comparable, Comparable, int)
    405         */
    406            public Number getPercentComplete(int row, int column, int subinterval) {
    407                    return this.underlying.getPercentComplete(row,
    408                                    column + this.firstCategoryIndex, subinterval);
    409            }
    410
    411        /**
    412         * Returns the start value of a sub-interval for a given item.
    413         *
    414         * @param rowKey  the row key.
    415         * @param columnKey  the column key.
    416         * @param subinterval  the sub-interval.
    417         *
    418         * @return The start value (possibly <code>null</code>).
    419         *
    420         * @see #getEndValue(Comparable, Comparable, int)
    421         */
    422            public Number getStartValue(Comparable rowKey, Comparable columnKey,
    423                            int subinterval) {
    424                    int r = getRowIndex(rowKey);
    425                    int c = getColumnIndex(columnKey);
    426                    if (c != -1) {
    427                return this.underlying.getStartValue(r,
    428                            c + this.firstCategoryIndex, subinterval);
    429                    }
    430                    else {
    431                            throw new UnknownKeyException("Unknown columnKey: " + columnKey);
    432                    }
    433            }
    434
    435        /**
    436         * Returns the start value of a sub-interval for a given item.
    437         *
    438         * @param row  the row index (zero-based).
    439         * @param column  the column index (zero-based).
    440         * @param subinterval  the sub-interval index (zero-based).
    441         *
    442         * @return The start value (possibly <code>null</code>).
    443         *
    444         * @see #getEndValue(int, int, int)
    445         */
    446            public Number getStartValue(int row, int column, int subinterval) {
    447                    return this.underlying.getStartValue(row,
    448                                    column + this.firstCategoryIndex, subinterval);
    449            }
    450
    451        /**
    452         * Returns the number of sub-intervals for a given item.
    453         *
    454         * @param rowKey  the row key.
    455         * @param columnKey  the column key.
    456         *
    457         * @return The sub-interval count.
    458         *
    459         * @see #getSubIntervalCount(int, int)
    460         */
    461            public int getSubIntervalCount(Comparable rowKey, Comparable columnKey) {
    462                    int r = getRowIndex(rowKey);
    463                    int c = getColumnIndex(columnKey);
    464                    if (c != -1) {
    465                return this.underlying.getSubIntervalCount(r,
    466                            c + this.firstCategoryIndex);
    467                    }
    468                    else {
    469                            throw new UnknownKeyException("Unknown columnKey: " + columnKey);
    470                    }
    471            }
    472
    473        /**
    474         * Returns the number of sub-intervals for a given item.
    475         *
    476         * @param row  the row index (zero-based).
    477         * @param column  the column index (zero-based).
    478         *
    479         * @return The sub-interval count.
    480         *
    481         * @see #getSubIntervalCount(Comparable, Comparable)
    482         */
    483            public int getSubIntervalCount(int row, int column) {
    484                    return this.underlying.getSubIntervalCount(row,
    485                                    column + this.firstCategoryIndex);
    486            }
    487
    488        /**
    489         * Returns the start value for the interval for a given series and category.
    490         *
    491         * @param rowKey  the series key.
    492         * @param columnKey  the category key.
    493         *
    494         * @return The start value (possibly <code>null</code>).
    495         *
    496         * @see #getEndValue(Comparable, Comparable)
    497         */
    498            public Number getStartValue(Comparable rowKey, Comparable columnKey) {
    499                    int r = getRowIndex(rowKey);
    500                    int c = getColumnIndex(columnKey);
    501                    if (c != -1) {
    502                return this.underlying.getStartValue(r, c + this.firstCategoryIndex);
    503                    }
    504                    else {
    505                            throw new UnknownKeyException("Unknown columnKey: " + columnKey);
    506                    }
    507            }
    508
    509        /**
    510         * Returns the start value for the interval for a given series and category.
    511         *
    512         * @param row  the series (zero-based index).
    513         * @param column  the category (zero-based index).
    514         *
    515         * @return The start value (possibly <code>null</code>).
    516         *
    517         * @see #getEndValue(int, int)
    518         */
    519            public Number getStartValue(int row, int column) {
    520                    return this.underlying.getStartValue(row,
    521                                    column + this.firstCategoryIndex);
    522            }
    523
    524        /**
    525         * Returns the end value for the interval for a given series and category.
    526         *
    527         * @param rowKey  the series key.
    528         * @param columnKey  the category key.
    529         *
    530         * @return The end value (possibly <code>null</code>).
    531         *
    532         * @see #getStartValue(Comparable, Comparable)
    533         */
    534            public Number getEndValue(Comparable rowKey, Comparable columnKey) {
    535                    int r = getRowIndex(rowKey);
    536                    int c = getColumnIndex(columnKey);
    537                    if (c != -1) {
    538                return this.underlying.getEndValue(r, c + this.firstCategoryIndex);
    539                    }
    540                    else {
    541                            throw new UnknownKeyException("Unknown columnKey: " + columnKey);
    542                    }
    543            }
    544
    545        /**
    546         * Returns the end value for the interval for a given series and category.
    547         *
    548         * @param series  the series (zero-based index).
    549         * @param category  the category (zero-based index).
    550         *
    551         * @return The end value (possibly <code>null</code>).
    552         */
    553            public Number getEndValue(int series, int category) {
    554                    return this.underlying.getEndValue(series,
    555                                    category + this.firstCategoryIndex);
    556            }
    557
    558            /**
    559             * Tests this <code>SlidingCategoryDataset</code> for equality with an
    560             * arbitrary object.
    561             *
    562             * @param obj  the object (<code>null</code> permitted).
    563             *
    564             * @return A boolean.
    565             */
    566            public boolean equals(Object obj) {
    567                    if (obj == this) {
    568                            return true;
    569                    }
    570                    if (!(obj instanceof SlidingGanttCategoryDataset)) {
    571                            return false;
    572                    }
    573                    SlidingGanttCategoryDataset that = (SlidingGanttCategoryDataset) obj;
    574                    if (this.firstCategoryIndex != that.firstCategoryIndex) {
    575                            return false;
    576                    }
    577                    if (this.maximumCategoryCount != that.maximumCategoryCount) {
    578                            return false;
    579                    }
    580                    if (!this.underlying.equals(that.underlying)) {
    581                            return false;
    582                    }
    583                    return true;
    584            }
    585
    586        /**
    587         * Returns an independent copy of the dataset.  Note that:
    588         * <ul>
    589         * <li>the underlying dataset is only cloned if it implements the
    590         * {@link PublicCloneable} interface;</li>
    591         * <li>the listeners registered with this dataset are not carried over to
    592         * the cloned dataset.</li>
    593         * </ul>
    594         *
    595         * @return An independent copy of the dataset.
    596         *
    597         * @throws CloneNotSupportedException if the dataset cannot be cloned for
    598         *         any reason.
    599         */
    600        public Object clone() throws CloneNotSupportedException {
    601            SlidingGanttCategoryDataset clone
    602                    = (SlidingGanttCategoryDataset) super.clone();
    603            if (this.underlying instanceof PublicCloneable) {
    604                    PublicCloneable pc = (PublicCloneable) this.underlying;
    605                clone.underlying = (GanttCategoryDataset) pc.clone();
    606            }
    607            return clone;
    608        }
    609
    610    }