/*
* The Broad Institute
* SOFTWARE COPYRIGHT NOTICE AGREEMENT
* This software and its documentation are copyright 2009 by the
* Broad Institute/Massachusetts Institute of Technology. All rights are reserved.
*
* This software is supplied without any warranty or guaranteed support whatsoever. Neither
* the Broad Institute nor MIT can be responsible for its use, misuse, or functionality.
*/
package org.broadinstitute.sting.secondarybase;

import net.sf.picard.PicardException;
import edu.mit.broad.picard.util.BasicTextFileParser;
import net.sf.picard.util.FormatUtil;

import java.io.File;

/**
 * Class to parse the data in an Illumina Firecrest directory and return an iterator over that data, in order
 * by tile.
 *
 * @author Kiran Garimella
 */
public class FirecrestFileParser extends AbstractFirecrestFileParser {

    private BasicTextFileParser parser;
    private final FormatUtil formatter = new FormatUtil();
    private final File[] intensityFiles;

    /**
     * Constructor
     *
     * @param firecrestDirectory  directory where the Firecrest files can be located
     * @param lane                the lane to parse
     */
    public FirecrestFileParser(final File firecrestDirectory, final int lane) {
        super(firecrestDirectory, lane);
        intensityFiles = getFilesMatchingRegexp("s_" + lane + "_\\d{4}_int.txt(.gz)?");
    }

    @Override
    public boolean isValidFirecrestDirectory() {
        return (intensityFiles.length > 0);
    }

    /**
     * Sorts the relevant files in the firecrestDirectory.  Does some basic sanity checking to ensure that some files
     * are found and that they are the expected multiple for paired-end or not.
     */
    @Override
    protected void prepareToIterate() {
        // Some basic sanity checking on file counts
        if (intensityFiles.length == 0) {
            throw new PicardException("No Firecrest 1.3 intensity files found in " + firecrestDirectory.getAbsolutePath() + " for lane " + lane);
        }

        // Sort each set of reads and create a text parser for it
        parser = makeParserForTextFiles(true, intensityFiles);
    }

    /**
     * Parses the next line from the parser and constructs a FirecrestReadData object from it
     * The first 4 fields are position information for the read, and the remaining value are
     * the intensities data.
     *
     * @return  a fully populated FirecrestReadData object
     */
    protected FirecrestReadData readNext() {
        if (!parser.hasNext()) {
                return null;
        }
        final String[] data = parser.next();
        final int lane = formatter.parseInt(data[0]);
        final int tile = formatter.parseInt(data[1]);
        final int x = formatter.parseInt(data[2]);
        final int y = formatter.parseInt(data[3]);

        int intensityOffset = 4;
        int numIntensities = (data.length - 4)/4;

        double[][] intensities = new double[numIntensities][4];

        for (int cycle = 0, index = intensityOffset; cycle < numIntensities; cycle++) {
            for (int channel = 0; channel < 4; channel++, index++) {
                intensities[cycle][channel] = formatter.parseFloat(data[index]);
            }
        }

        return new FirecrestReadData(lane, tile, x, y, intensities);
    }

    /**
     * Closes the underlying PasteParser
     */
    @Override
    public void close() {
        if (parser != null) {
            parser.close();
        }
    }

}