Add setCell method the DataTble

Author: [email protected]

src/main/java/com/google/visualization/datasource/datatable/DataTable.java

@@ -316,6 +316,31 @@ public List<TableCell> getColumnCells(String columnId) {
   public TableCell getCell(int rowIndex, int colIndex) {
     return getRow(rowIndex).getCell(colIndex);
   }
+  
+  /**
+   * Replaces an existing cell at a specified position in this table with the specified cell.
+   * The value type of the new cell must match the type of the existing one.
+   *  
+   * @param rowIndex The row index.
+   * @param colIndex The column index.
+   * @param cell The cell to be stored at the specified position.
+   * 
+   * @return The cell that was replaced.
+   * 
+   * @throws TypeMismatchException Thrown if the new cell value type doesn't match
+   *    the table column value type.
+   * @throws IndexOutOfBoundsException Thrown if the position is out of range.
+   */
+  public TableCell setCell(int rowIndex, int colIndex, TableCell cell)
+      throws TypeMismatchException, IndexOutOfBoundsException {
+    TableRow row = rows.get(rowIndex);
+    if (!row.getCell(colIndex).getType().equals(cell.getType())) {
+      throw new TypeMismatchException("New cell value type does not match expected value type." +
+          " Expected type: " + row.getCell(colIndex).getType() +
+          " but was: " + cell.getType().toString());
+    }
+    return row.setCell(colIndex, cell);
+  }
 
   /**
    * Returns the value in the cell at the specified row and column indexes.

src/main/java/com/google/visualization/datasource/datatable/TableRow.java

@@ -115,6 +115,22 @@ public List<TableCell> getCells() {
   public TableCell getCell(int index) {
     return cells.get(index);
   }
+  
+  /**
+   * Package protected function.
+   * Replaces the cell at the specified position in this row with the specified cell.
+   * The value type of the new cell must match the that of the replaced one.
+   * 
+   * @param index The index of the cell to replace.
+   * @param cell The cell to be stored at the specified position.
+   * 
+   * @return The cell that was replaced.
+   * 
+   * @throws IndexOutOfBoundsException Thrown if the index out of range.
+   */
+  TableCell setCell(int index, TableCell cell) throws IndexOutOfBoundsException {
+    return cells.set(index, cell);
+  }
 
   /**
    * Retrieves a custom property. Returns null if it does not exist.

src/test/java/com/google/visualization/datasource/datatable/DataTableTest.java

@@ -333,4 +333,48 @@ public void testGetCellAndGetValue() {
     TableCell cell = testData.getCell(1, 0);
     assertEquals("$ccc", cell.getFormattedValue());
   }
+  
+  public void testReplaceCell() throws Exception {
+    DataTable dataTable = testData.clone();
+    
+    TableCell newNumberValueCell = new TableCell(new NumberValue(5699), "5699$");
+    newNumberValueCell.setCustomProperty("testProperty", "A test property");
+    TableCell newBooleanValueCell = new TableCell(false);
+    TableCell newDateValueCell = new TableCell(new DateValue(2011, 3, 6), "03/06/2011");
+    
+    dataTable.setCell(1, 1, newNumberValueCell);
+    dataTable.setCell(2, 2, newBooleanValueCell);
+    dataTable.setCell(2, 3, newDateValueCell);
+    
+    assertEquals(newNumberValueCell.getValue(), dataTable.getValue(1, 1));
+    assertEquals(newNumberValueCell.getFormattedValue(),
+        dataTable.getCell(1,1).getFormattedValue());
+    assertEquals(newNumberValueCell.getCustomProperty("testProperty"),
+        dataTable.getCell(1,1).getCustomProperty("testProperty"));
+    assertEquals(newBooleanValueCell.getValue(), dataTable.getValue(2, 2));
+    assertEquals(newDateValueCell.getValue(), dataTable.getValue(2, 3));
+    assertEquals(newDateValueCell.getFormattedValue(),
+        dataTable.getCell(2,3).getFormattedValue());
+    
+    try {
+      dataTable.setCell(3, 3, new TableCell(100));
+      fail();
+    } catch (TypeMismatchException e) {
+      // Expected behavior.
+    }
+    
+    try {
+      dataTable.setCell(1, 1, new TableCell(false));
+      fail();
+    } catch (TypeMismatchException e) {
+      // Expected behavior.
+    }
+    
+    try {
+      dataTable.setCell(20, 20, new TableCell(100));
+      fail();
+    } catch (IndexOutOfBoundsException e) {
+      // Expected behavior.
+    }
+  }
 }