aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/libs/llvm14/tools/llvm-xray/xray-graph-diff.h
blob: 5d12c563f47c497a415ad3c2be55ed10192e57a1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
//===-- xray-graph-diff.h - XRay Graph Diff Renderer ------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// Generate a DOT file to represent the difference between the function call
// graph of two differnent traces.
//
//===----------------------------------------------------------------------===//

#ifndef XRAY_GRAPH_DIFF_H
#define XRAY_GRAPH_DIFF_H

#include "xray-graph.h"
#include "llvm/ADT/StringMap.h"
#include "llvm/XRay/Graph.h"

namespace llvm {
namespace xray {

// This class creates a graph representing the difference between two
// xray-graphs And allows you to print it to a dot file, with optional color
// coding.
class GraphDiffRenderer {
  static const int N = 2;

public:
  using StatType = GraphRenderer::StatType;
  using TimeStat = GraphRenderer::TimeStat;

  using GREdgeValueType = GraphRenderer::GraphT::EdgeValueType;
  using GRVertexValueType = GraphRenderer::GraphT::VertexValueType;

  struct EdgeAttribute {
    std::array<const GREdgeValueType *, N> CorrEdgePtr = {};
  };

  struct VertexAttribute {
    std::array<const GRVertexValueType *, N> CorrVertexPtr = {};
  };

  using GraphT = Graph<VertexAttribute, EdgeAttribute, StringRef>;

  class Factory {
    std::array<std::reference_wrapper<const GraphRenderer::GraphT>, N> G;

  public:
    template <typename... Ts> Factory(Ts &... Args) : G{{Args...}} {}

    Expected<GraphDiffRenderer> getGraphDiffRenderer();
  };

private:
  GraphT G;

  GraphDiffRenderer() = default;

public:
  void exportGraphAsDOT(raw_ostream &OS, StatType EdgeLabel = StatType::NONE,
                        StatType EdgeColor = StatType::NONE,
                        StatType VertexLabel = StatType::NONE,
                        StatType VertexColor = StatType::NONE,
                        int TruncLen = 40);

  const GraphT &getGraph() { return G; }
};
} // namespace xray
} // namespace llvm

#endif