global-vg-pyramid-maker.h

Go to the documentation of this file.

// Copyright 2007, Massachusetts Institute of Technology.
// The use of this code is permitted for research only. There is
// absolutely no warranty for this software.
//
// Author: John Lee (jjl@mit.edu)
//

#ifndef PYRAMIDS_GLOBAL_VG_PYRAMID_MAKER_H
#define PYRAMIDS_GLOBAL_VG_PYRAMID_MAKER_H

#include <vector>
#include <iostream>
#include <set>
#include <ext/hash_map>

#include "clustering/hierarchical-clusterer.h"
#include "histograms/multi-resolution-histogram.h"
#include "point_set/point-set-list.h"
#include "pyramids/vg-pyramid-maker.h"
#include "tree/sparse-tree-node.h"
#include "tree/sparse-tree.cc"
#include "util/distance-computer.h"
#include "util/sparse-vector.h"

using namespace std;
using namespace __gnu_cxx;

namespace libpmk {

class GlobalVGPyramidMaker : public VGPyramidMaker {
public:
  GlobalVGPyramidMaker(const HierarchicalClusterer& clusterer,
                       const DistanceComputer& distance_computer);
  ~GlobalVGPyramidMaker();


  void Preprocess(const PointSetList& point_sets);

  virtual MultiResolutionHistogram* MakePyramid(const PointSet& point_set);


  void ReadFromStream(istream& input_stream);


  void ReadFromFile(const char* filename);


  void WriteToStream(ostream& output_stream) const;


  void WriteToFile(const char* filename) const;

protected:

  virtual bool GetMembershipPath(const Point& f, LargeIndex* out_path,
                                 vector<double>* out_distances);

private:
  // This maps a node ID to a double representing the size of that
  // node. Recall that the data used to preprocess a
  // GlobalVGPyramidMaker does not have to be the same as the data
  // used to cluster. Thus, there could be some nodes in the tree
  // which own no points. Those nodes simply do not show up in this
  // hash_map. (having an entry with size=0 is allowable; this is
  // the case for a node with 1 point in it).
  hash_map<int, double> node_sizes_;

  bool preprocessed_;
};
}  // namespace libpmk

#endif  // PYRAMIDS_GLOBAL_VG_PYRAMID_MAKER_H

---