1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96
|
#include "libMUSCLE/muscle.h"
#include "libMUSCLE/textfile.h"
#include "libMUSCLE/msa.h"
#include "libMUSCLE/tree.h"
#include "libMUSCLE/profile.h"
#include "libMUSCLE/objscore.h"
namespace muscle {
TLS<bool> g_bTracePPScore(false);
TLS<MSA *> g_ptrPPScoreMSA1(0);
TLS<MSA *> g_ptrPPScoreMSA2(0);
static ProfPos *ProfileFromMSALocal(MSA &msa, Tree &tree)
{
const unsigned uSeqCount = msa.GetSeqCount();
for (unsigned uSeqIndex = 0; uSeqIndex < uSeqCount; ++uSeqIndex)
msa.SetSeqId(uSeqIndex, uSeqIndex);
TreeFromMSA(msa, tree, g_Cluster2.get(), g_Distance2.get(), g_Root1.get());
SetMuscleTree(tree);
return ProfileFromMSA(msa);
}
void PPScore()
{
if (0 == g_pstrFileName1.get() || 0 == g_pstrFileName2.get())
Quit("-ppscore needs -in1 and -in2");
SetSeqWeightMethod(g_SeqWeight1.get());
TextFile file1(g_pstrFileName1.get());
TextFile file2(g_pstrFileName2.get());
MSA msa1;
MSA msa2;
msa1.FromFile(file1);
msa2.FromFile(file2);
const unsigned uLength1 = msa1.GetColCount();
const unsigned uLength2 = msa2.GetColCount();
if (uLength1 != uLength2)
Quit("Profiles must have the same length");
ALPHA Alpha = ALPHA_Undefined;
switch (g_SeqType.get())
{
case SEQTYPE_Auto:
Alpha = msa1.GuessAlpha();
break;
case SEQTYPE_Protein:
Alpha = ALPHA_Amino;
break;
case SEQTYPE_DNA:
Alpha = ALPHA_DNA;
break;
case SEQTYPE_RNA:
Alpha = ALPHA_RNA;
break;
default:
Quit("Invalid SeqType");
}
SetAlpha(Alpha);
msa1.FixAlpha();
msa2.FixAlpha();
if (ALPHA_DNA == Alpha || ALPHA_RNA == Alpha)
SetPPScore(PPSCORE_SPN);
const unsigned uSeqCount1 = msa1.GetSeqCount();
const unsigned uSeqCount2 = msa2.GetSeqCount();
const unsigned uMaxSeqCount = (uSeqCount1 > uSeqCount2 ? uSeqCount1 : uSeqCount2);
MSA::SetIdCount(uMaxSeqCount);
Tree tree1;
Tree tree2;
ProfPos *Prof1 = ProfileFromMSALocal(msa1, tree1);
ProfPos *Prof2 = ProfileFromMSALocal(msa2, tree2);
g_bTracePPScore.get() = true;
g_ptrPPScoreMSA1.get() = &msa1;
g_ptrPPScoreMSA2.get() = &msa2;
SCORE Score = ObjScoreDP_Profs(Prof1, Prof2, uLength1);
Log("Score=%.4g\n", Score);
printf("Score=%.4g\n", Score);
}
}
|