001/*******************************************************************************
002 * This software is provided as a supplement to the authors' textbooks on digital
003 * image processing published by Springer-Verlag in various languages and editions.
004 * Permission to use and distribute this software is granted under the BSD 2-Clause
005 * "Simplified" License (see http://opensource.org/licenses/BSD-2-Clause).
006 * Copyright (c) 2006-2023 Wilhelm Burger, Mark J. Burge. All rights reserved.
007 * Visit https://imagingbook.com for additional details.
008 ******************************************************************************/
009package Ch21_Geometric_Operations;
010
011import ij.IJ;
012import ij.ImagePlus;
013import ij.gui.GenericDialog;
014import ij.plugin.filter.PlugInFilter;
015import ij.process.ImageProcessor;
016import imagingbook.common.geometry.mappings.linear.AffineMapping2D;
017import imagingbook.common.ij.DialogUtils;
018import imagingbook.common.ij.IjUtils;
019import imagingbook.common.image.ImageMapper;
020import imagingbook.core.jdoc.JavaDocHelp;
021import imagingbook.sampleimages.GeneralSampleImage;
022
023import java.awt.GridLayout;
024import java.awt.Label;
025import java.awt.Panel;
026import java.awt.TextField;
027import java.util.Locale;
028
029/**
030 * <p>
031 * ImageJ plugin, applies a configurable affine transformation to the current image. See Sec. 21.1.3 of [1] for details.
032 * Optionally opens a sample image if no image is currently open.
033 * </p>
034 * <p>
035 * [1] W. Burger, M.J. Burge, <em>Digital Image Processing &ndash; An Algorithmic Introduction</em>, 3rd ed, Springer
036 * (2022).
037 * </p>
038 *
039 * @author WB
040 * @version 2022/11/28
041 * @see ImageMapper
042 * @see AffineMapping2D
043 */
044public class Map_Affine_Matrix implements PlugInFilter, JavaDocHelp {
045
046        private static String[][] ElemNames = {
047                        { "a00", "a01", "a02" },
048                        { "a10", "a11", "a12" }};
049
050        private static double[][] A = {
051                        { 1, 0, 0 },
052                        { 0, 1, 0 }};
053        
054        /**
055         * Constructor, asks to open a predefined sample image if no other image
056         * is currently open.
057         */
058        public Map_Affine_Matrix() {
059                if (IjUtils.noCurrentImage()) {
060                        DialogUtils.askForSampleImage(GeneralSampleImage.Kepler);
061                }
062        }
063                
064        @Override
065        public int setup(String arg, ImagePlus imp) {
066                return DOES_ALL;
067        }
068
069        @Override
070        public void run(ImageProcessor ip) {
071                if (!runDialog()) {
072                        return;
073                }
074                AffineMapping2D imap = new AffineMapping2D(A).getInverse();
075                new ImageMapper(imap).map(ip);
076        }
077
078        // --------------------------------------------------------------------------------------
079
080        private boolean runDialog() {
081                GenericDialog gd = new GenericDialog(this.getClass().getSimpleName());
082                gd.addHelp(getJavaDocUrl());
083                gd.addMessage("Affine forward transformation matrix (source to target):");
084                TextField[] txtField = new TextField[A[0].length * A.length];
085                Panel panel = makePanel(A, ElemNames, txtField);
086                gd.addPanel(panel);
087                
088                gd.showDialog();
089                if (gd.wasCanceled()) {
090                        return false;
091                }
092                
093                return getPanelValues(A, txtField);
094        }
095
096        private Panel makePanel(double[][] vals, String[][] names, TextField[] textfield) {
097                Panel panel = new Panel();
098                panel.setLayout(new GridLayout(vals.length, vals[0].length * 2));
099                int i = 0;
100                for (int row = 0; row < vals.length; row++) {
101                        for (int col = 0; col < vals[row].length; col++) {
102                                textfield[i] = new TextField(String.format(Locale.US, "%.2f", vals[row][col]));
103                                panel.add(textfield[i]);
104                                panel.add(new Label(names[row][col]));
105                                i++;
106                        }
107                }
108                return panel;
109        }
110
111        private boolean getPanelValues(double[][] A, TextField[] tf) {
112                int i = 0; 
113                for (int r = 0; r < A.length; r++) {
114                        for (int c = 0; c < A[0].length; c++) {
115                                try {
116                                        A[r][c] = Double.valueOf(tf[i].getText());
117                                } catch (NumberFormatException e) {     
118                                        IJ.log("NumberFormatException: " + e.getMessage());
119                                        return false;
120                                }
121                                i++;
122                        }
123                }
124                return true;
125        }
126
127}