--- /dev/null
+/* -*- mode: c; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
+
+/* Module for deriving sequence weights from a tree. Largely based on
+ * Bob Edgar's Muscle (mainly clwwt.cpp; version 3.7). Ported to pure
+ * C. Most functions where apparently based on Clustal 1.8 anyway.
+ *
+ * Muscle's code is public domain and so is this code here.
+ *
+ * From http://www.drive5.com/muscle/license.htm:
+ * """
+ * MUSCLE is public domain software
+ *
+ * The MUSCLE software, including object and source code and
+ * documentation, is hereby donated to the public domain.
+ *
+ * Disclaimer of warranty
+ * THIS SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * """
+ *
+ */
+
+/*
+ * RCS $Id: weights.c 231 2011-04-09 17:13:06Z andreas $
+ */
+
+/*
+ * Documentation from Muscle
+ *
+ * """
+ * Compute weights by the CLUSTALW method.
+ * Thompson, Higgins and Gibson (1994), CABIOS (10) 19-29;
+ * see also CLUSTALW paper.
+ *
+ * Weights are computed from the edge lengths of a rooted tree.
+ *
+ * Define the strength of an edge to be its length divided by the number
+ * of leaves under that edge. The weight of a sequence is then the sum
+ * of edge strengths on the path from the root to the leaf.
+ *
+ * Example.
+ *
+ * 0.2
+ * -----A 0.1
+ * -x ------- B 0.7
+ * --------y ----------- C
+ * 0.3 ----------z
+ * 0.4 -------------- D
+ * 0.8
+ *
+ * Edge Length Leaves Strength
+ * ---- ----- ------ --------
+ * xy 0.3 3 0.1
+ * xA 0.2 1 0.2
+ * yz 0.4 2 0.2
+ * yB 0.1 1 0.1
+ * zC 0.7 1 0.7
+ * zD 0.8 1 0.8
+ *
+ * Leaf Path Strengths Weight
+ * ---- ---- --------- ------
+ * A xA 0.2 0.2
+ * B xy-yB 0.1 + 0.1 0.2
+ * C xy-yz-zC 0.1 + 0.2 + 0.7 1.0
+ * D xy-yz-zD 0.1 + 0.2 + 0.8 1.1
+ * """
+ *
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <assert.h>
+#include "log.h"
+#include "muscle_tree.h"
+#include "weights.h"
+
+
+/*
+ #undef DEBUG
+*/
+
+
+/**
+ * @brief FIXME
+ *
+ * @param[out] puLeavesUnderNode
+ * FIXME
+ * @param[in] prTree
+ * FIXME
+ * @param[in] uNodeIndex
+ * FIXME
+ *
+ * @return The return value
+ *
+ * @note see Muscle3.7:clwwt.cpp
+ *
+ */
+uint
+CountLeaves(uint *puLeavesUnderNode, tree_t *prTree, uint uNodeIndex)
+{
+ uint uLeft;
+ uint uRight;
+ uint uRightCount;
+ uint uLeftCount;
+ uint uCount;
+
+
+ if (IsLeaf(uNodeIndex, prTree)) {
+ puLeavesUnderNode[uNodeIndex] = 1;
+ return 1;
+ }
+
+ uLeft = GetLeft(uNodeIndex, prTree);
+ uRight = GetRight(uNodeIndex, prTree);
+ uRightCount = CountLeaves(puLeavesUnderNode, prTree, uRight);
+ uLeftCount = CountLeaves(puLeavesUnderNode, prTree, uLeft);
+ uCount = uRightCount + uLeftCount;
+
+ puLeavesUnderNode[uNodeIndex] = uCount;
+
+ return uCount;
+}
+/*** end: CountLeaves() ***/
+
+
+
+
+/**
+ * @brief Normalise values in a double array to values between 0 and 1.
+ *
+ * @param[out] p
+ * double array with n elements
+ * @param[in] n
+ * number of elements in p
+ *
+ * @note From Muscle3.7: intmath.cpp:Normalize()
+ *
+ */
+void
+Normalise(double *p, uint n) {
+ unsigned i;
+ double dSum = 0.0;
+ for (i = 0; i < n; ++i) {
+ dSum += p[i];
+ }
+ if (0.0 == dSum) {
+ Log(&rLog, LOG_FATAL, "Normalise, sum=0");
+ }
+ for (i = 0; i < n; ++i) {
+ p[i] /= dSum;
+ }
+}
+/*** end: Normalise() ***/
+
+
+
+/**
+ * @brief Calculate "Clustal" weights from a tree.
+ *
+ * FIXME see doc in muscle:clwwt.cpp
+ *
+ * @param[out] pdWeights_p
+ * Will contain a weight for each leaf/sequence. Allocated here. User
+ * has to free
+ * @param[in] prTree
+ * Tree to derive weights from
+ *
+ * @return 0 on success, non-zero otherwise
+ *
+ * @note Largely copied from Muscle3.7: clwwt.cpp:CalcClustalWWeights()
+ *
+ * @warning FIXME Not sure if Muscle routines are most efficient here.
+ * Couldn't we do all this while traversing the tree and thereby safe
+ * time?
+ *
+ */
+int
+CalcClustalWeights(double **pdWeights_p, tree_t *prTree)
+{
+ int i; /* aux */
+ uint uLeafCount;
+ uint uNodeCount;
+ uint *puLeavesUnderNode;
+ uint uLeavesUnderRoot;
+ uint uRootNodeIndex;
+ double *pdStrengths;
+ uint uNodeIndex;
+ bool bLogWeights = FALSE; /* verbose output of weights */
+
+
+ assert(NULL != pdWeights_p);
+ assert(NULL != prTree);
+
+ if (rLog.iLogLevelEnabled <= LOG_DEBUG) {
+ bLogWeights = TRUE;
+ }
+
+ uLeafCount = GetLeafCount(prTree);
+ uNodeCount = GetNodeCount(prTree);
+
+
+ (*pdWeights_p) = (double *) CKMALLOC(uNodeCount * sizeof(double));
+
+ if (0 == uLeafCount) {
+ return 0;
+ } else if (1 == uLeafCount) {
+ (*pdWeights_p)[0] = 1.0;
+ return 0;
+ } else if (2 == uLeafCount) {
+ (*pdWeights_p)[0] = 0.5;
+ (*pdWeights_p)[1] = 0.5;
+ return 0;
+ }
+
+ if (!IsRooted(prTree)) {
+ Log(&rLog, LOG_ERROR, "Tree must be rooted to get weights");
+ CKFREE(pdWeights_p);
+ return -1;
+ }
+
+
+#ifdef TRACE
+ Log(&rLog, LOG_FORCED_DEBUG, "%s", "Weights follow");
+ fprintf(stderr, "Node Leaves Length Strength\n");
+ fprintf(stderr, "---- ------ -------- --------\n");
+ /* 1234 123456 12345678 12345678 */
+#endif
+
+ uRootNodeIndex = GetRootNodeIndex(prTree);
+ puLeavesUnderNode = (uint *) CKCALLOC(uNodeCount, sizeof(uint));
+
+ uLeavesUnderRoot = CountLeaves(puLeavesUnderNode, prTree, uRootNodeIndex);
+ if (uLeavesUnderRoot != uLeafCount) {
+ Log(&rLog, LOG_FATAL, "Internal error, root count %u %u",
+ uLeavesUnderRoot, uLeafCount);
+ }
+#if 0
+ for (uNodeIndex=0; uNodeIndex<uNodeCount; uNodeIndex++) {
+ Log(&rLog, LOG_FORCED_DEBUG, "LeavesUnderNode[%d]=%d", uNodeIndex, puLeavesUnderNode[uNodeIndex]);
+ }
+#endif
+
+ pdStrengths = (double *) CKMALLOC(uNodeCount * sizeof(double));
+
+ for (uNodeIndex=0; uNodeIndex < uNodeCount; uNodeIndex++) {
+ uint uParent;
+ double dLength;
+ uint uLeaves;
+ double dStrength;
+
+ if (IsRoot(uNodeIndex, prTree)) {
+ pdStrengths[uNodeIndex] = 0.0;
+ continue;
+ }
+
+ uParent = GetParent(uNodeIndex, prTree);
+ dLength = GetEdgeLength(uNodeIndex, uParent, prTree);
+ uLeaves = puLeavesUnderNode[uNodeIndex];
+ dStrength = dLength / (double) uLeaves;
+ pdStrengths[uNodeIndex] = dStrength;
+
+#ifdef TRACE
+ fprintf(stderr, "%4u %6u %8g %8g\n", uNodeIndex, uLeaves, dLength, dStrength);
+#endif
+ }
+
+
+
+
+
+ if (bLogWeights){
+ fprintf(stderr, "\n");
+ fprintf(stderr, " Seq Path..Weight\n");
+ fprintf(stderr, "-------------------- ------------\n");
+ }
+ for (i=0; i<uLeafCount; i++) {
+ double dWeight = 0.0;
+ unsigned uLeafNodeIndex;
+ unsigned uNode;
+
+ uLeafNodeIndex = LeafIndexToNodeIndex(i, prTree);
+ uNode = uLeafNodeIndex;
+
+ if (bLogWeights){
+ fprintf(stderr, "%20.20s %4u ", GetLeafName(uLeafNodeIndex, prTree), uLeafNodeIndex);
+ }
+ if (! IsLeaf(uLeafNodeIndex, prTree)) {
+ Log(&rLog, LOG_FATAL,
+ "Internal error: non-leaf-node %d", uLeafNodeIndex);
+ }
+
+ /*LOG_DEBUG("dWeight = %f", dWeight);*/
+ while (! IsRoot(uNode, prTree)) {
+ dWeight += pdStrengths[uNode];
+ /*LOG_DEBUG("dWeight +== %f", pdStrengths[uNode]);*/
+ uNode = GetParent(uNode, prTree);
+ if (bLogWeights){
+ fprintf(stderr, "->%u(%g)", uNode, pdStrengths[uNode]);
+ }
+ }
+ /* AW: no idea what this is, but it's done like this in Muscle */
+ if (dWeight < 0.0001) {
+#ifdef TRACE
+ fprintf(stderr, "zero->one");
+#endif
+ dWeight = 1.0;
+ }
+
+ /* @note: the only difference to the muscle code is here: we
+ * use the input index for storing weights, instead of the
+ * tree leaf index
+ */
+ (*pdWeights_p)[GetLeafId(uLeafNodeIndex, prTree)] = dWeight;
+ if (bLogWeights){
+ fprintf(stderr, " = %g\n", dWeight);
+ }
+ }
+
+#if 0
+ for (i=0; i<uLeafCount; i++) {
+ Log(&rLog, LOG_FORCED_DEBUG, "Weights before normalisation: pdWeights_p[%d]=%f", i, (*pdWeights_p)[i]);
+ /*LOG_DEBUG("Should be %d", GetLeafId(LeafIndexToNodeIndex(i, prTree), prTree));*/
+ }
+#endif
+
+ Normalise((*pdWeights_p), uLeafCount);
+
+
+ CKFREE(puLeavesUnderNode);
+ CKFREE(pdStrengths);
+
+ return 0;
+}
+/*** end: CalcWeights() ***/