X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=src%2Ftoolkit%2Fbamtools_split.cpp;h=4575d8596cc9c1dc8b808e8690f3ca2bfbb2d08d;hb=a847098d20d1bb39a45123c6b886479c6a97001a;hp=62dc1046c9587d86e76e69c0133a5f5e0cffca0c;hpb=0c50d5abdc43df228147df3e3fb166b78b303fe9;p=bamtools.git diff --git a/src/toolkit/bamtools_split.cpp b/src/toolkit/bamtools_split.cpp index 62dc104..4575d85 100644 --- a/src/toolkit/bamtools_split.cpp +++ b/src/toolkit/bamtools_split.cpp @@ -1,60 +1,63 @@ // *************************************************************************** // bamtools_split.cpp (c) 2010 Derek Barnett, Erik Garrison // Marth Lab, Department of Biology, Boston College -// All rights reserved. // --------------------------------------------------------------------------- -// Last modified: 20 September 2010 (DB) +// Last modified: 7 April 2011 (DB) // --------------------------------------------------------------------------- -// +// Splits a BAM file on user-specified property, creating a new BAM output +// file for each value found // *************************************************************************** +#include "bamtools_split.h" + +#include +#include +#include +#include +#include +using namespace BamTools; + #include #include #include #include #include #include -#include "bamtools_split.h" -#include "bamtools_options.h" -#include "bamtools_variant.h" -#include "BamReader.h" -#include "BamWriter.h" using namespace std; -using namespace BamTools; namespace BamTools { - // string constants - static const string SPLIT_MAPPED_TOKEN = ".MAPPED"; - static const string SPLIT_UNMAPPED_TOKEN = ".UNMAPPED"; - static const string SPLIT_PAIRED_TOKEN = ".PAIRED_END"; - static const string SPLIT_SINGLE_TOKEN = ".SINGLE_END"; - static const string SPLIT_REFERENCE_TOKEN = ".REF_"; - - string GetTimestampString(void) { - - // get human readable timestamp - time_t currentTime; - time(¤tTime); - stringstream timeStream(""); - timeStream << ctime(¤tTime); - - // convert whitespace to '_' - string timeString = timeStream.str(); - size_t found = timeString.find(" "); - while (found != string::npos) { - timeString.replace(found, 1, "_"); - found = timeString.find(" ", found+1); - } - return timeString; - } - - // remove copy of filename without extension - // (so /path/to/file.txt becomes /path/to/file ) - string RemoveFilenameExtension(const string& filename) { - size_t found = filename.rfind("."); - return filename.substr(0, found); +// string constants +static const string SPLIT_MAPPED_TOKEN = ".MAPPED"; +static const string SPLIT_UNMAPPED_TOKEN = ".UNMAPPED"; +static const string SPLIT_PAIRED_TOKEN = ".PAIRED_END"; +static const string SPLIT_SINGLE_TOKEN = ".SINGLE_END"; +static const string SPLIT_REFERENCE_TOKEN = ".REF_"; + +string GetTimestampString(void) { + + // get human readable timestamp + time_t currentTime; + time(¤tTime); + stringstream timeStream(""); + timeStream << ctime(¤tTime); + + // convert whitespace to '_' + string timeString = timeStream.str(); + size_t found = timeString.find(" "); + while (found != string::npos) { + timeString.replace(found, 1, "_"); + found = timeString.find(" ", found+1); } + return timeString; +} + +// remove copy of filename without extension +// (so /path/to/file.txt becomes /path/to/file ) +string RemoveFilenameExtension(const string& filename) { + size_t found = filename.rfind("."); + return filename.substr(0, found); +} } // namespace BamTools @@ -97,8 +100,13 @@ class SplitTool::SplitToolPrivate { // ctor & dtor public: - SplitToolPrivate(SplitTool::SplitSettings* settings); - ~SplitToolPrivate(void); + SplitToolPrivate(SplitTool::SplitSettings* settings) + : m_settings(settings) + { } + + ~SplitToolPrivate(void) { + m_reader.Close(); + } // 'public' interface public: @@ -136,16 +144,6 @@ class SplitTool::SplitToolPrivate { RefVector m_references; }; -// constructor -SplitTool::SplitToolPrivate::SplitToolPrivate(SplitTool::SplitSettings* settings) - : m_settings(settings) -{ } - -// destructor -SplitTool::SplitToolPrivate::~SplitToolPrivate(void) { - m_reader.Close(); -} - void SplitTool::SplitToolPrivate::DetermineOutputFilenameStub(void) { // if user supplied output filename stub, use that @@ -162,10 +160,14 @@ void SplitTool::SplitToolPrivate::DetermineOutputFilenameStub(void) { } bool SplitTool::SplitToolPrivate::OpenReader(void) { + + // attempt to open BAM file if ( !m_reader.Open(m_settings->InputFilename) ) { - cerr << "ERROR: SplitTool could not open BAM file: " << m_settings->InputFilename << endl; + cerr << "bamtools split ERROR: could not open BAM file: " << m_settings->InputFilename << endl; return false; } + + // save file 'metadata' & return success m_header = m_reader.GetHeaderText(); m_references = m_reader.GetReferenceData(); return true; @@ -177,7 +179,8 @@ bool SplitTool::SplitToolPrivate::Run(void) { DetermineOutputFilenameStub(); // open up BamReader - if ( !OpenReader() ) return false; + if ( !OpenReader() ) + return false; // determine split type from settings if ( m_settings->IsSplittingMapped ) return SplitMapped(); @@ -186,7 +189,8 @@ bool SplitTool::SplitToolPrivate::Run(void) { if ( m_settings->IsSplittingTag ) return SplitTag(); // if we get here, no property was specified - cerr << "No property given to split on... Please use -mapped, -paired, -reference, or -tag TAG to specifiy split behavior." << endl; + cerr << "bamtools split ERROR: no property given to split on... " << endl + << "Please use -mapped, -paired, -reference, or -tag TAG to specifiy desired split behavior." << endl; return false; } @@ -210,9 +214,15 @@ bool SplitTool::SplitToolPrivate::SplitMapped(void) { if ( writerIter == outputFiles.end() ) { // open new BamWriter + const string outputFilename = m_outputFilenameStub + ( isCurrentAlignmentMapped + ? SPLIT_MAPPED_TOKEN + : SPLIT_UNMAPPED_TOKEN ) + ".bam"; writer = new BamWriter; - const string outputFilename = m_outputFilenameStub + ( isCurrentAlignmentMapped ? SPLIT_MAPPED_TOKEN : SPLIT_UNMAPPED_TOKEN ) + ".bam"; - writer->Open(outputFilename, m_header, m_references); + if ( !writer->Open(outputFilename, m_header, m_references) ) { + cerr << "bamtools split ERROR: could not open " << outputFilename + << " for writing." << endl; + return false; + } // store in map outputFiles.insert( make_pair(isCurrentAlignmentMapped, writer) ); @@ -222,7 +232,7 @@ bool SplitTool::SplitToolPrivate::SplitMapped(void) { else writer = (*writerIter).second; // store alignment in proper BAM output file - if ( writer ) + if ( writer ) writer->SaveAlignment(al); } @@ -253,9 +263,15 @@ bool SplitTool::SplitToolPrivate::SplitPaired(void) { if ( writerIter == outputFiles.end() ) { // open new BamWriter + const string outputFilename = m_outputFilenameStub + ( isCurrentAlignmentPaired + ? SPLIT_PAIRED_TOKEN + : SPLIT_SINGLE_TOKEN ) + ".bam"; writer = new BamWriter; - const string outputFilename = m_outputFilenameStub + ( isCurrentAlignmentPaired ? SPLIT_PAIRED_TOKEN : SPLIT_SINGLE_TOKEN ) + ".bam"; - writer->Open(outputFilename, m_header, m_references); + if ( !writer->Open(outputFilename, m_header, m_references) ) { + cerr << "bamtool split ERROR: could not open " << outputFilename + << " for writing." << endl; + return false; + } // store in map outputFiles.insert( make_pair(isCurrentAlignmentPaired, writer) ); @@ -296,11 +312,15 @@ bool SplitTool::SplitToolPrivate::SplitReference(void) { if ( writerIter == outputFiles.end() ) { // open new BamWriter - writer = new BamWriter; const string refName = m_references.at(currentRefId).RefName; const string outputFilename = m_outputFilenameStub + SPLIT_REFERENCE_TOKEN + refName + ".bam"; - writer->Open(outputFilename, m_header, m_references); - + writer = new BamWriter; + if ( !writer->Open(outputFilename, m_header, m_references) ) { + cerr << "bamtools split ERROR: could not open " << outputFilename + << " for writing." << endl; + return false; + } + // store in map outputFiles.insert( make_pair(currentRefId, writer) ); } @@ -329,32 +349,33 @@ bool SplitTool::SplitToolPrivate::SplitTag(void) { // look for tag in this alignment and get tag type char tagType(0); - if ( !al.GetTagType(m_settings->TagToSplit, tagType) ) continue; + if ( !al.GetTagType(m_settings->TagToSplit, tagType) ) + continue; // request split method based on tag type // pass it the current alignment found - switch (tagType) { + switch ( tagType ) { - case 'c' : - case 's' : - case 'i' : + case (Constants::BAM_TAG_TYPE_INT8) : + case (Constants::BAM_TAG_TYPE_INT16) : + case (Constants::BAM_TAG_TYPE_INT32) : return SplitTagImpl(al); - case 'C' : - case 'S' : - case 'I' : + case (Constants::BAM_TAG_TYPE_UINT8) : + case (Constants::BAM_TAG_TYPE_UINT16) : + case (Constants::BAM_TAG_TYPE_UINT32) : return SplitTagImpl(al); - case 'f' : + case (Constants::BAM_TAG_TYPE_FLOAT) : return SplitTagImpl(al); - case 'A': - case 'Z': - case 'H': + case (Constants::BAM_TAG_TYPE_ASCII) : + case (Constants::BAM_TAG_TYPE_STRING) : + case (Constants::BAM_TAG_TYPE_HEX) : return SplitTagImpl(al); default: - fprintf(stderr, "ERROR: Unknown tag storage class encountered: [%c]\n", tagType); + fprintf(stderr, "bamtools split ERROR: unknown tag type encountered: [%c]\n", tagType); return false; } } @@ -365,8 +386,9 @@ bool SplitTool::SplitToolPrivate::SplitTag(void) { // -------------------------------------------------------------------------------- // template method implementation -// N.B. - *technical note* - use of template methods defined in ".cpp" goes against normal practices -// but works here because these are purely internal (no one can call from outside this file) +// *Technical Note* - use of template methods declared & defined in ".cpp" file +// goes against normal practices, but works here because these +// are purely internal (no one can call from outside this file) // close BamWriters & delete pointers template @@ -375,15 +397,22 @@ void SplitTool::SplitToolPrivate::CloseWriters(map& writers) { typedef map WriterMap; typedef typename WriterMap::iterator WriterMapIterator; + // iterate over writers WriterMapIterator writerIter = writers.begin(); WriterMapIterator writerEnd = writers.end(); for ( ; writerIter != writerEnd; ++writerIter ) { BamWriter* writer = (*writerIter).second; - if (writer == 0 ) continue; + if ( writer == 0 ) continue; + + // close BamWriter writer->Close(); + + // destroy BamWriter delete writer; writer = 0; } + + // clear the container (destroying the items doesn't remove them) writers.clear(); } @@ -409,9 +438,13 @@ bool SplitTool::SplitToolPrivate::SplitTagImpl(BamAlignment& al) { if ( al.GetTag(tag, currentValue) ) { // open new BamWriter, save first alignment - writer = new BamWriter; outputFilenameStream << m_outputFilenameStub << ".TAG_" << tag << "_" << currentValue << ".bam"; - writer->Open(outputFilenameStream.str(), m_header, m_references); + writer = new BamWriter; + if ( !writer->Open(outputFilenameStream.str(), m_header, m_references) ) { + cerr << "bamtools split ERROR: could not open " << outputFilenameStream.str() + << " for writing." << endl; + return false; + } writer->SaveAlignment(al); // store in map @@ -434,10 +467,14 @@ bool SplitTool::SplitToolPrivate::SplitTagImpl(BamAlignment& al) { if ( writerIter == outputFiles.end() ) { // open new BamWriter - writer = new BamWriter; outputFilenameStream << m_outputFilenameStub << ".TAG_" << tag << "_" << currentValue << ".bam"; - writer->Open(outputFilenameStream.str(), m_header, m_references); - + writer = new BamWriter; + if ( !writer->Open(outputFilenameStream.str(), m_header, m_references) ) { + cerr << "bamtool split ERROR: could not open " << outputFilenameStream.str() + << " for writing." << endl; + return false; + } + // store in map outputFiles.insert( make_pair(currentValue, writer) ); @@ -503,10 +540,10 @@ int SplitTool::Run(int argc, char* argv[]) { // parse command line arguments Options::Parse(argc, argv, 1); - // initialize internal implementation + // initialize SplitTool with settings m_impl = new SplitToolPrivate(m_settings); - // run tool, return success/fail + // run SplitTool, return success/fail if ( m_impl->Run() ) return 0; else