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