OutputFormat.java

/*
 * Copyright 2017 Laszlo Balazs-Csiki
 *
 * This file is part of Pixelitor. Pixelitor is free software: you
 * can redistribute it and/or modify it under the terms of the GNU
 * General Public License, version 3 as published by the Free
 * Software Foundation.
 *
 * Pixelitor is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Pixelitor. If not, see <http://www.gnu.org/licenses/>.
 */
package pixelitor.io;

import pixelitor.Composition;
import pixelitor.utils.ImageUtils;
import pixelitor.utils.Messages;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;

/**
 * The output file format
 */
public enum OutputFormat {
    JPG {
        @Override
        public void doSaveComposition(Composition comp, File file) {
            BufferedImage img = comp.getCompositeImage();
            BufferedImage finalImage = ImageUtils.convertToRGB(img, false); // no alpha support, convert first to RGB
            OpenSaveManager.saveImageToFile(file, finalImage, toString());
        }
    }, PNG {
        @Override
        public void doSaveComposition(Composition comp, File file) {
            BufferedImage finalImage = comp.getCompositeImage(); // the format supports alpha, no need to convert ARGB to RGB
            OpenSaveManager.saveImageToFile(file, finalImage, toString());
        }
    }, GIF {
        @Override
        public void doSaveComposition(Composition comp, File file) {
            BufferedImage img = comp.getCompositeImage();
            // the format supports alpha, but the default encoder has bugs
            BufferedImage finalImage = ImageUtils.convertToRGB(img, false);

            OpenSaveManager.saveImageToFile(file, finalImage, toString());
        }
    }, BMP {
        @Override
        public void doSaveComposition(Composition comp, File file) {
            BufferedImage compositeImage = comp.getCompositeImage();
            BufferedImage finalImage = ImageUtils.convertToRGB(compositeImage, false); // no alpha support, convert first to RGB
            OpenSaveManager.saveImageToFile(file, finalImage, toString());
        }
    }, PXC {
        @Override
        public void doSaveComposition(Composition comp, File file) {
            OpenSaveManager.serializePXC(comp, file);
        }
    }, ORA {
        @Override
        void doSaveComposition(Composition comp, File file) {
            try {
                OpenRaster.writeOpenRaster(comp, file, false);
            } catch (IOException e) {
                Messages.showException(e);
            }
        }
    };

    @Override
    public String toString() {
        return super.toString().toLowerCase();
    }

    public static OutputFormat fromFile(File file) {
        String fileName = file.getName();
        String extension = FileExtensionUtils.getExt(fileName);
        return fromExtension(extension);
    }

    public static OutputFormat fromExtension(String extension) {
        String extLC = extension.toLowerCase();
        switch (extLC) {
            case "jpg":
            case "jpeg":
                return JPG;
            case "png":
                return PNG;
            case "bmp":
                return BMP;
            case "gif":
                return GIF;
            case "pxc":
                return PXC;
            case "ora":
                return ORA;
            default:
                throw new IllegalArgumentException("extension = " + extension);
        }
    }

    public void saveComp(Composition comp, File file, boolean addToRecentMenus) {
        doSaveComposition(comp, file);
        OpenSaveManager.afterSaveActions(comp, file, addToRecentMenus);
    }

    abstract void doSaveComposition(Composition comp, File file);

    private static OutputFormat lastOutputFormat = JPG;

    public static OutputFormat getLastUsed() {
        return lastOutputFormat;
    }

    public static void setLastUsed(OutputFormat lastOutputFormat) {
        OutputFormat.lastOutputFormat = lastOutputFormat;
    }
}