X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2FMCview%2FPDBfile.java;h=58f8ed5b7a35f5b7242e24e8e1602bb4d4c3c860;hb=17e77c3f2949a0729322b4a8d907f3f34b6a9914;hp=c671ec2bb67085691c306d579fb2c009ff9aa17b;hpb=f15837f03963ef2d3e53d890cdb4f04f75cc1728;p=jalview.git diff --git a/src/MCview/PDBfile.java b/src/MCview/PDBfile.java index c671ec2..58f8ed5 100755 --- a/src/MCview/PDBfile.java +++ b/src/MCview/PDBfile.java @@ -1,50 +1,97 @@ /* - * Jalview - A Sequence Alignment Editor and Viewer (Development Version 2.4.1) - * Copyright (C) 2009 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle + * Jalview - A Sequence Alignment Editor and Viewer (Version 2.9) + * Copyright (C) 2015 The Jalview Authors * - * 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 file is part of Jalview. * - * 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. + * 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 this program; if not, write to the Free Software - * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA + * along with Jalview. If not, see . + * The Jalview Authors are detailed in the 'AUTHORS' file. */ package MCview; -import java.io.*; -import java.util.*; - -import java.awt.*; - -import jalview.datamodel.*; +import jalview.analysis.AlignSeq; +import jalview.datamodel.Alignment; +import jalview.datamodel.AlignmentAnnotation; +import jalview.datamodel.AlignmentI; +import jalview.datamodel.PDBEntry; +import jalview.datamodel.SequenceI; import jalview.io.FileParse; +import jalview.util.MessageManager; + +import java.awt.Color; +import java.io.IOException; +import java.lang.reflect.Constructor; +import java.util.ArrayList; +import java.util.Hashtable; +import java.util.List; +import java.util.Vector; public class PDBfile extends jalview.io.AlignFile { - public Vector chains; + private static String CALC_ID_PREFIX = "JalviewPDB"; + + public Vector chains; public String id; /** - * set to true to add chain alignment annotation as visible annotation. + * set to true to add derived sequence annotations (temp factor read from + * file, or computed secondary structure) to the alignment + */ + private boolean visibleChainAnnotation = false; + + /* + * Set true to predict secondary structure (using JMol for protein, Annotate3D + * for RNA) + */ + private boolean predictSecondaryStructure = true; + + /* + * Set true (with predictSecondaryStructure=true) to predict secondary + * structure using an external service (currently Annotate3D for RNA only) */ - boolean VisibleChainAnnotation = false; + private boolean externalSecondaryStructure = false; - public PDBfile(String inFile, String inType) throws IOException + public PDBfile(boolean addAlignmentAnnotations, + boolean predictSecondaryStructure, boolean externalSecStr) { - super(inFile, inType); + super(); + this.visibleChainAnnotation = addAlignmentAnnotations; + this.predictSecondaryStructure = predictSecondaryStructure; + this.externalSecondaryStructure = externalSecStr; } - public PDBfile(FileParse source) throws IOException + public PDBfile(boolean addAlignmentAnnotations, + boolean predictSecondaryStructure, boolean externalSecStr, + String file, String protocol) throws IOException { - super(source); + super(false, file, protocol); + this.visibleChainAnnotation = addAlignmentAnnotations; + this.predictSecondaryStructure = predictSecondaryStructure; + this.externalSecondaryStructure = externalSecStr; + doParse(); + } + + public PDBfile(boolean addAlignmentAnnotations, + boolean predictSecondaryStructure, boolean externalSecStr, + FileParse source) throws IOException + { + super(false, source); + this.visibleChainAnnotation = addAlignmentAnnotations; + this.predictSecondaryStructure = predictSecondaryStructure; + this.externalSecondaryStructure = externalSecStr; + doParse(); } public String print() @@ -54,28 +101,40 @@ public class PDBfile extends jalview.io.AlignFile public void parse() throws IOException { - // TODO set the filename sensibly - id = (inFile == null) ? "PDBFILE" : inFile.getName(); - try - { - chains = new Vector(); + // TODO set the filename sensibly - try using data source name. + id = safeName(getDataName()); - PDBChain tmpchain; - String line; - boolean modelFlag = false; - boolean terFlag = false; + chains = new Vector(); + List rna = new ArrayList(); + List prot = new ArrayList(); + PDBChain tmpchain; + String line = null; + boolean modelFlag = false; + boolean terFlag = false; + String lastID = ""; - int index = 0; + int indexx = 0; + String atomnam = null; + try + { while ((line = nextLine()) != null) { if (line.indexOf("HEADER") == 0) { - if (line.length()>62) + if (line.length() > 62) { - if (line.length()>67) { - id = line.substring(62, 67).trim(); - } else { - id=line.substring(62).trim(); + String tid; + if (line.length() > 67) + { + tid = line.substring(62, 67).trim(); + } + else + { + tid = line.substring(62).trim(); + } + if (tid.length() > 0) + { + id = tid; } continue; } @@ -105,7 +164,8 @@ public class PDBfile extends jalview.io.AlignFile terFlag = false; // Jalview is only interested in CA bonds???? - if (!line.substring(12, 15).trim().equals("CA")) + atomnam = line.substring(12, 15).trim(); + if (!atomnam.equals("CA") && !atomnam.equals("P")) { continue; } @@ -114,6 +174,11 @@ public class PDBfile extends jalview.io.AlignFile tmpchain = findChain(tmpatom.chain); if (tmpchain != null) { + if (tmpatom.resNumIns.trim().equals(lastID)) + { + // phosphorylated protein - seen both CA and P.. + continue; + } tmpchain.atoms.addElement(tmpatom); } else @@ -122,6 +187,7 @@ public class PDBfile extends jalview.io.AlignFile chains.addElement(tmpchain); tmpchain.atoms.addElement(tmpatom); } + lastID = tmpatom.resNumIns.trim(); } index++; } @@ -133,43 +199,304 @@ public class PDBfile extends jalview.io.AlignFile { id = inFile.getName(); } - for (int i = 0; i < chains.size(); i++) + for (PDBChain chain : chains) { - SequenceI dataset = ((PDBChain) chains.elementAt(i)).sequence; - dataset.setName(id + "|" + dataset.getName()); - PDBEntry entry = new PDBEntry(); - entry.setId(id); - if (inFile != null) + SequenceI chainseq = postProcessChain(chain); + if (isRNA(chainseq)) { - entry.setFile(inFile.getAbsolutePath()); + rna.add(chainseq); } - dataset.addPDBId(entry); - SequenceI chainseq = dataset.deriveSequence(); // PDBChain objects - // maintain reference to - // dataset - seqs.addElement(chainseq); - AlignmentAnnotation[] chainannot = chainseq.getAnnotation(); - if (chainannot != null) + else { - for (int ai = 0; ai < chainannot.length; ai++) - { - chainannot[ai].visible = VisibleChainAnnotation; - annotations.addElement(chainannot[ai]); - } + prot.add(chainseq); } } + if (predictSecondaryStructure) + { + predictSecondaryStructure(rna, prot); + } } catch (OutOfMemoryError er) { System.out.println("OUT OF MEMORY LOADING PDB FILE"); - throw new IOException("Out of memory loading PDB File"); + throw new IOException( + MessageManager + .getString("exception.outofmemory_loading_pdb_file")); + } catch (NumberFormatException ex) + { + if (line != null) + { + System.err.println("Couldn't read number from line:"); + System.err.println(line); + } + } + markCalcIds(); + } + + /** + * Predict secondary structure for RNA and/or protein sequences and add as + * annotations + * + * @param rnaSequences + * @param proteinSequences + */ + protected void predictSecondaryStructure(List rnaSequences, + List proteinSequences) + { + /* + * Currently using Annotate3D for RNA, but only if the 'use external + * prediction' flag is set + */ + if (externalSecondaryStructure && rnaSequences.size() > 0) + { + try + { + processPdbFileWithAnnotate3d(rnaSequences); + } catch (Exception x) + { + System.err.println("Exceptions when dealing with RNA in pdb file"); + x.printStackTrace(); + + } + } + + /* + * Currently using JMol PDB parser for peptide + */ + if (proteinSequences.size() > 0) + { + try + { + processPdbFileWithJmol(proteinSequences); + } catch (Exception x) + { + System.err + .println("Exceptions from Jmol when processing data in pdb file"); + x.printStackTrace(); + } + } + } + + /** + * Process a parsed chain to construct and return a Sequence, and add it to + * the list of sequences parsed. + * + * @param chain + * @return + */ + protected SequenceI postProcessChain(PDBChain chain) + { + SequenceI dataset = chain.sequence; + dataset.setName(id + "|" + dataset.getName()); + PDBEntry entry = new PDBEntry(); + entry.setId(id); + entry.setType(PDBEntry.Type.PDB); + entry.setProperty(new Hashtable()); + if (chain.id != null) + { + // entry.getProperty().put("CHAIN", chains.elementAt(i).id); + entry.setChainCode(String.valueOf(chain.id)); + } + if (inFile != null) + { + entry.setFile(inFile.getAbsolutePath()); + } + else + { + // TODO: decide if we should dump the datasource to disk + entry.setFile(getDataName()); } + dataset.addPDBId(entry); + // PDBChain objects maintain reference to dataset + SequenceI chainseq = dataset.deriveSequence(); + seqs.addElement(chainseq); + + AlignmentAnnotation[] chainannot = chainseq.getAnnotation(); + + if (chainannot != null && visibleChainAnnotation) + { + for (int ai = 0; ai < chainannot.length; ai++) + { + chainannot[ai].visible = visibleChainAnnotation; + annotations.addElement(chainannot[ai]); + } + } + return chainseq; + } + + public static boolean isCalcIdHandled(String calcId) + { + return calcId != null && (CALC_ID_PREFIX.equals(calcId)); + } + + public static boolean isCalcIdForFile(AlignmentAnnotation alan, + String pdbFile) + { + return alan.getCalcId() != null + && CALC_ID_PREFIX.equals(alan.getCalcId()) + && pdbFile.equals(alan.getProperty("PDBID")); + } + + public static String relocateCalcId(String calcId, + Hashtable alreadyLoadedPDB) throws Exception + { + int s = CALC_ID_PREFIX.length(), end = calcId + .indexOf(CALC_ID_PREFIX, s); + String between = calcId.substring(s, end - 1); + return CALC_ID_PREFIX + alreadyLoadedPDB.get(between) + ":" + + calcId.substring(end); + } + + private void markCalcIds() + { + for (SequenceI sq : seqs) + { + if (sq.getAnnotation() != null) + { + for (AlignmentAnnotation aa : sq.getAnnotation()) + { + String oldId = aa.getCalcId(); + if (oldId == null) + { + oldId = ""; + } + aa.setCalcId(CALC_ID_PREFIX); + aa.setProperty("PDBID", id); + aa.setProperty("oldCalcId", oldId); + } + } + } + } + + private void processPdbFileWithJmol(List prot) + throws Exception + { + try + { + Class cl = Class.forName("jalview.ext.jmol.PDBFileWithJmol"); + if (cl != null) + { + final Constructor constructor = cl + .getConstructor(new Class[] { FileParse.class }); + final Object[] args = new Object[] { new FileParse(getDataName(), + type) }; + Object jmf = constructor.newInstance(args); + AlignmentI al = new Alignment((SequenceI[]) cl.getMethod( + "getSeqsAsArray", new Class[] {}).invoke(jmf)); + cl.getMethod("addAnnotations", new Class[] { AlignmentI.class }) + .invoke(jmf, al); + for (SequenceI sq : al.getSequences()) + { + if (sq.getDatasetSequence() != null) + { + sq.getDatasetSequence().getAllPDBEntries().clear(); + } + else + { + sq.getAllPDBEntries().clear(); + } + } + replaceAndUpdateChains(prot, al, AlignSeq.PEP, false); + } + } catch (ClassNotFoundException q) + { + } + } + + private void replaceAndUpdateChains(List prot, AlignmentI al, + String pep, boolean b) + { + List> replaced = AlignSeq + .replaceMatchingSeqsWith(seqs, annotations, prot, al, pep, + false); + for (PDBChain ch : chains) + { + int p = 0; + for (SequenceI sq : (List) replaced.get(0)) + { + p++; + if (sq == ch.sequence || sq.getDatasetSequence() == ch.sequence) + { + p = -p; + break; + } + } + if (p < 0) + { + p = -p - 1; + // set shadow entry for chains + ch.shadow = (SequenceI) replaced.get(1).get(p); + ch.shadowMap = ((AlignSeq) replaced.get(2).get(p)) + .getMappingFromS1(false); + } + } + } + + private void processPdbFileWithAnnotate3d(List rna) + throws Exception + { + // System.out.println("this is a PDB format and RNA sequence"); + // note: we use reflection here so that the applet can compile and run + // without the HTTPClient bits and pieces needed for accessing Annotate3D + // web service + try + { + Class cl = Class.forName("jalview.ws.jws1.Annotate3D"); + if (cl != null) + { + // TODO: use the PDB ID of the structure if one is available, to save + // bandwidth and avoid uploading the whole structure to the service + Object annotate3d = cl.getConstructor(new Class[] {}).newInstance( + new Object[] {}); + AlignmentI al = ((AlignmentI) cl.getMethod("getRNAMLFor", + new Class[] { FileParse.class }).invoke(annotate3d, + new Object[] { new FileParse(getDataName(), type) })); + for (SequenceI sq : al.getSequences()) + { + if (sq.getDatasetSequence() != null) + { + if (sq.getDatasetSequence().getAllPDBEntries() != null) + { + sq.getDatasetSequence().getAllPDBEntries().clear(); + } + } + else + { + if (sq.getAllPDBEntries() != null) + { + sq.getAllPDBEntries().clear(); + } + } + } + replaceAndUpdateChains(rna, al, AlignSeq.DNA, false); + } + } catch (ClassNotFoundException x) + { + // ignore classnotfounds - occurs in applet + } + ; + } + + /** + * make a friendly ID string. + * + * @param dataName + * @return truncated dataName to after last '/' + */ + private String safeName(String dataName) + { + int p = 0; + while ((p = dataName.indexOf("/")) > -1 && p < dataName.length()) + { + dataName = dataName.substring(p + 1); + } + return dataName; } public void makeResidueList() { for (int i = 0; i < chains.size(); i++) { - ((PDBChain) chains.elementAt(i)).makeResidueList(); + chains.elementAt(i).makeResidueList(visibleChainAnnotation); } } @@ -177,7 +504,7 @@ public class PDBfile extends jalview.io.AlignFile { for (int i = 0; i < chains.size(); i++) { - ((PDBChain) chains.elementAt(i)).makeCaBondList(); + chains.elementAt(i).makeCaBondList(); } } @@ -185,9 +512,9 @@ public class PDBfile extends jalview.io.AlignFile { for (int i = 0; i < chains.size(); i++) { - if (((PDBChain) chains.elementAt(i)).id.equals(id)) + if (chains.elementAt(i).id.equals(id)) { - return (PDBChain) chains.elementAt(i); + return chains.elementAt(i); } } @@ -198,7 +525,7 @@ public class PDBfile extends jalview.io.AlignFile { for (int i = 0; i < chains.size(); i++) { - ((PDBChain) chains.elementAt(i)).setChargeColours(); + chains.elementAt(i).setChargeColours(); } } @@ -206,7 +533,7 @@ public class PDBfile extends jalview.io.AlignFile { for (int i = 0; i < chains.size(); i++) { - ((PDBChain) chains.elementAt(i)).setChainColours(cs); + chains.elementAt(i).setChainColours(cs); } } @@ -214,8 +541,23 @@ public class PDBfile extends jalview.io.AlignFile { for (int i = 0; i < chains.size(); i++) { - ((PDBChain) chains.elementAt(i)).setChainColours(Color.getHSBColor( - 1.0f / (float) i, .4f, 1.0f)); + // divide by zero --> infinity --> 255 ;-) + chains.elementAt(i).setChainColours( + Color.getHSBColor(1.0f / i, .4f, 1.0f)); } } + + public static boolean isRNA(SequenceI seq) + { + for (char c : seq.getSequence()) + { + if ((c != 'A') && (c != 'C') && (c != 'G') && (c != 'U')) + { + return false; + } + } + + return true; + + } }