webcam/IJ_webcam_plugin.java at master · biol75/webcam · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import ij.IJ;
import ij.ImagePlus;
import ij.ImageStack;
import ij.Macro;
import ij.WindowManager;
import ij.gui.GenericDialog;
import ij.plugin.PlugIn;
import ij.process.ColorProcessor;
import ij.process.ImageProcessor;

import java.awt.Dimension;
import java.awt.image.BufferedImage;

import com.github.sarxos.webcam.Webcam;

/**
 * @author Jerome Mutterer
 * @date 2014 12 15 first release
 * @date 2015 01 14 added macro support and custom resolutions thanks to Jarom Jackson
 * @date 2015 02 03 added proper macro support using GenericDialog for options thanks to Wayne Rasband
 * @date 2015 08 14 modified to capture avi (Chris Elliott, biol75@york.ac.uk)
 */

public class IJ_webcam_plugin implements PlugIn {

	Webcam camera;
	BufferedImage image;
	ImagePlus imp, imp2;
	ImageProcessor ip;
	int camID=0, width=0, height=0;
    int sample_interval = 1000 ;
    int iMax = 10;
	boolean grab=false, customSize=false;

	public void run(String s) {

		if (!showDialog())
			return;

		camera = Webcam.getWebcams().get(camID);

		if (null != camera) {
			Dimension[] sizes = camera.getViewSizes();
			Dimension s1 = sizes[sizes.length - 1];

			if (customSize && (width > 0) && (height > 0)) {
				Dimension[] customSizes = new Dimension[1];
				customSizes[0] = new Dimension(width, height);
				camera.setCustomViewSizes(customSizes);
				s1 = customSizes[0];
			}

			camera.setViewSize(s1);
			camera.open();
			ip = new ColorProcessor(s1.width, s1.height);
			imp = new ImagePlus("Live (press Escape to grab)", ip);

			WindowManager.addWindow(imp.getWindow());

			imp.show();


            int i=0;
            while (null != imp.getWindow() && i < iMax) {

				if (camera.isImageNew()) {
					image = camera.getImage();
					imp2 = new ImagePlus("tmp", image);
                    ip = imp2.getProcessor();
                    imp.setProcessor(ip);


                    if (IJ.escapePressed()) {

                        // Add image to stack of current Image Plus Object
                        ImageStack imageStack = imp.getImageStack();
                        imageStack.addSlice("", new ColorProcessor(image));
                        imp.setStack("", imageStack);
                        imp.setSlice(imageStack.getSize());

                        imp.setTitle("Done " + i);
                        imageStack = null;

                        i ++ ;
                        IJ.wait (sample_interval);
                    }
                  imp.updateAndDraw();

				}
			}

            /*ip = stack.getProcessor(); /*ImageProcessor */
            /*Calibration cal=ip.getCalibration();
            cal.fps=1000/sample_interval; /* bug when < 1 fps ??
            ip.setCalibration(cal); */
            if (i > 0)
            {
               IJ.saveAs (imp, "AVI",  "/Users/cje2/Data/IJTest_webcam/Snapmvi" + i + ".avi");
            }
			imp.setTitle("Done");
			camera.close();
		}
	}

	boolean showDialog() {
		int n = 0;
		String[] cameraNames = new String[Webcam.getWebcams().size()];

		for (Webcam c : Webcam.getWebcams()) {
			cameraNames[n] = c.getName();
			n++;
		}

		GenericDialog gd = new GenericDialog("IJ webcam plugin...");
		gd.addChoice("Camera name", cameraNames, cameraNames[0]);
		gd.addCheckbox("Grab and return", false);
		gd.addCheckbox("Custom size", false);

        gd.addNumericField("Sample interval:", sample_interval / 1000, 0, 5, "s");
        gd.addNumericField("No of frames:", iMax, 0, 5, "frames");

        gd.addNumericField("Width:", width, 0, 5, "pixels");
		gd.addNumericField("Height:", height, 0, 5, "pixels");

		gd.showDialog();
		if (gd.wasCanceled())
			return false;
		camID = (int) gd.getNextChoiceIndex();
		grab = (boolean) gd.getNextBoolean();
		customSize = (boolean) gd.getNextBoolean();
        sample_interval = (int) (1000.0 * gd.getNextNumber());
        iMax = (int) gd.getNextNumber();
		width = (int) gd.getNextNumber();
		height = (int) gd.getNextNumber();

		return true;
	}
}