net.jakubec.view.save.ImageSaver Maven / Gradle / Ivy

Go to download
package net.jakubec.view.save;

import java.awt.image.*;
import java.io.File;
import java.io.IOException;
import java.util.Locale;

import javax.imageio.IIOImage;
import javax.imageio.ImageIO;
import javax.imageio.ImageWriteParam;
import javax.imageio.ImageWriter;
import javax.imageio.stream.ImageOutputStream;
import javax.imageio.plugins.jpeg.JPEGImageWriteParam;
import javax.swing.JFileChooser;
import javax.swing.SwingUtilities;

public class ImageSaver {
	public static File save(BufferedImage image, String dirPath, String imgName) throws IOException{
		String[] n=imgName.split("\\.");
		if (n[n.length-1].equals("jpg")){
			return writeJPG(image,dirPath, imgName);
		}else{
			File file=new File(dirPath+File.separator+imgName);
			ImageIO.write(image, n[n.length-1], file);
			return file;
		}
		
	}
	
	public static File saveAs(BufferedImage image, String dirPath) throws IOException{
		JFileChooser chooser = new JFileChooser(dirPath);
		SwingUtilities.updateComponentTreeUI(chooser);
		chooser.setFileSelectionMode(JFileChooser.FILES_ONLY);
		
		chooser.showSaveDialog(null);
		
		File save=chooser.getSelectedFile();
		System.out.println(save.getName());
		String[] temp=(save.getName()).split("\\.");

		System.out.println(temp.length);
		for (int i=0;i