X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fjalview%2Fws%2FDBRefFetcher.java;h=8c8a717dc2928ed67fee511f5d9f8ea188c15f60;hb=182059d079ef2563d139ab1f0721c466cd648c02;hp=c2d7ad5957d50817d08d1e760f0ca0d51f64057f;hpb=08fbe7c3d40a4077ab519334952a079968c07cd1;p=jalview.git
diff --git a/src/jalview/ws/DBRefFetcher.java b/src/jalview/ws/DBRefFetcher.java
index c2d7ad5..8c8a717 100644
--- a/src/jalview/ws/DBRefFetcher.java
+++ b/src/jalview/ws/DBRefFetcher.java
@@ -1,471 +1,841 @@
-/*
- * Jalview - A Sequence Alignment Editor and Viewer
- * Copyright (C) 2007 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License
- * as published by the Free Software Foundation; either version 2
- * of the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
- */
-package jalview.ws;
-
-import java.io.*;
-import java.util.*;
-
-import org.exolab.castor.mapping.*;
-import org.exolab.castor.xml.*;
-import jalview.analysis.*;
-import jalview.datamodel.*;
-import jalview.datamodel.Mapping;
-import jalview.gui.*;
-import jalview.ws.dbsources.Uniprot;
-import jalview.ws.ebi.EBIFetchClient;
-
-/**
- * DOCUMENT ME!
- *
- * @author $author$
- * @version $Revision$
- */
-public class DBRefFetcher implements Runnable
-{
- SequenceI[] dataset;
-
- AlignFrame af;
-
- CutAndPasteTransfer output = new CutAndPasteTransfer();
-
- StringBuffer sbuffer = new StringBuffer();
-
- boolean running = false;
-
- // /This will be a collection of Vectors of sequenceI refs.
- // The key will be the seq name or accession id of the seq
- Hashtable seqRefs;
-
- String[] dbSources;
-
- SequenceFetcher sfetcher;
-
- public DBRefFetcher()
- {
- }
-
- /**
- * Creates a new SequenceFeatureFetcher object.
- *
- * @param seqs
- * fetch references for these sequences
- * @param af
- * the parent alignframe for progress bar monitoring.
- */
- public DBRefFetcher(SequenceI[] seqs, AlignFrame af)
- {
- this.af = af;
- SequenceI[] ds = new SequenceI[seqs.length];
- for (int i = 0; i < seqs.length; i++)
- {
- if (seqs[i].getDatasetSequence() != null)
- ds[i] = seqs[i].getDatasetSequence();
- else
- ds[i] = seqs[i];
- }
- this.dataset = ds;
- sfetcher = new SequenceFetcher();
- // select appropriate databases based on alignFrame context.
- if (af.getViewport().getAlignment().isNucleotide())
- {
- dbSources = DBRefSource.DNACODINGDBS;
- }
- else
- {
- dbSources = DBRefSource.PROTEINDBS;
- }
- }
-
- /**
- * start the fetcher thread
- *
- * @param waitTillFinished
- * true to block until the fetcher has finished
- */
- public void fetchDBRefs(boolean waitTillFinished)
- {
- Thread thread = new Thread(this);
- thread.start();
- running = true;
-
- if (waitTillFinished)
- {
- while (running)
- {
- try
- {
- Thread.sleep(500);
- } catch (Exception ex)
- {
- }
- }
- }
- }
-
- /**
- * The sequence will be added to a vector of sequences belonging to key which
- * could be either seq name or dbref id
- *
- * @param seq
- * SequenceI
- * @param key
- * String
- */
- void addSeqId(SequenceI seq, String key)
- {
- key = key.toUpperCase();
-
- Vector seqs;
- if (seqRefs.containsKey(key))
- {
- seqs = (Vector) seqRefs.get(key);
-
- if (seqs != null && !seqs.contains(seq))
- {
- seqs.addElement(seq);
- }
- else if (seqs == null)
- {
- seqs = new Vector();
- seqs.addElement(seq);
- }
-
- }
- else
- {
- seqs = new Vector();
- seqs.addElement(seq);
- }
-
- seqRefs.put(key, seqs);
- }
-
- /**
- * DOCUMENT ME!
- */
- public void run()
- {
- if (dbSources == null)
- {
- throw new Error("Implementation error. Must initialise dbSources");
- }
- long startTime = System.currentTimeMillis();
- af.setProgressBar("Fetching db refs", startTime);
- running = true;
- int db = 0;
- Vector sdataset = new Vector();
- for (int s = 0; s < dataset.length; s++)
- {
- sdataset.addElement(dataset[s]);
- }
- while (sdataset.size() > 0 && db < dbSources.length)
- {
- int maxqlen = 1; // default number of queries made to at one time
- System.err.println("Verifying against " + dbSources[db]);
- jalview.ws.seqfetcher.DbSourceProxy dbsource = sfetcher
- .getSourceProxy(dbSources[db]);
- if (dbsource == null)
- {
- System.err.println("No proxy for " + dbSources[db]);
- db++;
- continue;
- }
- if (dbsource.getDbSourceProperties()
- .containsKey(DBRefSource.MULTIACC))
- {
- maxqlen = ((Integer) dbsource.getDbSourceProperties().get(
- DBRefSource.MULTIACC)).intValue();
- }
- // iterate through db for each remaining un-verified sequence
- SequenceI[] currSeqs = new SequenceI[sdataset.size()];
- sdataset.copyInto(currSeqs);// seqs that are to be validated against
- // dbSources[db]
- Vector queries = new Vector(); // generated queries curSeq
- seqRefs = new Hashtable();
-
- int seqIndex = 0;
-
- while (queries.size() > 0 || seqIndex < currSeqs.length)
- {
- if (queries.size() > 0)
- {
- // Still queries to make for current seqIndex
- StringBuffer queryString = new StringBuffer("");
- int nqSize = (maxqlen > queries.size()) ? queries.size()
- : maxqlen;
- for (int nq = 0, numq = 0; nq < nqSize; nq++)
- {
- String query = (String) queries.elementAt(nq);
- if (dbsource.isValidReference(query))
- {
- queryString.append((nq == 0) ? "" : dbsource
- .getAccessionSeparator());
- queryString.append(query);
- numq++;
- }
- }
- for (int nq = 0; nq < nqSize; nq++)
- {
- queries.removeElementAt(0);
- }
- // make the queries and process the response
- AlignmentI retrieved = null;
- try
- {
- retrieved = dbsource.getSequenceRecords(queryString.toString());
- } catch (Exception ex)
- {
- ex.printStackTrace();
- }
- if (retrieved != null)
- {
- transferReferences(sdataset, dbSources[db], retrieved);
- }
- }
- else
- {
- // make some more strings for use as queries
- for (int i = 0; (seqIndex < dataset.length) && (i < 50); seqIndex++, i++)
- {
- SequenceI sequence = dataset[seqIndex];
- DBRefEntry[] uprefs = jalview.util.DBRefUtils.selectRefs(
- sequence.getDBRef(), new String[]
- { dbSources[db] }); // jalview.datamodel.DBRefSource.UNIPROT
- // });
- // check for existing dbrefs to use
- if (uprefs != null)
- {
- for (int j = 0; j < uprefs.length; j++)
- {
- addSeqId(sequence, uprefs[j].getAccessionId());
- queries
- .addElement(uprefs[j].getAccessionId()
- .toUpperCase());
- }
- }
- else
- {
- // generate queries from sequence ID string
- StringTokenizer st = new StringTokenizer(sequence.getName(),
- "|");
- while (st.hasMoreTokens())
- {
- String token = st.nextToken();
- addSeqId(sequence, token);
- queries.addElement(token.toUpperCase());
- }
- }
- }
- }
- }
- // advance to next database
- db++;
- } // all databases have been queries.
- if (sbuffer.length() > 0)
- {
- output
- .setText("Your sequences have been verified against known sequence databases. Some of the ids have been\n"
- + "altered, most likely the start/end residue will have been updated.\n"
- + "Save your alignment to maintain the updated id.\n\n"
- + sbuffer.toString());
- Desktop.addInternalFrame(output, "Sequence names updated ", 600, 300);
- // The above is the dataset, we must now find out the index
- // of the viewed sequence
-
- }
-
- af.setProgressBar("DBRef search completed", startTime);
- // promptBeforeBlast();
-
- running = false;
-
- }
-
- /**
- * Verify local sequences in seqRefs against the retrieved sequence database
- * records.
- *
- */
- void transferReferences(Vector sdataset, String dbSource,
- AlignmentI retrievedAl) // File
- // file)
- {
-
- if (retrievedAl == null || retrievedAl.getHeight() == 0)
- {
- return;
- }
- SequenceI[] retrieved = retrievedAl.getSequencesArray();
- SequenceI sequence = null;
-
- // Vector entries = new Uniprot().getUniprotEntries(file);
-
- int i, iSize = retrieved.length; // entries == null ? 0 : entries.size();
- // UniprotEntry entry;
- for (i = 0; i < iSize; i++)
- {
- SequenceI entry = retrieved[i]; // (UniprotEntry) entries.elementAt(i);
-
- // Work out which sequences this sequence matches,
- // taking into account all accessionIds and names in the file
- Vector sequenceMatches = new Vector();
- // look for corresponding accession ids
- DBRefEntry[] entryRefs = jalview.util.DBRefUtils.selectRefs(entry
- .getDBRef(), new String[]
- { dbSource });
- for (int j = 0; j < entryRefs.length; j++)
- {
- String accessionId = entryRefs[j].getAccessionId(); // .getAccession().elementAt(j).toString();
- // match up on accessionId
- if (seqRefs.containsKey(accessionId.toUpperCase()))
- {
- Vector seqs = (Vector) seqRefs.get(accessionId);
- for (int jj = 0; jj < seqs.size(); jj++)
- {
- sequence = (SequenceI) seqs.elementAt(jj);
- if (!sequenceMatches.contains(sequence))
- {
- sequenceMatches.addElement(sequence);
- }
- }
- }
- }
- if (sequenceMatches.size()==0)
- {
- // failed to match directly on accessionId==query so just compare all sequences to entry
- Enumeration e = seqRefs.keys();
- while (e.hasMoreElements())
- {
- Vector sqs = (Vector) seqRefs.get(e.nextElement());
- if (sqs!=null && sqs.size()>0)
- {
- Enumeration sqe = sqs.elements();
- while (sqe.hasMoreElements())
- {
- sequenceMatches.addElement(sqe.nextElement());
- }
- }
- }
- }
- // look for corresponding names
- // this is uniprot specific ?
- // could be useful to extend this so we try to find any 'significant'
- // information in common between two sequence objects.
- /*
- * DBRefEntry[] entryRefs =
- * jalview.util.DBRefUtils.selectRefs(entry.getDBRef(), new String[] {
- * dbSource }); for (int j = 0; j < entry.getName().size(); j++) { String
- * name = entry.getName().elementAt(j).toString(); if
- * (seqRefs.containsKey(name)) { Vector seqs = (Vector) seqRefs.get(name);
- * for (int jj = 0; jj < seqs.size(); jj++) { sequence = (SequenceI)
- * seqs.elementAt(jj); if (!sequenceMatches.contains(sequence)) {
- * sequenceMatches.addElement(sequence); } } } }
- */
- // sequenceMatches now contains the set of all sequences associated with
- // the returned db record
- String entrySeq = entry.getSequenceAsString().toUpperCase();
- for (int m = 0; m < sequenceMatches.size(); m++)
- {
- sequence = (SequenceI) sequenceMatches.elementAt(m);
- // only update start and end positions and shift features if there are no existing references
- // TODO: test for legacy where uniprot or EMBL refs exist but no mappings are made (but content matches retrieved set)
- boolean updateRefFrame = sequence.getDBRef()==null || sequence.getDBRef().length==0;
- // verify sequence against the entry sequence
-
- String nonGapped = AlignSeq.extractGaps("-. ",
- sequence.getSequenceAsString()).toUpperCase();
-
- int absStart = entrySeq.indexOf(nonGapped);
- int mapStart = entry.getStart();
- jalview.datamodel.Mapping mp;
-
- if (absStart == -1)
- {
- // Is local sequence contained in dataset sequence?
- absStart = nonGapped.indexOf(entrySeq);
- if (absStart == -1)
- { // verification failed.
- sbuffer.append(sequence.getName()
- + " SEQUENCE NOT %100 MATCH \n");
- continue;
- }
-
- sbuffer.append(sequence.getName() + " HAS " + absStart
- + " PREFIXED RESIDUES COMPARED TO " + dbSource+"\n");
- //
- // + " - ANY SEQUENCE FEATURES"
- // + " HAVE BEEN ADJUSTED ACCORDINGLY \n");
- // absStart = 0;
- // create valid mapping between matching region of local sequence and
- // the mapped sequence
- mp = new Mapping(null, new int[]
- { sequence.getStart()+absStart, sequence.getStart()+absStart+entrySeq.length()-1 }, new int[]
- { entry.getStart(),
- entry.getStart() + entrySeq.length() - 1 }, 1, 1);
- updateRefFrame=false; // mapping is based on current start/end so don't modify start and end
- }
- else
- {
- // update start and end of local sequence to place it in entry's
- // reference frame.
- // apply identity map map from whole of local sequence to matching
- // region of database
- // sequence
- mp = null; // Mapping.getIdentityMap();
- // new Mapping(null,
- // new int[] { absStart+sequence.getStart(),
- // absStart+sequence.getStart()+entrySeq.length()-1},
- // new int[] { entry.getStart(), entry.getEnd() }, 1, 1);
- // relocate local features for updated start
- if (updateRefFrame && sequence.getSequenceFeatures() != null)
- {
- SequenceFeature[] sf = sequence.getSequenceFeatures();
- int start = sequence.getStart();
- int end = sequence.getEnd();
- for (int sfi = 0; sfi < sf.length; sfi++)
- {
- if (sf[sfi].getBegin() >= start && sf[sfi].getEnd() <= end)
- {
- // shift feature along by absstart
- sf[sfi].setBegin(sf[sfi].getBegin() + absStart);
- sf[sfi].setEnd(sf[sfi].getEnd() + absStart);
- }
- }
- }
- }
-
- System.out.println("Adding dbrefs to " + sequence.getName()
- + " from " + dbSource + " sequence : " + entry.getName());
- sequence.transferAnnotation(entry, mp);
- // unknownSequences.remove(sequence);
- int absEnd = absStart + nonGapped.length();
- absStart += 1;
- if (updateRefFrame)
- {
- // finally, update local sequence reference frame if we're allowed
- sequence.setStart(absStart);
- sequence.setEnd(absEnd);
- }
- // and remove it from the rest
- // TODO: decide if we should remove annotated sequence from set
- sdataset.remove(sequence);
- }
- }
- }
-}
+/*
+ * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
+ * Copyright (C) $$Year-Rel$$ The Jalview Authors
+ *
+ * This file is part of Jalview.
+ *
+ * Jalview is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation, either version 3
+ * of the License, or (at your option) any later version.
+ *
+ * Jalview is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty
+ * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
+ * PURPOSE. See the GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Jalview. If not, see .
+ * The Jalview Authors are detailed in the 'AUTHORS' file.
+ */
+package jalview.ws;
+
+import jalview.analysis.AlignSeq;
+import jalview.bin.Cache;
+import jalview.datamodel.AlignmentI;
+import jalview.datamodel.DBRefEntry;
+import jalview.datamodel.DBRefSource;
+import jalview.datamodel.Mapping;
+import jalview.datamodel.SequenceI;
+import jalview.gui.CutAndPasteTransfer;
+import jalview.gui.DasSourceBrowser;
+import jalview.gui.Desktop;
+import jalview.gui.FeatureSettings;
+import jalview.gui.IProgressIndicator;
+import jalview.gui.OOMWarning;
+import jalview.util.DBRefUtils;
+import jalview.util.MessageManager;
+import jalview.ws.dbsources.das.api.jalviewSourceI;
+import jalview.ws.dbsources.das.datamodel.DasSequenceSource;
+import jalview.ws.seqfetcher.DbSourceProxy;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Enumeration;
+import java.util.Hashtable;
+import java.util.List;
+import java.util.StringTokenizer;
+import java.util.Vector;
+
+import uk.ac.ebi.picr.model.UPEntry;
+import uk.ac.ebi.www.picr.AccessionMappingService.AccessionMapperServiceLocator;
+
+/**
+ * Implements a runnable for validating a sequence against external databases
+ * and then propagating references and features onto the sequence(s)
+ *
+ * @author $author$
+ * @version $Revision$
+ */
+public class DBRefFetcher implements Runnable
+{
+ private static final String NEWLINE = System.lineSeparator();
+
+ public interface FetchFinishedListenerI
+ {
+ void finished();
+ }
+
+ SequenceI[] dataset;
+
+ IProgressIndicator progressWindow;
+
+ CutAndPasteTransfer output = new CutAndPasteTransfer();
+
+ boolean running = false;
+
+ /**
+ * picr client instance
+ */
+ uk.ac.ebi.www.picr.AccessionMappingService.AccessionMapperInterface picrClient = null;
+
+ // This will be a collection of Vectors of sequenceI refs.
+ // The key will be the seq name or accession id of the seq
+ Hashtable> seqRefs;
+
+ DbSourceProxy[] dbSources;
+
+ SequenceFetcher sfetcher;
+
+ private List listeners;
+
+ private SequenceI[] alseqs;
+
+ /*
+ * when true - retrieved sequences will be trimmed to cover longest derived
+ * alignment sequence
+ */
+ private boolean trimDsSeqs = true;
+
+ /**
+ * Creates a new DBRefFetcher object and fetches from the currently selected
+ * set of databases, if this is null then it fetches based on feature settings
+ *
+ * @param seqs
+ * fetch references for these SequenceI array
+ * @param progressIndicatorFrame
+ * the frame for progress bar monitoring
+ * @param sources
+ * array of DbSourceProxy to query references form
+ * @param featureSettings
+ * FeatureSettings to get alternative DbSourceProxy from
+ * @param isNucleotide
+ * indicates if the array of SequenceI are Nucleotides or not
+ */
+ public DBRefFetcher(SequenceI[] seqs,
+ IProgressIndicator progressIndicatorFrame,
+ DbSourceProxy[] sources, FeatureSettings featureSettings,
+ boolean isNucleotide)
+ {
+ listeners = new ArrayList();
+ this.progressWindow = progressIndicatorFrame;
+ alseqs = new SequenceI[seqs.length];
+ SequenceI[] ds = new SequenceI[seqs.length];
+ for (int i = 0; i < seqs.length; i++)
+ {
+ alseqs[i] = seqs[i];
+ if (seqs[i].getDatasetSequence() != null)
+ {
+ ds[i] = seqs[i].getDatasetSequence();
+ }
+ else
+ {
+ ds[i] = seqs[i];
+ }
+ }
+ this.dataset = ds;
+ // TODO Jalview 2.5 lots of this code should be in the gui package!
+ sfetcher = jalview.gui.SequenceFetcher
+ .getSequenceFetcherSingleton(progressIndicatorFrame);
+ // set default behaviour for transferring excess sequence data to the
+ // dataset
+ trimDsSeqs = Cache.getDefault("TRIM_FETCHED_DATASET_SEQS", true);
+ if (sources == null)
+ {
+ setDatabaseSources(featureSettings, isNucleotide);
+ }
+ else
+ {
+ // we assume the caller knows what they're doing and ensured that all the
+ // db source names are valid
+ dbSources = sources;
+ }
+ }
+
+ /**
+ * Helper method to configure the list of database sources to query
+ *
+ * @param featureSettings
+ * @param forNucleotide
+ */
+ void setDatabaseSources(FeatureSettings featureSettings,
+ boolean forNucleotide)
+ {
+ // af.featureSettings_actionPerformed(null);
+ String[] defdb = null;
+ List selsources = new ArrayList();
+ Vector dasselsrc = (featureSettings != null) ? featureSettings
+ .getSelectedSources() : new DasSourceBrowser()
+ .getSelectedSources();
+
+ for (jalviewSourceI src : dasselsrc)
+ {
+ List sp = src.getSequenceSourceProxies();
+ if (sp != null)
+ {
+ selsources.addAll(sp);
+ if (sp.size() > 1)
+ {
+ Cache.log.debug("Added many Db Sources for :" + src.getTitle());
+ }
+ }
+ }
+ // select appropriate databases based on alignFrame context.
+ if (forNucleotide)
+ {
+ defdb = DBRefSource.DNACODINGDBS;
+ }
+ else
+ {
+ defdb = DBRefSource.PROTEINDBS;
+ }
+ List srces = new ArrayList();
+ for (String ddb : defdb)
+ {
+ List srcesfordb = sfetcher.getSourceProxy(ddb);
+ if (srcesfordb != null)
+ {
+ for (DbSourceProxy src : srcesfordb)
+ {
+ if (!srces.contains(src))
+ {
+ srces.addAll(srcesfordb);
+ }
+ }
+ }
+ }
+ // append the PDB data source, since it is 'special', catering for both
+ // nucleotide and protein
+ // srces.addAll(sfetcher.getSourceProxy(DBRefSource.PDB));
+
+ srces.addAll(selsources);
+ dbSources = srces.toArray(new DbSourceProxy[srces.size()]);
+ }
+
+ /**
+ * Constructor with only sequences provided
+ *
+ * @param sequences
+ */
+ public DBRefFetcher(SequenceI[] sequences)
+ {
+ this(sequences, null, null, null, false);
+ }
+
+ /**
+ * Add a listener to be notified when sequence fetching is complete
+ *
+ * @param l
+ */
+ public void addListener(FetchFinishedListenerI l)
+ {
+ listeners.add(l);
+ }
+
+ /**
+ * retrieve all the das sequence sources and add them to the list of db
+ * sources to retrieve from
+ */
+ public void appendAllDasSources()
+ {
+ if (dbSources == null)
+ {
+ dbSources = new DbSourceProxy[0];
+ }
+ // append additional sources
+ DbSourceProxy[] otherdb = sfetcher
+ .getDbSourceProxyInstances(DasSequenceSource.class);
+ if (otherdb != null && otherdb.length > 0)
+ {
+ DbSourceProxy[] newsrc = new DbSourceProxy[dbSources.length
+ + otherdb.length];
+ System.arraycopy(dbSources, 0, newsrc, 0, dbSources.length);
+ System.arraycopy(otherdb, 0, newsrc, dbSources.length, otherdb.length);
+ dbSources = newsrc;
+ }
+ }
+
+ /**
+ * start the fetcher thread
+ *
+ * @param waitTillFinished
+ * true to block until the fetcher has finished
+ */
+ public void fetchDBRefs(boolean waitTillFinished)
+ {
+ // TODO can we not simply write
+ // if (waitTillFinished) { run(); } else { new Thread(this).start(); }
+
+ Thread thread = new Thread(this);
+ thread.start();
+ running = true;
+
+ if (waitTillFinished)
+ {
+ while (running)
+ {
+ try
+ {
+ Thread.sleep(500);
+ } catch (Exception ex)
+ {
+ }
+ }
+ }
+ }
+
+ /**
+ * The sequence will be added to a vector of sequences belonging to key which
+ * could be either seq name or dbref id
+ *
+ * @param seq
+ * SequenceI
+ * @param key
+ * String
+ */
+ void addSeqId(SequenceI seq, String key)
+ {
+ key = key.toUpperCase();
+
+ Vector seqs;
+ if (seqRefs.containsKey(key))
+ {
+ seqs = seqRefs.get(key);
+
+ if (seqs != null && !seqs.contains(seq))
+ {
+ seqs.addElement(seq);
+ }
+ else if (seqs == null)
+ {
+ seqs = new Vector();
+ seqs.addElement(seq);
+ }
+
+ }
+ else
+ {
+ seqs = new Vector();
+ seqs.addElement(seq);
+ }
+
+ seqRefs.put(key, seqs);
+ }
+
+ /**
+ * DOCUMENT ME!
+ */
+ @Override
+ public void run()
+ {
+ if (dbSources == null)
+ {
+ throw new Error(
+ MessageManager
+ .getString("error.implementation_error_must_init_dbsources"));
+ }
+ running = true;
+ long startTime = System.currentTimeMillis();
+ if (progressWindow != null)
+ {
+ progressWindow.setProgressBar(
+ MessageManager.getString("status.fetching_db_refs"),
+ startTime);
+ }
+ try
+ {
+ if (Cache.getDefault("DBREFFETCH_USEPICR", false))
+ {
+ picrClient = new AccessionMapperServiceLocator()
+ .getAccessionMapperPort();
+ }
+ } catch (Exception e)
+ {
+ System.err.println("Couldn't locate PICR service instance.\n");
+ e.printStackTrace();
+ }
+
+ Vector sdataset = new Vector(
+ Arrays.asList(dataset));
+ List warningMessages = new ArrayList();
+
+ int db = 0;
+ while (sdataset.size() > 0 && db < dbSources.length)
+ {
+ int maxqlen = 1; // default number of queries made at one time
+ System.out.println("Verifying against " + dbSources[db].getDbName());
+
+ // iterate through db for each remaining un-verified sequence
+ SequenceI[] currSeqs = new SequenceI[sdataset.size()];
+ sdataset.copyInto(currSeqs);// seqs that are to be validated against
+ // dbSources[db]
+ Vector queries = new Vector(); // generated queries curSeq
+ seqRefs = new Hashtable>();
+
+ int seqIndex = 0;
+
+ DbSourceProxy dbsource = dbSources[db];
+ // for moment, we dumbly iterate over all retrieval sources for a
+ // particular database
+ // TODO: introduce multithread multisource queries and logic to remove a
+ // query from other sources if any source for a database returns a
+ // record
+ maxqlen = dbsource.getMaximumQueryCount();
+
+ while (queries.size() > 0 || seqIndex < currSeqs.length)
+ {
+ if (queries.size() > 0)
+ {
+ // Still queries to make for current seqIndex
+ StringBuffer queryString = new StringBuffer("");
+ int numq = 0;
+ int nqSize = (maxqlen > queries.size()) ? queries.size()
+ : maxqlen;
+
+ while (queries.size() > 0 && numq < nqSize)
+ {
+ String query = queries.elementAt(0);
+ if (dbsource.isValidReference(query))
+ {
+ queryString.append((numq == 0) ? "" : dbsource
+ .getAccessionSeparator());
+ queryString.append(query);
+ numq++;
+ }
+ // remove the extracted query string
+ queries.removeElementAt(0);
+ }
+ // make the queries and process the response
+ AlignmentI retrieved = null;
+ try
+ {
+ if (Cache.log.isDebugEnabled())
+ {
+ Cache.log.debug("Querying " + dbsource.getDbName()
+ + " with : '" + queryString.toString() + "'");
+ }
+ retrieved = dbsource.getSequenceRecords(queryString.toString());
+ } catch (Exception ex)
+ {
+ ex.printStackTrace();
+ } catch (OutOfMemoryError err)
+ {
+ new OOMWarning("retrieving database references ("
+ + queryString.toString() + ")", err);
+ }
+ if (retrieved != null)
+ {
+ transferReferences(sdataset, dbsource.getDbSource(), retrieved,
+ trimDsSeqs, warningMessages);
+ }
+ }
+ else
+ {
+ // make some more strings for use as queries
+ for (int i = 0; (seqIndex < dataset.length) && (i < 50); seqIndex++, i++)
+ {
+ SequenceI sequence = dataset[seqIndex];
+ DBRefEntry[] uprefs = DBRefUtils.selectRefs(
+ sequence.getDBRefs(),
+ new String[] { dbsource.getDbSource() }); // jalview.datamodel.DBRefSource.UNIPROT
+ // });
+ // check for existing dbrefs to use
+ if (uprefs != null && uprefs.length > 0)
+ {
+ for (int j = 0; j < uprefs.length; j++)
+ {
+ addSeqId(sequence, uprefs[j].getAccessionId());
+ queries.addElement(uprefs[j].getAccessionId().toUpperCase());
+ }
+ }
+ else
+ {
+ // generate queries from sequence ID string
+ StringTokenizer st = new StringTokenizer(sequence.getName(),
+ "|");
+ while (st.hasMoreTokens())
+ {
+ String token = st.nextToken();
+ UPEntry[] presp = null;
+ if (picrClient != null)
+ {
+ // resolve the string against PICR to recover valid IDs
+ try
+ {
+ presp = picrClient
+ .getUPIForAccession(token, null,
+ picrClient.getMappedDatabaseNames(),
+ null, true);
+ } catch (Exception e)
+ {
+ System.err.println("Exception with Picr for '" + token
+ + "'\n");
+ e.printStackTrace();
+ }
+ }
+ if (presp != null && presp.length > 0)
+ {
+ for (int id = 0; id < presp.length; id++)
+ {
+ // construct sequences from response if sequences are
+ // present, and do a transferReferences
+ // otherwise transfer non sequence x-references directly.
+ }
+ System.out
+ .println("Validated ID against PICR... (for what its worth):"
+ + token);
+ addSeqId(sequence, token);
+ queries.addElement(token.toUpperCase());
+ }
+ else
+ {
+ // if ()
+ // System.out.println("Not querying source with token="+token+"\n");
+ addSeqId(sequence, token);
+ queries.addElement(token.toUpperCase());
+ }
+ }
+ }
+ }
+ }
+ }
+ // advance to next database
+ db++;
+ } // all databases have been queried
+ if (!warningMessages.isEmpty())
+ {
+ StringBuilder sb = new StringBuilder(warningMessages.size() * 30);
+ sb.append(MessageManager
+ .getString("label.your_sequences_have_been_verified"));
+ for (String msg : warningMessages)
+ {
+ sb.append(msg).append(NEWLINE);
+ }
+ output.setText(sb.toString());
+
+ Desktop.addInternalFrame(output,
+ MessageManager.getString("label.sequences_updated"), 600, 300);
+ // The above is the dataset, we must now find out the index
+ // of the viewed sequence
+
+ }
+ if (progressWindow != null)
+ {
+ progressWindow.setProgressBar(
+ MessageManager.getString("label.dbref_search_completed"),
+ startTime);
+ }
+
+ for (FetchFinishedListenerI listener : listeners)
+ {
+ listener.finished();
+ }
+ running = false;
+ }
+
+ /**
+ * Verify local sequences in seqRefs against the retrieved sequence database
+ * records. Returns true if any sequence was modified as a result (start/end
+ * changed and/or sequence enlarged), else false.
+ *
+ * @param sdataset
+ * dataset sequences we are retrieving for
+ * @param dbSource
+ * database source we are retrieving from
+ * @param retrievedAl
+ * retrieved sequences as alignment
+ * @param trimDatasetSeqs
+ * if true, sequences will not be enlarged to match longer retrieved
+ * sequences, only their start/end adjusted
+ * @param warningMessages
+ * a list of messages to add to
+ */
+ boolean transferReferences(Vector sdataset, String dbSource,
+ AlignmentI retrievedAl, boolean trimDatasetSeqs,
+ List warningMessages)
+ {
+ // System.out.println("trimming ? " + trimDatasetSeqs);
+ if (retrievedAl == null || retrievedAl.getHeight() == 0)
+ {
+ return false;
+ }
+
+ boolean modified = false;
+ SequenceI[] retrieved = recoverDbSequences(retrievedAl
+ .getSequencesArray());
+ SequenceI sequence = null;
+
+ for (SequenceI retrievedSeq : retrieved)
+ {
+ // Work out which sequences this sequence matches,
+ // taking into account all accessionIds and names in the file
+ Vector sequenceMatches = new Vector();
+ // look for corresponding accession ids
+ DBRefEntry[] entryRefs = DBRefUtils.selectRefs(
+ retrievedSeq.getDBRefs(), new String[] { dbSource });
+ if (entryRefs == null)
+ {
+ System.err
+ .println("Dud dbSource string ? no entryrefs selected for "
+ + dbSource + " on " + retrievedSeq.getName());
+ continue;
+ }
+ for (int j = 0; j < entryRefs.length; j++)
+ {
+ String accessionId = entryRefs[j].getAccessionId();
+ // match up on accessionId
+ if (seqRefs.containsKey(accessionId.toUpperCase()))
+ {
+ Vector seqs = seqRefs.get(accessionId);
+ for (int jj = 0; jj < seqs.size(); jj++)
+ {
+ sequence = seqs.elementAt(jj);
+ if (!sequenceMatches.contains(sequence))
+ {
+ sequenceMatches.addElement(sequence);
+ }
+ }
+ }
+ }
+ if (sequenceMatches.isEmpty())
+ {
+ // failed to match directly on accessionId==query so just compare all
+ // sequences to entry
+ Enumeration e = seqRefs.keys();
+ while (e.hasMoreElements())
+ {
+ Vector sqs = seqRefs.get(e.nextElement());
+ if (sqs != null && sqs.size() > 0)
+ {
+ Enumeration sqe = sqs.elements();
+ while (sqe.hasMoreElements())
+ {
+ sequenceMatches.addElement(sqe.nextElement());
+ }
+ }
+ }
+ }
+ // look for corresponding names
+ // this is uniprot specific ?
+ // could be useful to extend this so we try to find any 'significant'
+ // information in common between two sequence objects.
+ /*
+ * DBRefEntry[] entryRefs =
+ * jalview.util.DBRefUtils.selectRefs(entry.getDBRef(), new String[] {
+ * dbSource }); for (int j = 0; j < entry.getName().size(); j++) { String
+ * name = entry.getName().elementAt(j).toString(); if
+ * (seqRefs.containsKey(name)) { Vector seqs = (Vector) seqRefs.get(name);
+ * for (int jj = 0; jj < seqs.size(); jj++) { sequence = (SequenceI)
+ * seqs.elementAt(jj); if (!sequenceMatches.contains(sequence)) {
+ * sequenceMatches.addElement(sequence); } } } }
+ */
+ // sequenceMatches now contains the set of all sequences associated with
+ // the returned db record
+ final String retrievedSeqString = retrievedSeq.getSequenceAsString();
+ String entrySeq = retrievedSeqString.toUpperCase();
+ for (int m = 0; m < sequenceMatches.size(); m++)
+ {
+ sequence = sequenceMatches.elementAt(m);
+ // only update start and end positions and shift features if there are
+ // no existing references
+ // TODO: test for legacy where uniprot or EMBL refs exist but no
+ // mappings are made (but content matches retrieved set)
+ boolean updateRefFrame = sequence.getDBRefs() == null
+ || sequence.getDBRefs().length == 0;
+ // TODO:
+ // verify sequence against the entry sequence
+
+ Mapping mp;
+ final int sequenceStart = sequence.getStart();
+
+ boolean remoteEnclosesLocal = false;
+ String nonGapped = AlignSeq.extractGaps("-. ",
+ sequence.getSequenceAsString()).toUpperCase();
+ int absStart = entrySeq.indexOf(nonGapped);
+ if (absStart == -1)
+ {
+ // couldn't find local sequence in sequence from database, so check if
+ // the database sequence is a subsequence of local sequence
+ absStart = nonGapped.indexOf(entrySeq);
+ if (absStart == -1)
+ {
+ // verification failed. couldn't find any relationship between
+ // entrySeq and local sequence
+ // messages suppressed as many-to-many matches are confusing
+ // String msg = sequence.getName()
+ // + " Sequence not 100% match with "
+ // + retrievedSeq.getName();
+ // addWarningMessage(warningMessages, msg);
+ continue;
+ }
+ /*
+ * retrieved sequence is a proper subsequence of local sequence
+ */
+ String msg = sequence.getName() + " has " + absStart
+ + " prefixed residues compared to "
+ + retrievedSeq.getName();
+ addWarningMessage(warningMessages, msg);
+
+ /*
+ * So create a mapping to the external entry from the matching region of
+ * the local sequence, and leave local start/end untouched.
+ */
+ mp = new Mapping(null, new int[] { sequenceStart + absStart,
+ sequenceStart + absStart + entrySeq.length() - 1 }, new int[]
+ { retrievedSeq.getStart(),
+ retrievedSeq.getStart() + entrySeq.length() - 1 }, 1, 1);
+ updateRefFrame = false;
+ }
+ else
+ {
+ /*
+ * local sequence is a subsequence of (or matches) retrieved sequence
+ */
+ remoteEnclosesLocal = true;
+ mp = null;
+
+ if (updateRefFrame)
+ {
+ /*
+ * relocate existing sequence features by offset
+ */
+ int startShift = absStart - sequenceStart + 1;
+ if (startShift != 0)
+ {
+ modified |= sequence.getFeatures().shiftFeatures(startShift);
+ }
+ }
+ }
+
+ System.out.println("Adding dbrefs to " + sequence.getName()
+ + " from " + dbSource + " sequence : "
+ + retrievedSeq.getName());
+ sequence.transferAnnotation(retrievedSeq, mp);
+
+ absStart += retrievedSeq.getStart();
+ int absEnd = absStart + nonGapped.length() - 1;
+ if (!trimDatasetSeqs)
+ {
+ /*
+ * update start position and/or expand to longer retrieved sequence
+ */
+ if (!retrievedSeqString.equals(sequence.getSequenceAsString())
+ && remoteEnclosesLocal)
+ {
+ sequence.setSequence(retrievedSeqString);
+ modified = true;
+ addWarningMessage(warningMessages,
+ "Sequence for " + sequence.getName()
+ + " expanded from " + retrievedSeq.getName());
+ }
+ if (sequence.getStart() != retrievedSeq.getStart())
+ {
+ sequence.setStart(retrievedSeq.getStart());
+ modified = true;
+ if (absStart != sequenceStart)
+ {
+ addWarningMessage(warningMessages, "Start/end position for "
+ + sequence.getName() + " updated from "
+ + retrievedSeq.getName());
+ }
+ }
+ }
+ if (updateRefFrame)
+ {
+ // finally, update local sequence reference frame if we're allowed
+ if (trimDatasetSeqs)
+ {
+ // just fix start/end
+ if (sequence.getStart() != absStart
+ || sequence.getEnd() != absEnd)
+ {
+ sequence.setStart(absStart);
+ sequence.setEnd(absEnd);
+ modified = true;
+ addWarningMessage(warningMessages, "Start/end for "
+ + sequence.getName() + " updated from "
+ + retrievedSeq.getName());
+ }
+ }
+ // search for alignment sequences to update coordinate frame for
+ for (int alsq = 0; alsq < alseqs.length; alsq++)
+ {
+ if (alseqs[alsq].getDatasetSequence() == sequence)
+ {
+ String ngAlsq = AlignSeq.extractGaps("-. ",
+ alseqs[alsq].getSequenceAsString()).toUpperCase();
+ int oldstrt = alseqs[alsq].getStart();
+ alseqs[alsq].setStart(sequence.getSequenceAsString()
+ .toUpperCase().indexOf(ngAlsq)
+ + sequence.getStart());
+ if (oldstrt != alseqs[alsq].getStart())
+ {
+ alseqs[alsq].setEnd(ngAlsq.length()
+ + alseqs[alsq].getStart() - 1);
+ modified = true;
+ }
+ }
+ }
+ // TODO: search for all other references to this dataset sequence, and
+ // update start/end
+ // TODO: update all AlCodonMappings which involve this alignment
+ // sequence (e.g. Q30167 cdna translation from exon2 product (vamsas
+ // demo)
+ }
+ // and remove it from the rest
+ // TODO: decide if we should remove annotated sequence from set
+ sdataset.remove(sequence);
+ // TODO: should we make a note of sequences that have received new DB
+ // ids, so we can query all enabled DAS servers for them ?
+ }
+ }
+ return modified;
+ }
+
+ /**
+ * Adds the message to the list unless it already contains it
+ *
+ * @param messageList
+ * @param msg
+ */
+ void addWarningMessage(List messageList, String msg)
+ {
+ if (!messageList.contains(msg))
+ {
+ messageList.add(msg);
+ }
+ }
+
+ /**
+ * loop thru and collect additional sequences in Map.
+ *
+ * @param sequencesArray
+ * @return
+ */
+ private SequenceI[] recoverDbSequences(SequenceI[] sequencesArray)
+ {
+ Vector nseq = new Vector();
+ for (int i = 0; sequencesArray != null && i < sequencesArray.length; i++)
+ {
+ nseq.addElement(sequencesArray[i]);
+ DBRefEntry[] dbr = sequencesArray[i].getDBRefs();
+ Mapping map = null;
+ for (int r = 0; (dbr != null) && r < dbr.length; r++)
+ {
+ if ((map = dbr[r].getMap()) != null)
+ {
+ if (map.getTo() != null && !nseq.contains(map.getTo()))
+ {
+ nseq.addElement(map.getTo());
+ }
+ }
+ }
+ }
+ if (nseq.size() > 0)
+ {
+ sequencesArray = new SequenceI[nseq.size()];
+ nseq.toArray(sequencesArray);
+ }
+ return sequencesArray;
+ }
+}