in progress
[jalview.git] / forester / java / src / org / forester / ws / seqdb / EbiDbEntry.java
index 5f2cad2..df739ed 100644 (file)
 
 package org.forester.ws.seqdb;
 
-import java.util.ArrayList;
 import java.util.List;
+import java.util.SortedSet;
+import java.util.TreeSet;
+import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
 import org.forester.go.GoTerm;
 import org.forester.phylogeny.data.Accession;
+import org.forester.phylogeny.data.Annotation;
+import org.forester.sequence.MolecularSequence;
 import org.forester.util.ForesterUtil;
 
 public final class EbiDbEntry implements SequenceDatabaseEntry {
 
-    public static SequenceDatabaseEntry createInstanceFromPlainText( final List<String> lines ) {
-        final EbiDbEntry e = new EbiDbEntry();
-        for( final String line : lines ) {
-            if ( line.startsWith( "PA" ) ) {
-                e.setPA( SequenceDbWsTools.extractFrom( line, "PA" ) );
-            }
-            else if ( line.startsWith( "DE" ) ) {
-                e.setDe( SequenceDbWsTools.extractFrom( line, "DE" ) );
-            }
-            else if ( line.startsWith( "OS" ) ) {
-                if ( line.indexOf( "(" ) > 0 ) {
-                    e.setOs( SequenceDbWsTools.extractFromTo( line, "OS", "(" ) );
-                }
-                else {
-                    e.setOs( SequenceDbWsTools.extractFrom( line, "OS" ) );
-                }
-            }
-            else if ( line.startsWith( "OX" ) ) {
-                if ( line.indexOf( "NCBI_TaxID=" ) > 0 ) {
-                    e.setTaxId( SequenceDbWsTools.extractFromTo( line, "NCBI_TaxID=", ";" ) );
-                }
-            }
-        }
-        return e;
-    }
-
-    public static SequenceDatabaseEntry createInstanceFromPlainTextForRefSeq( final List<String> lines ) {
-        final Pattern X_PATTERN = Pattern.compile( "^[A-Z]+" );
-        final Pattern chromosome_PATTERN = Pattern.compile( "\\s+/chromosome=\"(\\w+)\"" );
-        final Pattern map_PATTERN = Pattern.compile( "\\s+/map=\"([\\w+\\.])\"" );
-        final Pattern gene_PATTERN = Pattern.compile( "\\s+/gene=\"(.+)\"" );
-        final Pattern mim_xref_PATTERN = Pattern.compile( "\\s+/db_xref=\"MIM:(\\d+)\"" );
-        final Pattern taxon_xref_PATTERN = Pattern.compile( "\\s+/db_xref=\"taxon:(\\d+)\"" );
-        final Pattern interpro_PATTERN = Pattern.compile( "\\s+/db_xref=\"InterPro:(IP\\d+)\"" );
-        final Pattern uniprot_PATTERN = Pattern.compile( "\\s+/db_xref=\"UniProtKB/TrEMBL:(\\w+)\"" );
-        final EbiDbEntry e = new EbiDbEntry();
-        final StringBuilder def = new StringBuilder();
-        boolean in_def = false;
-        boolean in_features = false;
-        boolean in_source = false;
-        boolean in_gene = false;
-        boolean in_cds = false;
-        boolean in_protein = false;
-        for( final String line : lines ) {
-            if ( line.startsWith( "ACCESSION " ) ) {
-                e.setPA( SequenceDbWsTools.extractFrom( line, "ACCESSION" ) );
-                in_def = false;
-            }
-            else if ( line.startsWith( "DEFINITION " ) ) {
-                if ( line.indexOf( "[" ) > 0 ) {
-                    def.append( SequenceDbWsTools.extractFromTo( line, "DEFINITION", "[" ) );
-                }
-                else if ( line.indexOf( "." ) > 0 ) {
-                    def.append( SequenceDbWsTools.extractFromTo( line, "DEFINITION", "." ) );
-                }
-                else {
-                    def.append( SequenceDbWsTools.extractFrom( line, "DEFINITION" ) );
-                }
-                in_def = true;
-            }
-            else if ( line.startsWith( "  ORGANISM " ) ) {
-                if ( line.indexOf( "(" ) > 0 ) {
-                    e.setOs( SequenceDbWsTools.extractFromTo( line, "  ORGANISM", "(" ) );
-                }
-                else {
-                    e.setOs( SequenceDbWsTools.extractFrom( line, "  ORGANISM" ) );
-                }
-                //  in_def = false;
-            }
-            else if ( line.startsWith( " " ) && in_def ) {
-                def.append( " " );
-                if ( line.indexOf( "[" ) > 0 ) {
-                    def.append( SequenceDbWsTools.extractTo( line, "[" ) );
-                }
-                else if ( line.indexOf( "." ) > 0 ) {
-                    def.append( SequenceDbWsTools.extractTo( line, "." ) );
-                }
-                else {
-                    def.append( line.trim() );
-                }
-            }
-            else {
-                in_def = false;
-            }
-            if ( X_PATTERN.matcher( line ).find() ) {
-                in_features = false;
-                in_source = false;
-                in_gene = false;
-                in_cds = false;
-                in_protein = false;
-                // in_def = false;
-            }
-            if ( line.startsWith( "FEATURES " ) ) {
-                in_features = true;
-            }
-            if ( in_features && line.startsWith( "     source " ) ) {
-                in_source = true;
-                in_gene = false;
-                in_cds = false;
-                in_protein = false;
-            }
-            if ( in_features && line.startsWith( "     gene " ) ) {
-                in_source = false;
-                in_gene = true;
-                in_cds = false;
-                in_protein = false;
-            }
-            if ( in_features && line.startsWith( "     CDS " ) ) {
-                in_source = false;
-                in_gene = false;
-                in_cds = true;
-                in_protein = false;
-            }
-            if ( in_features && line.startsWith( "     Protein " ) ) {
-                in_source = false;
-                in_gene = false;
-                in_cds = false;
-                in_protein = true;
-            }
-        }
-        if ( def.length() > 0 ) {
-            e.setDe( def.toString().trim() );
-        }
-        return e;
-    }
+    private SortedSet<Annotation> _annotations;
+    private String                _chromosome;
+    private SortedSet<Accession>  _cross_references;
+    private String                _de;
+    private String                _gene_name;
+    private String                _map;
+    private String                _os;
     // FIXME actually this is NCBI entry
     //http://www.ebi.ac.uk/Tools/dbfetch/dbfetch/emb/AAR37336/
-    private String               _pa;
-    private String               _de;
-    private String               _os;
-    private String               _tax_id;
-    private String               _symbol;
-    private String               _provider;
-    private ArrayList<Accession> _cross_references;
-    private String               _gene_name;
+    private String                _pa;
+    private String                _provider;
+    private String                _symbol;
+    private String                _tax_id;
 
     // TODO  PUBMED   15798186
     //TODO  (FEATURES) 
@@ -425,14 +307,6 @@ public final class EbiDbEntry implements SequenceDatabaseEntry {
     private EbiDbEntry() {
     }
 
-    private void addCrossReference( final Accession accession ) {
-        if ( _cross_references == null ) {
-            _cross_references = new ArrayList<Accession>();
-        }
-        System.out.println( "XREF ADDED: " + accession );
-        _cross_references.add( accession );
-    }
-
     @Override
     public Object clone() throws CloneNotSupportedException {
         throw new CloneNotSupportedException();
@@ -444,7 +318,17 @@ public final class EbiDbEntry implements SequenceDatabaseEntry {
     }
 
     @Override
-    public List<Accession> getCrossReferences() {
+    public SortedSet<Annotation> getAnnotations() {
+        return _annotations;
+    }
+
+    @Override
+    public String getChromosome() {
+        return _chromosome;
+    }
+
+    @Override
+    public SortedSet<Accession> getCrossReferences() {
         return _cross_references;
     }
 
@@ -454,11 +338,16 @@ public final class EbiDbEntry implements SequenceDatabaseEntry {
     }
 
     @Override
-    public List<GoTerm> getGoTerms() {
+    public SortedSet<GoTerm> getGoTerms() {
         return null;
     }
 
     @Override
+    public String getMap() {
+        return _map;
+    }
+
+    @Override
     public String getProvider() {
         return _provider;
     }
@@ -490,10 +379,33 @@ public final class EbiDbEntry implements SequenceDatabaseEntry {
                 && ForesterUtil.isEmpty( getTaxonomyIdentifier() ) && ForesterUtil.isEmpty( getSequenceSymbol() ) );
     }
 
-    private void setDe( final String rec_name ) {
-        if ( _de == null ) {
-            _de = rec_name;
+    public void setProvider( final String provider ) {
+        _provider = provider;
+    }
+
+    private void addAnnotation( final Annotation annotation ) {
+        if ( _annotations == null ) {
+            _annotations = new TreeSet<Annotation>();
+        }
+        _annotations.add( annotation );
+    }
+
+    private void addCrossReference( final Accession accession ) {
+        if ( _cross_references == null ) {
+            _cross_references = new TreeSet<Accession>();
         }
+        System.out.println( "XREF ADDED: " + accession );
+        _cross_references.add( accession );
+    }
+
+    private void setAccession( final String pa ) {
+        if ( _pa == null ) {
+            _pa = pa;
+        }
+    }
+
+    private void setChromosome( final String chromosome ) {
+        _chromosome = chromosome;
     }
 
     private void setGeneName( final String gene_name ) {
@@ -502,20 +414,18 @@ public final class EbiDbEntry implements SequenceDatabaseEntry {
         }
     }
 
-    private void setOs( final String os ) {
-        if ( _os == null ) {
-            _os = os;
-        }
+    private void setMap( final String map ) {
+        _map = map;
     }
 
-    private void setPA( final String pa ) {
-        if ( _pa == null ) {
-            _pa = pa;
+    private void setSequenceName( final String rec_name ) {
+        if ( _de == null ) {
+            _de = rec_name;
         }
     }
 
-    public void setProvider( final String provider ) {
-        _provider = provider;
+    private void setSequenceSymbol( final String symbol ) {
+        _symbol = symbol;
     }
 
     private void setTaxId( final String tax_id ) {
@@ -523,4 +433,253 @@ public final class EbiDbEntry implements SequenceDatabaseEntry {
             _tax_id = tax_id;
         }
     }
+
+    private void setTaxonomyScientificName( final String os ) {
+        if ( _os == null ) {
+            _os = os;
+        }
+    }
+
+    //    public static SequenceDatabaseEntry createInstanceFromPlainText( final List<String> lines ) {
+    //        final EbiDbEntry e = new EbiDbEntry();
+    //        for( final String line : lines ) {
+    //            if ( line.startsWith( "PA" ) ) {
+    //                e.setPA( SequenceDbWsTools.extractFrom( line, "PA" ) );
+    //            }
+    //            else if ( line.startsWith( "DE" ) ) {
+    //                e.setDe( SequenceDbWsTools.extractFrom( line, "DE" ) );
+    //            }
+    //            else if ( line.startsWith( "OS" ) ) {
+    //                if ( line.indexOf( "(" ) > 0 ) {
+    //                    e.setOs( SequenceDbWsTools.extractFromTo( line, "OS", "(" ) );
+    //                }
+    //                else {
+    //                    e.setOs( SequenceDbWsTools.extractFrom( line, "OS" ) );
+    //                }
+    //            }
+    //            else if ( line.startsWith( "OX" ) ) {
+    //                if ( line.indexOf( "NCBI_TaxID=" ) > 0 ) {
+    //                    e.setTaxId( SequenceDbWsTools.extractFromTo( line, "NCBI_TaxID=", ";" ) );
+    //                }
+    //            }
+    //        }
+    //        return e;
+    //    }
+    public static SequenceDatabaseEntry createInstanceFromPlainTextForRefSeq( final List<String> lines ) {
+        final Pattern X_PATTERN = Pattern.compile( "^[A-Z]+" );
+        final Pattern chromosome_PATTERN = Pattern.compile( "\\s+/chromosome=\"(\\w+)\"" );
+        final Pattern map_PATTERN = Pattern.compile( "\\s+/map=\"([\\w+\\.])\"" );
+        final Pattern gene_PATTERN = Pattern.compile( "\\s+/gene=\"(.+)\"" );
+        final Pattern mim_PATTERN = Pattern.compile( "\\s+/db_xref=\"MIM:(\\d+)\"" );
+        final Pattern taxon_PATTERN = Pattern.compile( "\\s+/db_xref=\"taxon:(\\d+)\"" );
+        final Pattern interpro_PATTERN = Pattern.compile( "\\s+/db_xref=\"InterPro:([A-Z0-9]+)\"" );
+        final Pattern uniprot_PATTERN = Pattern.compile( "\\s+/db_xref=\"UniProtKB/[A-Za-z-]*:(\\w+)\"" );
+        final Pattern hgnc_PATTERN = Pattern.compile( "\\s+/db_xref=\"[A-Z:]*HGNC:(\\d+)\"" );
+        final Pattern geneid_PATTERN = Pattern.compile( "\\s+/db_xref=\"GeneID:(\\d+)\"" );
+        final Pattern pdb_PATTERN = Pattern.compile( "\\s+/db_xref=\"PDB:([A-Z0-9]+)\"" );
+        final Pattern ec_PATTERN = Pattern.compile( "\\s+/EC_number=\"([\\.\\-\\d]+)\"" );
+        final Pattern product_PATTERN = Pattern.compile( "\\s+/product=\"(\\w{1,10})\"" );
+        final EbiDbEntry e = new EbiDbEntry();
+        final StringBuilder def = new StringBuilder();
+        boolean in_definition = false;
+        boolean in_features = false;
+        boolean in_source = false;
+        boolean in_gene = false;
+        boolean in_cds = false;
+        boolean in_mrna = false;
+        boolean in_protein = false;
+        for( final String line : lines ) {
+            if ( line.startsWith( "ACCESSION " ) ) {
+                e.setAccession( SequenceDbWsTools.extractFrom( line, "ACCESSION" ) );
+                in_definition = false;
+            }
+            else if ( line.startsWith( "ID " ) ) {
+                e.setAccession( SequenceDbWsTools.extractFromTo( line, "ID", ";" ) );
+                in_definition = false;
+            }
+            else if ( line.startsWith( "DEFINITION " ) || ( line.startsWith( "DE " ) ) ) {
+                boolean definiton = false;
+                if ( line.startsWith( "DEFINITION " ) ) {
+                    definiton = true;
+                }
+                if ( line.indexOf( "[" ) > 0 ) {
+                    if ( definiton ) {
+                        x( def, ( SequenceDbWsTools.extractFromTo( line, "DEFINITION", "[" ) ) );
+                    }
+                    else {
+                        x( def, ( SequenceDbWsTools.extractFromTo( line, "DE", "[" ) ) );
+                    }
+                }
+                else if ( line.indexOf( "." ) > 0 ) {
+                    if ( definiton ) {
+                        x( def, ( SequenceDbWsTools.extractFromTo( line, "DEFINITION", "." ) ) );
+                    }
+                    else {
+                        x( def, ( SequenceDbWsTools.extractFromTo( line, "DE", "." ) ) );
+                    }
+                }
+                else {
+                    if ( definiton ) {
+                        x( def, ( SequenceDbWsTools.extractFrom( line, "DEFINITION" ) ) );
+                    }
+                    else {
+                        x( def, ( SequenceDbWsTools.extractFrom( line, "DE" ) ) );
+                    }
+                }
+                if ( definiton ) {
+                    in_definition = true;
+                }
+            }
+            else if ( line.startsWith( "  ORGANISM " ) ) {
+                if ( line.indexOf( "(" ) > 0 ) {
+                    e.setTaxonomyScientificName( SequenceDbWsTools.extractFromTo( line, "  ORGANISM", "(" ) );
+                }
+                else {
+                    e.setTaxonomyScientificName( SequenceDbWsTools.extractFrom( line, "  ORGANISM" ) );
+                }
+                //  in_def = false;
+            }
+            else if ( line.startsWith( "OS " ) ) {
+                if ( line.indexOf( "(" ) > 0 ) {
+                    e.setTaxonomyScientificName( SequenceDbWsTools.extractFromTo( line, "OS", "(" ) );
+                }
+                else {
+                    e.setTaxonomyScientificName( SequenceDbWsTools.extractFrom( line, "OS" ) );
+                }
+            }
+            else if ( line.startsWith( " " ) && in_definition ) {
+                def.append( " " );
+                if ( line.indexOf( "[" ) > 0 ) {
+                    def.append( SequenceDbWsTools.extractTo( line, "[" ) );
+                }
+                else if ( line.indexOf( "." ) > 0 ) {
+                    def.append( SequenceDbWsTools.extractTo( line, "." ) );
+                }
+                else {
+                    def.append( line.trim() );
+                }
+            }
+            else {
+                in_definition = false;
+            }
+            if ( !line.startsWith( "FT " ) && X_PATTERN.matcher( line ).find() ) {
+                in_features = false;
+                in_source = false;
+                in_gene = false;
+                in_cds = false;
+                in_mrna = false;
+                in_protein = false;
+                // in_def = false;
+            }
+            if ( line.startsWith( "FEATURES " ) || line.startsWith( "FT " ) ) {
+                in_features = true;
+            }
+            if ( in_features && ( line.startsWith( "     source " ) || line.startsWith( "FT   source " ) ) ) {
+                in_source = true;
+                in_gene = false;
+                in_cds = false;
+                in_mrna = false;
+                in_protein = false;
+            }
+            if ( in_features && ( line.startsWith( "     gene " ) || line.startsWith( "FT   gene " ) ) ) {
+                in_source = false;
+                in_gene = true;
+                in_cds = false;
+                in_mrna = false;
+                in_protein = false;
+            }
+            if ( in_features && ( line.startsWith( "     CDS " ) || line.startsWith( "FT   CDS " ) ) ) {
+                in_source = false;
+                in_gene = false;
+                in_cds = true;
+                in_mrna = false;
+                in_protein = false;
+            }
+            if ( in_features && ( line.startsWith( "     Protein " ) || line.startsWith( "FT   Protein " ) ) ) {
+                in_source = false;
+                in_gene = false;
+                in_cds = false;
+                in_mrna = false;
+                in_protein = true;
+            }
+            if ( in_features && ( line.startsWith( "     mRNA " ) || line.startsWith( "FT   mRNA " ) ) ) {
+                in_source = false;
+                in_gene = false;
+                in_cds = false;
+                in_mrna = true;
+                in_protein = false;
+            }
+            if ( in_source ) {
+                final Matcher ti = taxon_PATTERN.matcher( line );
+                if ( ti.find() ) {
+                    e.setTaxId( ti.group( 1 ) );
+                }
+                final Matcher chr = chromosome_PATTERN.matcher( line );
+                if ( chr.find() ) {
+                    e.setChromosome( chr.group( 1 ) );
+                }
+                final Matcher map = map_PATTERN.matcher( line );
+                if ( map.find() ) {
+                    e.setMap( map.group( 1 ) );
+                }
+            }
+            if ( in_cds || in_gene ) {
+                final Matcher hgnc = hgnc_PATTERN.matcher( line );
+                if ( hgnc.find() ) {
+                    e.addCrossReference( new Accession( hgnc.group( 1 ), "hgnc" ) );
+                }
+                final Matcher geneid = geneid_PATTERN.matcher( line );
+                if ( geneid.find() ) {
+                    e.addCrossReference( new Accession( geneid.group( 1 ), "geneid" ) );
+                }
+            }
+            if ( in_protein || in_cds || in_gene || in_mrna ) {
+                final Matcher ec = ec_PATTERN.matcher( line );
+                if ( ec.find() ) {
+                    e.addAnnotation( new Annotation( "EC", ec.group( 1 ) ) );
+                }
+                final Matcher gene = gene_PATTERN.matcher( line );
+                if ( gene.find() ) {
+                    e.setGeneName( gene.group( 1 ) );
+                }
+                final Matcher uniprot = uniprot_PATTERN.matcher( line );
+                if ( uniprot.find() ) {
+                    e.addCrossReference( new Accession( uniprot.group( 1 ), "uniprot" ) );
+                }
+                final Matcher interpro = interpro_PATTERN.matcher( line );
+                if ( interpro.find() ) {
+                    e.addCrossReference( new Accession( interpro.group( 1 ), "interpro" ) );
+                }
+                final Matcher mim = mim_PATTERN.matcher( line );
+                if ( mim.find() ) {
+                    e.addCrossReference( new Accession( mim.group( 1 ), "mim" ) );
+                }
+                final Matcher product = product_PATTERN.matcher( line );
+                if ( product.find() ) {
+                    e.setSequenceSymbol( product.group( 1 ) );
+                }
+                final Matcher pdb = pdb_PATTERN.matcher( line );
+                if ( pdb.find() ) {
+                    e.addCrossReference( new Accession( pdb.group( 1 ), "pdb" ) );
+                }
+            }
+        }
+        if ( def.length() > 0 ) {
+            e.setSequenceName( def.toString().trim() );
+        }
+        return e;
+    }
+
+    private static void x( final StringBuilder sb, final String s ) {
+        if ( sb.length() > 0 ) {
+            sb.append( " " );
+        }
+        sb.append( s.trim() );
+    }
+
+    @Override
+    public MolecularSequence getMolecularSequence() {
+        // TODO Auto-generated method stub
+        return null;
+    }
 }