SteerableViewMap.h

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */
 
#pragma once
 
#include <map>
 
#include "../geometry/Geom.h"
 
#include "../system/FreestyleConfig.h"
 
#ifdef WITH_CXX_GUARDEDALLOC
#  include "MEM_guardedalloc.h"
#endif
 
using namespace std;
 
namespace Freestyle {
 
using namespace Geometry;
 
class FEdge;
class ImagePyramid;
class GrayImage;
 
class SteerableViewMap {
 protected:
  // for each vector the list of nbOrientations weights corresponding to its contributions
  // to the nbOrientations directional maps
  map<unsigned int, double *> _mapping;
  unsigned _nbOrientations;
  ImagePyramid **_imagesPyramids;  // the pyramids of images storing the different SVM
 
  // internal
  double _bound;  // cos(Pi/N)
  vector<Vec2d> _directions;
 
 public:
  SteerableViewMap(unsigned int nbOrientations = 4);
  SteerableViewMap(const SteerableViewMap &iBrother);
  virtual ~SteerableViewMap();
 
  virtual void Reset();
 
  double *AddFEdge(FEdge *iFEdge);
 
  double ComputeWeight(const Vec2d &dir, unsigned iNOrientation);
 
  unsigned getSVMNumber(Vec2f dir);
 
  unsigned getSVMNumber(unsigned id);
 
  void buildImagesPyramids(GrayImage **steerableBases,
                           bool copy = false,
                           unsigned iNbLevels = 4,
                           float iSigma = 1.0f);
 
  float readSteerableViewMapPixel(unsigned iOrientation, int iLevel, int x, int y);
 
  float readCompleteViewMapPixel(int iLevel, int x, int y);
 
  unsigned int getNumberOfPyramidLevels() const;
 
  unsigned int getNumberOfOrientations() const
  {
    return _nbOrientations;
  }
 
  void saveSteerableViewMap() const;
 
 protected:
  void Clear();
  void Build();
 
#ifdef WITH_CXX_GUARDEDALLOC
  MEM_CXX_CLASS_ALLOC_FUNCS("Freestyle:SteerableViewMap")
#endif
};
 
} /* namespace Freestyle */