summaryrefslogtreecommitdiff
path: root/src/bjc/imgchain/pipeline/stages/ThresholdStage.java
blob: 146c5c43df6b62e33ef8403bf973a5e332639450 (plain)
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
package bjc.imgchain.pipeline.stages;

import java.awt.GridLayout;

import javax.swing.JComponent;
import javax.swing.JPanel;
import javax.swing.border.BevelBorder;
import javax.swing.border.TitledBorder;

import bjc.imgchain.pipeline.StageType;
import bjc.imgchain.utils.LabeledInputPanel;

/**
 * Stage which converts an image into three-tone.
 * 
 * @author Ben Culkin
 *
 */
public class ThresholdStage extends AbstractPixelStage {
	/**
	 * Create a new threshold stage, where all of the thresholds are 0.
	 */
	public ThresholdStage() {
		this(0, 0, 0);
	}

	/**
	 * Creates a new threshold stage.
	 * 
	 * @param rr The red threshold.
	 * @param gg The green threshold.
	 * @param bb The blue threshold.
	 */
	public ThresholdStage(int rr, int gg, int bb) {
		super(StageType.IMGTRANS);

		this.redThreshold = rr;
		this.greenThreshold = gg;
		this.blueThreshold = bb;
	}

	private int redThreshold;
	private int greenThreshold;
	private int blueThreshold;

	@Override
	public int[] processPixel(int[] pix) {
		int[] ret = new int[4];

		ret[0] = pix[0];
		ret[1] = pix[1] > redThreshold ? 255 : 0;
		ret[2] = pix[2] > greenThreshold ? 255 : 0;
		ret[3] = pix[3] > blueThreshold ? 255 : 0;

		ret[1] = Math.max(0, Math.min(255, ret[1]));
		ret[2] = Math.max(0, Math.min(255, ret[2]));
		ret[3] = Math.max(0, Math.min(255, ret[3]));

		return ret;
	}

	@Override
	public String name() {
		return "Colorized Threshold";
	}

	@Override
	public String description() {
		return "Convert image into three-tone";
	}

	@Override
	public JComponent getEditor() {
		JPanel holder = new JPanel();
		holder.setLayout(new GridLayout(3, 1));

		JPanel rSkew = new JPanel();
		rSkew.setLayout(new GridLayout(1, 3));
		rSkew.setBorder(new TitledBorder(new BevelBorder(BevelBorder.LOWERED), "Red Threshold"));

		JPanel gSkew = new JPanel();
		gSkew.setLayout(new GridLayout(1, 3));
		gSkew.setBorder(new TitledBorder(new BevelBorder(BevelBorder.LOWERED), "Green Threshold"));

		JPanel bSkew = new JPanel();
		bSkew.setLayout(new GridLayout(1, 3));
		bSkew.setBorder(new TitledBorder(new BevelBorder(BevelBorder.LOWERED), "Blue Threshold"));

		LabeledInputPanel rpercRed = new LabeledInputPanel("Red", redThreshold);
		rSkew.add(rpercRed);

		LabeledInputPanel gpercGreen = new LabeledInputPanel("Green", greenThreshold);
		gSkew.add(gpercGreen);

		LabeledInputPanel bpercBlue = new LabeledInputPanel("Blue", blueThreshold);
		bSkew.add(bpercBlue);

		rpercRed.field.addPropertyChangeListener("value", (ev) -> {
			redThreshold = (Integer) rpercRed.field.getValue();
		});

		gpercGreen.field.addPropertyChangeListener("value", (ev) -> {
			greenThreshold = (Integer) gpercGreen.field.getValue();
		});

		bpercBlue.field.addPropertyChangeListener("value", (ev) -> {
			blueThreshold = (Integer) bpercBlue.field.getValue();
		});

		holder.add(rSkew);
		holder.add(gSkew);
		holder.add(bSkew);

		return holder;
	}
}