SmtOperation.java example

Explorer
evosuite-master
/**
 * Copyright (C) 2010-2017 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 <http://www.gnu.org/licenses/>.
 */
package org.evosuite.symbolic.solver.smt;

import java.util.Arrays;

public final class SmtOperation extends SmtExpr {

	public enum Operator {
		MUL("*"), //
		MINUS("-"), //
		ADD("+"), //
		MOD("mod"), //
		INT2BV32("(_ int2bv 32)"), //
		BVOR("bvor"), //
		BV2Nat("bv2nat"), //
		BVAND("bvand"), //
		BVXOR("bvxor"), //
		BV2INT("bv2int"), //
		BVSHL("bvshl"), //
		BVASHR("bvashr"), //
		BVLSHR("bvlshr"), //
		GT(">"), //
		ITE("ite"), // 
		LT("<"), // 
		GE(">="), // 
		REAL2INT("to_int"), // 
		INT2REAL("to_real"), //
		DIV("div"), //
		SLASH("/"), //
		STR_SUBSTR("str.substr"), // 
		STR_REPLACE("str.replace"), //
		STR_INDEXOF("str.indexof"), //
		EQ("="), //
		STR_CONCAT("str.++"), //
		INT_TO_STR("int.to.str"), //
		STR_SUFFIXOF("str.suffixof"), //
		STR_CONTAINS("str.contains"), //
		STR_AT("str.at"), //
		CHAR_TO_INT("char_to_int"), //
		STR_PREFIXOF("str.prefixof"), //
		INT_TO_CHAR("int_to_char"), //
		STR_LEN("str.len"), //
		LE("<="), //
		NOT("not"), //
		STR_TO_INT("str.to.int"), //
		ABS("abs"), //
		BVADD("bvadd"), //
		STR_IN_REG_EXP("str.in.re"), //
		STR_TO_REG_EXP("str.to.re"), //
		REG_EXP_CONCAT("re.++"), //
		REG_EXP_KLEENE_STAR("re.*"), //
		REG_EXP_UNION("re.union"), //
		REG_EXP_OPTIONAL("re.opt"), //
		REG_EXP_ALL_CHAR("re.allchar"), //
		REG_EXP_KLEENE_CROSS("re.+"), //
		REG_EXP_LOOP("re.loop"), //
		REG_EXP_RANGE("re.range"), //
		REM("rem"), //
		CONCAT("Concat"), //
		REPLACE("Replace"), //
		SUBSTRING("Substring"), //
		ENDSWITH("EndsWith"), //
		CONTAINS("Contains"), //
		STARTSWITH("StartsWith"), //
		INDEXOF("Indexof"), //
		LENGTH("Length");

		private final String rep;

		private Operator(String rep) {
			this.rep = rep;
		}

		@Override
		public String toString() {
			return rep;
		}
	}

	private final Operator operator;
	private final SmtExpr[] arguments;

	private final boolean hasSymbolicValues;

	/**
	 * Unary operation
	 * 
	 * @param op
	 * @param arg
	 */
	public SmtOperation(Operator op, SmtExpr... arg) {
		this.operator = op;
		this.arguments = arg;
		this.hasSymbolicValues = hasSymbolicValue(arg);
	}

	@Override
	public <K, V> K accept(SmtExprVisitor<K, V> v, V arg) {
		return v.visit(this, arg);
	}

	public SmtExpr[] getArguments() {
		return arguments;
	}

	public Operator getOperator() {
		return operator;
	}

	@Override
	public int hashCode() {
		final int prime = 31;
		int result = 1;
		result = prime * result + Arrays.hashCode(arguments);
		result = prime * result
				+ ((operator == null) ? 0 : operator.hashCode());
		return result;
	}

	@Override
	public boolean equals(Object obj) {
		if (this == obj)
			return true;
		if (obj == null)
			return false;
		if (getClass() != obj.getClass())
			return false;
		SmtOperation other = (SmtOperation) obj;
		if (!Arrays.equals(arguments, other.arguments))
			return false;
		if (operator != other.operator)
			return false;
		return true;
	}

	@Override
	public boolean isSymbolic() {
		return hasSymbolicValues;
	}

	private static boolean hasSymbolicValue(SmtExpr[] arguments) {
		for (SmtExpr smtExpr : arguments) {
			if (smtExpr.isSymbolic()) {
				return true;
			}
		}
		return false;
	}

}