package test.tuigwaa;
   
   import java.io.File;
   import java.io.IOException;
   import java.util.ArrayList;
   import java.util.HashMap;
   import java.util.List;
   
   import junit.framework.TestCase;
  
  import org.jfree.chart.ChartUtilities;
  import org.jfree.chart.JFreeChart;
  import org.seasar.tuigwaa.database.DataRow;
  import org.seasar.tuigwaa.database.DataTable;
  import org.seasar.tuigwaa.database.DataTableImpl;
  import org.seasar.tuigwaa.database.ExtraRowDataTable;
  import org.seasar.tuigwaa.database.SelectedColumnDataTableImpl;
  import org.seasar.tuigwaa.database.SpreadSheet;
  import org.seasar.tuigwaa.database.chart.TCharFactory;
  import org.seasar.tuigwaa.database.function.criteria.DataTableCriteria;
  import org.seasar.tuigwaa.plugin.database.component.EditTableDto;
  import org.seasar.tuigwaa.util.functor.Op;
  
  public class DataTableTest extends TestCase {
  
  	private DataTable SIMPLE_DATA_TABLE;
  
  	private DataTable SELECTED_DATA_TABLE;
  
  	protected void setUp() throws Exception {
  		// simple data table
  		List data = new ArrayList();
  		data.add(new Bean("hoge", "hoge2", new Integer(1)));
  		data.add(new Bean("hoge3", "hoge4", new Integer(2)));
  		SIMPLE_DATA_TABLE = new DataTableImpl("a", Bean.headers(), Bean
  				.headers(), data, DataTable.BEAN_DATA, new HashMap());
  
  		// selected data table
  		List header = new ArrayList();
  		header.add("prop1");
  		header.add("prop3");
  		SELECTED_DATA_TABLE = new SelectedColumnDataTableImpl(
  				SIMPLE_DATA_TABLE, header);
  	}
  
  	public void testSimple() {
  		DataTable table = SIMPLE_DATA_TABLE;
  		assertEquals(2, table.getRowSize());
  		assertEquals(3, table.getColumnSize());
  	}
  
  	public void testSelectedColumn() {
  		DataTable sTable = SELECTED_DATA_TABLE;
  
  		DataRow row0 = (DataRow) sTable.next();
  		assertEquals("hoge", row0.next());
  		assertTrue(row0.hasNext());
  		assertEquals(new Integer(1), row0.next());
  
  		assertFalse(row0.hasNext());
  		assertTrue(sTable.hasNext());
  
  		DataRow row1 = (DataRow) sTable.next();
  
  		assertEquals("hoge3", row1.next());
  		assertEquals(new Integer(2), row1.next());
  
  		assertFalse(sTable.hasNext());
  	}
  
  	public void testExtraRow(){
  		DataTable dataTable = new ExtraRowDataTable(SIMPLE_DATA_TABLE, Op.SUM);
  		System.out.println("---------------------------------------");
  		while(dataTable.hasNext()){
  			DataRow dataRow = dataTable.nextRow();
  			
  			while(dataRow.hasNext()){
  				System.out.print(dataRow.next());
  				System.out.print("\t");
  			}
  			System.out.println("");
  		}
  		System.out.println("---------------------------------------");
  	}
  	
  	
  	public void testNotSelectedColumn() {
  		DataRow row = (DataRow)SIMPLE_DATA_TABLE.next();
  		assertEquals("hoge2", row.getCell("prop2"));
  	}
  
  	public void testFilter(){
  		DataTableCriteria criteria = new DataTableCriteria(SIMPLE_DATA_TABLE);
  		criteria.addEq(Bean.HEADER0, "hoge");
  		assertEquals(1,criteria.list().getRowSize());
  	}
  	
  	
  	
 	public void test1DCreator() {
 
 		List list = new ArrayList();
 		list.add(new Object[] { "aaa1", new Integer(2) });
 		list.add(new Object[] { "aaa2", new Integer(2) });
 		list.add(new Object[] { "bbb3", new Integer(3) });
 		list.add(new Object[] { "bbb4", new Integer(4) });
 
 		List headers = new ArrayList();
 		headers.add("row");
 		headers.add("num");
 		DataTable dataTable = new DataTableImpl("‚Ù‚°", headers, list,
 				DataTable.PRIMITIVE_DATA);
 
 		JFreeChart pieChart = TCharFactory.createPieChart(dataTable);
 		// JFreeChart pieChart = TCharFactory.createBarChart(dataTable);
 		// JFreeChart pieChart = TCharFactory.createLineChart(dataTable);
 		File outFile = new File("./piechart.png");
 		try {
 			ChartUtilities.saveChartAsPNG(outFile, pieChart, 500, 500);
 		} catch (IOException e) {
 			e.printStackTrace();
 		}
 	}
 
 	public void test2DCreator() {
 		Object[] row1 = new Object[] { "aaa", "hoge", new Integer(1) };
 		Object[] row2 = new Object[] { "aaa", "hage", new Integer(2) };
 		Object[] row3 = new Object[] { "bbb", "hoge", new Integer(3) };
 		Object[] row4 = new Object[] { "bbb", "hage", new Integer(4) };
 		List list = new ArrayList();
 		list.add(row1);
 		list.add(row2);
 		list.add(row3);
 		list.add(row4);
 
 		List headers = new ArrayList();
 		headers.add("row");
 		headers.add("column");
 		headers.add("num");
 
 		DataTable dataTable = new DataTableImpl("b", headers, list,
 				DataTable.PRIMITIVE_DATA);
 		SpreadSheet sheet = ((DataTableImpl) dataTable).createSpreadSheet();
 
 		// while(sheet.hasNext()){
 		// DataRow dataRow = sheet.nextRow();
 		// while(dataRow.hasNext()){
 		// System.out.print(dataRow.next());
 		// System.out.print("\t");
 		// }
 		// System.out.println("");
 		// }
 
 		//DataTableComponent comp = new DataTableComponent(sheet);
 		//System.out.println(comp.getHtml());
 
 	}
 
 	public boolean isSpreadSheet() {
 		return this instanceof SpreadSheet;
 	}
 	
 	public void test2DChart() {
 		Object[] row1 = new Object[] { "aaa", "hoge", new Integer(1) };
 		Object[] row2 = new Object[] { "aaa", "hage", new Integer(2) };
 		Object[] row3 = new Object[] { "bbb", "hoge", new Integer(3) };
 		Object[] row4 = new Object[] { "bbb", "hage", new Integer(4) };
 		List list = new ArrayList();
 		list.add(row1);
 		list.add(row2);
 		list.add(row3);
 		list.add(row4);
 
 		List headers = new ArrayList();
 		headers.add("row");
 		headers.add("column");
 		headers.add("num");
 
 		DataTable dataTable = new DataTableImpl("b", headers, list,
 				DataTable.PRIMITIVE_DATA);
 		SpreadSheet sheet = ((DataTableImpl) dataTable).createSpreadSheet();
 
 		// JFreeChart pieChart = TCharFactory.createPieChart(dataTable);
 		// JFreeChart pieChart = TCharFactory.createBarChart(sheet);
 		JFreeChart pieChart = TCharFactory.createLineChart(sheet);
 		File outFile = new File("./piechart.png");
 		try {
 			ChartUtilities.saveChartAsPNG(outFile, pieChart, 500, 500);
 		} catch (IOException e) {
 			e.printStackTrace();
 		}
 
 	}
 
 	public static class Bean {
 
 		private String prop1;
 
 		private String prop2;
 
 		private Integer prop3;
 
 		private static final String HEADER0 = "prop1";
 
 		private static final String HEADER1 = "prop2";
 		
 		private static final String HEADER2 = "prop3";
 		
 		
 		
 		public Bean(String prop1, String prop2, Integer prop3) {
 			this.prop1 = prop1;
 			this.prop2 = prop2;
 			this.prop3 = prop3;
 		}
 
 		public static List headers() {
 			List headers = new ArrayList();
 			headers.add(HEADER0);
 			headers.add(HEADER1);
 			headers.add(HEADER2);
 			return headers;
 		}
 
 		public String getProp1() {
 			return prop1;
 		}
 
 		public void setProp1(String prop1) {
 			this.prop1 = prop1;
 		}
 
 		public String getProp2() {
 			return prop2;
 		}
 
 		public void setProp2(String prop2) {
 			this.prop2 = prop2;
 		}
 
 		public void setProp3(Integer prop3) {
 			this.prop3 = prop3;
 		}
 
 		public Integer getProp3() {
 			return prop3;
 		}
 	}
 }