// open BAM input file
void BAMWalkerEngine::InitializeBAMs(const ReferenceReader& ref_reader, const vector<string>& bam_filenames)
{
if (not bam_reader_.SetExplicitMergeOrder(BamMultiReader::MergeByCoordinate)) {
cerr << "ERROR: Could not set merge order to BamMultiReader::MergeByCoordinate" << endl;
exit(1);
}
if (not bam_reader_.Open(bam_filenames)) {
cerr << "ERROR: Could not open input BAM file(s) : " << bam_reader_.GetErrorString() << endl;
exit(1);
}
if (not bam_reader_.LocateIndexes()) {
cerr << "ERROR: Could not open BAM index file(s) : " << bam_reader_.GetErrorString() << endl;
exit(1);
}
// BAM multi reader combines the read group information of the different BAMs but does not merge comment sections
bam_header_ = bam_reader_.GetHeader();
if (!bam_header_.HasReadGroups()) {
cerr << "ERROR: there is no read group in BAM files specified" << endl;
exit(1);
}
// Manually merge comment sections of BAM files if we have more than one BAM file
if (bam_filenames.size() > 1) {
unsigned int num_duplicates = 0;
unsigned int num_merged = 0;
for (unsigned int bam_idx = 0; bam_idx < bam_filenames.size(); bam_idx++) {
BamReader reader;
if (not reader.Open(bam_filenames.at(bam_idx))) {
cerr << "TVC ERROR: Failed to open input BAM file " << reader.GetErrorString() << endl;
exit(1);
}
SamHeader header = reader.GetHeader();
for (unsigned int i_co = 0; i_co < header.Comments.size(); i_co++) {
// Step 1: Check if this comment is already part of the merged header
unsigned int m_co = 0;
while (m_co < bam_header_.Comments.size() and bam_header_.Comments.at(m_co) != header.Comments.at(i_co))
m_co++;
if (m_co < bam_header_.Comments.size()){
num_duplicates++;
continue;
}
// Add comment line to merged header if it is a new one
num_merged++;
bam_header_.Comments.push_back(header.Comments.at(i_co));
}
}
// Verbose what we did
cout << "Merged " << num_merged << " unique comment lines into combined BAM header. Encountered " << num_duplicates << " duplicate comments." << endl;
}
//
// Reference sequences in the bam file must match that in the fasta file
//
vector<RefData> referenceSequences = bam_reader_.GetReferenceData();
if ((int)referenceSequences.size() != ref_reader.chr_count()) {
cerr << "ERROR: Reference in BAM file does not match fasta file" << endl
<< " BAM has " << referenceSequences.size()
<< " chromosomes while fasta has " << ref_reader.chr_count() << endl;
exit(1);
}
for (int chr_idx = 0; chr_idx < ref_reader.chr_count(); ++chr_idx) {
if (referenceSequences[chr_idx].RefName != ref_reader.chr_str(chr_idx)) {
cerr << "ERROR: Reference in BAM file does not match fasta file" << endl
<< " Chromosome #" << (chr_idx+1) << "in BAM is " << referenceSequences[chr_idx].RefName
<< " while fasta has " << ref_reader.chr_str(chr_idx) << endl;
exit(1);
}
if (referenceSequences[chr_idx].RefLength != ref_reader.chr_size(chr_idx)) {
cerr << "ERROR: Reference in BAM file does not match fasta file" << endl
<< " Chromosome " << referenceSequences[chr_idx].RefName
<< "in BAM has length " << referenceSequences[chr_idx].RefLength
<< " while fasta has " << ref_reader.chr_size(chr_idx) << endl;
exit(1);
}
}
//
// Retrieve BaseCaller and TMAP version strings from BAM header
//
set<string> basecaller_versions;
set<string> tmap_versions;
for (SamProgramIterator I = bam_header_.Programs.Begin(); I != bam_header_.Programs.End(); ++I) {
if (I->ID.substr(0,2) == "bc")
basecaller_versions.insert(I->Version);
if (I->ID.substr(0,4) == "tmap")
tmap_versions.insert(I->Version);
}
basecaller_version_ = "";
for (set<string>::const_iterator I = basecaller_versions.begin(); I != basecaller_versions.end(); ++I) {
if (not basecaller_version_.empty())
basecaller_version_ += ", ";
basecaller_version_ += *I;
}
tmap_version_ = "";
for (set<string>::const_iterator I = tmap_versions.begin(); I != tmap_versions.end(); ++I) {
if (not tmap_version_.empty())
tmap_version_ += ", ";
tmap_version_ += *I;
}
}
// open BAM input file
void BAMWalkerEngine::InitializeBAMs(const ReferenceReader& ref_reader, const vector<string>& bam_filenames)
{
if (not bam_reader_.SetExplicitMergeOrder(BamMultiReader::MergeByCoordinate)) {
cerr << "ERROR: Could not set merge order to BamMultiReader::MergeByCoordinate" << endl;
exit(1);
}
if (not bam_reader_.Open(bam_filenames)) {
cerr << "ERROR: Could not open input BAM file(s) : " << bam_reader_.GetErrorString();
exit(1);
}
if (not bam_reader_.LocateIndexes()) {
cerr << "ERROR: Could not open BAM index file(s) : " << bam_reader_.GetErrorString();
exit(1);
}
bam_header_ = bam_reader_.GetHeader();
if (!bam_header_.HasReadGroups()) {
cerr << "ERROR: there is no read group in BAM files specified" << endl;
exit(1);
}
//
// Reference sequences in the bam file must match that in the fasta file
//
vector<RefData> referenceSequences = bam_reader_.GetReferenceData();
if ((int)referenceSequences.size() != ref_reader.chr_count()) {
cerr << "ERROR: Reference in BAM file does not match fasta file" << endl
<< " BAM has " << referenceSequences.size()
<< " chromosomes while fasta has " << ref_reader.chr_count() << endl;
exit(1);
}
for (int chr_idx = 0; chr_idx < ref_reader.chr_count(); ++chr_idx) {
if (referenceSequences[chr_idx].RefName != ref_reader.chr_str(chr_idx)) {
cerr << "ERROR: Reference in BAM file does not match fasta file" << endl
<< " Chromosome #" << (chr_idx+1) << "in BAM is " << referenceSequences[chr_idx].RefName
<< " while fasta has " << ref_reader.chr_str(chr_idx) << endl;
exit(1);
}
if (referenceSequences[chr_idx].RefLength != ref_reader.chr_size(chr_idx)) {
cerr << "ERROR: Reference in BAM file does not match fasta file" << endl
<< " Chromosome " << referenceSequences[chr_idx].RefName
<< "in BAM has length " << referenceSequences[chr_idx].RefLength
<< " while fasta has " << ref_reader.chr_size(chr_idx) << endl;
exit(1);
}
}
//
// Retrieve BaseCaller and TMAP version strings from BAM header
//
set<string> basecaller_versions;
set<string> tmap_versions;
for (SamProgramIterator I = bam_header_.Programs.Begin(); I != bam_header_.Programs.End(); ++I) {
if (I->ID.substr(0,2) == "bc")
basecaller_versions.insert(I->Version);
if (I->ID.substr(0,4) == "tmap")
tmap_versions.insert(I->Version);
}
basecaller_version_ = "";
for (set<string>::const_iterator I = basecaller_versions.begin(); I != basecaller_versions.end(); ++I) {
if (not basecaller_version_.empty())
basecaller_version_ += ", ";
basecaller_version_ += *I;
}
tmap_version_ = "";
for (set<string>::const_iterator I = tmap_versions.begin(); I != tmap_versions.end(); ++I) {
if (not tmap_version_.empty())
tmap_version_ += ", ";
tmap_version_ += *I;
}
}
