import ij.*;
import ij.plugin.filter.PlugInFilter;
import ij.process.*;
import ij.gui.*;
import java.awt.*;
import java.awt.geom.*;
import java.awt.image.*;
import java.util.*;
import ij.plugin.*;

public class Dotted_Line implements PlugInFilter {
    ImagePlus imp;
    ImageCanvas canvas;
    Graphics2D g2;
    static double width = 2;
    static String dashpattern = "12, 12";
    float[] dashvalues;

     public int setup(String arg, ImagePlus imp) {
          if (IJ.versionLessThan("1.31i"))
             return DONE;
          IJ.register(Dotted_Line.class);
          this.imp = imp;
          if (!isSelection()) {
             IJ.showMessage("Dotted Line", "Image with selection required.");
             return DONE;
          } else
              return DOES_ALL+NO_CHANGES;
   }

    public void run(ImageProcessor ip) {
       Undo.setup(Undo.TYPE_CONVERSION, imp);
       Integer id = new Integer(imp.getID());
       ImageWindow win = imp.getWindow();
       win.addWindowListener(win);
       canvas = win.getCanvas();
       draw();
    }

    //Gets the values for the width and the dash pattern 
    public void userinput() {
       GenericDialog gd = new GenericDialog("Dotted Line Tool");
       gd.addNumericField("Line Width", width, 2);
       gd.addStringField("Dash Pattern", dashpattern);
       gd.showDialog();
       width = gd.getNextNumber();
       //Convert String values into dashvalues float array
       dashpattern = gd.getNextString();
       StringTokenizer st = new StringTokenizer(dashpattern , ", \t");
       int tokens = st.countTokens();
       dashvalues = new float[tokens];
       for(int i=0; i<tokens; i++) {
           try {
               dashvalues[i] = Float.parseFloat(st.nextToken().trim());
           } catch (NumberFormatException e){IJ.showMessage("Error: Invalid Values for Dash Pattern");}
       }
    }

    //Draws various dashed patterns with different roi type selections 
    public void draw() {
       userinput();
       BufferedImage bi = new BufferedImage(imp.getWidth(),
       imp.getHeight(),
       BufferedImage.TYPE_INT_RGB);

       g2 = (Graphics2D)bi.getGraphics();
       g2.drawImage(imp.getImage(), 0, 0, null);
       Roi roi = imp.getRoi();
       imp.setRoi(roi);
       if (!isSelection())
          return;

       Color c = Toolbar.getForegroundColor();
       Stroke stroke = new BasicStroke((float)width, BasicStroke.CAP_BUTT, BasicStroke.JOIN_BEVEL, 0, dashvalues, 0);
       g2.setStroke(stroke);
       g2.setColor(c);
       int type = roi.getType();
       Rectangle r = roi.getBoundingRect();
       switch (type) {
          case 0 :
             //Rectangular Selections
             Rectangle2D rect = new Rectangle2D.Double(r.x, r.y, r.width, r.height);
             g2.draw(rect);
             break;
          case 1 :
             //Oval Selections
             Ellipse2D ellipse = new Ellipse2D.Double(r.x, r.y, r.width, r.height);
             g2.draw(ellipse);
             break;
          case 2 :
             //Polygon Selections
             PolygonRoi pr = (PolygonRoi) roi;
             int x1Points[] = pr.getXCoordinates();
             int y1Points[] = pr.getYCoordinates();
             int Length = pr.getNCoordinates();
             GeneralPath polygon = new GeneralPath(GeneralPath.WIND_EVEN_ODD, Length);
             polygon.moveTo((x1Points[0] + r.x), (y1Points[0] + r.y));
             for (int index = 1; index < Length; index++) {
                polygon.lineTo((x1Points[index] + r.x), (y1Points[index] + r.y));
             };
             polygon.closePath();
             g2.draw(polygon);
             break;
          case 3 :
             //Freehand Selections
             PolygonRoi freehandpr = (PolygonRoi) roi;
             int fhx1Points[] = freehandpr.getXCoordinates();
             int fhy1Points[] = freehandpr.getYCoordinates();
             int freehandLength = freehandpr.getNCoordinates();
             GeneralPath freehandpolygon = new GeneralPath(GeneralPath.WIND_EVEN_ODD, freehandLength);
             freehandpolygon.moveTo((fhx1Points[0] + r.x), (fhy1Points[0] + r.y));
             for (int index = 1; index < freehandLength; index++) {
                freehandpolygon.lineTo((fhx1Points[index] + r.x), (fhy1Points[index] + r.y));
             };
             freehandpolygon.closePath();
             g2.draw(freehandpolygon);
             break;
          case 5 :
             //Straight Line Selections
             Line L = (Line) roi;
             Line2D line = new Line2D.Double((double)L.x1, (double)L.y1, (double)L.x2, (double)L.y2);
             g2.draw(line);
             break;
          case 6 :
             //Segmented Line Selections
             PolygonRoi seglinpr = (PolygonRoi) roi;
             int slx1Points[] = seglinpr.getXCoordinates();
             int sly1Points[] = seglinpr.getYCoordinates();
             int seglinLength = seglinpr.getNCoordinates();
             GeneralPath seglinpolygon = new GeneralPath(GeneralPath.WIND_EVEN_ODD, seglinLength);
             seglinpolygon.moveTo((slx1Points[0] + r.x), (sly1Points[0] + r.y));
             for (int index = 1; index < seglinLength; index++) {
                seglinpolygon.lineTo((slx1Points[index] + r.x), (sly1Points[index] + r.y));
             };
             g2.draw(seglinpolygon);
             break;
          case 7 :
             //Freehand Line Selections
             PolygonRoi fhlinpr = (PolygonRoi) roi;
             int fhlx1Points[] = fhlinpr.getXCoordinates();
             int fhly1Points[] = fhlinpr.getYCoordinates();
             int fhlinLength = fhlinpr.getNCoordinates();
             GeneralPath freehandlinpolygon = new GeneralPath(GeneralPath.WIND_EVEN_ODD, fhlinLength);
             freehandlinpolygon.moveTo((fhlx1Points[0] + r.x), (fhly1Points[0] + r.y));
             for (int index = 1; index < fhlinLength; index++) {
                freehandlinpolygon.lineTo((fhlx1Points[index] + r.x), (fhly1Points[index] + r.y));
             };
             g2.draw(freehandlinpolygon);
             break;
          case 8 :
             //Angle Selections
             PolygonRoi apr = (PolygonRoi) roi;
             int ax1Points[] = apr.getXCoordinates();
             int ay1Points[] = apr.getYCoordinates();
             int aLength = apr.getNCoordinates();
             GeneralPath apolygon = new GeneralPath(GeneralPath.WIND_EVEN_ODD, aLength);
             apolygon.moveTo((ax1Points[0] + r.x), (ay1Points[0] + r.y));
             for (int index = 1; index < aLength; index++) {
                apolygon.lineTo((ax1Points[index] + r.x), (ay1Points[index] + r.y));
             };
             g2.draw(apolygon);
             break;
          }
       imp.setImage(bi);
       imp.setRoi(roi);
}

    //Tests to see if a selections has been made, this test is executed prior to displaying the dialog box.
    boolean isSelection() {
         if (imp==null)
             return false;
         Roi roi = imp.getRoi();
         if (roi==null)
             return false;
         int roiType = roi.getType();
         if (roiType<=Roi.ANGLE)
             return true;
        else
             return false;
     }
}