PointSmootherBlockFilter.hpp

#pragma once
#include "RasterPointScanner/Headers/ViewportPointSmoother.hpp"
#include "BlockModel/Headers/BlockModelRaster.h"
namespace NDEVR
{
    template<class t_point_type, class t_smooth_type>
    class PointSmootherBlockFilter : public BlockModelFilteringCache<t_point_type, uint04>
    {
    public:
        PointSmootherBlockFilter(ViewportPointSmoother<t_smooth_type>* smoother)
            : m_smoother(smoother)
        {}
        bool shouldFilter(const Vertex<3, uint04>& location, const t_point_type& node, uint04) const final override
        {
            return m_smoother->contains(location.as<3, fltp04>(), node);
        }
        void onFiltered(const Vertex<3, uint04>& location, const t_point_type& node, uint04 weight) const final override
        {
            Vector<3, fltp04> flocation = location.as<3, fltp04>();
            if constexpr (t_point_type::HasOffset())
                flocation += node.offset();
            m_filtered_locations.add(flocation);
            m_filtered_colors.add(node.template getAs<t_smooth_type>());
            m_filtered_weights.add(weight);
        }
        void setSmoother(ViewportPointSmoother<t_smooth_type>* smoother)
        {
            m_smoother = smoother;
        }
        IntersectionTypes classifyToFilter(const Bounds<3, uint04>& value) const override
        {
            return m_smoother->classifyBlock(value.as<3, fltp04>());
        }
        void* gridLocationLockPtr()
        {
            return &m_filtered_locations;
        }
        void transferActiveLocations()
        {
            WLock lock(gridLocationLockPtr());
            m_smoother->addBlockPoints(m_filtered_locations, m_filtered_colors, m_filtered_weights);
            m_filtered_locations.clear();
            m_filtered_colors.clear();
            m_filtered_weights.clear();
        }
        void clearAll()
        {
            WLock lock(gridLocationLockPtr());
            m_filtered_locations.clear();
            m_filtered_colors.clear();
            m_filtered_weights.clear();
        }
        mutable Buffer<Vertex<3, fltp04>> m_filtered_locations;
        mutable Buffer<t_smooth_type> m_filtered_colors;
        mutable Buffer<uint04> m_filtered_weights;
        ViewportPointSmoother<t_smooth_type>* m_smoother;
    };
}