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 */ 030/** 031 * 032 */ 033package org.openimaj.examples.video.processing.motion; 034 035import java.io.IOException; 036 037import org.openimaj.image.MBFImage; 038import org.openimaj.math.geometry.point.Point2d; 039import org.openimaj.video.VideoDisplay; 040import org.openimaj.video.VideoDisplayListener; 041import org.openimaj.video.capture.VideoCapture; 042import org.openimaj.video.processing.motion.GridMotionEstimator; 043import org.openimaj.video.processing.motion.MotionEstimator; 044import org.openimaj.video.processing.motion.MotionEstimatorAlgorithm; 045import org.openimaj.video.translator.FImageToMBFImageVideoTranslator; 046import org.openimaj.video.translator.MBFImageToFImageVideoTranslator; 047 048/** 049 * Example showing how to estimate a motion field in a video using phase 050 * correlation. 051 * 052 * @author David Dupplaw (dpd@ecs.soton.ac.uk) 053 */ 054public class PhaseCorrelationExample { 055 /** 056 * @param args 057 * @throws IOException 058 */ 059 public static void main(String[] args) throws IOException { 060 final VideoCapture vc = new VideoCapture(320, 240); 061 final MotionEstimator me = new GridMotionEstimator( 062 new MBFImageToFImageVideoTranslator(vc), 063 new MotionEstimatorAlgorithm.PHASE_CORRELATION(), 10, 10, true); 064 065 final VideoDisplay<MBFImage> vd = VideoDisplay.createVideoDisplay( 066 new FImageToMBFImageVideoTranslator(me)); 067 vd.addVideoListener(new VideoDisplayListener<MBFImage>() 068 { 069 @Override 070 public void afterUpdate(VideoDisplay<MBFImage> display) 071 { 072 } 073 074 @Override 075 public void beforeUpdate(MBFImage frame) 076 { 077 for (final Point2d p : me.motionVectors.keySet()) 078 { 079 final Point2d p2 = me.motionVectors.get(p); 080 frame.drawLine((int) p.getX(), (int) p.getY(), 081 (int) (p.getX() + p2.getX()), 082 (int) (p.getY() + p2.getY()), 083 2, new Float[] { 1f, 0f, 0f }); 084 } 085 } 086 }); 087 } 088}