summaryrefslogtreecommitdiff
path: root/src/proof/proof_node_to_sexpr.h
blob: 83d719aaf3d84e76acd175240d6c74f08b8b857b (plain)
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/******************************************************************************
 * Top contributors (to current version):
 *   Andrew Reynolds, Haniel Barbosa, Gereon Kremer
 *
 * This file is part of the cvc5 project.
 *
 * Copyright (c) 2009-2021 by the authors listed in the file AUTHORS
 * in the top-level source directory and their institutional affiliations.
 * All rights reserved.  See the file COPYING in the top-level source
 * directory for licensing information.
 * ****************************************************************************
 *
 * Conversion from ProofNode to s-expressions.
 */

#include "cvc5_private.h"

#ifndef CVC5__PROOF__PROOF_NODE_TO_SEXPR_H
#define CVC5__PROOF__PROOF_NODE_TO_SEXPR_H

#include <map>

#include "expr/kind.h"
#include "expr/node.h"
#include "proof/method_id.h"
#include "proof/proof_rule.h"
#include "theory/inference_id.h"
#include "theory/theory_id.h"

namespace cvc5 {

class ProofNode;

/** A class to convert ProofNode objects to s-expressions */
class ProofNodeToSExpr
{
 public:
  ProofNodeToSExpr();
  ~ProofNodeToSExpr() {}
  /** Convert the given proof node to an s-expression
   *
   * This is useful for operations where it is useful to view a ProofNode as
   * a Node. Printing is one such example, where a ProofNode can be printed
   * as a dag after this conversion.
   *
   * The s-expression for a ProofNode has the form:
   *   (SEXPR (VAR "<d_rule>") S1 ... Sn (VAR ":args") (SEXPR <d_args>))
   * where S1, ..., Sn are the s-expressions for its <d_children>.
   */
  Node convertToSExpr(const ProofNode* pn);

 private:
  /** argument format, determines how to print an argument */
  enum class ArgFormat
  {
    // just use the argument itself
    DEFAULT,
    // print the argument as a kind
    KIND,
    // print the argument as a theory id
    THEORY_ID,
    // print the argument as a method id
    METHOD_ID,
    // print the argument as an inference id
    INFERENCE_ID,
    // print a variable whose name is the term (see getOrMkNodeVariable)
    NODE_VAR
  };
  /** map proof rules to a variable */
  std::map<PfRule, Node> d_pfrMap;
  /** map kind to a variable displaying the kind they represent */
  std::map<Kind, Node> d_kindMap;
  /** map theory ids to a variable displaying the theory id they represent */
  std::map<theory::TheoryId, Node> d_tidMap;
  /** map method ids to a variable displaying the method id they represent */
  std::map<MethodId, Node> d_midMap;
  /** map infer ids to a variable displaying the method id they represent */
  std::map<theory::InferenceId, Node> d_iidMap;
  /** Dummy ":args" marker */
  Node d_argsMarker;
  /** Dummy ":conclusion" marker */
  Node d_conclusionMarker;
  /** map proof nodes to their s-expression */
  std::map<const ProofNode*, Node> d_pnMap;
  /**
   * map nodes to a bound variable, used for nodes that have special AST status
   * like builtin operators
   */
  std::map<TNode, Node> d_nodeMap;
  /** get or make pf rule variable */
  Node getOrMkPfRuleVariable(PfRule r);
  /** get or make kind variable from the kind embedded in n */
  Node getOrMkKindVariable(TNode n);
  /** get or make theory id variable */
  Node getOrMkTheoryIdVariable(TNode n);
  /** get or make method id variable */
  Node getOrMkMethodIdVariable(TNode n);
  /** get or make inference id variable */
  Node getOrMkInferenceIdVariable(TNode n);
  /**
   * Get or make node variable that prints the same as n but has SEXPR type.
   * This is used to ensure the type checker does not complain when trying to
   * print e.g. builtin operators as first-class terms in the SEXPR.
   */
  Node getOrMkNodeVariable(TNode n);
  /** get argument based on the provided format */
  Node getArgument(Node arg, ArgFormat f);
  /** get argument format for proof node */
  ArgFormat getArgumentFormat(const ProofNode* pn, size_t i);
};

}  // namespace cvc5

#endif /* CVC5__PROOF__PROOF_RULE_H */
generated by cgit on debian on lair
contact matthew@masot.net with questions or feedback