MRChangePointCloudNormalsAction.h

Go to the documentation of this file.

#pragma once
#include "MRHistoryAction.h"
#include "MRObjectPoints.h"
#include "MRPointCloud.h"
#include "MRHeapBytes.h"
#include <memory>
 
 
namespace MR
{
 
class ChangePointCloudNormalsAction : public HistoryAction
{
public:
    using Obj = ObjectPoints;
 
    ChangePointCloudNormalsAction( std::string name, const std::shared_ptr<ObjectPoints>& obj ) :
        objPoints_{ obj },
        name_{ std::move( name ) }
    {
        if ( obj )
        {
            if ( auto pc = obj->pointCloud() )
                backupNormals_ = pc->normals;
        }
    }
 
    ChangePointCloudNormalsAction( std::string name, const std::shared_ptr<ObjectPoints>& obj, VertNormals && newNormals ) :
        objPoints_{ obj },
        backupNormals_{ std::move( newNormals ) },
        name_{ std::move( name ) }
    {
        action( HistoryAction::Type::Redo );
    }
 
    virtual std::string name() const override
    {
        return name_;
    }
 
    virtual void action( HistoryAction::Type ) override
    {
        if ( !objPoints_ || !objPoints_->varPointCloud() )
            return;
        
        std::swap( objPoints_->varPointCloud()->normals, backupNormals_ );
        setObjectDirty( objPoints_ );
    }
 
    static void setObjectDirty( const std::shared_ptr<ObjectPoints>& obj )
    {
        if ( obj )
            obj->setDirtyFlags( DIRTY_RENDER_NORMALS );
    }
 
    [[nodiscard]] virtual size_t heapBytes() const override
    {
        return name_.capacity() + backupNormals_.heapBytes();
    }
 
private:
    std::shared_ptr<ObjectPoints> objPoints_;
    VertNormals backupNormals_;
 
    std::string name_;
};
 
class ChangeOneNormalInCloudAction : public HistoryAction
{
public:
    using Obj = ObjectPoints;
 
    ChangeOneNormalInCloudAction( std::string name, const std::shared_ptr<ObjectPoints>& obj, VertId pointId ) :
        objPoints_{ obj },
        pointId_{ pointId },
        name_{ std::move( name ) }
    {
        if ( obj )
        {
            if ( auto m = obj->pointCloud() )
                if ( m->normals.size() > pointId_ )
                    safeNormal_ = m->normals[pointId_];
        }
    }
 
    ChangeOneNormalInCloudAction( std::string name, const std::shared_ptr<ObjectPoints>& obj, VertId pointId, const Vector3f & newNormal ) :
        objPoints_{ obj },
        pointId_{ pointId },
        safeNormal_{ newNormal },
        name_{ std::move( name ) }
    {
        action( HistoryAction::Type::Redo );
    }
 
    virtual std::string name() const override
    {
        return name_;
    }
 
    virtual void action( HistoryAction::Type ) override
    {
        if ( !objPoints_ )
            return;
 
        if ( auto m = objPoints_->varPointCloud() )
        {
            if ( m->normals.size() > pointId_ )
            {
                std::swap( safeNormal_, m->normals[pointId_] );
                objPoints_->setDirtyFlags( DIRTY_RENDER_NORMALS );
            }
        }
    }
 
    static void setObjectDirty( const std::shared_ptr<ObjectPoints>& obj )
    {
        if ( obj )
            obj->setDirtyFlags( DIRTY_RENDER_NORMALS );
    }
 
    [[nodiscard]] virtual size_t heapBytes() const override
    {
        return name_.capacity();
    }
 
private:
    std::shared_ptr<ObjectPoints> objPoints_;
    VertId pointId_;
    Vector3f safeNormal_;
 
    std::string name_;
};
 
} //namespace MR