AbstractExtendedVisualizer.h

Go to the documentation of this file.

#pragma once

// Global includes
#include <string>

// Package includes
#include <boost/shared_ptr.hpp>

#include <std_msgs/ColorRGBA.h>

// Local includes
#include "visualization/psm/helper/AbstractVisualizer.h"

namespace Visualization {
  
  class AbstractExtendedVisualizer : public AbstractVisualizer {
  public:
    
    AbstractExtendedVisualizer();
    
    virtual ~AbstractExtendedVisualizer();
    
    void setColors(const std_msgs::ColorRGBA& pFirstColor,
                   const std_msgs::ColorRGBA& pSecondColor,
                   const std_msgs::ColorRGBA& pThirdColor);
    
    void setScaleFactor(const double pScaleFactor);
    
  protected:
    
    const std_msgs::ColorRGBA getColor(unsigned int pNumber);
    
    double getScaleFactor();
    
  private:
    
    double mScaleFactor;
    
    std::vector<std_msgs::ColorRGBA> mColors;
  };
}