Makelangelo-software

package com.marginallyclever.makelangelo.plotter;

import com.hopding.jrpicam.RPiCamera;
import com.hopding.jrpicam.enums.AWB;
import com.hopding.jrpicam.enums.DRC;
import com.hopding.jrpicam.enums.Encoding;
import com.hopding.jrpicam.enums.Exposure;
import com.hopding.jrpicam.exceptions.FailedToRunRaspistillException;
import com.marginallyclever.makelangelo.MainFrame;
import com.marginallyclever.makelangelo.Translator;
import com.marginallyclever.makelangelo.actions.LoadFileAction;
import com.marginallyclever.makelangelo.paper.Paper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.imageio.ImageIO;
import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.Objects;

/**
 * Raspi camera capture to file for image processing
 */
public class PiCaptureAction {
    private static final int BUTTON_HEIGHT = 25;
	private static final Logger logger = LoggerFactory.getLogger(PiCaptureAction.class);
	private final RPiCamera piCamera = new RPiCamera("/home/pi/Pictures");
	
	// picam controls
	private BufferedImage buffImg;
	private boolean useImage;
	private int awb=1;
	private int drc=1;
	private int exp=11;
	private int contrast = 0;
	private int quality = 75;
	private int sharpness = 0;
	
	public PiCaptureAction() throws FailedToRunRaspistillException {
		super();
	}
	
	public void run(MainFrame mainFrame, Paper myPaper) {
        // let's make the image the correct width and height for the paper
		useImage = false;
        double aspectRatio = myPaper.getPaperWidth() / myPaper.getPaperHeight();
		final int captureH = 650;
        final int captureW = (int) ((double) captureH * aspectRatio);

		final JDialog dialog = new JDialog(mainFrame,Translator.get("PiCaptureAction.CaptureImageTitle"), true);
        dialog.setLocation(mainFrame.getLocation());

        JPanel panel = new JPanel();
        panel.setLayout(new GridBagLayout());
		final GridBagConstraints cMain = new GridBagConstraints();
		cMain.fill=GridBagConstraints.HORIZONTAL;
		cMain.anchor=GridBagConstraints.NORTH;
		cMain.gridx=0;
		cMain.gridy=0;
        cMain.gridheight = 1;
        cMain.gridwidth = 1;

		// create a frame to adjust the image

		panel.setBounds(1024, 100, 700, captureH);

        // if you add more things to the right side, you must increase this.
        cMain.gridheight = 16;
        final JLabel imageLabel = new JLabel();
        imageLabel.setPreferredSize(new Dimension(captureW, captureH));
  		panel.add(imageLabel, cMain);
        cMain.gridheight = 1;

        // all controls to the right
		cMain.gridx++;

		JLabel label = new JLabel(Translator.get("PiCaptureAction.AWB"));
		label.setPreferredSize(new Dimension(100,BUTTON_HEIGHT));
		panel.add(label, cMain);
		cMain.gridy++;

		String[] awbComboBoxChoices = {
		        Translator.get("PiCaptureAction.Off"),
                Translator.get("PiCaptureAction.Auto"),
                Translator.get("PiCaptureAction.Sun"),
                Translator.get("PiCaptureAction.Cloud"),
                Translator.get("PiCaptureAction.Shade"),
                Translator.get("PiCaptureAction.Tungsten"),
                Translator.get("PiCaptureAction.Fluorescent"),
                Translator.get("PiCaptureAction.Incandescent"),
                Translator.get("PiCaptureAction.Flash"),
                Translator.get("PiCaptureAction.Horizon") };
		final JComboBox<String> awbComboBox = new JComboBox<String>(awbComboBoxChoices);
		awbComboBox.setPreferredSize(new Dimension(100,BUTTON_HEIGHT));
		awbComboBox.setSelectedIndex(awb);
		panel.add(awbComboBox, cMain);
		cMain.gridy++;

		JLabel lblNewLabel = new JLabel(Translator.get("PiCaptureAction.DRC"));
		lblNewLabel.setPreferredSize(new Dimension(100,BUTTON_HEIGHT));
		panel.add(lblNewLabel, cMain);
		cMain.gridy++;

		String[] drcComboBoxChoices = {
                Translator.get("PiCaptureAction.Off"),
                Translator.get("PiCaptureAction.High"),
                Translator.get("PiCaptureAction.Medium"),
                Translator.get("PiCaptureAction.Low") };
		final JComboBox<String> drcComboBox = new JComboBox<>(drcComboBoxChoices);
		drcComboBox.setPreferredSize(new Dimension(100,BUTTON_HEIGHT));
		drcComboBox.setSelectedIndex(drc);
		panel.add(drcComboBox, cMain);
		cMain.gridy++;

		JLabel label_1 = new JLabel(Translator.get("PiCaptureAction.Exposure"));
		label_1.setPreferredSize(new Dimension(100,BUTTON_HEIGHT));
		panel.add(label_1, cMain);
		cMain.gridy++;

		String[] expComboBoxChoices = {
                Translator.get("PiCaptureAction.Antishake"),
                Translator.get("PiCaptureAction.Auto"),
                Translator.get("PiCaptureAction.Backlight"),
                Translator.get("PiCaptureAction.Beach"),
                Translator.get("PiCaptureAction.Fireworks"),
                Translator.get("PiCaptureAction.FixedFPS"),
                Translator.get("PiCaptureAction.Night"),
                Translator.get("PiCaptureAction.NightPreview"),
                Translator.get("PiCaptureAction.Snow"),
                Translator.get("PiCaptureAction.Sports"),
                Translator.get("PiCaptureAction.Spotlight"),
                Translator.get("PiCaptureAction.Verylong") };
		final JComboBox<String> expComboBox = new JComboBox<String>(expComboBoxChoices);
//		expComboBox.setBounds(584, 362, 90, 20);
		expComboBox.setPreferredSize(new Dimension(100,BUTTON_HEIGHT));
		expComboBox.setSelectedIndex(exp);
		panel.add(expComboBox, cMain);
		cMain.gridy++;

		JLabel lblContrast = new JLabel(Translator.get("PiCaptureAction.Contrast"));
//		lblContrast.setBounds(588, 393, 67, 14);
		lblContrast.setPreferredSize(new Dimension(100,BUTTON_HEIGHT));
		panel.add(lblContrast, cMain);
		cMain.gridy++;

		final JSlider contrastSlider = new JSlider();
		contrastSlider.setMinimum(-100);
//		contrastSlider.setBounds(588, 418, 90, 23);
		contrastSlider.setValue(contrast);
		panel.add(contrastSlider, cMain);
		cMain.gridy++;

		JLabel lblQuality = new JLabel(Translator.get("PiCaptureAction.Quality"));
//		lblQuality.setBounds(588, 452, 46, 14);
		lblQuality.setPreferredSize(new Dimension(100,BUTTON_HEIGHT));
		panel.add(lblQuality, cMain);
		cMain.gridy++;

		final JSlider qualitySlider = new JSlider();
		qualitySlider.setValue(quality);
//		qualitySlider.setBounds(584, 477, 90, 29);
		panel.add(qualitySlider, cMain);
		cMain.gridy++;

		JLabel lblSharpness = new JLabel(Translator.get("PiCaptureAction.Sharpness"));
//		lblSharpness.setBounds(585, 517, 66, 14);
		lblSharpness.setPreferredSize(new Dimension(100,BUTTON_HEIGHT));
		panel.add(lblSharpness, cMain);
		cMain.gridy++;

		final JSlider sharpnessSlider = new JSlider();
		sharpnessSlider.setMinimum(-100);
		sharpnessSlider.setValue(sharpness);
//		sharpnessSlider.setBounds(588, 542, 90, 23);
		panel.add(sharpnessSlider, cMain);
		cMain.gridy++;

		// I need 3 buttons one for Capture and one for Use if we have captured an image and one to just Cancel

        // a little space between everything else
        cMain.insets = new Insets(10,0,0,0);  //top padding

		JButton buttonCaptureImage = new JButton(Translator.get("PiCaptureAction.CaptureImage"));
		JButton buttonUseCapture = new JButton(Translator.get("PiCaptureAction.UseCapture"));

		buttonCaptureImage.addActionListener((arg0)->{
			try {
				piCamera.turnOnPreview(
						mainFrame.getLocationOnScreen().x + 50,
						mainFrame.getLocationOnScreen().y + 100,
						captureW,
						captureH);
				piCamera.setAWB(AWB.valueOf(((String) Objects.requireNonNull(awbComboBox.getSelectedItem())).toUpperCase()));
				piCamera.setDRC(DRC.valueOf(((String) Objects.requireNonNull(drcComboBox.getSelectedItem())).toUpperCase()));
				piCamera.setExposure(Exposure.valueOf(((String) Objects.requireNonNull(expComboBox.getSelectedItem())).toUpperCase()));
				piCamera.setEncoding(Encoding.JPG);
				piCamera.setWidth(captureW);
				piCamera.setHeight(captureH);
				piCamera.setContrast(contrastSlider.getValue());
				piCamera.setQuality(qualitySlider.getValue());
				piCamera.setSharpness(sharpnessSlider.getValue());
				piCamera.setTimeout(3000);
				buffImg = piCamera.takeBufferedStill();
				logger.debug("Executed this command:\n\t{}", piCamera.getPrevCommand());
				ImageIcon icon = new ImageIcon(buffImg);
				imageLabel.setIcon(icon);
				buttonUseCapture.setEnabled(true);
			} catch (Exception e) {
				logger.error("PiCaptureAction: ", e);
				JOptionPane.showMessageDialog(mainFrame, e.getLocalizedMessage(), Translator.get("ErrorTitle"), JOptionPane.ERROR_MESSAGE);
			}
		});
		buttonCaptureImage.setPreferredSize(new Dimension(89, BUTTON_HEIGHT));
		panel.add(buttonCaptureImage, cMain);
		cMain.gridy++;
        cMain.insets = new Insets(2,0,0,0);  //top padding

		buttonUseCapture.addActionListener(new ActionListener() {
        	@Override
			public void actionPerformed(ActionEvent arg0) {
				// we like this image, save off the parameters used.
				awb = awbComboBox.getSelectedIndex();
				drc = drcComboBox.getSelectedIndex();
				exp = expComboBox.getSelectedIndex();
				contrast = contrastSlider.getValue();
				quality = qualitySlider.getValue();
				sharpness = sharpnessSlider.getValue();

				File saveFile = new File("/home/pi/Pictures/capture.jpg");
				try {
					ImageIO.write(buffImg, "jpg", saveFile);
					useImage = true;
				} catch (IOException e) {
					logger.error("Error while saving {}", saveFile, e);
				}

				dialog.dispose();
			};
		});
		buttonUseCapture.setPreferredSize(new Dimension(89, BUTTON_HEIGHT));
		buttonUseCapture.setEnabled(false);
		panel.add(buttonUseCapture, cMain);
		cMain.gridy++;

		JButton buttonCancelCapture = new JButton(Translator.get("PiCaptureAction.CancelCapture"));
		buttonCancelCapture.addActionListener(new ActionListener() {
			public void actionPerformed(ActionEvent arg0) {
				dialog.dispose();
				useImage = false;
			};
		});
		buttonCancelCapture.setPreferredSize(new Dimension(89, BUTTON_HEIGHT));
		buttonCancelCapture.setEnabled(true);
		panel.add(buttonCancelCapture, cMain);

//		piCamera.setAWB(AWB.AUTO);	    // Change Automatic White Balance setting to automatic
//		piCamera.setDRC(DRC.OFF); 			// Turn off Dynamic Range Compression
//		piCamera.setContrast(100); 			// Set maximum contrast
//		piCamera.setSharpness(100);		    // Set maximum sharpness
//		piCamera.setQuality(100); 		    // Set maximum quality
//		piCamera.setTimeout(10000);		    // Wait 1 second to take the image
//		piCamera.turnOnPreview(200, 200, captureW, captureH);            // Turn on image preview
//		piCamera.setEncoding(Encoding.JPG); // Change encoding of images to PNG

		// Take a still image and save it as "/home/pi/Pictures/cameraCapture.jpg"

		logger.debug("We are about to display dialog\n");
		dialog.add(panel);
		dialog.pack();
		dialog.setVisible(true);
//			logger.debug("We are about to take a still image\n");
//			File image = piCamera.takeStill("cameraCapture.jpg", captureW, captureH);
//			logger.debug("New JPG capture saved to:\n\t" + image.getAbsolutePath());
//			piCamera.turnOffPreview();
		// setup for reopen

		if (useImage) {
			// process the image
			(new LoadFileAction(null,mainFrame,null)).openFile("/home/pi/Pictures/capture.jpg");
		}
	}
}