@Override
public void onTraversalStart() {
sampleCollection = new SampleCollection(getHeaderForReads());
logger.log(Level.INFO, "Reading targets locations from intervals...");
targetCollection = resolveTargetCollection();
// Initializing count and count column management member fields:
countColumns = groupBy.countColumns(this);
final int columnCount = countColumns.columnCount();
counts = new int[columnCount][targetCollection.targetCount()];
// Open output files and write headers:
outputWriter =
openOutputWriter(
output,
composeMatrixOutputHeader(
getCommandLine(), targetOutInfo, groupBy, countColumns.columnNames()));
if (columnSummaryOutput != null) {
columnSummaryOutputWriter =
openOutputWriter(
columnSummaryOutput,
composeColumnSummaryHeader(
getCommandLine(),
groupBy,
targetCollection.targetCount(),
targetCollection.totalSize()));
}
if (rowSummaryOutput != null) {
rowSummaryOutputWriter =
openOutputWriter(
rowSummaryOutput,
composeRowOutputHeader(
getCommandLine(), targetOutInfo, groupBy, countColumns.columnCount()));
}
// Next we start the traversal:
logger.log(Level.INFO, "Collecting read counts ...");
}
@Override
public void apply(
final GATKRead read,
final ReferenceContext referenceContext,
final FeatureContext featureContext) {
final SimpleInterval readLocation = referenceContext.getInterval();
final int columnIndex = countColumns.columnIndex(read);
if (columnIndex >= 0) { // < 0 would means that the read is to be ignored.
targetCollection.indexRange(readLocation).forEach(i -> counts[columnIndex][i]++);
}
}
/** Writes the column summary output table. */
private void writeColumnSummaryOutput() {
if (columnSummaryOutputWriter == null) {
return;
}
final long totalSize = targetCollection.totalSize();
final List<String> columnNames = countColumns.columnNames();
for (int i = 0; i < columnNames.size(); i++) {
final long sum = IntStream.of(counts[i]).sum();
columnSummaryOutputWriter.println(
String.join(
COLUMN_SEPARATOR,
columnNames.get(i),
String.valueOf(sum),
String.format(AVERAGE_DOUBLE_FORMAT, sum / (double) totalSize)));
}
}
/**
* Writes the row in the main matrix output file for a target and, if requested, the corresponding
* row in the row summary output file.
*
* @param countBuffer the counts for the target.
* @param index the index of target within the target collection.
*/
private void writeOutputRows(
final int[] countBuffer, final long[] columnTotals, final int index) {
final String countString =
IntStream.range(0, countBuffer.length)
.mapToObj(i -> transform.apply(countBuffer[i], columnTotals[i]))
.collect(Collectors.joining(COLUMN_SEPARATOR));
final String targetInfoString =
targetOutInfo.composeTargetOutInfoString(index, targetCollection);
outputWriter.println(String.join(COLUMN_SEPARATOR, targetInfoString, countString));
if (rowSummaryOutputWriter != null) {
final long sum = MathUtils.sum(countBuffer);
final SimpleInterval location = targetCollection.location(index);
final int targetSize = location.size();
rowSummaryOutputWriter.println(
String.join(
COLUMN_SEPARATOR,
targetInfoString,
Long.toString(sum),
String.format(
AVERAGE_DOUBLE_FORMAT, sum / ((float) countColumns.columnCount() * targetSize))));
}
}
