import jalview.analysis.TreeBuilder;
import jalview.datamodel.SequenceI;
-import jalview.math.MatrixI;
+import jalview.ext.forester.ForesterConversions;
+import jalview.ext.forester.ForesterMatrix;
+import jalview.util.MessageManager;
+import java.util.HashMap;
+import java.util.Map;
+
+import org.forester.evoinference.matrix.distance.DistanceMatrix;
import org.forester.phylogeny.Phylogeny;
import org.forester.phylogeny.PhylogenyNode;
+import org.forester.phylogeny.data.NodeData;
+import org.forester.phylogeny.data.Sequence;
-public class ArchaeopteryxTreeConverter
+public class ArchaeopteryxTreeConverter // implements PhylogenyFactory
{
- protected SequenceI[] sequences;
+ protected final SequenceI[] sequences;
- private Phylogeny aptxTree;
+ protected final DistanceMatrix distances;
+
+ protected final TreeBuilder jalviewTree;
+
+ public String treeTitle;
- private PhylogenyNode rootNode;
+ private final Phylogeny aptxTree;
- protected MatrixI distances;
+ private PhylogenyNode rootNode;
- {
- this.aptxTree = new Phylogeny();
- }
+ private final Map<SequenceI, PhylogenyNode> alignmentBoundNodes;
+ private final Map<PhylogenyNode, SequenceI> nodesBoundAlignment;
- public ArchaeopteryxTreeConverter(TreeBuilder calculatedTree)
+ public ArchaeopteryxTreeConverter(final TreeBuilder calculatedTree)
{
- this.sequences = calculatedTree.getSequences();
- this.distances = calculatedTree.getDistances();
+ jalviewTree = calculatedTree;
+ sequences = jalviewTree.getSequences();
+ distances = ForesterMatrix.convertJalviewToForester(
+ jalviewTree.getDistances(), sequences);
+ aptxTree = new Phylogeny();
+ rootNode = new PhylogenyNode();
+ alignmentBoundNodes = new HashMap<>(distances.getSize());
+ nodesBoundAlignment = new HashMap<>(distances.getSize());
- this.rootNode = new PhylogenyNode();
- this.rootNode.setName("root");
}
- public ArchaeopteryxTreeConverter(PhylogenyNode treeRoot,
- TreeBuilder calculatedTree)
+ public Phylogeny buildAptxTree(final PhylogenyNode treeRoot)
{
- this(calculatedTree);
- this.rootNode = treeRoot;
+
+ if (treeRoot != null)
+ {
+ rootNode = treeRoot;
+ }
+
+ buildAptxTree();
+
+ return aptxTree;
}
+
public Phylogeny buildAptxTree()
{
+ for (SequenceI sequence : sequences)
+ {
+ Sequence seq = ForesterConversions
+ .createForesterSequence(sequence);
+ PhylogenyNode sequenceNode = new PhylogenyNode(sequence.getName());
+ NodeData nodeData = sequenceNode.getNodeData();
+ nodeData.setSequence(seq);
+
+ nodesBoundAlignment.put(sequenceNode, sequence);
+ alignmentBoundNodes.put(sequence, sequenceNode);
+ rootNode.addAsChild(sequenceNode);
+ }
- this.aptxTree.setName(
- "PLEASE FIX ME reheaheth35yheqhb3q5hyq3bt3q5u4jwqjwuh6");
- // final NeighborJoiningF nj = NeighborJoiningF.createInstance(false, 5);
+ aptxTree.setRoot(rootNode);
- //
- // final Phylogeny phy = nj.execute(JalviewMatrixToForesterMatrix
- // .convertJalviewToForester(distances));
+ treeTitle = generateTreeName();
+ aptxTree.setName(treeTitle);
- return buildAptxTree(sequences);
+ return aptxTree;
}
- // testing method to be removed
- public Phylogeny buildAptxTree(SequenceI[] sequences)
+ public Map<SequenceI, PhylogenyNode> getAlignmentBoundNodes()
{
+ return alignmentBoundNodes;
+ }
- for (SequenceI sequence : sequences)
- {
- PhylogenyNode treeNode = new PhylogenyNode(sequence.getName());
- rootNode.addAsChild(treeNode);
-
- }
+ public Map<PhylogenyNode, SequenceI> getNodesBoundAlignment()
+ {
+ return nodesBoundAlignment;
+ }
- aptxTree.setRoot(rootNode);
+ private Phylogeny clusterNodes()
+ {
return aptxTree;
}
-
-
/**
* Formats a localised title for the tree panel, like
* <p>
*
* @return
*/
- // public String getPanelTitle()
- // {
- // if (treeTitle != null)
- // {
- // return treeTitle;
- // }
- // else
- // {
- // /*
- // * i18n description of Neighbour Joining or Average Distance method
- // */
- // String treecalcnm = MessageManager
- // .getString("label.tree_calc_" + treeType.toLowerCase());
- //
- // /*
- // * short score model name (long description can be too long)
- // */
- // String smn = substitutionMatrix;
- //
- // /*
- // * put them together as <method> Using <model>
- // */
- // final String ttl = MessageManager
- // .formatMessage("label.treecalc_title", treecalcnm, smn);
- // return ttl;
- // }
- // }
+ public String generateTreeName() // Move this and add selection region to the
+ // title when applicable
+ {
+ if (treeTitle != null) // will currently never happen, loaded tree file will
+ // take a different path
+ {
+ return treeTitle;
+ }
+ else
+ {
+ /*
+ * i18n description of Neighbour Joining or Average Distance method
+ */
+ String treecalcnm = MessageManager
+ .getString("label.tree_calc_" + jalviewTree.getClass()
+ .getSimpleName().substring(0, 2).toLowerCase());
+ /*
+ * short score model name (long description can be too long)
+ */
+ String smn = jalviewTree.getScoreModel().getName();
+
+ /*
+ * put them together as <method> Using <model>
+ */
+ final String ttl = MessageManager
+ .formatMessage("label.treecalc_title", treecalcnm, smn);
+ return ttl;
+ }
+ }
}