vocabmap.h

来自「这是一款很好用的工具包」· C头文件 代码 · 共 83 行

/*
 * VocabMap.h --
 *	Probabilistic mappings between vocabularies
 *
 * Copyright (c) 1995,1998,2003 SRI International.  All Rights Reserved.
 *
 * @(#)$Header: /home/srilm/devel/lm/src/RCS/VocabMap.h,v 1.7 2005/05/20 23:49:02 stolcke Exp $
 *
 */

#ifndef _VocabMap_h_
#define _VocabMap_h_

#include "Boolean.h"
#include "Prob.h"
#include "Vocab.h"
#include "Map2.cc"

class VocabMap
{
    friend class VocabMapIter;

public:
    VocabMap(Vocab &v1, Vocab &v2, Boolean logmap = false);
    
    Prob get(VocabIndex w1, VocabIndex w2);
    void put(VocabIndex w1, VocabIndex w2, Prob prob);
    void remove(VocabIndex w1, VocabIndex w2);
    void remove(VocabIndex w1);

    virtual Boolean read(File &file);
    virtual Boolean readClasses(File &file);
    virtual Boolean write(File &file);
    virtual Boolean writeBigrams(File &file);
    
    Vocab &vocab1;
    Vocab &vocab2;

protected:
    /*
     * The map is implemented by a two-level map where the first index is
     * from vocab1 and the second from vocab2
     */
    Map2<VocabIndex,VocabIndex,Prob> map;

    Boolean logmap;			/* treat probabilities as log probs */
};

/*
 * Map from an integer (`position') to a Vocab
 */
class PosVocabMap: public VocabMap
{
public:
    PosVocabMap(Vocab &vocab, Boolean logmap = false)
        : VocabMap(vocab, vocab, logmap)
	{ map.clear(); };

    /* 
     * not implemented yet (or ever)
     */
    Boolean read(File &file) { return false; };
    Boolean write(File &file) { return false; };
};

/*
 * Iteration over the mappings of a word
 */
class VocabMapIter
{
public:
    VocabMapIter(VocabMap &vmap, VocabIndex w);

    void init();
    Boolean next(VocabIndex &w, Prob &prob);

private:
    Map2Iter2<VocabIndex,VocabIndex,Prob> mapIter;
};

#endif /* _VocabMap_h_ */