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}