/*
* Copyright (C) 2008-2015 by Holger Arndt
*
* This file is part of the Universal Java Matrix Package (UJMP).
* See the NOTICE file distributed with this work for additional
* information regarding copyright ownership and licensing.
*
* UJMP 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 2
* of the License, or (at your option) any later version.
*
* UJMP 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 General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with UJMP; if not, write to the
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
* Boston, MA 02110-1301 USA
*/
package org.ujmp.core.objectmatrix.calculation;
import java.util.Collection;
import java.util.List;
import org.ujmp.core.Matrix;
import org.ujmp.core.util.MathUtil;
import org.ujmp.core.util.StringUtil;
public class Deletion extends AbstractObjectCalculation {
private static final long serialVersionUID = 3714270132906708701L;
private long[][] selection = null;
public Deletion(Matrix m, String deletionString) {
this(m, StringUtil.parseSelection(deletionString, m.getSize()));
}
public Deletion(Matrix m, Collection<? extends Number>... deletion) {
super(m);
List<Long> rows = MathUtil.sequenceListLong(0, getSource().getRowCount());
List<Long> columns = MathUtil.sequenceListLong(0, getSource().getColumnCount());
for (Number n : deletion[ROW]) {
rows.remove(n.longValue());
}
for (Number n : deletion[COLUMN]) {
columns.remove(n.longValue());
}
selection = new long[2][];
selection[ROW] = MathUtil.collectionToLongArray(rows);
selection[COLUMN] = MathUtil.collectionToLongArray(columns);
}
public Deletion(Matrix m, long[]... deletion) {
super(m);
List<Long> rows = MathUtil.sequenceListLong(0, getSource().getRowCount());
List<Long> columns = MathUtil.sequenceListLong(0, getSource().getColumnCount());
if (deletion != null && deletion[ROW] != null) {
for (int r = 0; r < deletion[ROW].length; r++) {
rows.remove(deletion[ROW][r]);
}
}
if (deletion.length > 1 && deletion[COLUMN] != null) {
for (int c = 0; c < deletion[COLUMN].length; c++) {
columns.remove(deletion[COLUMN][c]);
}
}
selection = new long[2][];
selection[ROW] = MathUtil.collectionToLongArray(rows);
selection[COLUMN] = MathUtil.collectionToLongArray(columns);
}
public Object getObject(long... coordinates) {
return getSource().getAsObject(selection[ROW][(int) coordinates[ROW]],
selection[COLUMN][(int) coordinates[COLUMN]]);
}
public long[] getSize() {
return new long[] { selection[ROW].length, selection[COLUMN].length };
}
}