ijfx.core.batch.item.SaveToFileWrapper.java Source code

Introduction

Here is the source code for ijfx.core.batch.item.SaveToFileWrapper.java

Source

/*
This file is part of ImageJ FX.
    
ImageJ FX is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
    
ImageJ FX 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 ImageJ FX.  If not, see <http://www.gnu.org/licenses/>. 
    
 Copyright 2015,2016 Cyril MONGIS, Michael Knop
       
 */
package ijfx.core.batch.item;

import ijfx.core.batch.BatchSingleInput;
import ijfx.ui.main.ImageJFX;
import io.scif.services.DatasetIOService;
import java.io.File;
import java.io.IOException;
import java.util.logging.Level;
import java.util.logging.Logger;
import mongis.utils.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.scijava.Context;
import org.scijava.plugin.Parameter;

/**
 *
 * @author Cyril MONGIS, 2016
 */
public class SaveToFileWrapper extends AbstractSaverWrapper {

    private File saveTo;

    @Parameter
    private DatasetIOService datasetIoService;

    private String suffix = null;

    private SaveToFileWrapper(Context context, BatchSingleInput input) {
        super(input);
        context.inject(this);
    }

    public SaveToFileWrapper(Context context, BatchSingleInput input, File saveFile) {

        this(context, input, saveFile, null);

    }

    public SaveToFileWrapper(Context context, BatchSingleInput input, File saveFile, String suffix) {

        this(context, input);
        String baseName;
        if (input.getDefaultSaveName() != null) {
            baseName = input.getDefaultSaveName();
        } else {
            baseName = FilenameUtils.getBaseName(saveFile.getName());
        }

        String ext = FilenameUtils.getExtension(saveFile.getName());

        String finalName;

        if (suffix != null) {
            finalName = new StringBuilder().append(baseName).append("_").append(suffix).append(".").append(ext)
                    .toString();
        } else {
            finalName = new StringBuilder().append(baseName).append(".").append(ext).toString();
        }
        saveTo = new File(saveFile.getParentFile(), finalName);
    }

    @Override
    public void save() {

        String savePath = saveTo.getAbsolutePath();

        try {
            if (datasetIoService.canSave(savePath) == false) {
                savePath = FileUtils.changeExtensionTo(savePath, "tif");
            }
        } catch (Exception e) {
            savePath = FileUtils.changeExtensionTo(savePath, "tif");
        }

        try {
            ImageJFX.getLogger().info("Saving to ..." + savePath);
            datasetIoService.save(getWrappedObject().getDataset(), savePath);
        } catch (IOException ex) {
            Logger.getLogger(SaveToFileWrapper.class.getName()).log(Level.SEVERE, null, ex);
        }

        getWrappedObject().save();
    }

}