main page modules namespaces classes files
Gecode/J home

VarMatrix.java

Go to the documentation of this file.

/*
 *  Main authors:
 *     Mikael Lagerkvist <lagerkvist@gecode.org>
 *     Guido Tack <tack@gecode.org>
 *
 *  Copyright:
 *     Mikael Lagerkvist, 2006
 *     Guido Tack, 2006 
 *  
 *
 *  Last modified:
 *     $Date: 2006-04-26 11:04:05 +0200 (Wed, 26 Apr 2006) $ by $Author: zayenz $
 *     $Revision: 3217 $
 *
 *  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;
import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;

public class VarMatrix<Var extends GecodeVar> extends VarArray<Var> {
    private int height, width;

    private int getPos(int r, int c) {
        return r*width + c;
    }



    public VarMatrix() {
        super();

    }

    public VarMatrix(int h, int w) {
        super(h * w);
        height = h; width = w;
    }

    @SuppressWarnings("unchecked") 
    public VarMatrix(JavaSpace newHome,
                     boolean share, VarMatrix<Var> va) {
        super(va.size());
        height = va.height; width = va.width;
        for (Var v: va) {
            add((Var)v.copy(newHome, share));
        }
    }

    public <C extends Iterable<? extends Var>> VarMatrix(int h, int w, C va) {
        super(h*w);
        height = h; width = w;
        int cnt = 0;
        for (Var v: va) {
            if (++cnt > h*w) throw new ArgumentSizeMismatchException();
            add(v);
        }
    }

    public <V2 extends Var> VarMatrix(int h, int w, V2... args) {
        super(args.length);
        height = h; width = w;
        if (args.length != h*w) throw new ArgumentSizeMismatchException();
        for (V2 v: args) add(v);
    }

    public VarMatrix(JavaSpace home, int h, int w, Class<? extends Var> type) {
        super(h*w);
        height = h; width = w;
        try {
            Constructor<? extends Var> ctor = type.getConstructor(JavaSpace.class);
            
            for (int i = 0; i < h*w; ++i) {
                add(ctor.newInstance(home));
            }
        } catch(NoSuchMethodException ex) {
            throw new FatalException(ex);
        } catch(InstantiationException ex) {
            throw new FatalException(ex);
        } catch(IllegalAccessException ex) {
            throw new FatalException(ex);
        } catch(InvocationTargetException ex) {
            throw new FatalException(ex);
        }
    }

    public VarMatrix(JavaSpace home, int h, int w, Class<? extends Var> type, Object arg) {
        super(h*w);
        height = h; width = w;
        try {
            Constructor<? extends Var> ctor = type.getConstructor(JavaSpace.class, arg.getClass());
            
            for (int i = 0; i < h*w; ++i) {
                add(ctor.newInstance(home, arg));
            }
        } catch(NoSuchMethodException ex) {
            throw new FatalException(ex);
        } catch(InstantiationException ex) {
            throw new FatalException(ex);
        } catch(IllegalAccessException ex) {
            throw new FatalException(ex);
        } catch(InvocationTargetException ex) {
            throw new FatalException(ex);
        }
    }

    public VarMatrix(JavaSpace home, int h, int w, Class<? extends Var> type, 
                    Object arg1, Object arg2) {
        super(h*w);
        height = h; width = w;
        try {
            Constructor<? extends Var> ctor = type.getConstructor(JavaSpace.class, 
                                                        arg1.getClass(),
                                                        arg2.getClass());
            
            for (int i = 0; i < h*w; ++i) {
                add(ctor.newInstance(home, arg1, arg2));
            }
        } catch(NoSuchMethodException ex) {
            throw new FatalException(ex);
        } catch(InstantiationException ex) {
            throw new FatalException(ex);
        } catch(IllegalAccessException ex) {
            throw new FatalException(ex);
        } catch(InvocationTargetException ex) {
            throw new FatalException(ex);
        }
    }

    public VarMatrix(JavaSpace home, int h, int w, Class<? extends Var> type, 
                    Object arg1, Object arg2, Object arg3) {
        super(h*w);
        height = h; width = w;
        try {
            Constructor<? extends Var> ctor = type.getConstructor(JavaSpace.class,
                                                        arg1.getClass(),
                                                        arg2.getClass(),
                                                        arg3.getClass());
            
            for (int i = 0; i < h*w; ++i) {
                add(ctor.newInstance(home, arg1, arg2, arg3));
            }
        } catch(NoSuchMethodException ex) {
            throw new FatalException(ex);
        } catch(InstantiationException ex) {
            throw new FatalException(ex);
        } catch(IllegalAccessException ex) {
            throw new FatalException(ex);
        } catch(InvocationTargetException ex) {
            throw new FatalException(ex);
        }
    }
    
    /* * Construct and array with as.length elements, the elements having
     * class type. The element at position i is initialized as Var(home, as[i]).
     * /
    public <A> VarMatrix(JavaSpace home, Class<? extends Var> type, A[][] ass) {
        super(ass.length > 0 ? ass.length * ass[0].length : 0);
        if (ass.length == 0 || ass[0].length == 0) return;
        try {
            Constructor<? extends Var> ctor = type.getConstructor(JavaSpace.class, ass[0][0].getClass());
            for (A[] as: ass)
                for (A a: as) {
                    add(ctor.newInstance(home, a));
                }
        } catch(NoSuchMethodException ex) {
            throw new FatalException(ex);
        } catch(InstantiationException ex) {
            throw new FatalException(ex);
        } catch(IllegalAccessException ex) {
            throw new FatalException(ex);
        } catch(InvocationTargetException ex) {
            throw new FatalException(ex);
        }
    }

    public void swap(int a, int b) {
        Var tmp = get(a);
        set(a, get(b));
        set(b, tmp);
    }

    public void swap(int r1, int c1, int r2, int c2) {
        swap(getPos(r1, c1), getPos(r2, c2));
    }
 
    public Var get(int r, int c) {
        return get(getPos(r,c));
    }

    public void set(int r, int c, Var v) {
        set(getPos(r,c), v);
    }

    public VarArray<Var> row(int r) {
        VarArray<Var> res = new VarArray<Var>(width);
        for (int i = 0; i < width; ++i)
            res.add(get(r, i));
        return res;
    }

    public VarArray<Var> col(int c) {
        VarArray<Var> res = new VarArray<Var>(height);
        for (int i = 0; i < height; ++i)
            res.add(get(i, c));
        return res;
    }

    public VarMatrix<Var> slice(int r1, int c1, int r2, int c2) {
        int h = r2-r1, w = c2-c1;
        VarMatrix<Var> res = new VarMatrix<Var>(h, w);
        for (int i=0; i<h; i++) {
            for (int j=0; j<w; j++) {
                res.add(get(r1+i, c1+j));
            }
        }
        return res;
    }

    public String toString() {
        String ret = "[[";
        int n = size();
        for (int i=0; i < n; i++) {
            GecodeVar v = get(i);
            String name = v.getName();
            if (!name.equals(""))
                ret += name+"=";
            ret += v.toString();
            if ((i+1)%height == 0) {
                if (i+height<size()) {
                    ret += "], [";
                } else {
                    ret += "]";
                }
            } else {
                ret += ",";
            }
        }
        return ret+"]";
    }
}