BowVector.h

Go to the documentation of this file.

#ifndef __D_T_BOW_VECTOR__
#define __D_T_BOW_VECTOR__

#include <iostream>
#include <map>
#include <vector>

namespace DBoW2 {

typedef unsigned int WordId;

typedef double WordValue;

typedef unsigned int NodeId;

enum LNorm
{
  L1,
  L2
};

enum WeightingType
{
  TF_IDF,
  TF,
  IDF,
  BINARY
};

enum ScoringType
{
  L1_NORM,
  L2_NORM,
  CHI_SQUARE,
  KL,
  BHATTACHARYYA,
  DOT_PRODUCT,
};

class BowVector: 
        public std::map<WordId, WordValue>
{
public:

        BowVector(void);

        ~BowVector(void);
        
        void addWeight(WordId id, WordValue v);
        
        void addIfNotExist(WordId id, WordValue v);

        void normalize(LNorm norm_type);
        
        friend std::ostream& operator<<(std::ostream &out, const BowVector &v);
        
        void saveM(const std::string &filename, size_t W) const;
};

} // namespace DBoW2

#endif