BrushID.h

Go to the documentation of this file.

/*

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

*/
#include "Maquina/BVHTriangle.h"
#include "Maquina/BrushPlugin.h"

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

using namespace Imath;

namespace maquina {

enum class MAQUINA_EXPORT SetBrushAttribute
{
  none = 0,
  radius
};

class MAQUINA_EXPORT BrushID : public BrushPlugin
{
public:
  BrushID(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() override;
  void deactivate() override;
  Array get_help(const char* tool) override;
  Array get_status(const char* tool) override;
  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) override;
  void stroke_update(bool first, bool finished) override;
  void update_weights(bool first, const Imath::M44f& previous_brush_matrix) final;
  float weight(float distance) final;
  float flow() final;

protected:
  virtual void _stroke_init_data(GeometryData& geometry);
  virtual void _stroke_update(GeometryData& geometry);

  // The currently edit attribute
  SetBrushAttribute __attribute_change = SetBrushAttribute::none;

  // The edited attributes
  Scene _ids;

  int _modifiers;

private:
  SetBrushAttribute _attribute_change(int32_t key);

  Plug _attribute();

  // Some cache to compute the weights
  float _radius;

  // The edit attribute data
  int _attribute_change_init = 0;
  float _attribute_init = 0.f;

protected:
  // transfer data
  std::string _input_path;
  std::string _attrib_name;
  std::string _source_path;
  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;
  SparseBufferConstUInt32 _source_ids = SparseBufferConstUInt32::default_value;
};

}