Visualization.hpp

Go to the documentation of this file.

#pragma once

 
#include "models/TreeCondition.hpp"
#include "models/TreeResponse.hpp"
#include "utils/Types.hpp"
 
#include <vector>
#include <utility>
#include <string>
 
namespace ppforest2::viz {
  // ===================================================================
  // Node Data Collection
  // ===================================================================

  struct NodeData {
    bool is_leaf;
    int depth;
    types::FeatureVector projector;
    types::Feature threshold;
    types::Response value;
    std::vector<types::Feature> projected_values;
    std::vector<types::Response> groups;
  };

  struct NodeDataVisitor : public TreeNode::Visitor {
    types::FeatureMatrix const& x;  
    types::ResponseVector const& y; 
    std::vector<int> indices;       
    int depth;                      
    std::vector<NodeData> nodes;    
 
    NodeDataVisitor(types::FeatureMatrix const& x, types::ResponseVector const& y);
 
    void visit(TreeCondition const& node) override;
    void visit(TreeResponse const& node) override;
  };
 
  // ===================================================================
  // Decision Boundary and Region Types
  // ===================================================================

  struct HalfSpace {
    types::FeatureVector projector;
    types::Feature threshold;
    bool is_lower; 
  };

  struct BoundarySegment {
    types::Feature x_start, y_start, x_end, y_end;
    int depth;
  };

  struct RegionPolygon {
    std::vector<std::pair<types::Feature, types::Feature>> vertices;
    types::Response predicted_group; 
  };
 
  // ===================================================================
  // 2D Projection Helpers
  // ===================================================================

  types::FeatureVector project_2d(types::FeatureVector const& full_proj, int var_i, int var_j);

  types::Feature adjust_threshold(types::FeatureVector const& full_proj,
                                  types::Feature thr,
                                  std::vector<std::pair<int, types::Feature>> const& fixed_vars);
 
  // ===================================================================
  // Parametric Line Clipping
  // ===================================================================

  bool clip_param_to_range(types::Feature origin,
                           types::Feature direction,
                           types::Feature range_min,
                           types::Feature range_max,
                           types::Feature& u_min,
                           types::Feature& u_max);

  bool clip_boundary_2d(types::FeatureVector const& a,
                        types::Feature threshold,
                        std::vector<HalfSpace> const& constraints,
                        types::Feature x_min,
                        types::Feature x_max,
                        types::Feature y_min,
                        types::Feature y_max,
                        BoundarySegment& segment,
                        int depth);
 
  // ===================================================================
  // Polygon Clipping (Sutherland–Hodgman)
  // ===================================================================

  using Polygon = std::vector<std::pair<types::Feature, types::Feature>>;

  Polygon clip_polygon_halfspace(Polygon const& polygon,
                                 types::FeatureVector const& normal,
                                 types::Feature threshold,
                                 bool is_lower);
 
  // ===================================================================
  // Boundary Visitor
  // ===================================================================

  struct BoundaryVisitor : public TreeNode::Visitor {
    int var_i, var_j;                                       
    std::vector<std::pair<int, types::Feature>> fixed_vars; 
    types::Feature x_min, x_max, y_min, y_max;              
    int depth;
    std::vector<HalfSpace> constraints;    
    std::vector<BoundarySegment> segments; 
 
    BoundaryVisitor(int var_i,
                    int var_j,
                    std::vector<std::pair<int, types::Feature>> const& fixed_vars,
                    types::Feature x_min,
                    types::Feature x_max,
                    types::Feature y_min,
                    types::Feature y_max);
 
    void visit(TreeCondition const& node) override;
    void visit(TreeResponse const& node) override;
  };
 
  // ===================================================================
  // Region Visitor
  // ===================================================================

  struct RegionVisitor : public TreeNode::Visitor {
    int var_i, var_j;                                       
    std::vector<std::pair<int, types::Feature>> fixed_vars; 
    types::Feature x_min, x_max, y_min, y_max;              
    std::vector<HalfSpace> constraints;                     
    std::vector<RegionPolygon> regions;                     
 
    RegionVisitor(int var_i,
                  int var_j,
                  std::vector<std::pair<int, types::Feature>> const& fixed_vars,
                  types::Feature x_min,
                  types::Feature x_max,
                  types::Feature y_min,
                  types::Feature y_max);
 
    void visit(TreeCondition const& node) override;
    void visit(TreeResponse const& node) override;
  };
 
  // ===================================================================
  // Tree Layout
  // ===================================================================

  struct LayoutParams {
    types::Feature y_spacing = 1.5f; 
    types::Feature node_w    = 0.8f; 
    types::Feature node_h    = 0.7f; 
    types::Feature leaf_w    = 0.5f; 
    types::Feature leaf_h    = 0.3f; 
    types::Feature gap       = 0.2f; 
  };

  struct LayoutNode {
    types::Feature x, y; 
    bool is_leaf;
    int node_idx; 
  };

  struct LayoutEdge {
    types::Feature from_x, from_y, to_x, to_y;
    std::string label; 
  };

  struct TreeLayout {
    std::vector<LayoutNode> nodes;
    std::vector<LayoutEdge> edges;
  };

  TreeLayout compute_tree_layout(TreeNode const& root, LayoutParams const& params = LayoutParams());
}