net.maizegenetics.analysis.modelfitter.AdditiveSiteStorePlugin Maven / Gradle / Ivy

package net.maizegenetics.analysis.modelfitter;

import java.awt.Frame;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Consumer;
import java.util.stream.IntStream;

import javax.swing.ImageIcon;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import net.maizegenetics.analysis.modelfitter.AdditiveSite.CRITERION;
import net.maizegenetics.dna.map.Position;
import net.maizegenetics.dna.snp.GenotypeTable;
import net.maizegenetics.plugindef.AbstractPlugin;
import net.maizegenetics.plugindef.DataSet;
import net.maizegenetics.plugindef.Datum;
import net.maizegenetics.plugindef.PluginParameter;
import net.maizegenetics.taxa.Taxon;

public class AdditiveSiteStorePlugin extends AbstractPlugin {
    private static Logger myLogger = LogManager.getLogger(AdditiveSiteStorePlugin.class);
    PluginParameter outFilename =
            new PluginParameter.Builder<>("filename", null, String.class)
                    .guiName("Output Filename")
                    .description("The name of file to which the additive sites will be written.")
                    .required(true)
                    .build();
    PluginParameter isGenotype =
            new PluginParameter.Builder<>("genotype", true, Boolean.class)
                    .guiName("Is Genotype?")
                    .description("Is the input nucleotide genotypes? If false, then ReferenceProbability will be used.")
                    .build();

    public AdditiveSiteStorePlugin() {
        super(null, false);
    }

    public AdditiveSiteStorePlugin(Frame parentFrame, boolean isInteractive) {
        super(parentFrame, isInteractive);
    }

    @Override
    protected void preProcessParameters(DataSet input) {
        List datumList = input.getDataOfType(GenotypeTable.class);
        if (datumList.size() != 1)
            throw new IllegalArgumentException("Exactly one genotype dataset will is required as input.");
    }

    @Override
    public DataSet processData(DataSet input) {
        GenotypeTable myGenotype =
                (GenotypeTable) input.getDataOfType(GenotypeTable.class).get(0).getData();
        int numberOfSites = myGenotype.numberOfSites();

        List siteList = new ArrayList<>(numberOfSites);
        long start = System.nanoTime();
        if (isGenotype.value()) {
            for (int s = 0; s < numberOfSites; s++) {
                siteList.add(new GenotypeAdditiveSite(s, myGenotype.chromosomeName(s), myGenotype.chromosomalPosition(s), myGenotype.siteName(s), CRITERION.pval, myGenotype.genotypeAllTaxa(s), myGenotype.majorAllele(s), myGenotype.majorAlleleFrequency(s)));
            }
        } else {
            int ntaxa = myGenotype.numberOfTaxa();
            for (int s = 0; s < numberOfSites; s++) {
                float[] refprobs = new float[ntaxa];
                for (int t = 0; t < ntaxa; t++)
                    refprobs[t] = myGenotype.referenceProbability(t, s);
                siteList.add(new RefProbAdditiveSite(s, myGenotype.chromosomeName(s), myGenotype.chromosomalPosition(s), myGenotype.siteName(s), CRITERION.pval, refprobs));
            }

        }

        myLogger.debug(String.format("site list created with %d sites in %d ms.", siteList.size(), (System.nanoTime() - start) / 1000000));

        start = System.nanoTime();
        String outfile = outFilename.value();
        ObjectOutputStream out;
        try {
            out = new ObjectOutputStream(new FileOutputStream(outfile));

            Consumer writeSite = s -> {
                try {
                    out.writeObject(s);
                } catch (IOException ioe) {
                    throw new RuntimeException("Error writing to SiteList object store.", ioe);
                } 
            };

            out.writeObject(new Integer(siteList.size()));
            
            ArrayList taxaNameList = new ArrayList<>();
            for (Taxon taxon : myGenotype.taxa()) taxaNameList.add(taxon.getName());
            
            out.writeObject(taxaNameList);
            siteList.stream().forEach(writeSite);
            out.flush();
            out.close();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        myLogger.debug(String.format("Sites written to outfile in %d ms.\n", (System.nanoTime() - start) / 1000000));

        return null;
    }

    @Override
    public String pluginDescription() {
        return "This plugin serializes a genotype as a list of additive sites. The objects stored are an Integer equal to the number of sites, "
                + "the Genotype TaxaList, then all of the additive sites";
    }

    @Override
    public ImageIcon getIcon() {
        return null;
    }

    @Override
    public String getButtonName() {
        return "Write additive site file";
    }

    @Override
    public String getToolTipText() {
        return "Stores genotypes coded as additive sites";
    }

    public static Object[] readSiteStore(String filename) {
    		try {
				ObjectInputStream ois = new ObjectInputStream(new FileInputStream(filename));
				int numberOfSites = ((Integer) ois.readObject()).intValue();
				Object[] retrievedObjects = new Object[2 + numberOfSites];
				retrievedObjects[0] = ois.readObject();
				List mySites = new ArrayList<>();
				for (int s = 0; s < numberOfSites; s++) {
					mySites.add((AdditiveSite) ois.readObject());
				}
				retrievedObjects[1] = mySites;
				ois.close();
				System.out.println("Additive Sites successfully read.");
				return retrievedObjects;
			} catch (IOException e) {
				throw new RuntimeException(e);
			} catch (ClassNotFoundException e) {
				throw new RuntimeException(e);
			}
    }
}