trf-alg.h

Go to the documentation of this file.

// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Copyright 2014-2015 Tsinghua University
// Author: wb.th08@gmail.com (Bin Wang), ozj@tsinghua.edu.cn (Zhijian Ou) 
//
// All h, cpp, cc, and script files (e.g. bat, sh, pl, py) should include the above 
// license declaration. Different coding language may use different comment styles.


#pragma once
#include "trf-def.h"


namespace trf
{

    class Msg;
    class VecIter;
    class Algfb;


    class Algfb
    {
    public:
        int m_nOrder; 
        int m_nLen; 
        int m_nValueLimit; 
        Array<Msg*> m_aAlpha;    
        Array<Msg*> m_aBeta;       

    public:
        Algfb() : m_nOrder(0), m_nLen(0), m_nValueLimit(0) {};
        ~Algfb();
        void Prepare(int nLen, int nOrder, int nValueLimit);
        void ForwardBackward(int nLen, int nOrder, int nValueLimit);
        LogP GetMarginalLogProb(int nPos, int *pSubSeq, int nSubLen, double logz = 0);
        LogP GetLogSummation();
        virtual LogP ClusterSum(int *pSeq, int nLen, int nPos, int nOrder) = 0;
    };

    class Msg
    {
    private:
        float *m_pbuf; 
        int m_dim; 
        int m_size; 

        //Model *m_pmodel; ///< the pointer to the model
    public:
        Msg(int nMsgDim, int nSize);
        ~Msg();
        void Fill(float v);
        void Copy(Msg &m);
        float& Get(int *pIdx, int nDim);
        int GetBufSize() const { return pow(m_size, m_dim); }
    };

    class VecIter
    {
    public:
        int *m_pBuf; 
        int m_nDim;  
        int m_nMin; 
        int m_nMax; 
    public:
        VecIter(int *p, int nDim, int nMin, int nMax);
        void Reset();
        bool Next();
    };

}