main page modules namespaces classes files
Gecode/J home

Expr.java

Go to the documentation of this file.

/*
 *  Main authors:
 *     Mikael Lagerkvist <lagerkvist@gecode.org>
 *
 *  Copyright:
 *     Mikael Lagerkvist, 2006
 *
 *  Last modified:
 *     $Date: 2006-09-11 17:06:50 +0200 (Mon, 11 Sep 2006) $ by $Author: zayenz $
 *     $Revision: 3650 $
 *
 *  This file is part of Gecode, the generic constraint
 *  development environment:
 *     http://www.gecode.org
 *
 *  See the file "LICENSE" for information on usage and
 *  redistribution of this file, and for a
 *     DISCLAIMER OF ALL WARRANTIES.
 *
 */

package org.gecode;

import java.util.ArrayList;

public class Expr {
    int d;
    ArrayList<Integer> cs;
    ArrayList<IntVar> vs;
    ArrayList<Integer> cbs;
    ArrayList<BExpr> vbs;

    public Expr() {
        d = 0;
        cs = new ArrayList<Integer>();
        vs = new ArrayList<IntVar>();
        cbs = new ArrayList<Integer>();
        vbs = new ArrayList<BExpr>();   
    }

    public Expr(Expr e) {
        d = e.d;
        cs = new ArrayList<Integer>(e.cs);
        vs = new ArrayList<IntVar>(e.vs);
        cbs = new ArrayList<Integer>(e.cbs);
        vbs = new ArrayList<BExpr>(e.vbs);      
    }
  
    public Expr(int c, IntVar v) {
        d = 0;
        cs = new ArrayList<Integer>(); cs.add(c);
        vs = new ArrayList<IntVar>();  vs.add(v);
        cbs = new ArrayList<Integer>();
        vbs = new ArrayList<BExpr>();   
    }
  
    public Expr(IntVar v) {
        d = 0;
        cs = new ArrayList<Integer>(); cs.add(1);
        vs = new ArrayList<IntVar>();  vs.add(v);
        cbs = new ArrayList<Integer>();
        vbs = new ArrayList<BExpr>();   
    }
  
    public Expr(int d) {
        this.d = d;
        cs = new ArrayList<Integer>();
        vs = new ArrayList<IntVar>();
        cbs = new ArrayList<Integer>();
        vbs = new ArrayList<BExpr>();   
    }
  
    public Expr(int c, BExpr b) {
        this.d = 0;
        cs = new ArrayList<Integer>();
        vs = new ArrayList<IntVar>();
        cbs = new ArrayList<Integer>(); cbs.add(c);
        vbs = new ArrayList<BExpr>();   vbs.add(b);
    }
  
    public Expr(BExpr b) {
        this.d = 0;
        cs = new ArrayList<Integer>();
        vs = new ArrayList<IntVar>();
        cbs = new ArrayList<Integer>(); cbs.add(1);
        vbs = new ArrayList<BExpr>();   vbs.add(b);
    }
  
    public Expr plus(int c, IntVar v) {
        cs.add(c);
        vs.add(v);
        return this;
    }

    public Expr minus(int c, IntVar v) {
        cs.add(-c);
        vs.add(v);
        return this;
    }

    public Expr plus(int c) {
        d += c;
        return this;
    }

    public Expr minus(int c) {
        d -= c;
        return this;
    }

    public Expr plus(Expr e) {
        cs.addAll(e.cs);
        vs.addAll(e.vs);
        cbs.addAll(e.cbs);
        vbs.addAll(e.vbs);
        d += e.d;
        return this;
    }

    public Expr minus(Expr e) {
        for (int i: e.cs)
            cs.add(-i);
        for (int i: e.cbs)
            cbs.add(-i);
        vs.addAll(e.vs);
        vbs.addAll(e.vbs);
        d -= e.d;
        return this;
    }

    public Expr plus(int c, BExpr b) {
        cbs.add(c);
        vbs.add(b);
        return this;
    }

    public Expr minus(int c, BExpr b) {
        cbs.add(-c);
        vbs.add(b);
        return this;
    }

    //  Convenience methods

    public Expr plus(IntVar v) {
        return plus(1, v);
    }

    public Expr minus(IntVar v) {
        return minus(1, v);
    }

    public Expr plus(BExpr b) {
        return plus(1, b);
    }

    public Expr minus(BExpr b) {
        return minus(1, b);
    }
    

    public Expr p(int c, IntVar v) {
        return plus(c, v);
    }

    public Expr m(int c, IntVar v) {
        return minus(c, v);
    }

    public Expr p(IntVar v) {
        return plus(1, v);
    }

    public Expr m(IntVar v) {
        return minus(1, v);
    }

    public Expr p(int c) {
        return plus(c);
    }

    public Expr m(int c) {
        return minus(c);
    }

    public Expr p(Expr e) {
        return plus(e);
    }

    public Expr m(Expr e) {
        return minus(e);
    }


    public Expr p(int c, BExpr b) {
        return plus(c, b);
    }

    public Expr m(int c, BExpr b) {
        return minus(c, b);
    }

    public Expr p(BExpr b) {
        return plus(1, b);
    }

    public Expr m(BExpr b) {
        return minus(1, b);
    }

    //  Debug methods
    public String toString() {
        String res = "";
        for (int i = 0; i < cs.size(); ++i) {
            int c = cs.get(i);
            if (c == 0) continue; 
            res += (i==0?"":
                    ((i<0)?" - ":" + "));
            if (Math.abs(c) != 1) res += "" + c + "*";
            res += "" + vs.get(i).getName() + vs.get(i);
        }
        if (d != 0 || res.equals(""))
            res += (res.equals("")?"":" + ") + d;
        return res;
    }

    //  Package methods
    void clear_bexpr(Space home) {
        for (int i = 0; i < cbs.size(); ++i) {
            cs.add(cbs.get(i));
            vs.add(vbs.get(i).e.post(home));
        }
        cbs.clear(); vbs.clear();
    }
}