+++ /dev/null
-#include "muscle.h"\r
-#include "tree.h"\r
-\r
-#define TRACE 0\r
-\r
-// Return false when done\r
-bool PhyEnumEdges(const Tree &tree, PhyEnumEdgeState &ES)\r
- {\r
- unsigned uNode1 = uInsane;\r
-\r
- if (!ES.m_bInit)\r
- {\r
- if (tree.GetNodeCount() <= 1)\r
- {\r
- ES.m_uNodeIndex1 = NULL_NEIGHBOR;\r
- ES.m_uNodeIndex2 = NULL_NEIGHBOR;\r
- return false;\r
- }\r
- uNode1 = tree.FirstDepthFirstNode();\r
- ES.m_bInit = true;\r
- }\r
- else\r
- {\r
- uNode1 = tree.NextDepthFirstNode(ES.m_uNodeIndex1);\r
- if (NULL_NEIGHBOR == uNode1)\r
- return false;\r
- if (tree.IsRooted() && tree.IsRoot(uNode1))\r
- {\r
- uNode1 = tree.NextDepthFirstNode(uNode1);\r
- if (NULL_NEIGHBOR == uNode1)\r
- return false;\r
- }\r
- }\r
- unsigned uNode2 = tree.GetParent(uNode1);\r
-\r
- ES.m_uNodeIndex1 = uNode1;\r
- ES.m_uNodeIndex2 = uNode2;\r
- return true;\r
- }\r
-\r
-bool PhyEnumEdgesR(const Tree &tree, PhyEnumEdgeState &ES)\r
- {\r
- unsigned uNode1 = uInsane;\r
-\r
- if (!ES.m_bInit)\r
- {\r
- if (tree.GetNodeCount() <= 1)\r
- {\r
- ES.m_uNodeIndex1 = NULL_NEIGHBOR;\r
- ES.m_uNodeIndex2 = NULL_NEIGHBOR;\r
- return false;\r
- }\r
- uNode1 = tree.FirstDepthFirstNodeR();\r
- ES.m_bInit = true;\r
- }\r
- else\r
- {\r
- uNode1 = tree.NextDepthFirstNodeR(ES.m_uNodeIndex1);\r
- if (NULL_NEIGHBOR == uNode1)\r
- return false;\r
- if (tree.IsRooted() && tree.IsRoot(uNode1))\r
- {\r
- uNode1 = tree.NextDepthFirstNode(uNode1);\r
- if (NULL_NEIGHBOR == uNode1)\r
- return false;\r
- }\r
- }\r
- unsigned uNode2 = tree.GetParent(uNode1);\r
-\r
- ES.m_uNodeIndex1 = uNode1;\r
- ES.m_uNodeIndex2 = uNode2;\r
- return true;\r
- }\r
-\r
-static void GetLeavesSubtree(const Tree &tree, unsigned uNodeIndex1,\r
- const unsigned uNodeIndex2, unsigned Leaves[], unsigned *ptruCount)\r
- {\r
- if (tree.IsLeaf(uNodeIndex1))\r
- {\r
- Leaves[*ptruCount] = uNodeIndex1;\r
- ++(*ptruCount);\r
- return;\r
- }\r
-\r
- const unsigned uLeft = tree.GetFirstNeighbor(uNodeIndex1, uNodeIndex2);\r
- const unsigned uRight = tree.GetSecondNeighbor(uNodeIndex1, uNodeIndex2);\r
- if (NULL_NEIGHBOR != uLeft)\r
- GetLeavesSubtree(tree, uLeft, uNodeIndex1, Leaves, ptruCount);\r
- if (NULL_NEIGHBOR != uRight)\r
- GetLeavesSubtree(tree, uRight, uNodeIndex1, Leaves, ptruCount);\r
- }\r
-\r
-static void PhyGetLeaves(const Tree &tree, unsigned uNodeIndex1, unsigned uNodeIndex2,\r
- unsigned Leaves[], unsigned *ptruCount)\r
- {\r
- *ptruCount = 0;\r
- GetLeavesSubtree(tree, uNodeIndex1, uNodeIndex2, Leaves, ptruCount);\r
- }\r
-\r
-bool PhyEnumBiParts(const Tree &tree, PhyEnumEdgeState &ES,\r
- unsigned Leaves1[], unsigned *ptruCount1,\r
- unsigned Leaves2[], unsigned *ptruCount2)\r
- {\r
- bool bOk = PhyEnumEdges(tree, ES);\r
- if (!bOk)\r
- {\r
- *ptruCount1 = 0;\r
- *ptruCount2 = 0;\r
- return false;\r
- }\r
-\r
-// Special case: in a rooted tree, both edges from the root\r
-// give the same bipartition, so skip one of them.\r
- if (tree.IsRooted() && tree.IsRoot(ES.m_uNodeIndex2)\r
- && tree.GetRight(ES.m_uNodeIndex2) == ES.m_uNodeIndex1)\r
- {\r
- bOk = PhyEnumEdges(tree, ES);\r
- if (!bOk)\r
- return false;\r
- }\r
-\r
- PhyGetLeaves(tree, ES.m_uNodeIndex1, ES.m_uNodeIndex2, Leaves1, ptruCount1);\r
- PhyGetLeaves(tree, ES.m_uNodeIndex2, ES.m_uNodeIndex1, Leaves2, ptruCount2);\r
-\r
- if (*ptruCount1 + *ptruCount2 != tree.GetLeafCount())\r
- Quit("PhyEnumBiParts %u + %u != %u",\r
- *ptruCount1, *ptruCount2, tree.GetLeafCount());\r
-#if DEBUG\r
- {\r
- for (unsigned i = 0; i < *ptruCount1; ++i)\r
- {\r
- if (!tree.IsLeaf(Leaves1[i]))\r
- Quit("PhyEnumByParts: not leaf");\r
- for (unsigned j = 0; j < *ptruCount2; ++j)\r
- {\r
- if (!tree.IsLeaf(Leaves2[j]))\r
- Quit("PhyEnumByParts: not leaf");\r
- if (Leaves1[i] == Leaves2[j])\r
- Quit("PhyEnumByParts: dupe");\r
- }\r
- }\r
- }\r
-#endif\r
-\r
- return true;\r
- }\r
-\r
-#if 0\r
-void TestBiPart()\r
- {\r
- SetListFileName("c:\\tmp\\lobster.log", false);\r
- Tree tree;\r
- TextFile fileIn("c:\\tmp\\test.phy");\r
- tree.FromFile(fileIn);\r
- tree.LogMe();\r
-\r
- const unsigned uNodeCount = tree.GetNodeCount();\r
- unsigned *Leaves1 = new unsigned[uNodeCount];\r
- unsigned *Leaves2 = new unsigned[uNodeCount];\r
-\r
- PhyEnumEdgeState ES;\r
- bool bDone = false;\r
- for (;;)\r
- {\r
- unsigned uCount1 = uInsane;\r
- unsigned uCount2 = uInsane;\r
- bool bOk = PhyEnumBiParts(tree, ES, Leaves1, &uCount1, Leaves2, &uCount2);\r
- Log("PEBP=%d ES.Init=%d ES.ni1=%d ES.ni2=%d\n",\r
- bOk,\r
- ES.m_bInit,\r
- ES.m_uNodeIndex1,\r
- ES.m_uNodeIndex2);\r
- if (!bOk)\r
- break;\r
- Log("\n");\r
- Log("Part1: ");\r
- for (unsigned n = 0; n < uCount1; ++n)\r
- Log(" %d(%s)", Leaves1[n], tree.GetLeafName(Leaves1[n]));\r
- Log("\n");\r
- Log("Part2: ");\r
- for (unsigned n = 0; n < uCount2; ++n)\r
- Log(" %d(%s)", Leaves2[n], tree.GetLeafName(Leaves2[n]));\r
- Log("\n");\r
- }\r
- }\r
-#endif\r
-\r
-static void GetLeavesSubtreeExcluding(const Tree &tree, unsigned uNodeIndex,\r
- unsigned uExclude, unsigned Leaves[], unsigned *ptruCount)\r
- {\r
- if (uNodeIndex == uExclude)\r
- return;\r
-\r
- if (tree.IsLeaf(uNodeIndex))\r
- {\r
- Leaves[*ptruCount] = uNodeIndex;\r
- ++(*ptruCount);\r
- return;\r
- }\r
-\r
- const unsigned uLeft = tree.GetLeft(uNodeIndex);\r
- const unsigned uRight = tree.GetRight(uNodeIndex);\r
- if (NULL_NEIGHBOR != uLeft)\r
- GetLeavesSubtreeExcluding(tree, uLeft, uExclude, Leaves, ptruCount);\r
- if (NULL_NEIGHBOR != uRight)\r
- GetLeavesSubtreeExcluding(tree, uRight, uExclude, Leaves, ptruCount);\r
- }\r
-\r
-void GetLeavesExcluding(const Tree &tree, unsigned uNodeIndex,\r
- unsigned uExclude, unsigned Leaves[], unsigned *ptruCount)\r
- {\r
- *ptruCount = 0;\r
- GetLeavesSubtreeExcluding(tree, uNodeIndex, uExclude, Leaves, ptruCount);\r
- }\r
-\r
-void GetInternalNodesInHeightOrder(const Tree &tree, unsigned NodeIndexes[])\r
- {\r
- const unsigned uNodeCount = tree.GetNodeCount();\r
- if (uNodeCount < 3)\r
- Quit("GetInternalNodesInHeightOrder: %u nodes, none are internal",\r
- uNodeCount);\r
- const unsigned uInternalNodeCount = (uNodeCount - 1)/2;\r
- double *Heights = new double[uInternalNodeCount];\r
-\r
- unsigned uIndex = 0;\r
- for (unsigned uNodeIndex = 0; uNodeIndex < uNodeCount; ++uNodeIndex)\r
- {\r
- if (tree.IsLeaf(uNodeIndex))\r
- continue;\r
- NodeIndexes[uIndex] = uNodeIndex;\r
- Heights[uIndex] = tree.GetNodeHeight(uNodeIndex);\r
- ++uIndex;\r
- }\r
- if (uIndex != uInternalNodeCount)\r
- Quit("Internal error: GetInternalNodesInHeightOrder");\r
-\r
-// Simple but slow bubble sort (probably don't care about speed here)\r
- bool bDone = false;\r
- while (!bDone)\r
- {\r
- bDone = true;\r
- for (unsigned i = 0; i < uInternalNodeCount - 1; ++i)\r
- {\r
- if (Heights[i] > Heights[i+1])\r
- {\r
- double dTmp = Heights[i];\r
- Heights[i] = Heights[i+1];\r
- Heights[i+1] = dTmp;\r
-\r
- unsigned uTmp = NodeIndexes[i];\r
- NodeIndexes[i] = NodeIndexes[i+1];\r
- NodeIndexes[i+1] = uTmp;\r
- bDone = false;\r
- }\r
- }\r
- }\r
-#if TRACE\r
- Log("Internal node index Height\n");\r
- Log("------------------- --------\n");\r
- // 1234567890123456789 123456789\r
- for (unsigned n = 0; n < uInternalNodeCount; ++n)\r
- Log("%19u %9.3f\n", NodeIndexes[n], Heights[n]);\r
-#endif\r
- delete[] Heights;\r
- }\r
-\r
-void ApplyMinEdgeLength(Tree &tree, double dMinEdgeLength)\r
- {\r
- const unsigned uNodeCount = tree.GetNodeCount();\r
- for (unsigned uNodeIndex = 0; uNodeIndex < uNodeCount; ++uNodeIndex)\r
- {\r
- const unsigned uNeighborCount = tree.GetNeighborCount(uNodeIndex);\r
- for (unsigned n = 0; n < uNeighborCount; ++n)\r
- {\r
- const unsigned uNeighborNodeIndex = tree.GetNeighbor(uNodeIndex, n);\r
- if (!tree.HasEdgeLength(uNodeIndex, uNeighborNodeIndex))\r
- continue;\r
- if (tree.GetEdgeLength(uNodeIndex, uNeighborNodeIndex) < dMinEdgeLength)\r
- tree.SetEdgeLength(uNodeIndex, uNeighborNodeIndex, dMinEdgeLength);\r
- }\r
- }\r
- }\r