1 /** 2 * Copyright (c) 2011, The University of Southampton and the individual contributors. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without modification, 6 * are permitted provided that the following conditions are met: 7 * 8 * * Redistributions of source code must retain the above copyright notice, 9 * this list of conditions and the following disclaimer. 10 * 11 * * Redistributions in binary form must reproduce the above copyright notice, 12 * this list of conditions and the following disclaimer in the documentation 13 * and/or other materials provided with the distribution. 14 * 15 * * Neither the name of the University of Southampton nor the names of its 16 * contributors may be used to endorse or promote products derived from this 17 * software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 21 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 22 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR 23 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 24 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 25 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON 26 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 28 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29 */ 30 package org.openimaj.image.pixel.statistics; 31 32 import org.openimaj.image.FImage; 33 import org.openimaj.image.MBFImage; 34 import org.openimaj.math.statistics.distribution.MultidimensionalHistogram; 35 36 37 /** 38 * An array of multidimensional histograms calculated from image pixels using a mask 39 * (assumes image is in 0-1 range) 40 * 41 * @author Jonathon Hare 42 * 43 */ 44 public class MaskingBlockHistogramModel extends BlockHistogramModel { 45 private static final long serialVersionUID = 1L; 46 private FImage mask; 47 48 /** 49 * Construct with the given parameters 50 * @param mask the mask image 51 * @param blocks_x the number of blocks in the x-direction 52 * @param blocks_y the number of blocks in the y-direction 53 * @param nbins the number of bins in each dimension for the histograms 54 */ 55 public MaskingBlockHistogramModel(FImage mask, int blocks_x, int blocks_y, int... nbins) { 56 super(blocks_x, blocks_y, nbins); 57 58 this.mask = mask; 59 } 60 61 @Override 62 protected void accum(MBFImage im, int bx, int by) { 63 assert (im.numBands() == ndims); 64 65 MultidimensionalHistogram histogram = histograms[by][bx]; 66 int height = im.getHeight(); 67 int width = im.getWidth(); 68 69 int cols_per_block = width / blocks_x; 70 int startx = bx*cols_per_block; 71 int stopx = (1+bx)*cols_per_block; 72 73 int rows_per_block = height / blocks_y; 74 int starty = by*rows_per_block; 75 int stopy = (1+by)*rows_per_block; 76 77 if (stopx >= width) stopx = width; 78 if (stopy >= height) stopy = height; 79 80 for (int y=starty; y<stopy; y++) { 81 for (int x=startx; x<stopx; x++) { 82 if (mask.pixels[y][x] != 1) 83 continue; 84 85 int [] bins = new int[ndims]; 86 87 for (int i=0; i<ndims; i++) { 88 bins[i] = (int)(im.getBand(i).pixels[y][x] * (histogram.nbins[i])); 89 if (bins[i] >= histogram.nbins[i]) bins[i] = histogram.nbins[i] - 1; 90 } 91 92 int bin = 0; 93 for (int i=0; i<ndims; i++) { 94 int f = 1; 95 for (int j=0; j<i; j++) 96 f *= histogram.nbins[j]; 97 98 bin += f * bins[i]; 99 } 100 101 histogram.values[bin]++; 102 } 103 } 104 } 105 }