Mercurial > repos > timpalpant > java_genomics_toolkit
diff src/edu/unc/genomics/wigmath/Scale.java @ 2:e16016635b2a
Uploaded
author | timpalpant |
---|---|
date | Mon, 13 Feb 2012 22:12:06 -0500 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/edu/unc/genomics/wigmath/Scale.java Mon Feb 13 22:12:06 2012 -0500 @@ -0,0 +1,57 @@ +package edu.unc.genomics.wigmath; + +import java.io.IOException; +import java.util.Iterator; + +import org.apache.log4j.Logger; +import org.broad.igv.bbfile.WigItem; + +import com.beust.jcommander.Parameter; + +import edu.unc.genomics.io.WigFile; +import edu.unc.genomics.io.WigFileException; + +public class Scale extends WigMathTool { + + private static final Logger log = Logger.getLogger(Scale.class); + + @Parameter(names = {"-i", "--input"}, description = "Input file", required = true) + public WigFile inputFile; + @Parameter(names = {"-m", "--multiplier"}, description = "Multiplier (scale factor, default = 1/mean)") + public Double multiplier; + + + @Override + public void setup() { + inputs.add(inputFile); + + if (multiplier == null) { + multiplier = inputFile.numBases() / inputFile.total(); + } + } + + @Override + public float[] compute(String chr, int start, int stop) throws IOException, WigFileException { + log.debug("Computing difference for chunk "+chr+":"+start+"-"+stop); + + Iterator<WigItem> data = inputFile.query(chr, start, stop); + float[] result = WigFile.flattenData(data, start, stop); + + for (int i = 0; i < result.length; i++) { + result[i] = (float) (multiplier * result[i]); + } + + return result; + } + + + /** + * @param args + * @throws WigFileException + * @throws IOException + */ + public static void main(String[] args) throws IOException, WigFileException { + new Scale().instanceMain(args); + } + +}