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.tools.imagecollection.collection.video; 31 32 import java.util.HashMap; 33 import java.util.Iterator; 34 35 import org.openimaj.image.Image; 36 import org.openimaj.tools.imagecollection.collection.ImageCollectionEntry; 37 import org.openimaj.tools.imagecollection.collection.ImageCollectionEntrySelection; 38 import org.openimaj.video.Video; 39 40 public class MetadataVideoIterator<T extends Image<?, T>> implements Iterator<ImageCollectionEntry<T>> { 41 42 private Video<T> video; 43 private ImageCollectionEntrySelection<T> selection; 44 private int frameCount = 0; 45 46 public MetadataVideoIterator(ImageCollectionEntrySelection<T> selection, Video<T> video) { 47 this.video = video; 48 this.selection = selection; 49 } 50 51 @Override 52 public boolean hasNext() { 53 return frameCount >= 0 && video.hasNextFrame(); 54 } 55 56 @Override 57 public ImageCollectionEntry<T> next() { 58 final T image = video.getNextFrame(); 59 final ImageCollectionEntry<T> entry = new ImageCollectionEntry<T>(); 60 entry.meta = new HashMap<String, String>(); 61 entry.meta.put("timestamp", "" + this.frameCount / this.video.getFPS()); 62 entry.accepted = selection.acceptEntry(image); 63 entry.image = image; 64 this.frameCount++; 65 66 // hack to stop the iterator at the end until hasNext works properly 67 if (image == null) 68 frameCount = -1; 69 70 return entry; 71 } 72 73 @Override 74 public void remove() { 75 // TODO Auto-generated method stub 76 77 } 78 79 }