package cn.ogsu.vod.util;


import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;


/**
 * 解析excel的輔助類
 * @author enter
 * @date 2016年10月10日
 */
public class ExcelUtil {

/**
* 匯出excel 資料來源於data
* @param data
* @throws Exception
*/
@SuppressWarnings({ "rawtypes", "resource" })
public static void exportExcel(List<PageData> data,OutputStream output) throws Exception{
//建立excel的文件物件
HSSFWorkbook excel=new HSSFWorkbook();
//建立新的sheet物件（excel的表單）
HSSFSheet sheet=excel.createSheet();
//代表excel的行物件
HSSFRow row=null;
//代表excel的列物件
HSSFCell column=null;
int index=0;
for (int i = 0; i <data.size()+1; i++) {
//建立excel行
row=sheet.createRow(i);
PageData pd=data.get(index);
//獲取map集合中所有的key
Set collect=pd.keySet();
int j=collect.size()-1;
//建立表格標題
if(i==0){
for (Object obj:collect) {
column=row.createCell(j);
column.setCellValue(obj.toString());
j--;
}
}else{
j=collect.size()-1;
for (Object obj:collect) {
//建立excel列
column=row.createCell(j);
//設定單元格的值
column.setCellValue(pd.get(obj).toString());
j--;
}
index++;
}
}
excel.write(output);
output.flush();
}

/**
* 匯入excel 資料來源於excel
* @param stream
* @return
* @throws Exception
*/
public static List<PageData> importExcel(InputStream stream) throws Exception{
List<PageData> pds=new ArrayList<>();
List<String> title=new ArrayList<>();
PageData pd=null;
//根據指定的檔案輸入流匯入Excel從而產生Workbook物件
Workbook excel = new HSSFWorkbook(stream);
//獲取Excel文件中的第一個表單
Sheet sheet = excel.getSheetAt(0);
//是否在讀取表格的第一行
boolean rowInOne=true;
//對Sheet中的每一行進行迭代
        for (Row row : sheet) {
        if(!rowInOne){
        int rowNum=0;
        pd=new PageData();
        for (Cell cell : row) {
    String cellValue=cell.getStringCellValue();
        String cellTitle=title.get(rowNum);
        pd.put(cellTitle, cellValue);
        rowNum++;
    }
        pds.add(pd);
        }else{
        for (Cell cell : row) {
            String cellTitle=cell.getStringCellValue();
            //新增標題
                title.add(cellTitle);
    }
        rowInOne=false;
        }
}
        excel.close();
return pds;
}
}