src/bjc/imgchain/pipeline/stages/PipeStage.java


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

package bjc.imgchain.pipeline.stages;

import java.awt.BorderLayout;
import java.awt.Image;

import javax.swing.JButton;
import javax.swing.JComponent;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JTextField;

import bjc.imgchain.ImgChain;
import bjc.imgchain.pipeline.Pipeline;
import bjc.imgchain.pipeline.PipelinePicker;
import bjc.imgchain.pipeline.StageType;

/**
 * Stage which runs a sub-pipeline.
 * @author Ben Culkin
 *
 */
public class PipeStage extends AbstractPipelineStage {
	private String pipeName;

	/**
	 * Create a new sub-pipeline stage.
	 */
	public PipeStage() {
		super(StageType.IMGTRANS);
	}

	@Override
	public Image process(Image inp) {
		Pipeline pipeline = ImgChain.chan.pipelineRepo.get(pipeName);

		return pipeline.process(inp);
	}

	@Override
	public String name() {
		return "Sub-pipeline";
	}

	@Override
	public String description() {
		return "Run the steps of another pipeline";
	}

	@Override
	public JComponent getEditor() {
		JPanel holder = new JPanel();
		holder.setLayout(new BorderLayout());

		JLabel pipeLabel = new JLabel("Pipeline");

		JTextField pipeField = new JTextField(80);
		pipeField.addPropertyChangeListener("value", (ev) -> {
			pipeName = pipeField.getText();
		});

		JButton pipeButton = new JButton("Select Pipeline");
		pipeButton.addActionListener((ev) -> {
			PipelinePicker pick = new PipelinePicker();

			pick.pack();
			pick.setVisible(true);

			if (pick.pipeName == null) {
				System.out.println("WARN: picked null pipe");
				return;
			}

			pipeField.setText(pick.pipeName);
		});

		holder.add(pipeLabel, BorderLayout.LINE_START);
		holder.add(pipeField, BorderLayout.CENTER);
		holder.add(pipeButton, BorderLayout.LINE_END);

		return holder;
	}
}