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"
 
#include "MEM_guardedalloc.h"
 
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();
 
  MEM_CXX_CLASS_ALLOC_FUNCS("Freestyle:SteerableViewMap")
};
 
} /* namespace Freestyle */