COM_Debug.h

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2013 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once
 
#include <map>
#include <string>
 
#include "BLI_vector.hh"
 
#include "COM_ExecutionSystem.h"
#include "COM_MemoryBuffer.h"
#include "COM_Node.h"
 
namespace blender::compositor {
 
static constexpr bool COM_EXPORT_GRAPHVIZ = false;
static constexpr bool COM_GRAPHVIZ_SHOW_NODE_NAME = false;
 
/* Saves operations results to image files. */
static constexpr bool COM_EXPORT_OPERATION_BUFFERS = false;
 
class Node;
class NodeOperation;
class ExecutionSystem;
 
class DebugInfo {
 public:
  typedef std::map<const Node *, std::string> NodeNameMap;
  typedef std::map<const NodeOperation *, std::string> OpNameMap;
 
  static std::string node_name(const Node *node);
  static std::string operation_name(const NodeOperation *op);
 
 private:
  static int file_index_;
  static NodeNameMap node_names_;
  static OpNameMap op_names_;
  static std::string current_node_name_;
  static std::string current_op_name_;
 
 public:
  static void convert_started()
  {
    if (COM_EXPORT_GRAPHVIZ) {
      op_names_.clear();
    }
  }
 
  static void execute_started()
  {
    if (COM_EXPORT_GRAPHVIZ) {
      file_index_ = 1;
    }
    if (COM_EXPORT_OPERATION_BUFFERS) {
      delete_operation_exports();
    }
  };
 
  static void node_added(const Node *node)
  {
    if (COM_EXPORT_GRAPHVIZ) {
      node_names_[node] = std::string(node->get_bnode() ? node->get_bnode()->name : "");
    }
  }
 
  static void node_to_operations(const Node *node)
  {
    if (COM_EXPORT_GRAPHVIZ) {
      current_node_name_ = node_names_[node];
    }
  }
 
  static void operation_added(const NodeOperation *operation)
  {
    if (COM_EXPORT_GRAPHVIZ) {
      op_names_[operation] = current_node_name_;
    }
  };
 
  static void operation_read_write_buffer(const NodeOperation *operation)
  {
    if (COM_EXPORT_GRAPHVIZ) {
      current_op_name_ = op_names_[operation];
    }
  };
 
  static void operation_rendered(const NodeOperation *op, MemoryBuffer *render)
  {
    /* Don't export constant operations as there are too many and it's rarely useful. */
    if (COM_EXPORT_OPERATION_BUFFERS && render && !render->is_a_single_elem()) {
      export_operation(op, render);
    }
  }
 
  static void graphviz(const ExecutionSystem *system, StringRefNull name = "");
 
 protected:
  static int graphviz_operation(const ExecutionSystem *system,
                                NodeOperation *operation,
                                char *str,
                                int maxlen);
  static int graphviz_legend_color(const char *name, const char *color, char *str, int maxlen);
  static int graphviz_legend_line(
      const char *name, const char *color, const char *style, char *str, int maxlen);
  static int graphviz_legend_group(
      const char *name, const char *color, const char *style, char *str, int maxlen);
  static int graphviz_legend(char *str, int maxlen);
  static bool graphviz_system(const ExecutionSystem *system, char *str, int maxlen);
 
  static void export_operation(const NodeOperation *op, MemoryBuffer *render);
  static void delete_operation_exports();
};
 
}  // namespace blender::compositor