com.compomics.pepshell.controllers.dataexport.DataExport.java Source code

Introduction

Here is the source code for com.compomics.pepshell.controllers.dataexport.DataExport.java

Source

/*
 * Copyright 2014 Davy Maddelein.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.compomics.pepshell.controllers.dataexport;

import com.compomics.pepshell.model.protein.proteinimplementations.PepshellProtein;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.IOException;
import javax.imageio.ImageIO;
import javax.swing.JList;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Workbook;

/**
 *
 * @author Davy Maddelein
 */
class DataExport {

    public void exportToTSV(JList listOfProteins, File saveLocation) throws IOException {
        PepshellProtein pepshellProtein;
        StringBuilder tsvOutput = new StringBuilder();
        for (int i = 0; i < listOfProteins.getModel().getSize(); i++) {
            pepshellProtein = (PepshellProtein) listOfProteins.getModel().getElementAt(i);
            tsvOutput.append(pepshellProtein.getVisibleAccession()).append("\t");
            tsvOutput.append(DataPreparationForExport.returnListCommaSeparated(pepshellProtein.getPeptideGroups()))
                    .append("\t");
            tsvOutput.append(DataPreparationForExport.returnListCommaSeparated(pepshellProtein.getPdbFilesInfo()))
                    .append("\t");
            tsvOutput.append("\n");
        }
        writeFileToDisk(tsvOutput.toString(), saveLocation);
    }

    public void exportPNG(BufferedImage image, File saveLocation) throws IOException {
        ImageIO.write(image, "png", saveLocation);
    }

    public void excelProteinExport(JList listOfProteins, File saveLocation)
            throws FileNotFoundException, IOException {
        Workbook workBook = new HSSFWorkbook();
        ExcelExport.exportProteinToExcelWorkBook(listOfProteins, workBook);
        writeFileToDisk(workBook, saveLocation);
    }

    private void writeFileToDisk(Workbook workBook, File saveLocation) throws FileNotFoundException, IOException {
        try (FileOutputStream fos = new FileOutputStream(saveLocation)) {
            workBook.write(fos);
            fos.flush();
        }
    }

    private void writeFileToDisk(String output, File saveLocation) throws IOException {
        try (FileWriter fileWriter = new FileWriter(saveLocation)) {
            fileWriter.write(output);
            fileWriter.flush();
        }
    }
}