001    /*
002     * Copyright (C) 2008-2010 by Holger Arndt
003     *
004     * This file is part of the Universal Java Matrix Package (UJMP).
005     * See the NOTICE file distributed with this work for additional
006     * information regarding copyright ownership and licensing.
007     *
008     * UJMP is free software; you can redistribute it and/or modify
009     * it under the terms of the GNU Lesser General Public License as
010     * published by the Free Software Foundation; either version 2
011     * of the License, or (at your option) any later version.
012     *
013     * UJMP is distributed in the hope that it will be useful,
014     * but WITHOUT ANY WARRANTY; without even the implied warranty of
015     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
016     * GNU Lesser General Public License for more details.
017     *
018     * You should have received a copy of the GNU Lesser General Public
019     * License along with UJMP; if not, write to the
020     * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
021     * Boston, MA  02110-1301  USA
022     */
023    
024    package org.ujmp.core.objectmatrix.calculation;
025    
026    import org.ujmp.core.Coordinates;
027    import org.ujmp.core.Matrix;
028    import org.ujmp.core.annotation.Annotation;
029    import org.ujmp.core.annotation.DefaultAnnotation;
030    import org.ujmp.core.exceptions.MatrixException;
031    
032    public class ExtractAnnotation extends AbstractObjectCalculation {
033            private static final long serialVersionUID = 1461447576658284276L;
034    
035            private long[] size = null;
036    
037            public ExtractAnnotation(Matrix m, int dim) {
038                    super(dim, m);
039                    size = Coordinates.copyOf(m.getSize());
040                    size[dim]--;
041                    setAnnotation(new DefaultAnnotation(size));
042                    getAnnotation().setMatrixAnnotation(m.getMatrixAnnotation());
043    
044                    if (dim == ROW) {
045                            Annotation a = m.getAnnotation();
046                            if (a != null) {
047                                    Matrix ai = a.getDimensionMatrix(COLUMN);
048                                    ai = ai.deleteRows(Ret.NEW, 0);
049                                    getAnnotation().setDimensionMatrix(COLUMN, ai);
050                            }
051                            getAnnotation().setDimensionMatrix(ROW, m.selectRows(Ret.NEW, 0));
052                    } else if (dim == COLUMN) {
053                            Annotation a = m.getAnnotation();
054                            if (a != null) {
055                                    Matrix ai = a.getDimensionMatrix(ROW);
056                                    ai = ai.selectColumns(Ret.NEW, 0);
057                                    getAnnotation().setDimensionMatrix(ROW, ai);
058                            }
059                            getAnnotation().setDimensionMatrix(COLUMN, m.selectColumns(Ret.NEW, 0));
060                    } else {
061                            throw new MatrixException("only supported for 2D matrices");
062                    }
063            }
064    
065            public Object getObject(long... coordinates) throws MatrixException {
066                    coordinates = Coordinates.copyOf(coordinates);
067                    coordinates[getDimension()]++;
068                    return getSource().getAsObject(coordinates);
069            }
070    
071            public long[] getSize() {
072                    return size;
073            }
074    
075    }