codac_VIBesFigTube.h

Go to the documentation of this file.

 
#ifndef __CODAC_VIBESFIGTUBE_H__
#define __CODAC_VIBESFIGTUBE_H__
 
#include "codac_VIBesFig.h"
#include "codac_Tube.h"
#include "codac_Slice.h"
#include "codac_Trajectory.h"
 
namespace codac
{
  #define DEFAULT_TUBE_NAME         "[?](·)"
  #define DEFAULT_TRAJ_NAME         "?(·)"
  #define TRAJ_NB_DISPLAYED_POINTS  10000
  
  // HTML color codes:
  #define DEFAULT_TRAJ_COLOR        "#004257"
  #define DEFAULT_FRGRND_COLOR      "#a2a2a2[#a2a2a2]"
  #define DEFAULT_BCKGRND_COLOR     "#d2d2d2[#d2d2d2]"
  #define DEFAULT_SLICES_COLOR      "#828282[#F0F0F0]"
  #define DEFAULT_GATES_COLOR       "#0084AF[#0084AF]"
  #define DEFAULT_POLYGONS_COLOR    "#00536E[#2696BA]"
 
  enum class TubeColorType { FOREGROUND, BACKGROUND, SLICES, GATES, POLYGONS };
 
  class VIBesFigTube : public VIBesFig
  {
    public:
 
 
      VIBesFigTube(const std::string& fig_name, const Tube *tube = nullptr, const Trajectory *traj = nullptr);
 
      ~VIBesFigTube();
 
      void show();
 
      void show(bool detail_slices);
 
      void set_cursor(double t);
 
      void show_cursor(bool display = true);
 
 
      void add_tube(const Tube *tube, const std::string& name, const std::string& color_frgrnd = DEFAULT_FRGRND_COLOR, const std::string& color_bckgrnd = DEFAULT_BCKGRND_COLOR);
 
      void add_tubes(const TubeVector *tubevector, const std::string& name, const std::string& color_frgrnd = DEFAULT_FRGRND_COLOR, const std::string& color_bckgrnd = DEFAULT_BCKGRND_COLOR);
 
      void add_tubes(const TubeVector *tubevector, int start_index, int end_index, const std::string& name, const std::string& color_frgrnd = DEFAULT_FRGRND_COLOR, const std::string& color_bckgrnd = DEFAULT_BCKGRND_COLOR);
 
      void set_tube_name(const Tube *tube, const std::string& name);
 
      void set_tube_derivative(const Tube *tube, const Tube *derivative);
 
      void set_tube_color(const Tube *tube, const std::string& color_frgrnd, const std::string& color_bckgrnd = DEFAULT_BCKGRND_COLOR);
 
      void set_tube_color(const Tube *tube, TubeColorType color_type, const std::string& color);
 
      void reset_tube_background(const Tube *tube);
 
      void remove_tube(const Tube *tube);
 
 
      void add_trajectory(const Trajectory *traj, const std::string& name, const std::string& color = DEFAULT_TRAJ_COLOR);
 
      void add_trajectories(const TrajectoryVector *trajvector, const std::string& name, const std::string& color = DEFAULT_TRAJ_COLOR);
 
      void add_trajectories(const TrajectoryVector *trajvector, int start_index, int end_index, const std::string& name, const std::string& color = DEFAULT_TRAJ_COLOR);
 
      void set_trajectory_name(const Trajectory *traj, const std::string& name);
 
      void set_trajectory_color(const Trajectory *traj, const std::string& color);
      
      void set_trajectory_points_size(const Trajectory *traj, double points_size);
 
      void remove_trajectory(const Trajectory *traj);
 
 
    protected:
 
      void create_group_color(const Tube *tube, TubeColorType color_type);
 
      void create_groups_color(const Tube *tube);
 
      const IntervalVector draw_tube(const Tube *tube, bool detail_slices = false);
 
      void draw_slice(const Slice& slice, const vibes::Params& params);
 
      void draw_slice(const Slice& slice, const Slice& deriv_slice, const vibes::Params& params_slice, const vibes::Params& params_polygon);
 
      void draw_gate(const Interval& gate, double t, const vibes::Params& params);
 
      const IntervalVector draw_trajectory(const Trajectory *traj);
 
    protected:
 
      struct FigTubeParams
      {
        std::string name; 
        std::map<TubeColorType,std::string> m_colors; 
        const Tube *tube_copy = nullptr; 
        const Tube *tube_derivative = nullptr; 
      };
 
      struct FigTrajParams
      {
        std::string name; 
        std::string color; 
        float points_size = 0.; 
      };
 
      std::map<const Tube*,FigTubeParams> m_map_tubes; 
      std::map<const Trajectory*,FigTrajParams> m_map_trajs; 
      bool m_display_cursor = false; 
      double m_cursor; 
 
      friend class VIBesFigTubeVector;
  };
}
 
#endif