001/** 002 * Copyright (c) 2011, The University of Southampton and the individual contributors. 003 * All rights reserved. 004 * 005 * Redistribution and use in source and binary forms, with or without modification, 006 * are permitted provided that the following conditions are met: 007 * 008 * * Redistributions of source code must retain the above copyright notice, 009 * this list of conditions and the following disclaimer. 010 * 011 * * Redistributions in binary form must reproduce the above copyright notice, 012 * this list of conditions and the following disclaimer in the documentation 013 * and/or other materials provided with the distribution. 014 * 015 * * Neither the name of the University of Southampton nor the names of its 016 * contributors may be used to endorse or promote products derived from this 017 * software without specific prior written permission. 018 * 019 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 020 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 021 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 022 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR 023 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 024 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 025 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON 026 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 027 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 028 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 029 */ 030package org.openimaj.demos.acmmm11.presentation.slides.tutorial; 031 032import org.openimaj.image.MBFImage; 033import org.openimaj.image.colour.ColourSpace; 034import org.openimaj.image.colour.Transforms; 035import org.openimaj.image.pixel.statistics.HistogramModel; 036import org.openimaj.math.geometry.shape.Rectangle; 037import org.openimaj.math.statistics.distribution.MultidimensionalHistogram; 038import org.openimaj.video.Video; 039 040/** 041 * Slide illustrating colour histogram extraction. 042 * 043 * @author Sina Samangooei (ss@ecs.soton.ac.uk) 044 * @author Jonathon Hare (jsh2@ecs.soton.ac.uk) 045 * 046 */ 047public class ColourHistogramGrid extends TutorialPanel { 048 private static final long serialVersionUID = 4894581289602770940L; 049 050 /** 051 * Default constructor 052 * 053 * @param capture 054 * @param width 055 * @param height 056 */ 057 public ColourHistogramGrid(Video<MBFImage> capture, int width, int height) { 058 super("Colour Histogram", capture, width, height); 059 } 060 061 @Override 062 public void doTutorial(MBFImage image) { 063 HistogramModel model = new HistogramModel(10,4,1); 064 MBFImage space = Transforms.RGB_TO_HSV(image); 065 model.estimateModel(space); 066 MultidimensionalHistogram feature = model.histogram; 067 Float[][] colours = buildBinCols(feature); 068 MBFImage colourGrid = new MBFImage(80,image.getHeight(),3); 069 int sqW = (colourGrid.getWidth()/4); 070 int sqH = (colourGrid.getHeight()/10); 071 for(int y = 0; y < 4; y++){ 072 for(int k = 0; k < 10; k++){ 073 Rectangle draw = new Rectangle(y * sqW,sqH*k,sqW,sqH); 074 colourGrid.drawShapeFilled(draw, colours[y * 10 + k]); 075 } 076 } 077 078// DisplayUtilities.displayName(colourGrid, "wang"); 079 image.drawImage(colourGrid, image.getWidth()-colourGrid.getWidth(), 0); 080 } 081 082 Float[][] buildBinCols(MultidimensionalHistogram feature) { 083 Float[][] binCols = new Float[10*4*1][]; 084 double maxFeature = feature.max(); 085 if(maxFeature == 0) maxFeature = 1; 086 for (int k=0; k<10; k++) { 087 for (int j=0; j<4; j++) { 088 float s = (float)j/4 + (0.5f/4); 089 float h = (float)k/10 + (0.5f/10); 090 091 MBFImage img = new MBFImage(1,1,ColourSpace.HSV); 092 img.setPixel(0, 0, new Float[] {h,s,(float) (feature.get(k,j,0) / maxFeature)}); 093// img.setPixel(0, 0, new Float[] {h,s,1f}); 094 095 img = Transforms.HSV_TO_RGB(img); 096 097 binCols[j* 10 + k] = img.getPixel(0, 0); 098 } 099 } 100 return binCols; 101 } 102 103}