JAL-2685 use actual selection for report of pairwise alignment, help
[jalview.git] / src / jalview / gui / PairwiseAlignPanel.java
1 /*
2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
4  * 
5  * This file is part of Jalview.
6  * 
7  * Jalview is free software: you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License 
9  * as published by the Free Software Foundation, either version 3
10  * of the License, or (at your option) any later version.
11  *  
12  * Jalview is distributed in the hope that it will be useful, but 
13  * WITHOUT ANY WARRANTY; without even the implied warranty 
14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
15  * PURPOSE.  See the GNU General Public License for more details.
16  * 
17  * You should have received a copy of the GNU General Public License
18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
19  * The Jalview Authors are detailed in the 'AUTHORS' file.
20  */
21 package jalview.gui;
22
23 import jalview.analysis.AlignSeq;
24 import jalview.datamodel.Alignment;
25 import jalview.datamodel.AlignmentView;
26 import jalview.datamodel.SequenceGroup;
27 import jalview.datamodel.SequenceI;
28 import jalview.jbgui.GPairwiseAlignPanel;
29 import jalview.util.MessageManager;
30 import jalview.viewmodel.AlignmentViewport;
31
32 import java.awt.event.ActionEvent;
33 import java.util.Vector;
34
35 /**
36  * DOCUMENT ME!
37  * 
38  * @author $author$
39  * @version $Revision$
40  */
41 public class PairwiseAlignPanel extends GPairwiseAlignPanel
42 {
43
44   private static final String DASHES = "---------------------\n";
45
46   AlignmentViewport av;
47
48   Vector<SequenceI> sequences;
49
50   /**
51    * Creates a new PairwiseAlignPanel object.
52    * 
53    * @param viewport
54    *          DOCUMENT ME!
55    */
56   public PairwiseAlignPanel(AlignmentViewport viewport)
57   {
58     super();
59     this.av = viewport;
60
61     sequences = new Vector<SequenceI>();
62
63     SequenceGroup selectionGroup = viewport.getSelectionGroup();
64     boolean isSelection = selectionGroup != null
65             && selectionGroup.getSize() > 0;
66     AlignmentView view = viewport.getAlignmentView(isSelection);
67     // String[] seqStrings = viewport.getViewAsString(true);
68     String[] seqStrings = view.getSequenceStrings(viewport
69             .getGapCharacter());
70
71     SequenceI[] seqs;
72     if (isSelection)
73     {
74       seqs = (SequenceI[]) view.getAlignmentAndHiddenColumns(viewport
75               .getGapCharacter())[0];
76     }
77     else
78     {
79       seqs = av.getAlignment().getSequencesArray();
80     }
81
82     String type = (viewport.getAlignment().isNucleotide()) ? AlignSeq.DNA
83             : AlignSeq.PEP;
84
85     float[][] scores = new float[seqs.length][seqs.length];
86     double totscore = 0;
87     int count = seqs.length;
88     boolean first = true;
89
90     for (int i = 1; i < count; i++)
91     {
92       for (int j = 0; j < i; j++)
93       {
94         AlignSeq as = new AlignSeq(seqs[i], seqStrings[i], seqs[j],
95                 seqStrings[j], type);
96
97         if (as.s1str.length() == 0 || as.s2str.length() == 0)
98         {
99           continue;
100         }
101
102         as.calcScoreMatrix();
103         as.traceAlignment();
104
105         if (!first)
106         {
107           System.out.println(DASHES);
108           textarea.append(DASHES);
109         }
110         first = false;
111         as.printAlignment(System.out);
112         scores[i][j] = as.getMaxScore()
113                 / as.getASeq1().length;
114         totscore = totscore + scores[i][j];
115
116         textarea.append(as.getOutput());
117         sequences.add(as.getAlignedSeq1());
118         sequences.add(as.getAlignedSeq2());
119       }
120     }
121
122     if (count > 2)
123     {
124       System.out.println(
125               "Pairwise alignment scaled similarity score matrix\n");
126
127       for (int i = 0; i < count; i++)
128       {
129         System.out.println(String.format("%d %s", i,
130                 seqs[i].getDisplayId(true)));
131       }
132
133       for (int i = 0; i < count; i++)
134       {
135         for (int j = 0; j < i; j++)
136         {
137           System.out.print(String.format("%7.3f", scores[i][j] / totscore));
138         }
139         System.out.println();
140       }
141
142       System.out.println("\n");
143     }
144   }
145
146   /**
147    * DOCUMENT ME!
148    * 
149    * @param e
150    *          DOCUMENT ME!
151    */
152   @Override
153   protected void viewInEditorButton_actionPerformed(ActionEvent e)
154   {
155     SequenceI[] seq = new SequenceI[sequences.size()];
156
157     for (int i = 0; i < sequences.size(); i++)
158     {
159       seq[i] = sequences.elementAt(i);
160     }
161
162     AlignFrame af = new AlignFrame(new Alignment(seq),
163             AlignFrame.DEFAULT_WIDTH, AlignFrame.DEFAULT_HEIGHT);
164
165     Desktop.addInternalFrame(af,
166             MessageManager.getString("label.pairwise_aligned_sequences"),
167             AlignFrame.DEFAULT_WIDTH, AlignFrame.DEFAULT_HEIGHT);
168   }
169 }