cd7fafedb9eb5be0819a2144f536540599b70f17
[jalview.git] / test / jalview / structure / Mapping.java
1 package jalview.structure;
2
3 import static org.junit.Assert.assertEquals;
4 import static org.junit.Assert.assertTrue;
5 import jalview.datamodel.AlignmentAnnotation;
6 import jalview.datamodel.Annotation;
7 import jalview.datamodel.Sequence;
8 import jalview.datamodel.SequenceI;
9
10 import org.junit.Test;
11
12 import MCview.PDBfile;
13
14 public class Mapping
15 {
16
17   @Test
18   public void testPDBentryMapping() throws Exception
19   {
20     Sequence sq = new Sequence(
21             "1GAQ A subseq 126 to 219",
22             "EIVKGVCSNFLCDLQPGDNVQITGPVGKEMLMPKDPNATIIMLATGTGIAPFRSFLWKMFFEKHDDYKFNGLGWLFLGVPTSSSLLYKEEFGKM");
23     Sequence sq1 = new Sequence(sq);
24     String inFile;
25     StructureSelectionManager ssm = new jalview.structure.StructureSelectionManager();
26     // Associate the 1GAQ pdb file with the subsequence 'imported' from another
27     // source
28     PDBfile pde = ssm.setMapping(new SequenceI[]
29     { sq }, new String[]
30     { "A" }, inFile = "examples/1gaq.txt", jalview.io.FormatAdapter.FILE);
31     assertTrue("PDB File couldn't be found", pde != null);
32     StructureMapping[] mp = ssm.getMapping(inFile);
33     assertTrue("No mappings made.", mp != null && mp.length > 0);
34     boolean hasSecStr=false,hasTemp=false;
35     for (StructureMapping origMap:mp)
36     {
37       if (origMap.getSequence()==sq)
38       {
39         assertEquals("Mapping was incomplete.", sq.getLength() - 1,
40                 (origMap.getPDBResNum(sq.getEnd()) - origMap
41                         .getPDBResNum(sq.getStart())));
42         // sanity check - if this fails, mapping from first position in sequence
43         // we want to transfer to is not where we expect
44         assertEquals(1, origMap.getSeqPos(126));
45         SequenceI firstChain = pde.getSeqs().get(0);
46         // Compare the annotated positions on the PDB chain sequence with the
47         // annotation on the associated sequence
48         for (AlignmentAnnotation alan : firstChain.getAnnotation())
49         {
50           if (alan.hasIcons)
51           {
52             hasSecStr = true;
53           }
54           if (alan.graph == alan.LINE_GRAPH)
55           {
56             hasTemp = true;
57           }
58
59           AlignmentAnnotation transfer = origMap.transfer(alan);
60           System.out.println("pdb:" + firstChain.getSequenceAsString());
61           System.out.println("ann:" + alan.toString());
62           System.out.println("pdb:" + sq.getSequenceAsString());
63           System.out.println("ann:" + transfer.toString());
64
65           for (int p = 0, pSize = firstChain.getLength() - 1; p < pSize; p++)
66           {
67             // walk along the pdb chain's jalview sequence
68             int rseqpos;
69             int fpos = origMap.getSeqPos(rseqpos = firstChain
70                     .findPosition(p));
71             // only look at positions where there is a corresponding position in
72             // mapping
73             if (fpos < 1)
74             {
75               continue;
76             }
77             // p is index into PDB residue entries
78             // rseqpos is pdb sequence position for position p
79             // fpos is sequence position for associated position for rseqpos
80             int tanpos = sq.findIndex(fpos);
81             if (transfer.annotations.length <= tanpos)
82             {
83               // gone beyond mapping to the sequence
84               break;
85             }
86             Annotation a = transfer.annotations[sq.findIndex(fpos)], b = alan.annotations[p];
87             assertEquals("Non-equivalent annotation element at " + p + "("
88                     + rseqpos + ")"
89                             + " expected at " + fpos + " (alIndex "
90                             + sq.findIndex(fpos) + ")",
91                     a == null ? a : a.toString(),
92                     b == null ? b : b.toString());
93             System.out.print("(" + a + "|" + b + ")");
94           }
95
96         }
97       }
98     }
99     assertTrue("No secondary structure transferred to associated sequence.",hasSecStr);
100     assertTrue("No temperature factor transferred to associated sequence.",hasTemp);
101   }
102
103 }