org.chocosolver.solver.expression.discrete.logical.UnLoExpression Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of choco-solver Show documentation

Open-source constraint solver.

The newest version!

/*
 * This file is part of choco-solver, http://choco-solver.org/
 *
 * Copyright (c) 2024, IMT Atlantique. All rights reserved.
 *
 * Licensed under the BSD 4-clause license.
 *
 * See LICENSE file in the project root for full license information.
 */
package org.chocosolver.solver.expression.discrete.logical;

import org.chocosolver.solver.Model;
import org.chocosolver.solver.constraints.Constraint;
import org.chocosolver.solver.exception.SolverException;
import org.chocosolver.solver.expression.discrete.relational.ReExpression;
import org.chocosolver.solver.variables.BoolVar;
import org.chocosolver.solver.variables.IntVar;

import java.util.HashSet;
import java.util.Map;

/**
 * Binary arithmetic expression
 * 
 * Project: choco-solver.
 *
 * @author Charles Prud'homme
 * @since 28/04/2016.
 */
public class UnLoExpression extends LoExpression {

    /**
     * The first expression this expression relies on
     */
    private final ReExpression e;
    /**
     * Builds a n-ary expression
     *
     * @param op an operator
     * @param e an expression
     */
    public UnLoExpression(Operator op, ReExpression e) {
        super(e.getModel(), op);
        this.e = e;
    }

    @Override
    public Model getModel() {
        return model;
    }

    @Override
    public BoolVar boolVar() {
        if (me == null) {
            BoolVar b = e.boolVar();
            if (op == Operator.NOT) {
                me = model.boolNotView(b);
            } else {
                throw new UnsupportedOperationException(
                    "Unary logical expressions does not support " + op.name());
            }
        }
        return me;
    }

    @Override
    public void extractVar(HashSet variables) {
        e.extractVar(variables);
    }

    @Override
    public Constraint decompose() {
        BoolVar v1 = e.boolVar();
        Model model = v1.getModel();
        if (op == Operator.NOT) {
            return model.arithm(v1, "<", 1);
        }
        throw new SolverException("Unexpected case");
    }

    @Override
    public boolean beval(int[] values, Map map) {
        return op.eval(e.beval(values, map), true);
    }

    @Override
    public String toString() {
        return op.name() + "(" + e.toString()+ ")";
    }
}