bundles/org.simantics.spreadsheet.common/src/org/simantics/spreadsheet/common/matrix/VariantMatrix.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.spreadsheet.common.matrix;
  13
  14 import java.util.ArrayList;
  15 import java.util.Collection;
  16
  17 import org.simantics.databoard.binding.mutable.Variant;
  18
  19 public class VariantMatrix {
  20
  21         public int rows;
  22         public int columns;
  23         public Variant[] data;
  24
  25         public VariantMatrix() {
  26
  27         }
  28
  29         public VariantMatrix(int rows, int columns) {
  30                 this.rows = rows;
  31                 this.columns = columns;
  32                 this.data = new Variant[rows*columns];
  33         }
  34
  35         public int getColumnCount() {
  36                 return columns;
  37         }
  38
  39         public int getRowCount() {
  40                 return rows;
  41         }
  42
  43         public Variant get(int row, int column) {
  44                 return data[column*rows + row];
  45         }
  46
  47         public void setColumn(int column, Collection<Variant> objects) {
  48                 int index = rows*column;
  49                 for(Variant o : objects) {
  50                         data[index++] = o;
  51                 }
  52         }
  53
  54         public void set(int row, int column, Variant object) {
  55                 data[rows*column + row] = object;
  56         }
  57
  58         public Collection<Variant> getColumn(int column) {
  59                 ArrayList<Variant> result = new ArrayList<Variant>();
  60                 for(int i=0;i<rows;i++) {
  61                         result.add(data[rows*column+i]);
  62                 }
  63                 return result;
  64         }
  65
  66         public Collection<Variant> getRow(int row) {
  67                 ArrayList<Variant> result = new ArrayList<Variant>();
  68                 for(int i=0;i<columns;i++) {
  69                         result.add(data[row + columns*i]);
  70                 }
  71                 return result;
  72         }
  73
  74 }