ModelSelectProbAll.cpp

#include "Headers.h"
#include "ModelSelectProbAll.h"

CModelSelectProbAll::CModelSelectProbAll(
    unsigned int    numChannels,
    unsigned int    numLayers,
    unsigned int    numPeers,
    CProb*          prob
    ) : CModelSelect(numChannels, numLayers, numPeers)
{
    this->prob = prob;
}

CModelSelectProbAll::~CModelSelectProbAll()
{
}

CPeer** CModelSelectProbAll::Select(__time time, __uint32 channel, __uint32 child)
{
    assert(channel < this->numChannels);
    assert(child < this->numPeers);

    CPeer* peerChild = &this->peers[child];

    for(__uint32 layer = 0; layer < this->numLayers; layer++)
    {
        double maxProb = 0;
        CPeer* maxPeer = 0;

        for(set<CPeer*>::iterator iter = this->index[channel].begin(); iter != this->index[channel].end(); iter++)
        {
            if(!this->IsResult(*iter, layer))
            {
                if(peerChild->VerifyParent(channel, layer, *iter))
                {
                    double prob = this->prob->Prob((*iter)->Elapsed(time, channel), peerChild->Elapsed(time, channel));
                    if(maxProb < prob)
                    {
                        maxProb = prob;
                        maxPeer = *iter;
                    }
                }
            }
        }

        this->result[layer] = maxPeer;
    }

    return this->result;
}

CPeer* CModelSelectProbAll::Select(__time time, __uint32 channel, __uint32 layer, __uint32 child)
{
    assert(channel < this->numChannels);
    assert(child < this->numPeers);

    CPeer* peerChild = &this->peers[child];

    double maxProb = 0;
    CPeer* maxPeer = 0;

    for(set<CPeer*>::iterator iter = this->index[channel].begin(); iter != this->index[channel].end(); iter++)
    {
        if(!this->IsResult(*iter, layer))
        {
            if(peerChild->VerifyParent(channel, layer, *iter))
            {
                double prob = this->prob->Prob((*iter)->Elapsed(time, channel), peerChild->Elapsed(time, channel));
                if(maxProb < prob)
                {
                    maxProb = prob;
                    maxPeer = *iter;
                }
            }
        }
    }

    return maxPeer;
}