void SequenceSet::UnpackSeqSet(const CBioseq_set& bss)
{
CBioseq_set::TSeq_set::const_iterator q, qe = bss.GetSeq_set().end();
for (q=bss.GetSeq_set().begin(); q!=qe; ++q) {
if (q->GetObject().IsSeq()) {
// only store amino acid or nucleotide sequences
if (q->GetObject().GetSeq().GetInst().GetMol() != CSeq_inst::eMol_aa &&
q->GetObject().GetSeq().GetInst().GetMol() != CSeq_inst::eMol_dna &&
q->GetObject().GetSeq().GetInst().GetMol() != CSeq_inst::eMol_rna &&
q->GetObject().GetSeq().GetInst().GetMol() != CSeq_inst::eMol_na)
continue;
const Sequence *sequence = new Sequence(q->GetObject().GetSeq());
if (!sequence || sequence->Status() != CAV_SUCCESS) {
status = sequence->Status();
return;
}
sequences.push_back(sequence);
} else { // Bioseq-set
UnpackSeqSet(q->GetObject().GetSet());
}
}
}
bool
CShortReadFastaInputSource::x_ReadFromTwoFiles(CBioseq_set& bioseq_set,
TSeqPos batch_size,
CShortReadFastaInputSource::EInputFormat format)
{
if (format == eFastc) {
NCBI_THROW(CInputException, eInvalidInput, "FASTC format cannot be "
"used with two files");
}
// tags to indicate paired sequences
CRef<CSeqdesc> seqdesc_first(new CSeqdesc);
seqdesc_first->SetUser().SetType().SetStr("Mapping");
seqdesc_first->SetUser().AddField("has_pair", eFirstSegment);
CRef<CSeqdesc> seqdesc_last(new CSeqdesc);
seqdesc_last->SetUser().SetType().SetStr("Mapping");
seqdesc_last->SetUser().AddField("has_pair", eLastSegment);
m_BasesAdded = 0;
while (m_BasesAdded < batch_size && !m_LineReader->AtEOF() &&
!m_SecondLineReader->AtEOF()) {
CRef<CSeq_entry> first;
CRef<CSeq_entry> second;
if (format == eFasta) {
first = x_ReadFastaOneSeq(m_LineReader);
second = x_ReadFastaOneSeq(m_SecondLineReader);
}
else {
first = x_ReadFastqOneSeq(m_LineReader);
second = x_ReadFastqOneSeq(m_SecondLineReader);
}
if (first.NotEmpty()) {
if (second.NotEmpty()) {
first->SetSeq().SetDescr().Set().push_back(seqdesc_first);
}
bioseq_set.SetSeq_set().push_back(first);
}
if (second.NotEmpty()) {
if (first.NotEmpty()) {
second->SetSeq().SetDescr().Set().push_back(seqdesc_last);
}
bioseq_set.SetSeq_set().push_back(second);
}
}
return true;
}
static void UnpackSeqSet(CBioseq_set& bss, SequenceSet *parent, SequenceSet::SequenceList& seqlist)
{
CBioseq_set::TSeq_set::iterator q, qe = bss.SetSeq_set().end();
for (q=bss.SetSeq_set().begin(); q!=qe; ++q) {
if (q->GetObject().IsSeq()) {
// only store amino acid or nucleotide sequences
if (q->GetObject().GetSeq().GetInst().GetMol() != CSeq_inst::eMol_aa &&
q->GetObject().GetSeq().GetInst().GetMol() != CSeq_inst::eMol_dna &&
q->GetObject().GetSeq().GetInst().GetMol() != CSeq_inst::eMol_rna &&
q->GetObject().GetSeq().GetInst().GetMol() != CSeq_inst::eMol_na)
continue;
const Sequence *sequence = new Sequence(parent, q->GetObject().SetSeq());
if (!sequence || !sequence->identifier)
FATALMSG("Can't create Sequence object, aborting...");
seqlist.push_back(sequence);
} else { // Bioseq-set
UnpackSeqSet(q->GetObject().SetSet(), parent, seqlist);
}
}
}
void
CShortReadFastaInputSource::x_ReadFastc(CBioseq_set& bioseq_set,
TSeqPos batch_size)
{
string id;
// tags to indicate paired sequences
CRef<CSeqdesc> seqdesc_first(new CSeqdesc);
seqdesc_first->SetUser().SetType().SetStr("Mapping");
seqdesc_first->SetUser().AddField("has_pair", eFirstSegment);
CRef<CSeqdesc> seqdesc_last(new CSeqdesc);
seqdesc_last->SetUser().SetType().SetStr("Mapping");
seqdesc_last->SetUser().AddField("has_pair", eLastSegment);
m_BasesAdded = 0;
while (m_BasesAdded < batch_size && !m_LineReader->AtEOF()) {
++(*m_LineReader);
m_Line = **m_LineReader;
// ignore empty lines
if (m_Line.empty()) {
continue;
}
// if defline
if (m_Line[0] == '>') {
id = x_ParseDefline(m_Line);
}
else {
// otherwise sequence
// make sure that a defline was read first
if (id.empty()) {
NCBI_THROW(CInputException, eInvalidInput,
(string)"Missing defline before line: " +
NStr::IntToString(m_LineReader->GetLineNumber()));
}
// find '><' that separate reads of a pair
size_t p = m_Line.find('>');
if (p == CTempString::npos || m_Line[p + 1] != '<') {
NCBI_THROW(CInputException, eInvalidInput,
(string)"FASTC parse error: Sequence separator '><'"
" was not found in line: " +
NStr::IntToString(m_LineReader->GetLineNumber()));
}
// set up reads, there are two sequences in the same line separated
char* first = (char*)m_Line.data();
char* second = (char*)m_Line.data() + p + 2;
size_t first_len = p;
size_t second_len = m_Line.length() - p - 2;
{{
CRef<CSeq_id> seqid(new CSeq_id);
seqid->Set(CSeq_id::e_Local, id + ".1");
CRef<CSeq_entry> seq_entry(new CSeq_entry);
CBioseq& bioseq = seq_entry->SetSeq();
bioseq.SetInst().SetMol(CSeq_inst::eMol_na);
bioseq.SetInst().SetRepr(CSeq_inst::eRepr_raw);
bioseq.SetId().clear();
bioseq.SetId().push_back(seqid);
bioseq.SetInst().SetLength(first_len);
first[first_len] = 0;
bioseq.SetInst().SetSeq_data().SetIupacna(CIUPACna(first));
bioseq.SetDescr().Set().push_back(seqdesc_first);
// add a sequence to the batch
bioseq_set.SetSeq_set().push_back(seq_entry);
}}
{{
CRef<CSeq_id> seqid(new CSeq_id);
seqid->Set(CSeq_id::e_Local, id + ".2");
CRef<CSeq_entry> seq_entry(new CSeq_entry);
CBioseq& bioseq = seq_entry->SetSeq();
bioseq.SetInst().SetMol(CSeq_inst::eMol_na);
bioseq.SetInst().SetRepr(CSeq_inst::eRepr_raw);
bioseq.SetId().clear();
bioseq.SetId().push_back(seqid);
bioseq.SetInst().SetLength(second_len);
second[second_len] = 0;
bioseq.SetInst().SetSeq_data().SetIupacna(CIUPACna(second));
bioseq.SetDescr().Set().push_back(seqdesc_last);
// add a sequence to the batch
bioseq_set.SetSeq_set().push_back(seq_entry);
}}
m_BasesAdded += first_len + second_len;
id.clear();
}
}
}
void
CShortReadFastaInputSource::x_ReadFastaOrFastq(CBioseq_set& bioseq_set,
TSeqPos batch_size)
{
// tags to indicate paired sequences
CRef<CSeqdesc> seqdesc_first(new CSeqdesc);
seqdesc_first->SetUser().SetType().SetStr("Mapping");
seqdesc_first->SetUser().AddField("has_pair", eFirstSegment);
CRef<CSeqdesc> seqdesc_last(new CSeqdesc);
seqdesc_last->SetUser().SetType().SetStr("Mapping");
seqdesc_last->SetUser().AddField("has_pair", eLastSegment);
m_BasesAdded = 0;
while (m_BasesAdded < batch_size && !m_LineReader->AtEOF()) {
CRef<CSeq_entry> first;
CRef<CSeq_entry> second;
switch (m_Format) {
case eFasta:
first = x_ReadFastaOneSeq(m_LineReader);
break;
case eFastq:
first = x_ReadFastqOneSeq(m_LineReader);
break;
default:
NCBI_THROW(CInputException, eInvalidInput, "Invalid input file "
"format x_ReadFastaOrFastq read either FASTA or FASTQ");
}
// if paired read the next sequence and mark a pair
if (m_IsPaired) {
switch (m_Format) {
case eFasta:
second = x_ReadFastaOneSeq(m_LineReader);
break;
case eFastq:
second = x_ReadFastqOneSeq(m_LineReader);
break;
default:
NCBI_THROW(CInputException, eInvalidInput, "Invalid input file "
"format x_ReadFastaOrFastq read either FASTA or "
"FASTQ");
}
if (first.NotEmpty()) {
if (second.NotEmpty()) {
first->SetSeq().SetDescr().Set().push_back(seqdesc_first);
}
bioseq_set.SetSeq_set().push_back(first);
}
if (second.NotEmpty()) {
if (first.NotEmpty()) {
second->SetSeq().SetDescr().Set().push_back(seqdesc_last);
}
bioseq_set.SetSeq_set().push_back(second);
}
}
else {
// otherwise just add the read sequence
if (first.NotEmpty()) {
bioseq_set.SetSeq_set().push_back(first);
}
}
}
}
