FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Add option to return sequences from the Genbank file as a stream of sequences by emckee2006 · Pull Request #870 · biojava/biojava · GitHub

/ biojava Public

Add option to return sequences from the Genbank file as a stream of sequences #870

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
emckee2006 wants to merge 3 commits into from
Conversation 18 Commits 3 Checks 0 Files changed
Closed

Add option to return sequences from the Genbank file as a stream of sequences #870

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
e9fec3c
Add option to return sequences from the Genbank file as a stream of s…
emckee2006 Apr 15, 2020
4eb1c20
Merge branch 'GenbankReader_SequenceStream'
emckee2006 Apr 15, 2020
7327923
Merge branch 'upstream_master'
emckee2006 Apr 15, 2020
File filter

Filter by extension

Filter by extension .java  (2) All 1 file type selected
Viewed files
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Unified
Split
Hide whitespace
Diff view
Unified
Split
Hide whitespace
60 changes: 41 additions & 19 deletions biojava-core/src/main/java/org/biojava/nbio/core/sequence/io/GenbankReader.java
Show comments
View file
Open in desktop
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Show hidden characters
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.stream.Stream;

/**
* Use {@link GenbankReaderHelper} as an example of how to use this class where {@link GenbankReaderHelper} should be the
Expand Down Expand Up @@ -153,35 +154,56 @@ public LinkedHashMap<String,S> process(final int max) throws IOException, Compou
}

LinkedHashMap<String,S> sequences = new LinkedHashMap<>();
@SuppressWarnings("unchecked")
int i=0;
while(true) {

lafita Apr 22, 2020

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low Quality

This is what is hanging the build. Because the stream returned is null (the file does not exist, see comment below), sequence is always null and this while loop never finishes.

The while loop is not needed anymore, simply take an action if the sequence is null.
if(max>0 && i>=max) break;
i++;
String seqString = genbankParser.getSequence(bufferedReader, 0);
//reached end of file?
if(seqString==null) break;
@SuppressWarnings("unchecked")
S sequence = (S) sequenceCreator.getSequence(seqString, 0);
genbankParser.getSequenceHeaderParser().parseHeader(genbankParser.getHeader(), sequence);

// add features to new sequence
genbankParser.getFeatures().values().stream()
.flatMap(List::stream)
.forEach(sequence::addFeature);

// add taxonomy ID to new sequence
ArrayList<DBReferenceInfo> dbQualifier = genbankParser.getDatabaseReferences().get("db_xref");
if (dbQualifier != null){
DBReferenceInfo q = dbQualifier.get(0);
sequence.setTaxonomy(new TaxonomyID(q.getDatabase()+":"+q.getId(), DataSource.GENBANK));
S sequence = getSequence();
if(null == sequence) {
break;
}

sequences.put(sequence.getAccession().getID(), sequence);
}

return sequences;
}

public Stream<S> getSequencesAsStream() {
return Stream.generate(() -> {
try {
return getSequence();
} catch (IOException | CompoundNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
return null;

josemduarte Apr 15, 2020

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low Quality

I'd rather throw the exception forward and let the caller handle it. Returning null means all callers have to handle the null and that they lose the ability to provide a good error message.
}
});
}

private S getSequence() throws IOException, CompoundNotFoundException {
String seqString = genbankParser.getSequence(bufferedReader, 0);
//reached end of file?
if(seqString==null) {
return null;
}
@SuppressWarnings("unchecked")
S sequence = (S) sequenceCreator.getSequence(seqString, 0);
genbankParser.getSequenceHeaderParser().parseHeader(genbankParser.getHeader(), sequence);

// add features to new sequence
genbankParser.getFeatures().values().stream()
.flatMap(List::stream)
.forEach(sequence::addFeature);

// add taxonomy ID to new sequence
ArrayList<DBReferenceInfo> dbQualifier = genbankParser.getDatabaseReferences().get("db_xref");
if (dbQualifier != null){
DBReferenceInfo q = dbQualifier.get(0);
sequence.setTaxonomy(new TaxonomyID(q.getDatabase()+":"+q.getId(), DataSource.GENBANK));
}

return sequence;
}

public void close() {
try {
Expand Down
22 changes: 22 additions & 0 deletions biojava-core/src/test/java/org/biojava/nbio/core/sequence/io/GenbankReaderTest.java
Show comments
View file
Open in desktop
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Show hidden characters
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Stream;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertEquals;
Expand Down Expand Up @@ -163,6 +164,27 @@ public void testProcess() throws Exception {
assertEquals(3, dnaSequence.getAccession().getVersion().intValue());
assertTrue(genbankDNA.isClosed());
}

@Test
public void testSequenceStream() {
CheckableInputStream inStream = new CheckableInputStream(this.getClass().getResourceAsStream("/two-dnaseqs.gb"));

lafita Apr 22, 2020

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low Quality

Did you also include this file in the test resources directory? It seems not to be in this PR

GenbankReader<DNASequence, NucleotideCompound> genbankDNA
= new GenbankReader<>(
inStream,
new GenericGenbankHeaderParser<>(),
new DNASequenceCreator(DNACompoundSet.getDNACompoundSet())
);

Stream<DNASequence> seqStream = genbankDNA.getSequencesAsStream();
assertEquals(seqStream.count(),2);

assertFalse(genbankDNA.isClosed());
genbankDNA.close();
assertTrue(genbankDNA.isClosed());
assertTrue(inStream.isclosed());

}

/**
* Test the process method with a number of sequences to be read at each call.
Expand Down
Back | FazBrowse Home | New Git URL