Head

GCC Code Coverage Report

Directory:	.		Exec	Total	Coverage
File:	src/prop/skolem_def_manager.cpp	Lines:	70	83	84.3 %
Date:	2021-05-22	Branches:	92	248	37.1 %


/******************************************************************************
 * Top contributors (to current version):
 *   Andrew Reynolds
 *
 * 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.
 * ****************************************************************************
 *
 * Skolem definition manager.
 */

#include "prop/skolem_def_manager.h"

#include "expr/attribute.h"

namespace cvc5 {
namespace prop {

SkolemDefManager::SkolemDefManager(context::Context* context,
                                   context::UserContext* userContext)
    : d_skDefs(userContext), d_skActive(context)
{
}

SkolemDefManager::~SkolemDefManager() {}

void SkolemDefManager::notifySkolemDefinition(TNode skolem, Node def)
{
  // Notice that skolem may have kind SKOLEM or BOOLEAN_TERM_VARIABLE
  Trace("sk-defs") << "notifySkolemDefinition: " << def << " for " << skolem
                   << std::endl;
  // in very rare cases, a skolem may be generated twice for terms that are
  // equivalent up to purification
  if (d_skDefs.find(skolem) == d_skDefs.end())
  {
    d_skDefs.insert(skolem, def);
  }
}

TNode SkolemDefManager::getDefinitionForSkolem(TNode skolem) const
{
  NodeNodeMap::const_iterator it = d_skDefs.find(skolem);
  Assert(it != d_skDefs.end()) << "No skolem def for " << skolem;
  return it->second;
}

void SkolemDefManager::notifyAsserted(TNode literal,
                                      std::vector<TNode>& activatedSkolems,
                                      bool useDefs)
{
  std::unordered_set<Node> skolems;
  getSkolems(literal, skolems);
  for (const Node& k : skolems)
  {
    if (d_skActive.find(k) != d_skActive.end())
    {
      // already active
      continue;
    }
    d_skActive.insert(k);
    if (useDefs)
    {
      // add its definition to the activated list
      NodeNodeMap::const_iterator it = d_skDefs.find(k);
      Assert(it != d_skDefs.end());
      activatedSkolems.push_back(it->second);
    }
    else
    {
      // add to the activated list
      activatedSkolems.push_back(k);
    }
  }
}

struct HasSkolemTag
{
};
struct HasSkolemComputedTag
{
};
/** Attribute true for nodes with skolems in them */
typedef expr::Attribute<HasSkolemTag, bool> HasSkolemAttr;
/** Attribute true for nodes where we have computed the above attribute */
typedef expr::Attribute<HasSkolemComputedTag, bool> HasSkolemComputedAttr;

bool SkolemDefManager::hasSkolems(TNode n) const
{
  std::unordered_set<TNode> visited;
  std::unordered_set<TNode>::iterator it;
  std::vector<TNode> visit;
  TNode cur;
  visit.push_back(n);
  do
  {
    cur = visit.back();
    if (cur.getAttribute(HasSkolemComputedAttr()))
    {
      visit.pop_back();
      // already computed
      continue;
    }
    it = visited.find(cur);
    if (it == visited.end())
    {
      visited.insert(cur);
      if (cur.getNumChildren() == 0)
      {
        visit.pop_back();
        bool hasSkolem = false;
        if (cur.isVar())
        {
          hasSkolem = (d_skDefs.find(cur) != d_skDefs.end());
        }
        cur.setAttribute(HasSkolemAttr(), hasSkolem);
        cur.setAttribute(HasSkolemComputedAttr(), true);
      }
      else
      {
        visit.insert(visit.end(), cur.begin(), cur.end());
      }
    }
    else
    {
      visit.pop_back();
      bool hasSkolem = false;
      for (TNode i : cur)
      {
        Assert(i.getAttribute(HasSkolemComputedAttr()));
        if (i.getAttribute(HasSkolemAttr()))
        {
          hasSkolem = true;
          break;
        }
      }
      cur.setAttribute(HasSkolemAttr(), hasSkolem);
      cur.setAttribute(HasSkolemComputedAttr(), true);
    }
  } while (!visit.empty());
  Assert(n.getAttribute(HasSkolemComputedAttr()));
  return n.getAttribute(HasSkolemAttr());
}

void SkolemDefManager::getSkolems(TNode n,
                                  std::unordered_set<Node>& skolems) const
{
  std::unordered_set<TNode> visited;
  std::unordered_set<TNode>::iterator it;
  std::vector<TNode> visit;
  TNode cur;
  visit.push_back(n);
  do
  {
    cur = visit.back();
    visit.pop_back();
    if (!hasSkolems(cur))
    {
      // does not have skolems, continue
      continue;
    }
    it = visited.find(cur);
    if (it == visited.end())
    {
      visited.insert(cur);
      if (cur.isVar())
      {
        if (d_skDefs.find(cur) != d_skDefs.end())
        {
          skolems.insert(cur);
        }
      }
      else
      {
        visit.insert(visit.end(), cur.begin(), cur.end());
      }
    }
  } while (!visit.empty());
}

}  // namespace prop
}  // namespace cvc5


Generated by: GCOVR (Version 3.2)

Line	Exec	Source
1		/******************************************************************************
2		* Top contributors (to current version):
3		* Andrew Reynolds
4		*
5		* This file is part of the cvc5 project.
6		*
7		* Copyright (c) 2009-2021 by the authors listed in the file AUTHORS
8		* in the top-level source directory and their institutional affiliations.
9		* All rights reserved. See the file COPYING in the top-level source
10		* directory for licensing information.
11		* ****************************************************************************
12		*
13		* Skolem definition manager.
14		*/
15
16		#include "prop/skolem_def_manager.h"
17
18		#include "expr/attribute.h"
19
20		namespace cvc5 {
21		namespace prop {
22
23	9494	SkolemDefManager::SkolemDefManager(context::Context* context,
24	9494	context::UserContext* userContext)
25	9494	: d_skDefs(userContext), d_skActive(context)
26		{
27	9494	}
28
29	9494	SkolemDefManager::~SkolemDefManager() {}
30
31	30870	void SkolemDefManager::notifySkolemDefinition(TNode skolem, Node def)
32		{
33		// Notice that skolem may have kind SKOLEM or BOOLEAN_TERM_VARIABLE
34	61740	Trace("sk-defs") << "notifySkolemDefinition: " << def << " for " << skolem
35	30870	<< std::endl;
36		// in very rare cases, a skolem may be generated twice for terms that are
37		// equivalent up to purification
38	30870	if (d_skDefs.find(skolem) == d_skDefs.end())
39		{
40	30864	d_skDefs.insert(skolem, def);
41		}
42	30870	}
43
44	2185	TNode SkolemDefManager::getDefinitionForSkolem(TNode skolem) const
45		{
46	2185	NodeNodeMap::const_iterator it = d_skDefs.find(skolem);
47	2185	Assert(it != d_skDefs.end()) << "No skolem def for " << skolem;
48	2185	return it->second;
49		}
50
51		void SkolemDefManager::notifyAsserted(TNode literal,
52		std::vector<TNode>& activatedSkolems,
53		bool useDefs)
54		{
55		std::unordered_set<Node> skolems;
56		getSkolems(literal, skolems);
57		for (const Node& k : skolems)
58		{
59		if (d_skActive.find(k) != d_skActive.end())
60		{
61		// already active
62		continue;
63		}
64		d_skActive.insert(k);
65		if (useDefs)
66		{
67		// add its definition to the activated list
68		NodeNodeMap::const_iterator it = d_skDefs.find(k);
69		Assert(it != d_skDefs.end());
70		activatedSkolems.push_back(it->second);
71		}
72		else
73		{
74		// add to the activated list
75		activatedSkolems.push_back(k);
76		}
77		}
78		}
79
80		struct HasSkolemTag
81		{
82		};
83		struct HasSkolemComputedTag
84		{
85		};
86		/** Attribute true for nodes with skolems in them */
87		typedef expr::Attribute<HasSkolemTag, bool> HasSkolemAttr;
88		/** Attribute true for nodes where we have computed the above attribute */
89		typedef expr::Attribute<HasSkolemComputedTag, bool> HasSkolemComputedAttr;
90
91	16708	bool SkolemDefManager::hasSkolems(TNode n) const
92		{
93	33416	std::unordered_set<TNode> visited;
94	16708	std::unordered_set<TNode>::iterator it;
95	33416	std::vector<TNode> visit;
96	33416	TNode cur;
97	16708	visit.push_back(n);
98	113783	do
99		{
100	130491	cur = visit.back();
101	176865	if (cur.getAttribute(HasSkolemComputedAttr()))
102		{
103	46374	visit.pop_back();
104		// already computed
105	46374	continue;
106		}
107	84117	it = visited.find(cur);
108	84117	if (it == visited.end())
109		{
110	47429	visited.insert(cur);
111	47429	if (cur.getNumChildren() == 0)
112		{
113	10741	visit.pop_back();
114	10741	bool hasSkolem = false;
115	10741	if (cur.isVar())
116		{
117	7494	hasSkolem = (d_skDefs.find(cur) != d_skDefs.end());
118		}
119	10741	cur.setAttribute(HasSkolemAttr(), hasSkolem);
120	10741	cur.setAttribute(HasSkolemComputedAttr(), true);
121		}
122		else
123		{
124	36688	visit.insert(visit.end(), cur.begin(), cur.end());
125		}
126		}
127		else
128		{
129	36688	visit.pop_back();
130	36688	bool hasSkolem = false;
131	103140	for (TNode i : cur)
132		{
133	72471	Assert(i.getAttribute(HasSkolemComputedAttr()));
134	72471	if (i.getAttribute(HasSkolemAttr()))
135		{
136	6019	hasSkolem = true;
137	6019	break;
138		}
139		}
140	36688	cur.setAttribute(HasSkolemAttr(), hasSkolem);
141	36688	cur.setAttribute(HasSkolemComputedAttr(), true);
142		}
143	130491	} while (!visit.empty());
144	16708	Assert(n.getAttribute(HasSkolemComputedAttr()));
145	33416	return n.getAttribute(HasSkolemAttr());
146		}
147
148	2619	void SkolemDefManager::getSkolems(TNode n,
149		std::unordered_set<Node>& skolems) const
150		{
151	5238	std::unordered_set<TNode> visited;
152	2619	std::unordered_set<TNode>::iterator it;
153	5238	std::vector<TNode> visit;
154	5238	TNode cur;
155	2619	visit.push_back(n);
156	14089	do
157		{
158	16708	cur = visit.back();
159	16708	visit.pop_back();
160	16708	if (!hasSkolems(cur))
161		{
162		// does not have skolems, continue
163	6599	continue;
164		}
165	10109	it = visited.find(cur);
166	10109	if (it == visited.end())
167		{
168	8392	visited.insert(cur);
169	8392	if (cur.isVar())
170		{
171	2185	if (d_skDefs.find(cur) != d_skDefs.end())
172		{
173	2185	skolems.insert(cur);
174		}
175		}
176		else
177		{
178	6207	visit.insert(visit.end(), cur.begin(), cur.end());
179		}
180		}
181	16708	} while (!visit.empty());
182	2619	}
183
184		} // namespace prop
185	28191	} // namespace cvc5