1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193
|
/*
* sam.h
*
* Created on: Sep 23, 2009
* Author: Ben Langmead
*/
#ifndef SAM_H_
#define SAM_H_
#include "refmap.h"
#include "annot.h"
#include "pat.h"
#include "random_source.h"
#include "btypes.h"
class ReferenceMap;
class AnnotationMap;
class PatternSourcePerThread;
enum {
SAM_FLAG_PAIRED = 1,
SAM_FLAG_MAPPED_PAIRED = 2,
SAM_FLAG_UNMAPPED = 4,
SAM_FLAG_MATE_UNMAPPED = 8,
SAM_FLAG_QUERY_STRAND = 16,
SAM_FLAG_MATE_STRAND = 32,
SAM_FLAG_FIRST_IN_PAIR = 64,
SAM_FLAG_SECOND_IN_PAIR = 128,
SAM_FLAG_NOT_PRIMARY = 256,
SAM_FLAG_FAILS_CHECKS = 512,
SAM_FLAG_DUPLICATE = 1024
};
/**
* Sink that prints lines in SAM format:
*/
class SAMHitSink : public HitSink {
public:
/**
* Construct a single-stream VerboseHitSink (default)
*/
SAMHitSink(OutFileBuf* out,
int offBase,
ReferenceMap *rmap,
AnnotationMap *amap,
bool fullRef,
bool noQnameTrunc,
int defaultMapq,
DECL_HIT_DUMPS2) :
HitSink(out, PASS_HIT_DUMPS2),
offBase_(offBase), defaultMapq_(defaultMapq),
rmap_(rmap), amap_(amap), fullRef_(fullRef),
noQnameTrunc_(noQnameTrunc) { }
/**
* Construct a multi-stream VerboseHitSink with one stream per
* reference string (see --refout)
*/
SAMHitSink(size_t numOuts,
int offBase,
ReferenceMap *rmap,
AnnotationMap *amap,
bool fullRef,
int defaultMapq,
DECL_HIT_DUMPS2) :
HitSink(numOuts, PASS_HIT_DUMPS2),
offBase_(offBase), defaultMapq_(defaultMapq),
rmap_(rmap), amap_(amap), fullRef_(fullRef) { }
/**
* Append a SAM alignment to the given output stream.
*/
static void append(ostream& ss,
const Hit& h,
int mapq,
int xms,
const vector<string>* refnames,
ReferenceMap *rmap,
AnnotationMap *amap,
bool fullRef,
bool noQnameTrunc,
int offBase);
/**
* Append a SAM alignment for an aligned read to the given output
* stream.
*/
static void appendAligned(ostream& ss,
const Hit& h,
int mapq,
int xms,
const vector<string>* refnames,
ReferenceMap *rmap,
AnnotationMap *amap,
bool fullRef,
bool noQnameTrunc,
int offBase);
/**
* Append a verbose, readable hit to the output stream
* corresponding to the hit.
*/
virtual void append(ostream& ss, const Hit& h) {
SAMHitSink::append(ss, h, defaultMapq_, 0, _refnames, rmap_, amap_, fullRef_, noQnameTrunc_, offBase_);
}
/**
* Append a verbose, readable hit to the output stream
* corresponding to the hit.
*/
virtual void append(ostream& ss, const Hit& h, int mapq, int xms) {
SAMHitSink::append(ss, h, mapq, xms, _refnames, rmap_, amap_, fullRef_, noQnameTrunc_, offBase_);
}
/**
* Write the SAM header lines.
*/
void appendHeaders(OutFileBuf& os,
size_t numRefs,
const vector<string>& refnames,
bool color,
bool nosq,
ReferenceMap *rmap,
const TIndexOffU* plen,
bool fullRef,
bool noQnameTrunc,
const char *cmdline,
const char *rgline);
protected:
/**
*
*/
void reportUnOrMax(
PatternSourcePerThread& p,
vector<Hit>* hs,
bool un);
/**
* Report a verbose, human-readable alignment to the appropriate
* output stream.
*/
virtual void reportHit(const Hit& h) {
reportSamHit(h, defaultMapq_, 0);
}
/**
* Report a SAM alignment with the given mapping quality and XM
* field.
*/
virtual void reportSamHit(
const Hit& h,
int mapq,
int xms);
/**
* Report a batch of SAM alignments (e.g. two mates that should be
* printed together) with the given mapping quality and XM field.
*/
virtual void reportSamHits(
vector<Hit>& hs,
size_t start,
size_t end,
int mapq,
int xms);
/**
* See sam.cpp
*/
virtual void reportMaxed(vector<Hit>& hs, PatternSourcePerThread& p);
/**
* See sam.cpp
*/
virtual void reportUnaligned(PatternSourcePerThread& p) {
reportUnOrMax(p, NULL, true);
}
private:
int offBase_; /// Add this to reference offsets before outputting.
/// (An easy way to make things 1-based instead of
/// 0-based)
int defaultMapq_; /// Default mapping quality to report when one is
/// not specified
ReferenceMap *rmap_; /// mapping to reference coordinate system.
AnnotationMap *amap_; ///
bool fullRef_; /// print full reference name, not just up to whitespace
bool noQnameTrunc_; /// true -> don't truncate QNAME at first whitespace
};
#endif /* SAM_H_ */
|