SteerableViewMap.h

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#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<uint, double *> _mapping;
  uint _nbOrientations;
  ImagePyramid **_imagesPyramids;  // the pyramids of images storing the different SVM
 
  // internal
  double _bound;  // cos(Pi/N)
  vector<Vec2d> _directions;
 
 public:
  SteerableViewMap(uint nbOrientations = 4);
  SteerableViewMap(const SteerableViewMap &iBrother);
  virtual ~SteerableViewMap();
 
  virtual void Reset();
 
  double *AddFEdge(FEdge *iFEdge);
 
  double ComputeWeight(const Vec2d &dir, uint iNOrientation);
 
  uint getSVMNumber(Vec2f dir);
 
  uint getSVMNumber(uint id);
 
  void buildImagesPyramids(GrayImage **steerableBases,
                           bool copy = false,
                           uint iNbLevels = 4,
                           float iSigma = 1.0f);
 
  float readSteerableViewMapPixel(uint iOrientation, int iLevel, int x, int y);
 
  float readCompleteViewMapPixel(int iLevel, int x, int y);
 
  uint getNumberOfPyramidLevels() const;
 
  uint 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 */