ClusterPainter.hpp

Go to the documentation of this file.

/*
 * ClusterPainter.hpp
 *
 *  @date 18.06.2017
 *  @author Johan M. von Behren <johan@vonbehren.eu>
 */

#ifndef LVR2_ALGORITHM_CLUSTERPAINTER_H_
#define LVR2_ALGORITHM_CLUSTERPAINTER_H_

#include <cstdint>
#include <array>

using std::array;

#include "lvr2/geometry/Handles.hpp"
#include "lvr2/util/ClusterBiMap.hpp"
#include "lvr2/reconstruction/PointsetSurface.hpp"
#include "lvr2/attrmaps/AttrMaps.hpp"
#include "lvr2/algorithm/ColorAlgorithms.hpp"

namespace lvr2
{

class ClusterPainter
{
public:
    ClusterPainter(const ClusterBiMap<FaceHandle>& clusterBiMap) : m_clusterBiMap(clusterBiMap) {};

    template<typename BaseVecT>
    DenseClusterMap<Rgb8Color> simpsons(const BaseMesh<BaseVecT>& mesh) const;

private:
    ClusterBiMap<FaceHandle> m_clusterBiMap;
    inline Rgb8Color getSimpsonColorForIdx(size_t idx) const
    {
      return {
          static_cast<uint8_t>(fabs(cos(idx)) * 255),
          static_cast<uint8_t>(fabs(sin(idx * 30)) * 255),
          static_cast<uint8_t>(fabs(sin(idx * 2)) * 255)
      };
    }
};

} // namespace lvr2

#include "lvr2/algorithm/ClusterPainter.tcc"

#endif /* LVR2_ALGORITHM_CLUSTERPAINTER_H_ */