BrushSculpt.h

Go to the documentation of this file.

/*

             *
            ***
           *****
   *********************       Mercenaries Engineering SARL
     *****************              Copyright (C) 2018
       *************
         *********        http://www.mercenaries-engineering.com
        ***********
       ****     ****
      **           **

*/
#include "Maquina/AnimCurve.h"
#include "Maquina/BVHTriangle.h"
#include "Maquina/BrushSculptPlugin.h"

MAQUINA_IGNORE_WARNINGS_BEGIN
#include <vector>
#include <tbb/parallel_for.h>
MAQUINA_IGNORE_WARNINGS_END

class TransferSculptGeometryUserData;

namespace maquina {

class MAQUINA_EXPORT BrushSculpt : public BrushSculptPlugin
{
public:

  enum class Operation
  {
    pull = 0,
    inflate,
    smooth_vertices,
    smooth_delta,
    relax,
    reset,
    flatten,
    pinch,
    grab,
    mirror,
    transfer,
    count
  };

  enum class SetBrushAttribute
  {
    none = 0,
    radius,
    amount,
    smoothness,
    flow
  };

  BrushSculpt(Node* node);

  // From ToolPlugin
  void on_mouse_move( const CameraContext& camera, const InteractionContext& interaction ) final;
  void on_mouse_press( const CameraContext& camera, const InteractionContext& interaction ) final;
  void on_mouse_release( const CameraContext& camera, const InteractionContext& interaction ) final;
  void activate();
  void deactivate();
  Array get_help(const char *tool) final;
  Array get_status(const char *tool) final;
  bool on_key_press(int32_t key, int32_t modifiers) final;
  bool on_key_release(int32_t key, int32_t modifiers) final;

  // From BrushPlugin
  void stroke_begin(int modifiers);
  void update_brush_matrix(const CameraContext& camera, const InteractionContext& interaction, bool& visible, Imath::M44f& brush_matrix, bool on_press) final;
  void update_weights(bool first, const Imath::M44f&) final;
  float weight(float distance) final;
  float flow() final;

  // From BrushSculptPlugin
  std::shared_ptr<BrushSculptUserData> user_data(
    const SparseBufferV3f& delta_points,
    const BrushPlugin::GeometryData& geometry,
    const BufferConstM33f& vertex_space,
    const std::shared_ptr<SceneConst>& reference
  );
  SceneConst delta_points() const;
  void set_delta_points(const SceneConst&);
  SceneConst vertex_space() final;
  SceneConst reference() final;
  bool preserve_edges() final;

  Operation operation(bool* invert=nullptr) const;

protected:
  int _modifiers;

private:
  SetBrushAttribute _attribute_change(int32_t key);
  Plug _attribute();
  bool _attribute_exponential();

  // Some cache to compute the weights
  float _radius;
  AnimCurveFloat _profile;

  // The currently edit attribute
  SetBrushAttribute __attribute_change = SetBrushAttribute::none;
  int _attribute_change_init = 0;
  float _attribute_init = 0.f;

  // transfer data
protected:
  std::string _input_path;
  std::string _source_path;
  Imath::M44d _source_matrix;
  Imath::M44d _source_matrix_inv;
  Mesh _source_mesh;
  BufferV3i _source_topo;
  std::vector<uint32_t> _source_tri_to_face;
  SpanSpan<const uint32_t, const uint32_t> _source_topo_face;
  BVHTriangle _source_bvh_triangle;
  SparseBufferConstV3f _source_sculpt_data = SparseBufferConstV3f::default_value;

private:
  friend class ::TransferSculptGeometryUserData;
};

}