zzh
/
gatk-3.8
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

vast speedup 

git-svn-id: file:///humgen/gsa-scr1/gsa-engineering/svn_contents/trunk@729 348d0f76-0448-11de-a6fe-93d51630548a
This commit is contained in:
jmaguire 2009-05-16 15:27:58 +00:00
parent 6f1559bd77
commit 947bac5cdc
1 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
java/src/org/broadinstitute/sting/playground/gatk/walkers/CoverageBySample.java
View File

@ -18,17 +18,22 @@ public class CoverageBySample extends LocusWalker<String, String>
public boolean requiresReads() { return true; }
private SAMFileHeader header;
public void initialize()
{
GenomeAnalysisEngine toolkit = this.getToolkit();
SAMFileHeader header = toolkit.getSamReader().getFileHeader();
this.header = toolkit.getEngine().getSAMHeader();
List<SAMReadGroupRecord> read_groups = header.getReadGroups();
sample_names = new ArrayList<String>();
HashSet<String> unique_sample_names = new HashSet<String>();
for (int i = 0; i < read_groups.size(); i++)
{
String sample_name = read_groups.get(i).getSample();
if (unique_sample_names.contains(sample_name)) { continue; }
unique_sample_names.add(sample_name);
sample_names.add(sample_name);
}
}
@ -36,30 +41,31 @@ public class CoverageBySample extends LocusWalker<String, String>
public String map(RefMetaDataTracker tracker, char ref, LocusContext context)
{
String line = context.getLocation().getContig() + " " + context.getLocation().getStart() + " " ;
HashMap<String,Integer> counts = countReadsBySample(context);
for (int i = 0; i < sample_names.size(); i++)
{
int count = countReadsBySample(context, sample_names.get(i));
int count = counts.get(sample_names.get(i));
line += " " + count;
}
line += "\n";
return line;
}
private int countReadsBySample(LocusContext context, String sample_name)
private HashMap<String,Integer> countReadsBySample(LocusContext context)
{
int count = 0;
HashMap<String,Integer> counts = new HashMap<String,Integer>();
for (int i = 0; i < sample_names.size(); i++)
{
counts.put(sample_names.get(i), 0);
}
for (int i = 0; i < context.getReads().size(); i++)
{
SAMRecord read = context.getReads().get(i);
Integer offset = context.getOffsets().get(i);
String RG = (String)(read.getAttribute("RG"));
String sample = read.getHeader().getReadGroup(RG).getSample();
if (sample == sample_name)
{
count += 1;
}
String sample = header.getReadGroup(RG).getSample();
counts.put(sample, counts.get(sample)+1);
}
return count;
return counts;
}
public void onTraversalDone()
@ -82,7 +88,7 @@ public class CoverageBySample extends LocusWalker<String, String>
public String reduce(String line, String sum)
{
out.print(line);
return sum + line;
return "";
}