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

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
   3  * in Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.spreadsheet.common.matrix;\r
  13 \r
  14 import java.io.Externalizable;\r
  15 import java.io.IOException;\r
  16 import java.io.ObjectInput;\r
  17 import java.io.ObjectOutput;\r
  18 import java.util.ArrayList;\r
  19 import java.util.Collection;\r
  20 \r
  21 import org.simantics.spreadsheet.Matrix;\r
  22 \r
  23 public class MatrixImpl implements Matrix, Externalizable {\r
  24 \r
  25         private int rows;\r
  26         private int columns;\r
  27         private Object[] data;\r
  28         \r
  29         public MatrixImpl() {\r
  30                 \r
  31         }\r
  32         \r
  33         public MatrixImpl(int rows, int columns) {\r
  34                 this.rows = rows;\r
  35                 this.columns = columns;\r
  36                 this.data = new Object[rows*columns];\r
  37         }\r
  38 \r
  39         @Override\r
  40         public int getColumnCount() {\r
  41                 return columns;\r
  42         }\r
  43 \r
  44         @Override\r
  45         public int getRowCount() {\r
  46                 return rows;\r
  47         }\r
  48 \r
  49         @Override\r
  50         public void readExternal(ObjectInput in) throws IOException,\r
  51                         ClassNotFoundException {\r
  52                 rows = in.readInt();\r
  53                 columns = in.readInt();\r
  54                 data = new Object[rows*columns];\r
  55                 for(int i=0;i<data.length;i++) data[i] = in.readObject();\r
  56         }\r
  57 \r
  58         @Override\r
  59         public void writeExternal(ObjectOutput out) throws IOException {\r
  60                 out.writeInt(rows);\r
  61                 out.writeInt(columns);\r
  62                 for(Object o : data) out.writeObject(o);\r
  63         }\r
  64 \r
  65         @Override\r
  66         public Object get(int row, int column) {\r
  67                 return data[column*rows + row];\r
  68         }\r
  69         \r
  70         public void setColumn(int column, Collection<Object> objects) {\r
  71                 int index = rows*column;\r
  72                 for(Object o : objects) {\r
  73                         data[index++] = o;\r
  74                 }\r
  75         }\r
  76 \r
  77         public void set(int row, int column, Object object) {\r
  78                 data[rows*column + row] = object;\r
  79         }\r
  80 \r
  81         public Collection<Object> getColumn(int column) {\r
  82                 ArrayList<Object> result = new ArrayList<Object>();\r
  83                 for(int i=0;i<rows;i++) {\r
  84                         result.add(data[rows*column+i]);\r
  85                 }\r
  86                 return result;\r
  87         }\r
  88 \r
  89         public Collection<Object> getRow(int row) {\r
  90                 ArrayList<Object> result = new ArrayList<Object>();\r
  91                 for(int i=0;i<columns;i++) {\r
  92                         result.add(data[row + columns*i]);\r
  93                 }\r
  94                 return result;\r
  95         }\r
  96         \r
  97 }\r