File: bundle.cpp

package info (click to toggle)
stringtie 3.0.3%2Bds-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 14,888 kB
  • sloc: cpp: 32,956; sh: 448; python: 429; makefile: 146
file content (242 lines) | stat: -rw-r--r-- 10,764 bytes parent folder | download
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
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
#include "bundle.h"

const byte txNASCENT_MASK=0x0C; // 2 bit flag for synthetic nascent transcripts
//0 = not nascent, 1 = nascent, 2 = nascent replacing a guide

const byte txSTATUS_MASK=0x03; // 2 bits mask for bundle status, former RC_TData::in_bundle
           // 1 if used in a bundle (guide added to keepguides, default value)
	         // 2 if all introns are covered by at least one read
					 // 3 if it is stored to be printed
GuideBundleStatus getGuideStatus(GffObj* guide) {
  return (GuideBundleStatus)guide->getUserFlags(txSTATUS_MASK);
}
void setGuideStatus(GffObj* guide, GuideBundleStatus status) {
  if (status>3) {
    GError("Error: invalid status value %d set for transcript %s\n", status, guide->getID());
    exit(1);
  }
  guide->clearUserFlags(txSTATUS_MASK);
  guide->setUserFlags(txSTATUS_MASK & status); // can be 1, 2 or 0 (clear)
}

// set/get synthetic nascent flag for a transcript
void setNascent(GffObj* guide, byte nasc_st) { // 0, 1 or 2
      if (nasc_st>2) {
        GError("Error: invalid nascent status value %d set for transcript %s\n", nasc_st, guide->getID());
        exit(1);
      }
      if (nasc_st) guide->setUserFlags((nasc_st<<2) & txNASCENT_MASK);
      else guide->clearUserFlags(txNASCENT_MASK);
}

byte isNascent(GffObj* guide) {
      byte v=guide->getUserFlags(txNASCENT_MASK);
      return (v>>2);
}

/* inline GffObj* nascentFrom(GffObj* ntx) {
    if (ntx && ntx->uptr) {
        return ((RC_TData*)(ntx->uptr))->gen_from;
    }
    return NULL;
}*/

// genNascent is included from rlink.h
void BundleData::keepGuide(GffObj* guide, Ref_RC_Data& ref_rc) {
	if (rc_data==NULL) {
	  rc_init(guide, ref_rc.rc_tdata, ref_rc.rc_edata,
               ref_rc.rc_idata);
	}
	keepguides.Add(guide);
  if (isNascent(guide)) {
    // add it to corresponding refdata.synrnas
    RC_TData* tdata=new RC_TData(*guide, ref_rc.rc_tdata->Count());
    // for nascents, uptr was used to store the gen_from guide
    tdata->gen_from=(GffObj*)(guide->uptr);
	  guide->uptr=tdata;
	  ref_rc.rc_tdata->Add(tdata);
    ref_rc.refdata->synrnas.Add(guide);
  }
	guide->udata=(int)rc_data->addTranscript(*guide);
  //this also adds exon/intron info
}

//generate all nascent transcripts for a guide
//keep nascents in 2 lists: tnlist for multi-exon, setnlist for single-exon
void genTxNascents(GffObj &guide, GList<GffObj>& tnlist) { //, GPVec<GffObj>& guides) {
    // DEBUG only: keeping track of the number of nascent transcripts generated vs redundant
    //GMessage("Generating %d nascents for %s(%c)\n", guide.exons.Count()-1, guide.getID(), guide.strand);
    int nxr=0; //number of redundant nascent transcripts discarded
    int n = guide.exons.Count() - 1; // number of introns
    for (int i = 1; i <= n; ++i) {
        GffObj* nt = new GffObj(true, nullptr, guide.gseq_id, guide.strand); // Create a new transcript
        nt->track_id = guide.track_id;
        nt->setGeneID(guide.getGeneID());
        nt->setGeneName(guide.getGeneName());
        if (guide.strand == '-') { // on the reverse strand, we need to add the exons in reverse order
          for (int j=n; j>n-i;--j) {
              nt->addExon(guide.exons[j]->start, guide.exons[j]->end);
          }
          // Extend the first exon of nt through the previous intron up to the start of the previous exon in guide
          nt->exons[0]->start = guide.exons[n-i]->end + 1;
          nt->start = nt->exons[0]->start;
        } else { //forward strand, add exons in the same order as in guide
          for (int j = 0; j < i; ++j) {
              // add i exons from guide to nt
              nt->addExon(guide.exons[j]->start, guide.exons[j]->end);
          }
          // Extend the last exon of nt through the next intron up to the start of the next exon in guide
          nt->exons.Last()->end = guide.exons[i]->start-1;
          nt->end = nt->exons.Last()->end;
        }

        bool keepNascent = true;
         //   set keepNascent to false if a matching previous nascent is found
        int ridx=-1;
        if (tnlist.Found(nt, ridx)) { //exact match found with a previous nascent
          keepNascent=false;
          nxr++;
        } else { // no exact match in tnlist
          // might still be a structural match with a previous nascent
          // backward search --------------
          // set a limit to the backward search, to avoid checking too distant nascents
          int64_t nt_bck_limit=((int64_t)nt->start - (int64_t)(nt->end - nt->start)/2);

          for (int i = ridx - 1; i >= 0; --i) {
              // break condition for the backward search is not as straightforward as the forward search
              // because the end coordinates do not influence the sort order.
              if ((int64_t)tnlist[i]->start < nt_bck_limit) {
                  // if the current transcript's start is before the middle of nt, no previous transcripts
                  // can structurally match nt, because they will start even earlier.
                  break;
              }
              if (txStructureMatch(*nt, *tnlist[i], 0.99, 70)) {
                  keepNascent = false;
                  nxr++;
                  break;
              }
              // Note: An optimization here might consider the spatial distance between transcripts, but
              // without a clear rule on how much overlap is needed for a match, it's safer not to break early.
          } // backward search

          if (keepNascent)
            //forward search
            for (int i = ridx; i < tnlist.Count(); ++i) {
                 if (tnlist[i]->start > nt->end) {
                        // If the current transcript's start is beyond the end of guide, no subsequent transcripts
                        // can structurally match guide, because they will start even later.
                        break;
                 }
                 if (txStructureMatch(*nt, *tnlist[i], 0.99, 70)) {
                      keepNascent = false;
                      nxr++;
                      break;
                 }
            }
        }

        if (keepNascent) {
          // insert it in the right location in the list
          tnlist.sortInsert(ridx, nt);
          GStr nid(guide.getID());
          nt->uptr = &guide; // store the parent guide temporarily - replace with RC_TData later
          nid.appendfmt(".nasc%03d", i);
          nt->setGeneID(guide.getGeneID());
          nt->setGeneName(guide.getGeneName());
          nt->setID(nid.chars());
          setNascent(nt, 1); // Set the nascent flag
        } else
           delete nt; //discard this nascent transcript (matching guide/other nascent found)
    } //for i
}

void BundleData::generateAllNascents(int from_guide_idx, Ref_RC_Data& ref_rc) {
  //GPVec<GffObj> oguides(keepguides); // copy list of pointers to guides
  // tnlist is a sorted list of nascent transcripts to be added to keepguides
  GList<GffObj> tnlist(true, true); //list to collect all nascents in this bundle
  tnlist.setSorted(txCmpByExons); //sort and match by exon coordinates
  for (int i=from_guide_idx;i<keepguides.Count();++i) {
    GffObj* guide=keepguides[i];
    if (guide->exons.Count()<2) continue;
    genTxNascents(*guide, tnlist); //, keepguides);
  }
  //add all nascents from tnlist to keepguides
  numNascents+=tnlist.Count();
  for (int i=0;i<tnlist.Count();++i) {
    keepGuide(tnlist[i], ref_rc);
    tnlist.Put(i, NULL); //don't delete nascent, it's now owned by keepguides
  }
  keepguides.Sort();
}

bool BundleData::evalReadAln(GReadAlnData& alndata, char& xstrand) {
	            //GSamRecord& brec, char& strand, int nh) {
 if (rc_data==NULL) {
	  return false; //no ref transcripts available for this reads' region
 }
 GSamRecord& brec=*(alndata.brec);
 int mate_pos=brec.mate_start();
 int nh=alndata.nh;
 if ((int)brec.end<rc_data->lmin || (int)brec.start>rc_data->rmax) {
	 return false; //hit outside coverage area
 }
 if (rc_data->g_exons.Count()==0 || rc_data->g_tdata.Count()==0)
	 return false; //nothing to do without transcripts
 //check this read alignment against guides exons and introns
 char strandbits=0;
 bool result=false;
 bool full_overlap=true; // exons and junctions are in reference transcripts but they might be in different guides
 for (int i=0;i<brec.exons.Count();i++) { //for each read exon
	 if (ballgown)
		 rc_data->updateCov(xstrand, nh, brec.exons[i].start, brec.exons[i].len());
	 GArray<RC_ExonOvl> exonOverlaps(true, true);
   //overlaps sorted by priority (guide/nascent) and length
   //   guide overlaps, if any, are always first
   // findOvlExons requires an overlap of at least 5 bp
	 if (rc_data->findOvlExons(exonOverlaps, brec.exons[i].start,
			 brec.exons[i].end, xstrand, mate_pos)) {
		 result=true;
		 int best_ovl=exonOverlaps[0].ovlen; //largest guide exon overlap (nascents have lower priority)
     int mate_ovl=exonOverlaps[0].mate_ovl;
     byte guide_ovl=exonOverlaps[0].in_guides; // is the ovl ref exon in a real guide, or just nascents?
		 if(full_overlap && (uint)best_ovl<brec.exons[i].len())
        full_overlap=false; // this exon is not fully overlapped by a guide exon

		 for (int k=0;k<exonOverlaps.Count();++k) { //check all overlaps of similar length
          if (k && (exonOverlaps[k].mate_ovl < mate_ovl
              || exonOverlaps[k].ovlen+5<best_ovl || exonOverlaps[k].in_guides < guide_ovl) )
            break; // NOTE ignore further overlaps after a mate matched
                    // or if they are shorter than max_overlap-5
          if (exonOverlaps[k].feature->strand=='+') strandbits |= 0x01;
          else if (exonOverlaps[k].feature->strand=='-') strandbits |= 0x02;
          //TODO: perhaps we could use a better approach for non-overlapping ref exons
          //      spanned by this same read alignment
          //counting this overlap for multiple exons if it's similarly large
          //(in the shared region of overlapping exons)
          if (exonOverlaps[k].in_guides)
            rc_updateExonCounts(exonOverlaps[k], nh);
     } //for each overlap
	 } //has guide exon overlaps
	 if (i>0) { //adjacent intron check for match to guide introns
		 int j_l=brec.exons[i-1].end+1;
		 int j_r=brec.exons[i].start-1;
		 alndata.juncs.Add(new CJunction(j_l, j_r));
		 RC_Feature* ri=rc_data->findIntron(j_l, j_r, xstrand);
		 if (ri) { //update guide intron counts
			 ri->rcount++;
			 ri->mrcount += (nh > 1) ? (1.0/nh) : 1;
			 if (nh==1)  ri->ucount++;
			 alndata.juncs.Last()->guide_match=1;
		 }
		 else full_overlap=false;

	 } // neighboring intron processing
 } //for each read exon
 if (xstrand=='.' && strandbits && strandbits<3) {
	xstrand = (strandbits==1) ? '+' : '-';
 }

 if(!mergeMode && full_overlap) alndata.in_guide=true;

 return result;
}