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}