Head

GCC Code Coverage Report

Directory:	.		Exec	Total	Coverage
File:	src/theory/strings/strings_rewriter.cpp	Lines:	162	162	100.0 %
Date:	2021-11-05	Branches:	421	872	48.3 %


/******************************************************************************
 * Top contributors (to current version):
 *   Andrew Reynolds, Andres Noetzli, 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.
 * ****************************************************************************
 *
 * Implementation of rewrite rules for string-specific operators in
 * theory of strings.
 */

#include "theory/strings/strings_rewriter.h"

#include "expr/node_builder.h"
#include "theory/strings/theory_strings_utils.h"
#include "util/rational.h"
#include "util/string.h"

using namespace cvc5::kind;

namespace cvc5 {
namespace theory {
namespace strings {

StringsRewriter::StringsRewriter(Rewriter* r,
                                 HistogramStat<Rewrite>* statistics,
                                 uint32_t alphaCard)
    : SequencesRewriter(r, statistics), d_alphaCard(alphaCard)
{
}

RewriteResponse StringsRewriter::postRewrite(TNode node)
{
  Trace("strings-postrewrite")
      << "Strings::StringsRewriter::postRewrite start " << node << std::endl;

  Node retNode = node;
  Kind nk = node.getKind();
  if (nk == kind::STRING_LT)
  {
    retNode = rewriteStringLt(node);
  }
  else if (nk == kind::STRING_LEQ)
  {
    retNode = rewriteStringLeq(node);
  }
  else if (nk == STRING_TOLOWER || nk == STRING_TOUPPER)
  {
    retNode = rewriteStrConvert(node);
  }
  else if (nk == STRING_IS_DIGIT)
  {
    retNode = rewriteStringIsDigit(node);
  }
  else if (nk == kind::STRING_ITOS)
  {
    retNode = rewriteIntToStr(node);
  }
  else if (nk == kind::STRING_STOI)
  {
    retNode = rewriteStrToInt(node);
  }
  else if (nk == STRING_TO_CODE)
  {
    retNode = rewriteStringToCode(node);
  }
  else if (nk == STRING_FROM_CODE)
  {
    retNode = rewriteStringFromCode(node);
  }
  else
  {
    return SequencesRewriter::postRewrite(node);
  }

  Trace("strings-postrewrite")
      << "Strings::StringsRewriter::postRewrite returning " << retNode
      << std::endl;
  if (node != retNode)
  {
    Trace("strings-rewrite-debug") << "Strings::StringsRewriter::postRewrite "
                                   << node << " to " << retNode << std::endl;
    return RewriteResponse(REWRITE_AGAIN_FULL, retNode);
  }
  return RewriteResponse(REWRITE_DONE, retNode);
}

Node StringsRewriter::rewriteStrToInt(Node node)
{
  Assert(node.getKind() == STRING_STOI);
  NodeManager* nm = NodeManager::currentNM();
  if (node[0].isConst())
  {
    Node ret;
    String s = node[0].getConst<String>();
    if (s.isNumber())
    {
      ret = nm->mkConst(s.toNumber());
    }
    else
    {
      ret = nm->mkConst(Rational(-1));
    }
    return returnRewrite(node, ret, Rewrite::STOI_EVAL);
  }
  else if (node[0].getKind() == STRING_CONCAT)
  {
    for (TNode nc : node[0])
    {
      if (nc.isConst())
      {
        String t = nc.getConst<String>();
        if (!t.isNumber())
        {
          Node ret = nm->mkConst(Rational(-1));
          return returnRewrite(node, ret, Rewrite::STOI_CONCAT_NONNUM);
        }
      }
    }
  }
  return node;
}

Node StringsRewriter::rewriteIntToStr(Node node)
{
  Assert(node.getKind() == STRING_ITOS);
  NodeManager* nm = NodeManager::currentNM();
  if (node[0].isConst())
  {
    Node ret;
    if (node[0].getConst<Rational>().sgn() == -1)
    {
      ret = nm->mkConst(String(""));
    }
    else
    {
      std::string stmp = node[0].getConst<Rational>().getNumerator().toString();
      Assert(stmp[0] != '-');
      ret = nm->mkConst(String(stmp));
    }
    return returnRewrite(node, ret, Rewrite::ITOS_EVAL);
  }
  return node;
}

Node StringsRewriter::rewriteStrConvert(Node node)
{
  Kind nk = node.getKind();
  Assert(nk == STRING_TOLOWER || nk == STRING_TOUPPER);
  NodeManager* nm = NodeManager::currentNM();
  if (node[0].isConst())
  {
    std::vector<unsigned> nvec = node[0].getConst<String>().getVec();
    for (unsigned i = 0, nvsize = nvec.size(); i < nvsize; i++)
    {
      unsigned newChar = nvec[i];
      // transform it
      // upper 65 ... 90
      // lower 97 ... 122
      if (nk == STRING_TOUPPER)
      {
        if (newChar >= 97 && newChar <= 122)
        {
          newChar = newChar - 32;
        }
      }
      else if (nk == STRING_TOLOWER)
      {
        if (newChar >= 65 && newChar <= 90)
        {
          newChar = newChar + 32;
        }
      }
      nvec[i] = newChar;
    }
    Node retNode = nm->mkConst(String(nvec));
    return returnRewrite(node, retNode, Rewrite::STR_CONV_CONST);
  }
  else if (node[0].getKind() == STRING_CONCAT)
  {
    NodeBuilder concatBuilder(STRING_CONCAT);
    for (const Node& nc : node[0])
    {
      concatBuilder << nm->mkNode(nk, nc);
    }
    // tolower( x1 ++ x2 ) --> tolower( x1 ) ++ tolower( x2 )
    Node retNode = concatBuilder.constructNode();
    return returnRewrite(node, retNode, Rewrite::STR_CONV_MINSCOPE_CONCAT);
  }
  else if (node[0].getKind() == STRING_TOLOWER
           || node[0].getKind() == STRING_TOUPPER)
  {
    // tolower( tolower( x ) ) --> tolower( x )
    // tolower( toupper( x ) ) --> tolower( x )
    Node retNode = nm->mkNode(nk, node[0][0]);
    return returnRewrite(node, retNode, Rewrite::STR_CONV_IDEM);
  }
  else if (node[0].getKind() == STRING_ITOS)
  {
    // tolower( str.from.int( x ) ) --> str.from.int( x )
    return returnRewrite(node, node[0], Rewrite::STR_CONV_ITOS);
  }
  return node;
}

Node StringsRewriter::rewriteStringLt(Node n)
{
  Assert(n.getKind() == kind::STRING_LT);
  NodeManager* nm = NodeManager::currentNM();
  // eliminate s < t ---> s != t AND s <= t
  Node retNode = nm->mkNode(
      AND, n[0].eqNode(n[1]).negate(), nm->mkNode(STRING_LEQ, n[0], n[1]));
  return returnRewrite(n, retNode, Rewrite::STR_LT_ELIM);
}

Node StringsRewriter::rewriteStringLeq(Node n)
{
  Assert(n.getKind() == kind::STRING_LEQ);
  NodeManager* nm = NodeManager::currentNM();
  if (n[0] == n[1])
  {
    Node ret = nm->mkConst(true);
    return returnRewrite(n, ret, Rewrite::STR_LEQ_ID);
  }
  if (n[0].isConst() && n[1].isConst())
  {
    String s = n[0].getConst<String>();
    String t = n[1].getConst<String>();
    Node ret = nm->mkConst(s.isLeq(t));
    return returnRewrite(n, ret, Rewrite::STR_LEQ_EVAL);
  }
  // empty strings
  for (unsigned i = 0; i < 2; i++)
  {
    if (n[i].isConst() && n[i].getConst<String>().empty())
    {
      Node ret = i == 0 ? nm->mkConst(true) : n[0].eqNode(n[1]);
      return returnRewrite(n, ret, Rewrite::STR_LEQ_EMPTY);
    }
  }

  std::vector<Node> n1;
  utils::getConcat(n[0], n1);
  std::vector<Node> n2;
  utils::getConcat(n[1], n2);
  Assert(!n1.empty() && !n2.empty());

  // constant prefixes
  if (n1[0].isConst() && n2[0].isConst() && n1[0] != n2[0])
  {
    String s = n1[0].getConst<String>();
    String t = n2[0].getConst<String>();
    size_t prefixLen = std::min(s.size(), t.size());
    s = s.prefix(prefixLen);
    t = t.prefix(prefixLen);
    // if the prefixes are not the same, then we can already decide the outcome
    if (s != t)
    {
      Node ret = nm->mkConst(s.isLeq(t));
      return returnRewrite(n, ret, Rewrite::STR_LEQ_CPREFIX);
    }
  }
  return n;
}

Node StringsRewriter::rewriteStringFromCode(Node n)
{
  Assert(n.getKind() == kind::STRING_FROM_CODE);
  NodeManager* nm = NodeManager::currentNM();

  if (n[0].isConst())
  {
    Integer i = n[0].getConst<Rational>().getNumerator();
    Node ret;
    if (i >= 0 && i < d_alphaCard)
    {
      std::vector<unsigned> svec = {i.toUnsignedInt()};
      ret = nm->mkConst(String(svec));
    }
    else
    {
      ret = nm->mkConst(String(""));
    }
    return returnRewrite(n, ret, Rewrite::FROM_CODE_EVAL);
  }
  return n;
}

Node StringsRewriter::rewriteStringToCode(Node n)
{
  Assert(n.getKind() == kind::STRING_TO_CODE);
  if (n[0].isConst())
  {
    NodeManager* nm = NodeManager::currentNM();
    String s = n[0].getConst<String>();
    Node ret;
    if (s.size() == 1)
    {
      std::vector<unsigned> vec = s.getVec();
      Assert(vec.size() == 1);
      ret = nm->mkConst(Rational(vec[0]));
    }
    else
    {
      ret = nm->mkConst(Rational(-1));
    }
    return returnRewrite(n, ret, Rewrite::TO_CODE_EVAL);
  }
  return n;
}

Node StringsRewriter::rewriteStringIsDigit(Node n)
{
  Assert(n.getKind() == kind::STRING_IS_DIGIT);
  NodeManager* nm = NodeManager::currentNM();
  // eliminate str.is_digit(s) ----> 48 <= str.to_code(s) <= 57
  Node t = nm->mkNode(STRING_TO_CODE, n[0]);
  Node retNode = nm->mkNode(AND,
                            nm->mkNode(LEQ, nm->mkConst(Rational(48)), t),
                            nm->mkNode(LEQ, t, nm->mkConst(Rational(57))));
  return returnRewrite(n, retNode, Rewrite::IS_DIGIT_ELIM);
}

}  // namespace strings
}  // namespace theory
}  // namespace cvc5


Generated by: GCOVR (Version 3.2)

Line	Exec	Source
1		/******************************************************************************
2		* Top contributors (to current version):
3		* Andrew Reynolds, Andres Noetzli, Gereon Kremer
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		* Implementation of rewrite rules for string-specific operators in
14		* theory of strings.
15		*/
16
17		#include "theory/strings/strings_rewriter.h"
18
19		#include "expr/node_builder.h"
20		#include "theory/strings/theory_strings_utils.h"
21		#include "util/rational.h"
22		#include "util/string.h"
23
24		using namespace cvc5::kind;
25
26		namespace cvc5 {
27		namespace theory {
28		namespace strings {
29
30	15273	StringsRewriter::StringsRewriter(Rewriter* r,
31		HistogramStat<Rewrite>* statistics,
32	15273	uint32_t alphaCard)
33	15273	: SequencesRewriter(r, statistics), d_alphaCard(alphaCard)
34		{
35	15273	}
36
37	479460	RewriteResponse StringsRewriter::postRewrite(TNode node)
38		{
39	958920	Trace("strings-postrewrite")
40	479460	<< "Strings::StringsRewriter::postRewrite start " << node << std::endl;
41
42	958920	Node retNode = node;
43	479460	Kind nk = node.getKind();
44	479460	if (nk == kind::STRING_LT)
45		{
46	23	retNode = rewriteStringLt(node);
47		}
48	479437	else if (nk == kind::STRING_LEQ)
49		{
50	404	retNode = rewriteStringLeq(node);
51		}
52	479033	else if (nk == STRING_TOLOWER \|\| nk == STRING_TOUPPER)
53		{
54	208	retNode = rewriteStrConvert(node);
55		}
56	478825	else if (nk == STRING_IS_DIGIT)
57		{
58	8	retNode = rewriteStringIsDigit(node);
59		}
60	478817	else if (nk == kind::STRING_ITOS)
61		{
62	619	retNode = rewriteIntToStr(node);
63		}
64	478198	else if (nk == kind::STRING_STOI)
65		{
66	1042	retNode = rewriteStrToInt(node);
67		}
68	477156	else if (nk == STRING_TO_CODE)
69		{
70	6180	retNode = rewriteStringToCode(node);
71		}
72	470976	else if (nk == STRING_FROM_CODE)
73		{
74	114	retNode = rewriteStringFromCode(node);
75		}
76		else
77		{
78	470862	return SequencesRewriter::postRewrite(node);
79		}
80
81	17196	Trace("strings-postrewrite")
82	8598	<< "Strings::StringsRewriter::postRewrite returning " << retNode
83	8598	<< std::endl;
84	8598	if (node != retNode)
85		{
86	4688	Trace("strings-rewrite-debug") << "Strings::StringsRewriter::postRewrite "
87	2344	<< node << " to " << retNode << std::endl;
88	2344	return RewriteResponse(REWRITE_AGAIN_FULL, retNode);
89		}
90	6254	return RewriteResponse(REWRITE_DONE, retNode);
91		}
92
93	1042	Node StringsRewriter::rewriteStrToInt(Node node)
94		{
95	1042	Assert(node.getKind() == STRING_STOI);
96	1042	NodeManager* nm = NodeManager::currentNM();
97	1042	if (node[0].isConst())
98		{
99	546	Node ret;
100	546	String s = node[0].getConst<String>();
101	273	if (s.isNumber())
102		{
103	124	ret = nm->mkConst(s.toNumber());
104		}
105		else
106		{
107	149	ret = nm->mkConst(Rational(-1));
108		}
109	273	return returnRewrite(node, ret, Rewrite::STOI_EVAL);
110		}
111	769	else if (node[0].getKind() == STRING_CONCAT)
112		{
113	1255	for (TNode nc : node[0])
114		{
115	959	if (nc.isConst())
116		{
117	229	String t = nc.getConst<String>();
118	125	if (!t.isNumber())
119		{
120	42	Node ret = nm->mkConst(Rational(-1));
121	21	return returnRewrite(node, ret, Rewrite::STOI_CONCAT_NONNUM);
122		}
123		}
124		}
125		}
126	748	return node;
127		}
128
129	619	Node StringsRewriter::rewriteIntToStr(Node node)
130		{
131	619	Assert(node.getKind() == STRING_ITOS);
132	619	NodeManager* nm = NodeManager::currentNM();
133	619	if (node[0].isConst())
134		{
135	550	Node ret;
136	275	if (node[0].getConst<Rational>().sgn() == -1)
137		{
138	126	ret = nm->mkConst(String(""));
139		}
140		else
141		{
142	298	std::string stmp = node[0].getConst<Rational>().getNumerator().toString();
143	149	Assert(stmp[0] != '-');
144	149	ret = nm->mkConst(String(stmp));
145		}
146	275	return returnRewrite(node, ret, Rewrite::ITOS_EVAL);
147		}
148	344	return node;
149		}
150
151	208	Node StringsRewriter::rewriteStrConvert(Node node)
152		{
153	208	Kind nk = node.getKind();
154	208	Assert(nk == STRING_TOLOWER \|\| nk == STRING_TOUPPER);
155	208	NodeManager* nm = NodeManager::currentNM();
156	208	if (node[0].isConst())
157		{
158	52	std::vector<unsigned> nvec = node[0].getConst<String>().getVec();
159	96	for (unsigned i = 0, nvsize = nvec.size(); i < nvsize; i++)
160		{
161	70	unsigned newChar = nvec[i];
162		// transform it
163		// upper 65 ... 90
164		// lower 97 ... 122
165	70	if (nk == STRING_TOUPPER)
166		{
167	36	if (newChar >= 97 && newChar <= 122)
168		{
169	22	newChar = newChar - 32;
170		}
171		}
172	34	else if (nk == STRING_TOLOWER)
173		{
174	34	if (newChar >= 65 && newChar <= 90)
175		{
176	25	newChar = newChar + 32;
177		}
178		}
179	70	nvec[i] = newChar;
180		}
181	52	Node retNode = nm->mkConst(String(nvec));
182	26	return returnRewrite(node, retNode, Rewrite::STR_CONV_CONST);
183		}
184	182	else if (node[0].getKind() == STRING_CONCAT)
185		{
186	92	NodeBuilder concatBuilder(STRING_CONCAT);
187	198	for (const Node& nc : node[0])
188		{
189	152	concatBuilder << nm->mkNode(nk, nc);
190		}
191		// tolower( x1 ++ x2 ) --> tolower( x1 ) ++ tolower( x2 )
192	92	Node retNode = concatBuilder.constructNode();
193	46	return returnRewrite(node, retNode, Rewrite::STR_CONV_MINSCOPE_CONCAT);
194		}
195	408	else if (node[0].getKind() == STRING_TOLOWER
196	408	\|\| node[0].getKind() == STRING_TOUPPER)
197		{
198		// tolower( tolower( x ) ) --> tolower( x )
199		// tolower( toupper( x ) ) --> tolower( x )
200	16	Node retNode = nm->mkNode(nk, node[0][0]);
201	8	return returnRewrite(node, retNode, Rewrite::STR_CONV_IDEM);
202		}
203	128	else if (node[0].getKind() == STRING_ITOS)
204		{
205		// tolower( str.from.int( x ) ) --> str.from.int( x )
206	4	return returnRewrite(node, node[0], Rewrite::STR_CONV_ITOS);
207		}
208	124	return node;
209		}
210
211	23	Node StringsRewriter::rewriteStringLt(Node n)
212		{
213	23	Assert(n.getKind() == kind::STRING_LT);
214	23	NodeManager* nm = NodeManager::currentNM();
215		// eliminate s < t ---> s != t AND s <= t
216		Node retNode = nm->mkNode(
217	46	AND, n[0].eqNode(n[1]).negate(), nm->mkNode(STRING_LEQ, n[0], n[1]));
218	46	return returnRewrite(n, retNode, Rewrite::STR_LT_ELIM);
219		}
220
221	404	Node StringsRewriter::rewriteStringLeq(Node n)
222		{
223	404	Assert(n.getKind() == kind::STRING_LEQ);
224	404	NodeManager* nm = NodeManager::currentNM();
225	404	if (n[0] == n[1])
226		{
227	56	Node ret = nm->mkConst(true);
228	28	return returnRewrite(n, ret, Rewrite::STR_LEQ_ID);
229		}
230	376	if (n[0].isConst() && n[1].isConst())
231		{
232	88	String s = n[0].getConst<String>();
233	88	String t = n[1].getConst<String>();
234	88	Node ret = nm->mkConst(s.isLeq(t));
235	44	return returnRewrite(n, ret, Rewrite::STR_LEQ_EVAL);
236		}
237		// empty strings
238	967	for (unsigned i = 0; i < 2; i++)
239		{
240	656	if (n[i].isConst() && n[i].getConst<String>().empty())
241		{
242	42	Node ret = i == 0 ? nm->mkConst(true) : n[0].eqNode(n[1]);
243	21	return returnRewrite(n, ret, Rewrite::STR_LEQ_EMPTY);
244		}
245		}
246
247	622	std::vector<Node> n1;
248	311	utils::getConcat(n[0], n1);
249	622	std::vector<Node> n2;
250	311	utils::getConcat(n[1], n2);
251	311	Assert(!n1.empty() && !n2.empty());
252
253		// constant prefixes
254	311	if (n1[0].isConst() && n2[0].isConst() && n1[0] != n2[0])
255		{
256	111	String s = n1[0].getConst<String>();
257	111	String t = n2[0].getConst<String>();
258	71	size_t prefixLen = std::min(s.size(), t.size());
259	71	s = s.prefix(prefixLen);
260	71	t = t.prefix(prefixLen);
261		// if the prefixes are not the same, then we can already decide the outcome
262	71	if (s != t)
263		{
264	62	Node ret = nm->mkConst(s.isLeq(t));
265	31	return returnRewrite(n, ret, Rewrite::STR_LEQ_CPREFIX);
266		}
267		}
268	280	return n;
269		}
270
271	114	Node StringsRewriter::rewriteStringFromCode(Node n)
272		{
273	114	Assert(n.getKind() == kind::STRING_FROM_CODE);
274	114	NodeManager* nm = NodeManager::currentNM();
275
276	114	if (n[0].isConst())
277		{
278	48	Integer i = n[0].getConst<Rational>().getNumerator();
279	48	Node ret;
280	24	if (i >= 0 && i < d_alphaCard)
281		{
282	22	std::vector<unsigned> svec = {i.toUnsignedInt()};
283	11	ret = nm->mkConst(String(svec));
284		}
285		else
286		{
287	13	ret = nm->mkConst(String(""));
288		}
289	24	return returnRewrite(n, ret, Rewrite::FROM_CODE_EVAL);
290		}
291	90	return n;
292		}
293
294	6180	Node StringsRewriter::rewriteStringToCode(Node n)
295		{
296	6180	Assert(n.getKind() == kind::STRING_TO_CODE);
297	6180	if (n[0].isConst())
298		{
299	1512	NodeManager* nm = NodeManager::currentNM();
300	3024	String s = n[0].getConst<String>();
301	3024	Node ret;
302	1512	if (s.size() == 1)
303		{
304	1296	std::vector<unsigned> vec = s.getVec();
305	648	Assert(vec.size() == 1);
306	648	ret = nm->mkConst(Rational(vec[0]));
307		}
308		else
309		{
310	864	ret = nm->mkConst(Rational(-1));
311		}
312	1512	return returnRewrite(n, ret, Rewrite::TO_CODE_EVAL);
313		}
314	4668	return n;
315		}
316
317	8	Node StringsRewriter::rewriteStringIsDigit(Node n)
318		{
319	8	Assert(n.getKind() == kind::STRING_IS_DIGIT);
320	8	NodeManager* nm = NodeManager::currentNM();
321		// eliminate str.is_digit(s) ----> 48 <= str.to_code(s) <= 57
322	16	Node t = nm->mkNode(STRING_TO_CODE, n[0]);
323		Node retNode = nm->mkNode(AND,
324	16	nm->mkNode(LEQ, nm->mkConst(Rational(48)), t),
325	32	nm->mkNode(LEQ, t, nm->mkConst(Rational(57))));
326	16	return returnRewrite(n, retNode, Rewrite::IS_DIGIT_ELIM);
327		}
328
329		} // namespace strings
330		} // namespace theory
331	31125	} // namespace cvc5