org.evosuite.symbolic.solver.smt.SmtExprPrinter Maven / Gradle / Ivy

Go to download
/**
 * Copyright (C) 2010-2018 Gordon Fraser, Andrea Arcuri and EvoSuite
 * contributors
 *
 * This file is part of EvoSuite.
 *
 * EvoSuite is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3.0 of the License, or
 * (at your option) any later version.
 *
 * EvoSuite is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with EvoSuite. If not, see .
 */
package org.evosuite.symbolic.solver.smt;

import java.text.DecimalFormat;
import java.util.LinkedList;
import java.util.List;

public final class SmtExprPrinter implements SmtExprVisitor {

	@Override
	public String visit(SmtIntConstant n, Void arg) {
		long longValue = n.getConstantValue();
		if (longValue == Long.MIN_VALUE) {
			return "(- " + String.valueOf(Long.MIN_VALUE).replace("-", "") + ")";
		} else if (longValue < 0) {
			long absoluteValue = Math.abs(longValue);
			return "(- " + String.valueOf(absoluteValue) + ")";
		} else {
			return String.valueOf(longValue);
		}
	}

	private static DecimalFormat DECIMAL_FORMAT = new DecimalFormat("################0.0################");

	@Override
	public String visit(SmtRealConstant n, Void arg) {
		double doubleVal = n.getConstantValue();
		if (doubleVal < 0) {
			String magnitudeStr = DECIMAL_FORMAT.format(Math.abs(doubleVal));
			return "(- " + magnitudeStr + ")";
		} else {
			String doubleStr = DECIMAL_FORMAT.format(doubleVal);
			return doubleStr;
		}
	}

	@Override
	public String visit(SmtStringConstant n, Void arg) {
		String str = encodeString(n.getConstantValue());
		return "\"" + str + "\"";
	}

	@Override
	public String visit(SmtIntVariable n, Void arg) {
		String varName = n.getName();
		return varName;
	}

	@Override
	public String visit(SmtRealVariable n, Void arg) {
		String varName = n.getName();
		return varName;
	}

	@Override
	public String visit(SmtStringVariable n, Void arg) {
		String varName = n.getName();
		return varName;
	}

	@Override
	public String visit(SmtOperation n, Void arg) {
		List retValues = new LinkedList();
		for (SmtExpr argument : n.getArguments()) {
			String retValue = argument.accept(this, null);
			retValues.add(retValue);
		}

		StringBuffer result = new StringBuffer();
		if (!retValues.isEmpty()) {
			result.append("(");
		}
		result.append(n.getOperator().toString());
		for (String retValue : retValues) {
			result.append(" ");
			result.append(retValue);
		}
		if (!retValues.isEmpty()) {
			result.append(")");
		}
		return result.toString();
	}

	@Override
	public String visit(SmtBooleanConstant n, Void arg) {
		if (n.booleanValue() == true) {
			return "true";
		} else {
			return "false";
		}
	}

	public static String encodeString(String str) {
		char[] charArray = str.toCharArray();
		String ret_val = "";
		for (int i = 0; i < charArray.length; i++) {
			char c = charArray[i];
			if (Character.isISOControl(c)) {
				if (Integer.toHexString(c).length() == 1) {
					// padding
					ret_val += "_x0" + Integer.toHexString(c);
				} else {
					ret_val += "_x" + Integer.toHexString(c);
				}
			} else {
				ret_val += c;
			}
		}
		return ret_val;
	}

}