001    /*
002     * Copyright (C) 2008-2010 by Holger Arndt
003     *
004     * This file is part of the Universal Java Matrix Package (UJMP).
005     * See the NOTICE file distributed with this work for additional
006     * information regarding copyright ownership and licensing.
007     *
008     * UJMP is free software; you can redistribute it and/or modify
009     * it under the terms of the GNU Lesser General Public License as
010     * published by the Free Software Foundation; either version 2
011     * of the License, or (at your option) any later version.
012     *
013     * UJMP is distributed in the hope that it will be useful,
014     * but WITHOUT ANY WARRANTY; without even the implied warranty of
015     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
016     * GNU Lesser General Public License for more details.
017     *
018     * You should have received a copy of the GNU Lesser General Public
019     * License along with UJMP; if not, write to the
020     * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
021     * Boston, MA  02110-1301  USA
022     */
023    
024    package org.ujmp.core.intmatrix.calculation;
025    
026    import org.ujmp.core.Matrix;
027    import org.ujmp.core.exceptions.MatrixException;
028    import org.ujmp.core.util.UJMPSettings;
029    
030    public class DiscretizeStandardBinning extends AbstractIntCalculation {
031            private static final long serialVersionUID = -2045926868254834270L;
032    
033            private Matrix min = null;
034    
035            private Matrix max = null;
036    
037            private int numberOfBins = 3;
038    
039            public DiscretizeStandardBinning(int dimension, Matrix matrix, int numberOfBins) {
040                    super(dimension, matrix);
041                    this.numberOfBins = numberOfBins;
042            }
043    
044            public int getInt(long... coordinates) throws MatrixException {
045                    if (min == null || max == null) {
046                            calculate();
047                    }
048                    double v = getSource().getAsDouble(coordinates);
049                    double mi = 0;
050                    double ma = 0;
051    
052                    switch (getDimension()) {
053                    case Matrix.ROW:
054                            mi = min.getAsDouble(0, coordinates[COLUMN]);
055                            ma = max.getAsDouble(0, coordinates[COLUMN]) + UJMPSettings.getTolerance();
056                            break;
057                    case Matrix.COLUMN:
058                            mi = min.getAsDouble(coordinates[ROW], 0);
059                            ma = max.getAsDouble(coordinates[ROW], 0) + UJMPSettings.getTolerance();
060                            break;
061                    default:
062                            mi = min.getAsDouble(0, 0);
063                            ma = max.getAsDouble(0, 0) + UJMPSettings.getTolerance();
064                            break;
065                    }
066    
067                    double bs = (ma - mi) / numberOfBins;
068                    int i = (int) Math.floor((v - mi) / bs);
069                    return i;
070            }
071    
072            private void calculate() {
073                    min = getSource().min(Ret.NEW, getDimension());
074                    max = getSource().max(Ret.NEW, getDimension());
075            }
076    
077    }