Head

GCC Code Coverage Report

Directory:	.		Exec	Total	Coverage
File:	src/theory/bv/theory_bv_rewrite_rules_core.h	Lines:	144	144	100.0 %
Date:	2021-09-07	Branches:	265	460	57.6 %


/******************************************************************************
 * Top contributors (to current version):
 *   Dejan Jovanovic, Liana Hadarean, Clark Barrett
 *
 * 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.
 * ****************************************************************************
 *
 * [[ Add one-line brief description here ]]
 *
 * [[ Add lengthier description here ]]
 * \todo document this file
 */

#include "cvc5_private.h"

#pragma once

#include "theory/bv/theory_bv_rewrite_rules.h"
#include "theory/bv/theory_bv_utils.h"

namespace cvc5 {
namespace theory {
namespace bv {

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<ConcatFlatten>::applies(TNode node) {
  return (node.getKind() == kind::BITVECTOR_CONCAT);
}

template<> inline
Node RewriteRule<ConcatFlatten>::apply(TNode node) {
  Debug("bv-rewrite") << "RewriteRule<ConcatFlatten>(" << node << ")" << std::endl;
  NodeBuilder result(kind::BITVECTOR_CONCAT);
  std::vector<Node> processing_stack;
  processing_stack.push_back(node);
  while (!processing_stack.empty()) {
    Node current = processing_stack.back();
    processing_stack.pop_back();
    if (current.getKind() == kind::BITVECTOR_CONCAT) {
      for (int i = current.getNumChildren() - 1; i >= 0; i--)
        processing_stack.push_back(current[i]);
    } else {
      result << current;
    }
  }
  Node resultNode = result;
  Debug("bv-rewrite") << "RewriteRule<ConcatFlatten>(" << resultNode << ")" << std::endl;
  return resultNode;
}

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<ConcatExtractMerge>::applies(TNode node) {
  return (node.getKind() == kind::BITVECTOR_CONCAT);
}

template<> inline
Node RewriteRule<ConcatExtractMerge>::apply(TNode node) {

  Debug("bv-rewrite") << "RewriteRule<ConcatExtractMerge>(" << node << ")" << std::endl;

  std::vector<Node> mergedExtracts;

  Node current = node[0];
  bool mergeStarted = false;
  unsigned currentHigh = 0;
  unsigned currentLow  = 0;

  for(size_t i = 1, end = node.getNumChildren(); i < end; ++ i) {
    // The next candidate for merging
    Node next = node[i];
    // If the current is not an extract we just go to the next
    if (current.getKind() != kind::BITVECTOR_EXTRACT) {
      mergedExtracts.push_back(current);
      current = next;
      continue;
    }
    // If it is an extract and the first one, get the extract parameters
    else if (!mergeStarted) {
      currentHigh = utils::getExtractHigh(current);
      currentLow = utils::getExtractLow(current);
    }

    // If the next one can be merged, try to merge
    bool merged = false;
    if (next.getKind() == kind::BITVECTOR_EXTRACT && current[0] == next[0]) {
      // x[i : j] @ x[j - 1 : k] -> c x[i : k]
      unsigned nextHigh = utils::getExtractHigh(next);
      unsigned nextLow  = utils::getExtractLow(next);
      if(nextHigh + 1 == currentLow) {
        currentLow = nextLow;
        mergeStarted = true;
        merged = true;
      }
    }
    // If we haven't merged anything, add the previous merge and continue with the next
    if (!merged) {
      if (!mergeStarted) mergedExtracts.push_back(current);
      else mergedExtracts.push_back(utils::mkExtract(current[0], currentHigh, currentLow));
      current = next;
      mergeStarted = false;
    }
  }

  // Add the last child
  if (!mergeStarted) mergedExtracts.push_back(current);
  else mergedExtracts.push_back(utils::mkExtract(current[0], currentHigh, currentLow));

  // Return the result
  return utils::mkConcat(mergedExtracts);
}

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<ConcatConstantMerge>::applies(TNode node) {
  return node.getKind() == kind::BITVECTOR_CONCAT;
}

template<> inline
Node RewriteRule<ConcatConstantMerge>::apply(TNode node) {

  Debug("bv-rewrite") << "RewriteRule<ConcatConstantMerge>(" << node << ")" << std::endl;

  std::vector<Node> mergedConstants;
  for (unsigned i = 0, end = node.getNumChildren(); i < end;) {
    if (node[i].getKind() != kind::CONST_BITVECTOR) {
      // If not a constant, just add it
      mergedConstants.push_back(node[i]);
      ++i;
    } else {
      // Find the longest sequence of constants
      unsigned j = i + 1;
      while (j < end) {
        if (node[j].getKind() != kind::CONST_BITVECTOR) {
          break;
        } else {
          ++ j;
        }
      }
      // Append all the constants
      BitVector current = node[i].getConst<BitVector>();
      for (unsigned k = i + 1; k < j; ++ k) {
        current = current.concat(node[k].getConst<BitVector>());
      }
      // Add the new merged constant
      mergedConstants.push_back(utils::mkConst(current));
      i = j;
    }
  }

  Debug("bv-rewrite") << "RewriteRule<ConcatConstantMerge>(" << node << ") => " << utils::mkConcat(mergedConstants) << std::endl;

  return utils::mkConcat(mergedConstants);
}

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<ExtractWhole>::applies(TNode node) {
  if (node.getKind() != kind::BITVECTOR_EXTRACT) return false;
  unsigned length = utils::getSize(node[0]);
  unsigned extractHigh = utils::getExtractHigh(node);
  if (extractHigh != length - 1) return false;
  unsigned extractLow  = utils::getExtractLow(node);
  if (extractLow != 0) return false;
  return true;
}

template<> inline
Node RewriteRule<ExtractWhole>::apply(TNode node) {
  Debug("bv-rewrite") << "RewriteRule<ExtractWhole>(" << node << ")" << std::endl;
  return node[0];
}

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<ExtractConstant>::applies(TNode node) {
  if (node.getKind() != kind::BITVECTOR_EXTRACT) return false;
  if (node[0].getKind() != kind::CONST_BITVECTOR) return false;
  return true;
}

template<> inline
Node RewriteRule<ExtractConstant>::apply(TNode node) {
  Debug("bv-rewrite") << "RewriteRule<ExtractConstant>(" << node << ")" << std::endl;
  Node child = node[0];
  BitVector childValue = child.getConst<BitVector>();
  return utils::mkConst(childValue.extract(utils::getExtractHigh(node), utils::getExtractLow(node)));
}

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<ExtractConcat>::applies(TNode node) {
  //Debug("bv-rewrite") << "RewriteRule<ExtractConcat>(" << node << ")" << std::endl;
  if (node.getKind() != kind::BITVECTOR_EXTRACT) return false;
  if (node[0].getKind() != kind::BITVECTOR_CONCAT) return false;
  return true;
}

template<> inline
Node RewriteRule<ExtractConcat>::apply(TNode node) {
  Debug("bv-rewrite") << "RewriteRule<ExtractConcat>(" << node << ")" << std::endl;
  int extract_high = utils::getExtractHigh(node);
  int extract_low = utils::getExtractLow(node);

  std::vector<Node> resultChildren;

  Node concat = node[0];
  for (int i = concat.getNumChildren() - 1; i >= 0 && extract_high >= 0; i--) {
    Node concatChild = concat[i];
    int concatChildSize = utils::getSize(concatChild);
    if (extract_low < concatChildSize) {
      int extract_start = extract_low < 0 ? 0 : extract_low;
      int extract_end = extract_high < concatChildSize ? extract_high : concatChildSize - 1;
      resultChildren.push_back(utils::mkExtract(concatChild, extract_end, extract_start));
    }
    extract_low -= concatChildSize;
    extract_high -= concatChildSize;
  }

  std::reverse(resultChildren.begin(), resultChildren.end());

  return utils::mkConcat(resultChildren);
}

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<ExtractExtract>::applies(TNode node) {
  if (node.getKind() != kind::BITVECTOR_EXTRACT) return false;
  if (node[0].getKind() != kind::BITVECTOR_EXTRACT) return false;
  return true;
}

template<> inline
Node RewriteRule<ExtractExtract>::apply(TNode node) {
  Debug("bv-rewrite") << "RewriteRule<ExtractExtract>(" << node << ")" << std::endl;

  // x[i:j][k:l] ~>  x[k+j:l+j]
  Node child = node[0];
  unsigned k = utils::getExtractHigh(node);
  unsigned l = utils::getExtractLow(node);
  unsigned j = utils::getExtractLow(child);

  Node result = utils::mkExtract(child[0], k + j, l + j);
  return result;
}

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<FailEq>::applies(TNode node) {
  //Debug("bv-rewrite") << "RewriteRule<FailEq>(" << node << ")" << std::endl;
  if (node.getKind() != kind::EQUAL) return false;
  if (node[0].getKind() != kind::CONST_BITVECTOR) return false;
  if (node[1].getKind() != kind::CONST_BITVECTOR) return false;
  return node[0] != node[1];
}

template<> inline
Node RewriteRule<FailEq>::apply(TNode node) {
  return utils::mkFalse();
}

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<SimplifyEq>::applies(TNode node) {
  if (node.getKind() != kind::EQUAL) return false;
  return node[0] == node[1];
}

template<> inline
Node RewriteRule<SimplifyEq>::apply(TNode node) {
  Debug("bv-rewrite") << "RewriteRule<SimplifyEq>(" << node << ")" << std::endl;
  return utils::mkTrue();
}

/* -------------------------------------------------------------------------- */

template<> inline
bool RewriteRule<ReflexivityEq>::applies(TNode node) {
  return (node.getKind() == kind::EQUAL && node[0] < node[1]);
}

template<> inline
Node RewriteRule<ReflexivityEq>::apply(TNode node) {
  Debug("bv-rewrite") << "RewriteRule<ReflexivityEq>(" << node << ")" << std::endl;
  Node res = node[1].eqNode(node[0]);
  return res;
}

}
}
}  // namespace cvc5


Generated by: GCOVR (Version 3.2)

Line	Exec	Source
1		/******************************************************************************
2		* Top contributors (to current version):
3		* Dejan Jovanovic, Liana Hadarean, Clark Barrett
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		* [[ Add one-line brief description here ]]
14		*
15		* [[ Add lengthier description here ]]
16		* \todo document this file
17		*/
18
19		#include "cvc5_private.h"
20
21		#pragma once
22
23		#include "theory/bv/theory_bv_rewrite_rules.h"
24		#include "theory/bv/theory_bv_utils.h"
25
26		namespace cvc5 {
27		namespace theory {
28		namespace bv {
29
30		/* -------------------------------------------------------------------------- */
31
32		template<> inline
33	336244	bool RewriteRule<ConcatFlatten>::applies(TNode node) {
34	336244	return (node.getKind() == kind::BITVECTOR_CONCAT);
35		}
36
37		template<> inline
38	168122	Node RewriteRule<ConcatFlatten>::apply(TNode node) {
39	168122	Debug("bv-rewrite") << "RewriteRule<ConcatFlatten>(" << node << ")" << std::endl;
40	336244	NodeBuilder result(kind::BITVECTOR_CONCAT);
41	336244	std::vector<Node> processing_stack;
42	168122	processing_stack.push_back(node);
43	1618930	while (!processing_stack.empty()) {
44	1450808	Node current = processing_stack.back();
45	725404	processing_stack.pop_back();
46	725404	if (current.getKind() == kind::BITVECTOR_CONCAT) {
47	731963	for (int i = current.getNumChildren() - 1; i >= 0; i--)
48	557282	processing_stack.push_back(current[i]);
49		} else {
50	550723	result << current;
51		}
52		}
53	168122	Node resultNode = result;
54	168122	Debug("bv-rewrite") << "RewriteRule<ConcatFlatten>(" << resultNode << ")" << std::endl;
55	336244	return resultNode;
56		}
57
58		/* -------------------------------------------------------------------------- */
59
60		template<> inline
61	336244	bool RewriteRule<ConcatExtractMerge>::applies(TNode node) {
62	336244	return (node.getKind() == kind::BITVECTOR_CONCAT);
63		}
64
65		template<> inline
66	168122	Node RewriteRule<ConcatExtractMerge>::apply(TNode node) {
67
68	168122	Debug("bv-rewrite") << "RewriteRule<ConcatExtractMerge>(" << node << ")" << std::endl;
69
70	336244	std::vector<Node> mergedExtracts;
71
72	336244	Node current = node[0];
73	168122	bool mergeStarted = false;
74	168122	unsigned currentHigh = 0;
75	168122	unsigned currentLow = 0;
76
77	550723	for(size_t i = 1, end = node.getNumChildren(); i < end; ++ i) {
78		// The next candidate for merging
79	453975	Node next = node[i];
80		// If the current is not an extract we just go to the next
81	693828	if (current.getKind() != kind::BITVECTOR_EXTRACT) {
82	311227	mergedExtracts.push_back(current);
83	311227	current = next;
84	311227	continue;
85		}
86		// If it is an extract and the first one, get the extract parameters
87	71374	else if (!mergeStarted) {
88	71341	currentHigh = utils::getExtractHigh(current);
89	71341	currentLow = utils::getExtractLow(current);
90		}
91
92		// If the next one can be merged, try to merge
93	71374	bool merged = false;
94	71374	if (next.getKind() == kind::BITVECTOR_EXTRACT && current[0] == next[0]) {
95		// x[i : j] @ x[j - 1 : k] -> c x[i : k]
96	24939	unsigned nextHigh = utils::getExtractHigh(next);
97	24939	unsigned nextLow = utils::getExtractLow(next);
98	24939	if(nextHigh + 1 == currentLow) {
99	96	currentLow = nextLow;
100	96	mergeStarted = true;
101	96	merged = true;
102		}
103		}
104		// If we haven't merged anything, add the previous merge and continue with the next
105	71374	if (!merged) {
106	71278	if (!mergeStarted) mergedExtracts.push_back(current);
107	9	else mergedExtracts.push_back(utils::mkExtract(current[0], currentHigh, currentLow));
108	71278	current = next;
109	71278	mergeStarted = false;
110		}
111		}
112
113		// Add the last child
114	168122	if (!mergeStarted) mergedExtracts.push_back(current);
115	63	else mergedExtracts.push_back(utils::mkExtract(current[0], currentHigh, currentLow));
116
117		// Return the result
118	336244	return utils::mkConcat(mergedExtracts);
119		}
120
121		/* -------------------------------------------------------------------------- */
122
123		template<> inline
124	336207	bool RewriteRule<ConcatConstantMerge>::applies(TNode node) {
125	336207	return node.getKind() == kind::BITVECTOR_CONCAT;
126		}
127
128		template<> inline
129	168085	Node RewriteRule<ConcatConstantMerge>::apply(TNode node) {
130
131	168085	Debug("bv-rewrite") << "RewriteRule<ConcatConstantMerge>(" << node << ")" << std::endl;
132
133	336170	std::vector<Node> mergedConstants;
134	682992	for (unsigned i = 0, end = node.getNumChildren(); i < end;) {
135	514907	if (node[i].getKind() != kind::CONST_BITVECTOR) {
136		// If not a constant, just add it
137	374119	mergedConstants.push_back(node[i]);
138	374119	++i;
139		} else {
140		// Find the longest sequence of constants
141	140788	unsigned j = i + 1;
142	212154	while (j < end) {
143	116930	if (node[j].getKind() != kind::CONST_BITVECTOR) {
144	81247	break;
145		} else {
146	35683	++ j;
147		}
148		}
149		// Append all the constants
150	281576	BitVector current = node[i].getConst<BitVector>();
151	176471	for (unsigned k = i + 1; k < j; ++ k) {
152	35683	current = current.concat(node[k].getConst<BitVector>());
153		}
154		// Add the new merged constant
155	140788	mergedConstants.push_back(utils::mkConst(current));
156	140788	i = j;
157		}
158		}
159
160	168085	Debug("bv-rewrite") << "RewriteRule<ConcatConstantMerge>(" << node << ") => " << utils::mkConcat(mergedConstants) << std::endl;
161
162	336170	return utils::mkConcat(mergedConstants);
163		}
164
165		/* -------------------------------------------------------------------------- */
166
167		template<> inline
168	660211	bool RewriteRule<ExtractWhole>::applies(TNode node) {
169	660211	if (node.getKind() != kind::BITVECTOR_EXTRACT) return false;
170	199917	unsigned length = utils::getSize(node[0]);
171	199917	unsigned extractHigh = utils::getExtractHigh(node);
172	199917	if (extractHigh != length - 1) return false;
173	87464	unsigned extractLow = utils::getExtractLow(node);
174	87464	if (extractLow != 0) return false;
175	19759	return true;
176		}
177
178		template<> inline
179	15118	Node RewriteRule<ExtractWhole>::apply(TNode node) {
180	15118	Debug("bv-rewrite") << "RewriteRule<ExtractWhole>(" << node << ")" << std::endl;
181	15118	return node[0];
182		}
183
184		/* -------------------------------------------------------------------------- */
185
186		template<> inline
187	184030	bool RewriteRule<ExtractConstant>::applies(TNode node) {
188	184030	if (node.getKind() != kind::BITVECTOR_EXTRACT) return false;
189	184030	if (node[0].getKind() != kind::CONST_BITVECTOR) return false;
190	86808	return true;
191		}
192
193		template<> inline
194	43404	Node RewriteRule<ExtractConstant>::apply(TNode node) {
195	43404	Debug("bv-rewrite") << "RewriteRule<ExtractConstant>(" << node << ")" << std::endl;
196	86808	Node child = node[0];
197	86808	BitVector childValue = child.getConst<BitVector>();
198	86808	return utils::mkConst(childValue.extract(utils::getExtractHigh(node), utils::getExtractLow(node)));
199		}
200
201		/* -------------------------------------------------------------------------- */
202
203		template<> inline
204	163231	bool RewriteRule<ExtractConcat>::applies(TNode node) {
205		//Debug("bv-rewrite") << "RewriteRule<ExtractConcat>(" << node << ")" << std::endl;
206	163231	if (node.getKind() != kind::BITVECTOR_EXTRACT) return false;
207	163231	if (node[0].getKind() != kind::BITVECTOR_CONCAT) return false;
208	20410	return true;
209		}
210
211		template<> inline
212	10205	Node RewriteRule<ExtractConcat>::apply(TNode node) {
213	10205	Debug("bv-rewrite") << "RewriteRule<ExtractConcat>(" << node << ")" << std::endl;
214	10205	int extract_high = utils::getExtractHigh(node);
215	10205	int extract_low = utils::getExtractLow(node);
216
217	20410	std::vector<Node> resultChildren;
218
219	20410	Node concat = node[0];
220	33441	for (int i = concat.getNumChildren() - 1; i >= 0 && extract_high >= 0; i--) {
221	46472	Node concatChild = concat[i];
222	23236	int concatChildSize = utils::getSize(concatChild);
223	23236	if (extract_low < concatChildSize) {
224	19857	int extract_start = extract_low < 0 ? 0 : extract_low;
225	19857	int extract_end = extract_high < concatChildSize ? extract_high : concatChildSize - 1;
226	19857	resultChildren.push_back(utils::mkExtract(concatChild, extract_end, extract_start));
227		}
228	23236	extract_low -= concatChildSize;
229	23236	extract_high -= concatChildSize;
230		}
231
232	10205	std::reverse(resultChildren.begin(), resultChildren.end());
233
234	20410	return utils::mkConcat(resultChildren);
235		}
236
237		/* -------------------------------------------------------------------------- */
238
239		template<> inline
240	142733	bool RewriteRule<ExtractExtract>::applies(TNode node) {
241	142733	if (node.getKind() != kind::BITVECTOR_EXTRACT) return false;
242	99329	if (node[0].getKind() != kind::BITVECTOR_EXTRACT) return false;
243	4214	return true;
244		}
245
246		template<> inline
247	2107	Node RewriteRule<ExtractExtract>::apply(TNode node) {
248	2107	Debug("bv-rewrite") << "RewriteRule<ExtractExtract>(" << node << ")" << std::endl;
249
250		// x[i:j][k:l] ~> x[k+j:l+j]
251	4214	Node child = node[0];
252	2107	unsigned k = utils::getExtractHigh(node);
253	2107	unsigned l = utils::getExtractLow(node);
254	2107	unsigned j = utils::getExtractLow(child);
255
256	2107	Node result = utils::mkExtract(child[0], k + j, l + j);
257	4214	return result;
258		}
259
260		/* -------------------------------------------------------------------------- */
261
262		template<> inline
263	466856	bool RewriteRule<FailEq>::applies(TNode node) {
264		//Debug("bv-rewrite") << "RewriteRule<FailEq>(" << node << ")" << std::endl;
265	466856	if (node.getKind() != kind::EQUAL) return false;
266	466856	if (node[0].getKind() != kind::CONST_BITVECTOR) return false;
267	104102	if (node[1].getKind() != kind::CONST_BITVECTOR) return false;
268	86107	return node[0] != node[1];
269		}
270
271		template<> inline
272	41364	Node RewriteRule<FailEq>::apply(TNode node) {
273	41364	return utils::mkFalse();
274		}
275
276		/* -------------------------------------------------------------------------- */
277
278		template<> inline
279	433254	bool RewriteRule<SimplifyEq>::applies(TNode node) {
280	433254	if (node.getKind() != kind::EQUAL) return false;
281	391890	return node[0] == node[1];
282		}
283
284		template<> inline
285	7762	Node RewriteRule<SimplifyEq>::apply(TNode node) {
286	7762	Debug("bv-rewrite") << "RewriteRule<SimplifyEq>(" << node << ")" << std::endl;
287	7762	return utils::mkTrue();
288		}
289
290		/* -------------------------------------------------------------------------- */
291
292		template<> inline
293	466950	bool RewriteRule<ReflexivityEq>::applies(TNode node) {
294	466950	return (node.getKind() == kind::EQUAL && node[0] < node[1]);
295		}
296
297		template<> inline
298	41458	Node RewriteRule<ReflexivityEq>::apply(TNode node) {
299	41458	Debug("bv-rewrite") << "RewriteRule<ReflexivityEq>(" << node << ")" << std::endl;
300	41458	Node res = node[1].eqNode(node[0]);
301	41458	return res;
302		}
303
304		}
305		}
306		} // namespace cvc5