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.sandbox;
031
032import java.io.IOException;
033import java.util.List;
034
035import org.openimaj.content.animation.animator.DoubleArrayValueAnimator;
036import org.openimaj.content.animation.animator.ValueAnimator;
037import org.openimaj.image.FImage;
038import org.openimaj.image.ImageUtilities;
039import org.openimaj.image.model.asm.datasets.AMToolsSampleDataset;
040import org.openimaj.image.model.asm.datasets.ShapeModelDataset;
041import org.openimaj.math.geometry.line.Line2d;
042import org.openimaj.math.geometry.point.Point2d;
043import org.openimaj.math.geometry.point.PointList;
044import org.openimaj.math.geometry.point.PointListConnections;
045import org.openimaj.math.geometry.shape.PointDistributionModel;
046import org.openimaj.math.geometry.transforms.TransformUtilities;
047import org.openimaj.video.AnimatedVideo;
048import org.openimaj.video.VideoDisplay;
049
050public class PDMAnimator {
051        public static void main(String[] args) throws IOException {
052                // final ShapeModelDataset<FImage> dataset =
053                // IMMFaceDatabase.load(ImageUtilities.FIMAGE_READER);
054                final ShapeModelDataset<FImage> dataset = AMToolsSampleDataset.load(ImageUtilities.FIMAGE_READER);
055
056                final PointListConnections connections = dataset.getConnections();
057                final List<PointList> pointData = dataset.getPointLists();
058
059                final PointDistributionModel pdm = new PointDistributionModel(pointData);
060                pdm.setNumComponents(4);
061
062                VideoDisplay.createVideoDisplay(new AnimatedVideo<FImage>(new FImage(600, 600)) {
063                        ValueAnimator<double[]> a = DoubleArrayValueAnimator.makeRandomLinear(60, pdm.getStandardDeviations(3));
064
065                        @Override
066                        protected void updateNextFrame(FImage frame) {
067                                frame.fill(0f);
068
069                                final PointList newShape = pdm.generateNewShape(a.nextValue());
070                                final PointList tfShape = newShape.transform(TransformUtilities.translateMatrix(300, 300).times(
071                                                TransformUtilities.scaleMatrix(150, 150)));
072
073                                final List<Line2d> lines = connections.getLines(tfShape);
074                                frame.drawLines(lines, 1, 1f);
075
076                                for (final Point2d pt : tfShape) {
077                                        final Line2d normal = connections.calculateNormalLine(pt, tfShape, 10f);
078
079                                        if (normal != null) {
080                                                frame.drawLine(normal, 1, 0.5f);
081                                        }
082                                }
083                        }
084                });
085        }
086}