uniform-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)
//
// TODO(jjl): what about turning new PointSets with min/max feature
// value beyond min_value_ or max_value_? check for it and add a note in
// the spec.
#ifndef PYRAMIDS_UNIFORM_PYRAMID_MAKER_H
#define PYRAMIDS_UNIFORM_PYRAMID_MAKER_H

#include <vector>
#include <map>
#include <iostream>
#include "pyramids/pyramid-maker.h"
#include "point_set/point-set-list.h"
#include "point_set/point-set.h"
#include "histograms/multi-resolution-histogram.h"

using namespace std;

namespace libpmk {


class UniformPyramidMaker : public PyramidMaker {
 public:
   UniformPyramidMaker();


   void Preprocess(const PointSetList& point_sets,
                   double finest_side_length,
                   double side_factor,
                   int discretize_factor,
                   bool do_translations,
                   bool global_translations);

   void Preprocess(const vector<PointRef>& points,
                   double finest_side_length,
                   double side_factor,
                   int discretize_factor,
                   bool do_translations,
                   bool global_translations);


   void ReadFromStream(istream& input_stream);


   void ReadFromFile(const char* filename);


   void WriteToStream(ostream& output_stream) const;


   void WriteToFile(const char* filename) const;


   int num_levels() const;


   virtual MultiResolutionHistogram* MakePyramid(const PointSet& ps);


 private:
   double finest_side_length_;
   double side_factor_;
   int discretize_factor_;
   int feature_dim_;
   bool do_translations_;
   bool global_translations_;
   bool preprocessed_;

   double min_value_;
   double max_value_;

   double diameter_;
   int num_levels_;

   // One vector per level. This is a num_levels * dim matrix.
   vector<vector<double> > translations_;

   // Make sure that the values (ints) are unique. We need to do this for
   // each level, so it's a vector of size num_levels_ of these mappings from
   // index to int.
   vector<map<LargeIndex, int> > vector_map_;

   // A vector of size num_levels. The ith entry tells you the last-used
   // value in vector_map_.
   vector<int> last_vector_map_values_;

   int GetSparseValue(int level, const LargeIndex& index);
   void ComputeHistogramSettings(int dim,
                                 double finest_side_length,
                                 double side_factor,
                                 int discretize_factor,
                                 bool do_translations,
                                 bool global_translations);
};
}  // namespace libpmk
#endif  // PYRAMIDS_UNIFORM_PYRAMID_MAKER_H

---