-
Notifications
You must be signed in to change notification settings - Fork 13
/
GFastaFile.h
763 lines (730 loc) · 23.7 KB
/
GFastaFile.h
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
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
#ifndef GFASTAFILE_H
#define GFASTAFILE_H
#include "GBase.h"
#define CAPINC 64
#define SEQCAPINC 256
#define DEF_FASTA_DELIM (char*)">"
class GFastaCharHandler;
class GFastaFile;
class FastaSeq { /* fasta record storage */
friend GFastaCharHandler;
friend GFastaFile;
protected:
int id_cap; /* allocated size of the sequence name string*/
int namelen; // real length of seq name
int d_cap; /* allocated size of the description */
int descrlen; /* real length of the description */
//-------actual sequence :
int s_cap; /* allocated length of the sequence string */
public:
int len; /* the actual string length of seq */
char *id; /* id only, up to first space */
char *descr; /* any comment on the defline, after the first space */
char* seq; /* the sequence buffer itself */
protected:
//----
void detach() {
//when pointers are taken over by another object
//clear and forget
id=NULL;
descr=NULL;
seq=NULL;
init();
}
void init(const char* cname, const char* cdescr=NULL,
const char* cseq=NULL, int sbeg=-1, int send=-1) {
//Warning: sbeg and send are 0-based!
int l=0;
if (cname==NULL) {
GMALLOC(id, CAPINC);
id_cap=CAPINC;
namelen=0;
id[0]='\0';
GMALLOC(descr, CAPINC);
}
else {
l=strlen(cname);
GMALLOC(id, l+1);strcpy(id,cname);
id_cap=l+1;
namelen=l;
}
if (cdescr==NULL) {
GMALLOC(descr, CAPINC);
descr[0]='\0';
d_cap=CAPINC;
descrlen=0;
}
else {//copy given description
l=strlen(cdescr);
GMALLOC(descr, l+1);
strcpy(descr,cdescr);
d_cap=l+1;
descrlen=l;
}
if (cseq==NULL) {
GMALLOC(seq, SEQCAPINC);
seq[0]='\0';
len=0;
s_cap=SEQCAPINC;
}
else { //sequence given
if (sbeg>=0) { //sequence range given
if (send<0) send=strlen(cseq)-1;
len=send-sbeg+1;
if (len>0) {
s_cap=len+1;
GMALLOC(seq, s_cap);
strncpy(seq, cseq+sbeg, len);
seq[len]=0;
}
else { //null range
GMALLOC(seq, SEQCAPINC);
seq[0]='\0';
len=0;
s_cap=SEQCAPINC;
}
}
else {// copy whole cseq
l=strlen(cseq);
GMALLOC(seq, l+1);
strcpy(seq,cseq);
len=l;
s_cap=l+1;
}
}
} //init(alldata, range)
void init(int seqalloc=0) {
//ntCompTableInit();
GMALLOC(id, CAPINC);
id_cap=CAPINC;
namelen=0;
id[0]='\0';
GMALLOC(descr, CAPINC);
descr[0]='\0';
d_cap=CAPINC;
descrlen=0;
if (seqalloc<=0) {
s_cap=SEQCAPINC;
GMALLOC(seq, SEQCAPINC);
}
else {
s_cap=seqalloc;
GMALLOC(seq, seqalloc);
}
seq[0]='\0';
len=0;
}
public:
FastaSeq(const char* cname, const char* cdescr=NULL,
const char* bseq=NULL,int bseq_len=0) {
if (bseq_len>0) init(cname, cdescr, bseq, 0, bseq_len-1);
else init(cname, cdescr, bseq);
}
FastaSeq(int seqalloc=0) {
init(seqalloc);
}
//copy constructor:
FastaSeq(const FastaSeq& fa,int sbeg=-1,int send=-1) {
if (sbeg<0) { sbeg=0; send=fa.len-1; }
else if (send<0) send=fa.len-1;
if (send>fa.len-1) send=fa.len-1;
init(fa.id, fa.descr, fa.seq, sbeg, send);
}
FastaSeq(FastaSeq& fa, bool takeover) {
if (takeover) {
id_cap=fa.id_cap;
id=fa.id;
namelen=fa.namelen;
descr=fa.descr;
d_cap=fa.d_cap;
descrlen=fa.descrlen;
s_cap=fa.s_cap;
len=fa.len;
seq=fa.seq;
fa.detach();
}else {
init(fa.id, fa.descr, fa.seq);
}
}
void clear() {
GFREE(id);id_cap=0;namelen=0;id=NULL;
GFREE(descr);d_cap=0;descrlen=0;descr=NULL;
GFREE(seq);s_cap=0;len=0;seq=NULL;
}
~FastaSeq() {
clear();
}
int getNameLen() { return namelen; }
const char* getName() { return (const char*) id; }
const char* name() { return (const char*) id; }
const char* getSeqName() { return (const char*) id; }
const char* getId() { return (const char*) id; }
const char* getDescr() { return (const char*) descr; }
int getDescrLen() { return descrlen; }
const char* getSeq() { return (const char*) seq; }
int getSeqLen() { return len; }
void extendId(char c) {
if (namelen+1 >= id_cap) {
id_cap += CAPINC;
GREALLOC(id, id_cap);
}
id[namelen]= c;
namelen++;
}
void extendSeqName(char c) { extendId(c); }
void extendName(char c) { extendId(c); }
void extendDescr(char c) {
if (descrlen+1 >= d_cap) {
d_cap += CAPINC;
GREALLOC(descr, d_cap);
}
descr[descrlen]= c;
descrlen++;
}
void endId() { id[namelen]=0; }
void endName() { id[namelen]=0; }
void endSeqName() { id[namelen]=0; }
void endDescr() { descr[descrlen]=0; }
void endSeq() { seq[len]=0; }
void extendSeq(char c) {
if (len+1 >= s_cap) {
s_cap += SEQCAPINC;
GREALLOC(seq, s_cap);
}
seq[len]= c;
len++;
}
void compactIdMem() { if (namelen>0) {
GREALLOC(id, namelen+1); id_cap=namelen+1;
} }
void compactDescrMem() { if (descrlen>0) {
GREALLOC(descr, descrlen+1); d_cap=descrlen+1; } }
void compactSeqMem() { if (len>0) {
GREALLOC(seq, len+1); s_cap=len+1; } }
void compactMem() {
compactIdMem();
compactDescrMem();
compactSeqMem();
}
char* detachSeqPtr() { //such that the sequence allocated memory is no longer
// freed when the FastaSeq object is destroyed
// the returned pointer MUST be deallocated by the the user, later!
char* p=seq;
GMALLOC(seq, SEQCAPINC);
s_cap=SEQCAPINC;
len=0;
return p;
}
char* setSeqPtr(char* newseq, int newlen=0, int newcap=0) {
if (newlen==0) newlen=strlen(newseq);
if (newcap<=newlen) newcap=newlen+1;
GFREE(seq);
seq=newseq;
len=newlen;
s_cap=newcap;
return seq;
}
void reset() {// allocated space remains the same!
namelen=0;id[0]=0;
descrlen=0;descr[0]=0;
len=0;seq[0]=0;
}
/*
//reverse-complement a nucleotide sequence:
// -- requires gdna.h
void reverseComplement() {
if (len==0) return;
//ntCompTableInit();
reverseChars(seq,len);
for (int i=0;i<len;i++) seq[i]=ntComplement(seq[i]);
}
*/
//printing fasta formatted sequence to a file stream
void fprint(FILE* fout, int line_len=60, bool defline=false) {
if (defline) {
if (descrlen>0) fprintf(fout, "%s %s\n", id, descr);
else fprintf(fout, ">%s\n", id);
}
int l=len;
char* p=seq;
while (l>0) {
int to_write=GMIN(line_len, l);
fwrite(p,1,to_write,fout);
fprintf(fout,"\n");
p+=line_len;
l-=line_len;
}
}
//
static void write(FILE *fh, const char* seqid, const char* descr, char* seq,
const int linelen=70, const int seqlen=0) {
writeFasta(fh, seqid, descr, seq, linelen, seqlen); //from GBase.cpp
}
};
typedef int charFunc(char c, int pos, FastaSeq* fseq); //char processing function
/* passes:
c = current sequence character (generally aminoacid or nucleotide)
pos = 0-based coordinate of the given character within the sequence
fseq = FastaSeq pointer (useful for retrieving sequence defline info)
the return value is not used yet
*/
//(for reading/writing variable length records, etc.)
enum fileMode {
fmRead,
fmWrite
};
class GFastaFile {
char* fname;
FILE* fh;
fileMode fmode;
long int rec_fpos; //the input stream offset of the current record to be read
long int cur_fpos; //the input stream offset of the current byte to be read
uint seqcoord; //1-based coordinate of the current record's sequence reading position
//(updated by getSeqRange() mostly)
protected:
void bad_fastafmt() {
GError("Error parsing file '%s'. Not a Fasta file?\n", fname);
}
void check_eof(int c) {
if (c == EOF) bad_fastafmt();
}
public:
GFastaFile(const char* filename, fileMode filemode=fmRead) {
fh=NULL;
cur_fpos=0;
rec_fpos=0;
fmode=filemode;
seqcoord=0;
const char *mode=(filemode==fmRead) ? "rb" : "wb";
if (filename == NULL || filename[0]=='\0') {
fh = (filemode == fmRead) ? stdin : stdout;
fname=NULL;
}
else {
if ((fh = fopen(filename, mode)) == NULL)
GError("Cannot open file '%s'!", filename);
fname=Gstrdup(filename);
}
/*
GCALLOC(curseqid, CAPINC);
curseqidlen=CAPINC;
GCALLOC(curdescr, CAPINC);
curdescrlen=CAPINC;*/
}
//attach a GFastaFile object to an already open handle
GFastaFile(FILE* fhandle, fileMode filemode=fmRead, const char* filename=NULL) {
fh=fhandle;
cur_fpos=ftell(fh);
fmode=filemode;
rec_fpos=cur_fpos;
seqcoord=0;
if (filename == NULL || filename[0]=='\0') {
fname=NULL;
}
else
fname=Gstrdup(filename);
}
void reset() {
if (fh!=NULL && fh!=stdout && fh!=stdin) {
fseeko(fh,0L, SEEK_SET);
cur_fpos=0;
rec_fpos=0;
}
else GError("Cannot use GFastaFile::reset() on stdin, stdout or NULL handles.\n");
}
void seek(int pos) {
if (fh!=NULL && fh!=stdout && fh!=stdin) {
fseeko(fh, pos, SEEK_SET);
cur_fpos=pos;
seqcoord=0; //seqcoord agnostic after a seek
}
else GError("Cannot use GFastaFile::seek() on stdin, stdout or NULL handles.\n");
}
~GFastaFile() {
if (fh!=NULL && fh!=stdout && fh!=stdin) fclose(fh);
fh=NULL;
GFREE(fname);
/*GFREE(curseqid);
GFREE(curdescr);*/
}
int getReadPos() { return cur_fpos; } /* returns current read position in the
input stream (can be used within callback) */
int ReadSeqPos() {return rec_fpos; } /* returns the input stream offset of the last fasta
record processed by getFastaSeq*/
bool readHeader(FastaSeq& seq) { return (readHeader(&seq)!=NULL); }
FastaSeq* readSeq(int seqalloc=0) {
//allocate a new FastaSeq, reads the next record and returns it
//caller is responsible for deallocating returned FastaSeq memory!
FastaSeq* r=readHeader(NULL, seqalloc);
int len=0;
int c=-1;
//load the whole sequence in FastaSeq
while ((c = getc(fh)) != EOF && c != '>') {
cur_fpos++;
//if (isspace(c) || c<31)
if (c<=32) {
//before = (c=='\n' || c=='\r')?1:0;
continue; /* skip spaces */
}
if (len >= r->s_cap-1) {
GREALLOC(r->seq, r->s_cap + SEQCAPINC);
r->s_cap+=SEQCAPINC;
}
r->seq[len] = c;
//before=0;
len++;
}
r->seq[len] = '\0';
r->len=len;
return r;
}
FastaSeq* readHeader(FastaSeq* seq=NULL, int seqalloc=0, const char* trim_delim=NULL) {
/* reads the Fasta sequence header
the first character must be '>' for this call, after any spaces,
if seq is NULL a new FastaSeq object is allocated and returned,
otherwise id and descr are updated */
seqcoord=0;
int* buflen;
int* buflenstr;
char** buf;
int before;
if (feof(fh)) return NULL;
int c = getc(fh);
if (c==EOF) return NULL;
cur_fpos++;
while (c!=EOF && c<=32) { c=getc(fh); cur_fpos++; }//skip spaces etc.
if (c == EOF) return NULL;
if (c != '>')
bad_fastafmt();
if (seq==NULL) seq=new FastaSeq(seqalloc);
else { seq->clear(); seq->init(seqalloc); }
int len = 0; //chars accumulated so far
buflen=&(seq->id_cap);
buf=&(seq->id);
buflenstr=&(seq->namelen);
before=1; //before seq ID was completed
bool trim_done=false;
int dt_len=0; //only set if defline trim was requested
int dt_match=0; // trim_delim match char idx+1
while ((c = getc(fh)) != EOF) {
cur_fpos++;
if (c=='\n' || c=='\r') break;
if (trim_done) continue;
if (len >= *buflen-1) {
GREALLOC(*buf, *buflen + CAPINC);
*buflen+=CAPINC;
}
if (before) {//seq ID parsing
if (c<=32) {
// space encountered => seq_name finished
before=0;
(*buf)[len]='\0';
*buflenstr=len;
buf=&seq->descr;
buflen=&seq->d_cap;
buflenstr=&seq->descrlen;
len=0;
if (trim_delim!=NULL) {
//trimming the defline was requested
dt_len=strlen(trim_delim);
if (c<32 && c==trim_delim[0]) { trim_done=true; continue; }
}
//if (c!=1) // special case, nrdb concatenation
continue; // skip this "space"
}
} else { //seq description parsing
if (c<32 && dt_len>0 && c==trim_delim[0]) {
//end the defline parsing here (e.g. nrdb concatenation)
trim_done=true; continue;
}
if (dt_len) {
if (dt_match==0) {
if (c==trim_delim[0]) {
//quick way out?
if (dt_len==1) { trim_done=true; continue; }
dt_match=1;
}
}
else if (c==trim_delim[dt_match-1]) {
if (dt_match==dt_len) { len-=dt_len-1; trim_done=true; continue; }
dt_match++;
} else dt_match=0; //cancel this match attempt
} //trim delimiter matching
}
(*buf)[len]=c;
len++;
} //while reading defline
(*buf)[len]='\0'; /* terminate the comment string */
*buflenstr = len;
check_eof(c); /* it's wrong to have eof here */
seqcoord=1;
return (seq->namelen==0) ? NULL : seq;
}
FastaSeq *getFastaSeq(bool& is_last, FastaSeq* seq, charFunc* callbackFn = NULL, const char* trim_descr_at=NULL) {
/* seq must be a pointer to a initialized FastaSeq structure
if seq is NULL, the sequence is not actually read,
but just skipped and the file pointer set accordingly, while
the returned "pointer" will not be a FastaSeq one but just NULL or not NULL
(depending if eof was encountered)
if callbackFn is NULL, the sequence is read entirely in memory in a FastaSeq.seq field
otherwise only the defline is parsed into FastaSeq::id and FastaSeq::descr but actual
sequence letters are passed one by one to the callback function
and the actual sequence is never stored in memory (unless the callback does it)
The optional trim_descr_at is there for nrdb-collapsed entries, it's supposed to discard any text from the
description line following the trim_descr_at string; special case for "\x01" string: ^A delimiter
*/
int c, len;
int before;
rec_fpos=cur_fpos;
len = 0; //chars accumulated so far
if (fh==NULL || feof(fh)) return NULL;
// -------- read the defline first
if (seq==NULL) { // navigate only! don't read/parse anything but the record delimiter
before=1;
while ((c = getc(fh)) != EOF && c != '\n' && c !='\r') cur_fpos++; // skip defline
if (c==EOF && cur_fpos<=rec_fpos+2) return NULL;
check_eof(c); /* it's wrong to have eof here! */
cur_fpos++; //to account for the '\n' read
/*----- read the sequence now: */
before=1; /* "newline before" flag */
while ((c = getc(fh)) != EOF && c != '>') {
cur_fpos++;
before = (c=='\n' || c=='\r') ? 1 : 0;
}
//we should end up at a '>' character here, or EOF
} /* fasta fmt navigation to next sequence, no seq storage */
else { // sequence storage:
if (!readHeader(seq, 0, trim_descr_at)) {
is_last=true;
return NULL;
}
/*----- read the actual sequence now: */
len=0;
before=1; //newline before indicator
if (callbackFn==NULL) { //load the whole sequence in FastaSeq
while ((c = getc(fh)) != EOF && c != '>') {
cur_fpos++;
//if (isspace(c) || c<31)
if (c<=32) {
before = (c=='\n' || c=='\r')?1:0;
continue; /* skip spaces */
}
if (len >= seq->s_cap-1) {
GREALLOC(seq->seq, seq->s_cap + CAPINC);
seq->s_cap+=CAPINC;
}
seq->seq[len] = c;
before=0;
len++;
}
seq->seq[len] = '\0';
seq->len=len;
} /* sequence storage */
else { //use the callback for each letter, do not store the whole sequence in FastaSeq
while ((c = getc(fh)) != EOF && c != '>') {
cur_fpos++;
if (c<=32) {
before = (c=='\n' || c=='\r')?1:0;
continue; /* skip spaces within sequence*/
}
(*callbackFn)(c, len, seq); //call the user function for each letter
before=0;
len++;
}
seq->len=len;
} /* callback sequence reading (no storage)*/
} /* sequence parsing */
if (c=='>') {
if (!before) bad_fastafmt(); /* '>' must only be at start of line,
never within the sequence ! */
is_last=false; /* FALSE - not the last one */
ungetc(c, fh);
}
else is_last=true; /* TRUE - eof() here */
return ((seq==NULL) ? (FastaSeq*)fh : seq); //alwayws return non NULL here!
} //getFastaSeq
//simplified call to ignore the is_last flag
FastaSeq *getFastaSeq(FastaSeq* seq, charFunc* callbackFn = NULL) {
bool b;
if (fh==NULL || feof(fh)) return NULL;
return getFastaSeq(b, seq, callbackFn);
}
FastaSeq *getFastaSeq(FastaSeq* seq, const char* trim_descr_at) {
//special case to trim nrdb collapsed descriptions
bool b;
if (fh==NULL || feof(fh)) return NULL;
return getFastaSeq(b, seq, NULL, trim_descr_at);
}
uint seqSkip(uint slen, int& c){
//assumes the header was read !
//skip exactly slen characters in the actual aa or nt sequence
//(spaces are not counted)
uint skipacc=0;
while (skipacc<slen && ((c=getc(fh))!= EOF && c != '>')) {
cur_fpos++;
if (c<=32) continue; //skip spaces and other non-ASCII characters
seqcoord++;
skipacc++;
}
return skipacc; //may terminate prematurely
}
/* read a sequence range from the current FASTA record
this is much faster when rcoord>=seqcoord (i.e. when sequence
ranges are read sequentially)
if rcoord>=seqcoord assumes the header has been read already!
Returns the actual length of the sequence returned (0 if rcoord>seq_length)
and updates seqcoord, cur_fpos accordingly (rec_fpos is unchanged)
*/
uint getSeqRange(FastaSeq& seq, uint rcoord, uint rlen=0) {
int c;
uint len;
rec_fpos=cur_fpos;
if (!seqcoord || seqcoord>rcoord) {
// slow -- go back to the beginning of the record
seek(rec_fpos);
readHeader(&seq); //this will also reset seqcoord to 1
}
if (rcoord!=seqcoord) {
seqSkip(rcoord-seqcoord, c);
check_eof(c);
if (c=='>')
GError("Error: '>' character found while skipping through sequence!\n");
}
len = 0; //chars accumulated so far
seq.seq[0]='\0';
seq.len=0;
//----- read the actual subsequence now:
len=0;
while ((c = getc(fh)) != EOF && c != '>') {
cur_fpos++;
if (c<=32) continue; // skip spaces
if (len >= (uint) (seq.s_cap-1)) {
GREALLOC(seq.seq, seq.s_cap + CAPINC);
seq.s_cap+=CAPINC;
}
seq.seq[len] = c;
len++;
seqcoord++;
if (rlen>0 && len==rlen) break;
}
seq.seq[len] = '\0';
seq.len=len;
if (c=='>') bad_fastafmt(); /* '>' must only be at start of line,
never within the sequence ! */
return len;
} //getSeqRange
//only for writing
void putFastaSeq(FastaSeq *fa, const int linelen=60) {
writeFasta(fh, fa->id, fa->descr, fa->seq, linelen);
}
/*
static void writeFasta(FILE *fh, char* seqid, char* descr, char* seq, const int linelen=60, const int seqlen=0) {
FastaSeq::write(fh, seqid, descr, seq, linelen, seqlen);
}
*/
};
// ------------- FASTA parser/handler ----
// REQUIRES the first character processed after init()
// to be the first character of the record delimiter
// (default: ">")
class GFastaCharHandler {
protected:
char* recdelim;
charFunc* seqCallBack;
bool in_delim;
int delim_pos;
bool in_seqname;
bool in_descr;
bool in_seq;
FastaSeq* rec;
unsigned int seq_pos;
void reset() {
in_delim=true;
delim_pos=0;
in_seqname=false;
in_descr=false;
in_seq=false;
seq_pos=0;
}
public:
GFastaCharHandler(char* recdel=DEF_FASTA_DELIM) {
reset();
rec=NULL;
recdelim=recdel;
seqCallBack=NULL;
}
GFastaCharHandler(charFunc* chrCallBack, FastaSeq* r=NULL, char* recdel=DEF_FASTA_DELIM) {
reset();
rec=r;
recdelim=recdel;
seqCallBack=chrCallBack;
if (rec!=NULL) rec->reset();
}
void init() {
init(rec, seqCallBack);
}
void init(charFunc* chrCallBack) {
init(rec,chrCallBack);
}
void init(FastaSeq* r) {
init(r,seqCallBack);
}
void init(FastaSeq* r, charFunc* chrCallBack) {
rec=r;
seqCallBack=chrCallBack;
if (rec==NULL)
GError("GFastaCharHandler::init() Error: cannot use NULL FastaSeq!\n");
rec->reset();
reset();
}
void done() {
if (rec==NULL)
GError("GFastaCharHandler::done() Error: cannot use NULL FastaSeq!\n");
rec->endId();
rec->endDescr();
rec->endSeq();
}
//~GFastaCharHandler();
void processChar(char c) {
if (in_delim) { //skip record delimiter -- but it must be there!
if (recdelim[delim_pos]!=c) {//the only way to detect an Id starting
in_seqname=true;
in_delim=false;
}
delim_pos++;
}
if (in_seqname) {
if (rec->namelen>0 && c<=32) {
//breaking out of seq_name
rec->endId();
if (c=='\n' || c=='\r') { //end defline
in_seqname=false;
in_seq=true;
}
else { //seqname break, not defline end
in_seqname=false;
in_descr=true;
}
} // seqname termination
else { //seqname continues
if (c>32) rec->extendId(c);
}
return;
} // in_seqname
if (in_descr) {
if (c=='\n' || c=='\r') { //end defline
rec->endDescr();
in_descr=false;
in_seq=true;
}
else rec->extendDescr(c);
return;
} // in_descr
if (in_seq && c>32) {
seq_pos++; // 1-based sequence position !
if (seqCallBack==NULL) rec->extendSeq(c);
else (*seqCallBack)(c,seq_pos,rec);
}
}
};
#endif