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 Ch11_Circle_Ellipse_Fitting;
010
011
012import ij.ImagePlus;
013import ij.gui.GenericDialog;
014import ij.gui.NewImage;
015import ij.gui.PointRoi;
016import ij.plugin.PlugIn;
017import ij.process.ImageProcessor;
018import imagingbook.common.color.sets.BasicAwtColor;
019import imagingbook.common.geometry.basic.Pnt2d;
020import imagingbook.common.geometry.ellipse.GeometricEllipse;
021import imagingbook.common.geometry.fitting.ellipse.utils.EllipseSampler;
022import imagingbook.common.ij.DialogUtils;
023import imagingbook.common.ij.DialogUtils.DialogLabel;
024import imagingbook.common.ij.DialogUtils.DialogStringColumns;
025import imagingbook.common.ij.RoiUtils;
026import imagingbook.common.ij.overlay.ColoredStroke;
027import imagingbook.common.ij.overlay.ShapeOverlayAdapter;
028import imagingbook.common.util.ParameterBundle;
029import imagingbook.core.jdoc.JavaDocHelp;
030
031import static imagingbook.common.ij.DialogUtils.addToDialog;
032import static imagingbook.common.ij.DialogUtils.getFromDialog;
033
034/**
035 * Samples points on a given (ideal) circle and creates a new image with the sample points marked and also contained in
036 * a {@link PointRoi}. Image size, ellipse parameters and noise can be specified. The result can be used as a test image
037 * for circle fitting. Note that the resulting image has an inverted LUT, i.e., the background value is 0 and marked
038 * points have value 255.
039 *
040 * @author WB
041 * @version 2022/10/03
042 */
043public class Ellipse_Make_Random implements PlugIn, JavaDocHelp {
044        
045        public static class Parameters implements ParameterBundle<Ellipse_Make_Random> {
046                
047                @DialogLabel("image title")@DialogStringColumns(12)
048                public String Title = "RandomEllipse"; // Ellipse_Make_Random.class.getSimpleName();
049                
050                @DialogLabel("image width")
051                public int W = 400;
052                @DialogLabel("image height")
053                public int H = 400;
054                
055                @DialogLabel("number of points")
056                public int n = 20;
057                
058                @DialogLabel("ellipse center (xc)")
059                public double xc = 200;
060                
061                @DialogLabel("ellipsecenter (yc)")
062                public double yc = 190;
063                
064                @DialogLabel("major axis radius (ra)")
065                public double ra = 170;
066                
067                @DialogLabel("minor axis radius (rb)")
068                public double rb = 120;
069                
070                @DialogLabel("start angle (deg)")
071                public double angle0 = 0;
072                
073                @DialogLabel("stop angle (deg)")
074                public double angle1 = 180; // was Math.PI/4;
075                
076                @DialogLabel("ellipse orientation (deg)")
077                public double theta = 45;
078                
079                @DialogLabel("x/y noise (sigma)")
080                public double sigma = 5.0; //2.0;
081                @DialogLabel("Random seed (0 = none)")
082                public int seed = 0;
083                
084                @DialogLabel("show real ellipse")
085                public boolean ShowRealCurve = true;
086                @DialogLabel("ellipse color")
087                public BasicAwtColor StrokeColor = BasicAwtColor.Green;
088                @DialogLabel("stroke width")
089                public double StrokeWidth = 1.0;
090        };
091        
092        
093        private static Parameters params = new Parameters();
094        
095        @Override
096        public void run(String arg) {
097                if (!runDialog()) {
098                        return;
099                }
100                
101                GeometricEllipse realEllipse = new GeometricEllipse(params.ra, params.rb, params.xc, params.yc, 
102                                Math.toRadians(params.theta));
103                EllipseSampler sampler = new EllipseSampler(realEllipse, params.seed);
104                Pnt2d[] points =
105                                sampler.getPoints(params.n, Math.toRadians(params.angle0), Math.toRadians(params.angle1), params.sigma);
106                
107                ImagePlus im = NewImage.createByteImage(params.Title, params.W, params.H, 1, NewImage.FILL_BLACK);
108                im.setRoi(RoiUtils.toPointRoi(points));
109                
110                ImageProcessor ip = im.getProcessor();
111                for (Pnt2d p : points) {
112                        int u = (int) Math.rint(p.getX());
113                        int v = (int) Math.rint(p.getY());
114                        ip.putPixel(u, v, 255);
115                }
116                ip.invertLut();
117                
118                if (params.ShowRealCurve) {
119                        ShapeOverlayAdapter ola = new ShapeOverlayAdapter();
120                        ColoredStroke circleStroke = new ColoredStroke(params.StrokeWidth, params.StrokeColor.getColor());
121                        ola.addShapes(realEllipse.getShapes(3), circleStroke);
122                        im.setOverlay(ola.getOverlay());
123                }
124                
125                im.show();
126        }
127        
128        // ------------------------------------------
129        
130        private boolean runDialog() {
131                GenericDialog gd = new GenericDialog(this.getClass().getSimpleName());
132                gd.addHelp(getJavaDocUrl());
133                gd.addMessage(DialogUtils.formatText(50,
134                                "This plugin samples points on a given (ideal) ellipse and",
135                                "creates a new image with the sample points marked and also",
136                                "contained in a ROI (float coordinates)."
137                                ));
138                
139                addToDialog(params, gd);
140
141                gd.showDialog();
142                if (gd.wasCanceled())
143                        return false;
144
145                getFromDialog(params, gd);
146
147                return params.validate();
148        }
149
150}