poi-examples/src/main/java/org/apache/poi/examples/hssf/usermodel/HSSFReadWrite.java - poi - Git at Google

 /* ====================================================================
    Licensed to the Apache Software Foundation (ASF) under one or more
    contributor license agreements.  See the NOTICE file distributed with
    this work for additional information regarding copyright ownership.
    The ASF licenses this file to You under the Apache License, Version 2.0
    (the "License"); you may not use this file except in compliance with
    the License.  You may obtain a copy of the License at

        http://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing, software
    distributed under the License is distributed on an "AS IS" BASIS,
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.
 ==================================================================== */

 package org.apache.poi.examples.hssf.usermodel;

 import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.IOException;

 import org.apache.poi.hssf.usermodel.HSSFCell;
 import org.apache.poi.hssf.usermodel.HSSFCellStyle;
 import org.apache.poi.hssf.usermodel.HSSFDataFormat;
 import org.apache.poi.hssf.usermodel.HSSFFont;
 import org.apache.poi.hssf.usermodel.HSSFRichTextString;
 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.BorderStyle;
 import org.apache.poi.ss.usermodel.FillPatternType;
 import org.apache.poi.ss.util.CellRangeAddress;

 /**
  * File for HSSF testing/examples
  *
  * THIS IS NOT THE MAIN HSSF FILE!! This is a utility for testing functionality.
  * It does contain sample API usage that may be educational to regular API
  * users.
  */
 @SuppressWarnings({"java:S106","java:S4823"})
 public final class HSSFReadWrite {

 	private HSSFReadWrite() {}

 	/**
 	 * creates an {@link HSSFWorkbook} with the specified OS filename.
 	 */
 	private static HSSFWorkbook readFile(String filename) throws IOException {
 		try (FileInputStream fis = new FileInputStream(filename)) {
 			return new HSSFWorkbook(fis);        // NOSONAR - should not be closed here
 		}
 	}

 	/**
 	 * given a filename this outputs a sample sheet with just a set of
 	 * rows/cells.
 	 */
 	private static void testCreateSampleSheet(String outputFilename) throws IOException {
 		try (HSSFWorkbook wb = new HSSFWorkbook();
 			 FileOutputStream out = new FileOutputStream(outputFilename)) {
 			HSSFSheet s = wb.createSheet();
 			HSSFCellStyle cs = wb.createCellStyle();
 			HSSFCellStyle cs2 = wb.createCellStyle();
 			HSSFCellStyle cs3 = wb.createCellStyle();
 			HSSFFont f = wb.createFont();
 			HSSFFont f2 = wb.createFont();

 			f.setFontHeightInPoints((short) 12);
 			f.setColor((short) 0xA);
 			f.setBold(true);
 			f2.setFontHeightInPoints((short) 10);
 			f2.setColor((short) 0xf);
 			f2.setBold(true);
 			cs.setFont(f);
 			cs.setDataFormat(HSSFDataFormat.getBuiltinFormat("($#,##0_);[Red]($#,##0)"));
 			cs2.setBorderBottom(BorderStyle.THIN);
 			cs2.setFillPattern(FillPatternType.SOLID_FOREGROUND);
 			cs2.setFillForegroundColor((short) 0xA);
 			cs2.setFont(f2);
 			wb.setSheetName(0, "HSSF Test");
 			int rownum;
 			for (rownum = 0; rownum < 300; rownum++) {
 				HSSFRow r = s.createRow(rownum);
 				if ((rownum % 2) == 0) {
 					r.setHeight((short) 0x249);
 				}

 				for (int cellnum = 0; cellnum < 50; cellnum += 2) {
 					HSSFCell c = r.createCell(cellnum);
 					c.setCellValue((rownum * 10000.0) + cellnum
 							+ (rownum / 1000.0) + (cellnum / 10000.0));
 					if ((rownum % 2) == 0) {
 						c.setCellStyle(cs);
 					}
 					c = r.createCell(cellnum + 1);
 					c.setCellValue(new HSSFRichTextString("TEST"));
 					// 50 characters divided by 1/20th of a point
 					s.setColumnWidth(cellnum + 1, (int) (50 * 8 / 0.05));
 					if ((rownum % 2) == 0) {
 						c.setCellStyle(cs2);
 					}
 				}
 			}

 			// draw a thick black border on the row at the bottom using BLANKS
 			rownum++;
 			rownum++;
 			HSSFRow r = s.createRow(rownum);
 			cs3.setBorderBottom(BorderStyle.THICK);
 			for (int cellnum = 0; cellnum < 50; cellnum++) {
 				HSSFCell c = r.createCell(cellnum);
 				c.setCellStyle(cs3);
 			}
 			s.addMergedRegion(new CellRangeAddress(0, 3, 0, 3));
 			s.addMergedRegion(new CellRangeAddress(100, 110, 100, 110));

 			// end draw thick black border
 			// create a sheet, set its title then delete it
 			wb.createSheet();
 			wb.setSheetName(1, "DeletedSheet");
 			wb.removeSheetAt(1);

 			// end deleted sheet
 			wb.write(out);
 		}
 	}

 	/**
      * Method main
      *
      * Given 1 argument takes that as the filename, inputs it and dumps the
      * cell values/types out to sys.out.<br>
      *
      * given 2 arguments where the second argument is the word "write" and the
      * first is the filename - writes out a sample (test) spreadsheet
      * see {@link HSSFReadWrite#testCreateSampleSheet(String)}.<br>
      *
      * given 2 arguments where the first is an input filename and the second
      * an output filename (not write), attempts to fully read in the
      * spreadsheet and fully write it out.<br>
      *
      * given 3 arguments where the first is an input filename and the second an
      * output filename (not write) and the third is "modify1", attempts to read in the
      * spreadsheet, deletes rows 0-24, 74-99.  Changes cell at row 39, col 3 to
      * "MODIFIED CELL" then writes it out.  Hence this is "modify test 1".  If you
      * take the output from the write test, you'll have a valid scenario.
      */
 	public static void main(String[] args) throws Exception {
 		if (args.length < 1) {
 			System.err.println("At least one argument expected");
 			return;
 		}

 		String fileName = args[0];
 		if (args.length < 2) {

 			try (HSSFWorkbook wb = HSSFReadWrite.readFile(fileName)) {
 				System.out.println("Data dump:\n");

 				for (int k = 0; k < wb.getNumberOfSheets(); k++) {
 					HSSFSheet sheet = wb.getSheetAt(k);
 					int rows = sheet.getPhysicalNumberOfRows();
 					System.out.println("Sheet " + k + " \"" + wb.getSheetName(k) + "\" has " + rows	+ " row(s).");
 					for (int r = 0; r < rows; r++) {
 						HSSFRow row = sheet.getRow(r);
 						if (row == null) {
 							continue;
 						}

 						System.out.println("\nROW " + row.getRowNum() + " has " + row.getPhysicalNumberOfCells() + " cell(s).");
 						for (int c = 0; c < row.getLastCellNum(); c++) {
 							HSSFCell cell = row.getCell(c);
 							String value;

 							if (cell != null) {
 								switch (cell.getCellType()) {

 									case FORMULA:
 										value = "FORMULA value=" + cell.getCellFormula();
 										break;

 									case NUMERIC:
 										value = "NUMERIC value=" + cell.getNumericCellValue();
 										break;

 									case STRING:
 										value = "STRING value=" + cell.getStringCellValue();
 										break;

 									case BLANK:
 										value = "<BLANK>";
 										break;

 									case BOOLEAN:
 										value = "BOOLEAN value-" + cell.getBooleanCellValue();
 										break;

 									case ERROR:
 										value = "ERROR value=" + cell.getErrorCellValue();
 										break;

 									default:
 										value = "UNKNOWN value of type " + cell.getCellType();
 								}
 								System.out.println("CELL col=" + cell.getColumnIndex() + " VALUE=" + value);
 							}
 						}
 					}
 				}
 			}
 		} else if (args.length == 2) {
 			if ("write".equalsIgnoreCase(args[1])) {
 				System.out.println("Write mode");
 				long time = System.currentTimeMillis();
 				HSSFReadWrite.testCreateSampleSheet(fileName);

 				System.out.println("" + (System.currentTimeMillis() - time) + " ms generation time");
 			} else {
 				System.out.println("readwrite test");
 				try (HSSFWorkbook wb = HSSFReadWrite.readFile(fileName);
 					 FileOutputStream stream = new FileOutputStream(args[1])) {
 					wb.write(stream);
 				}
 			}
 		} else if (args.length == 3 && "modify1".equalsIgnoreCase(args[2])) {
 			// delete row 0-24, row 74 - 99 && change cell 3 on row 39 to string "MODIFIED CELL!!"

 			try (HSSFWorkbook wb = HSSFReadWrite.readFile(fileName);
 				 FileOutputStream stream = new FileOutputStream(args[1])) {
 				HSSFSheet sheet = wb.getSheetAt(0);

 				for (int k = 0; k < 25; k++) {
 					HSSFRow row = sheet.getRow(k);
 					sheet.removeRow(row);
 				}
 				for (int k = 74; k < 100; k++) {
 					HSSFRow row = sheet.getRow(k);
 					sheet.removeRow(row);
 				}
 				HSSFRow row = sheet.getRow(39);
 				HSSFCell cell = row.getCell(3);
 				cell.setCellValue("MODIFIED CELL!!!!!");

 				wb.write(stream);
 			}
 		}
 	}
 }
	/* ====================================================================
	Licensed to the Apache Software Foundation (ASF) under one or more
	contributor license agreements. See the NOTICE file distributed with
	this work for additional information regarding copyright ownership.
	The ASF licenses this file to You under the Apache License, Version 2.0
	(the "License"); you may not use this file except in compliance with
	the License. You may obtain a copy of the License at

	http://www.apache.org/licenses/LICENSE-2.0

	Unless required by applicable law or agreed to in writing, software
	distributed under the License is distributed on an "AS IS" BASIS,
	WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	See the License for the specific language governing permissions and
	limitations under the License.
	==================================================================== */

	package org.apache.poi.examples.hssf.usermodel;

	import java.io.FileInputStream;
	import java.io.FileOutputStream;
	import java.io.IOException;

	import org.apache.poi.hssf.usermodel.HSSFCell;
	import org.apache.poi.hssf.usermodel.HSSFCellStyle;
	import org.apache.poi.hssf.usermodel.HSSFDataFormat;
	import org.apache.poi.hssf.usermodel.HSSFFont;
	import org.apache.poi.hssf.usermodel.HSSFRichTextString;
	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.BorderStyle;
	import org.apache.poi.ss.usermodel.FillPatternType;
	import org.apache.poi.ss.util.CellRangeAddress;

	/**
	* File for HSSF testing/examples
	*
	* THIS IS NOT THE MAIN HSSF FILE!! This is a utility for testing functionality.
	* It does contain sample API usage that may be educational to regular API
	* users.
	*/
	@SuppressWarnings({"java:S106","java:S4823"})
	public final class HSSFReadWrite {

	private HSSFReadWrite() {}

	/**
	* creates an {@link HSSFWorkbook} with the specified OS filename.
	*/
	private static HSSFWorkbook readFile(String filename) throws IOException {
	try (FileInputStream fis = new FileInputStream(filename)) {
	return new HSSFWorkbook(fis); // NOSONAR - should not be closed here
	}
	}

	/**
	* given a filename this outputs a sample sheet with just a set of
	* rows/cells.
	*/
	private static void testCreateSampleSheet(String outputFilename) throws IOException {
	try (HSSFWorkbook wb = new HSSFWorkbook();
	FileOutputStream out = new FileOutputStream(outputFilename)) {
	HSSFSheet s = wb.createSheet();
	HSSFCellStyle cs = wb.createCellStyle();
	HSSFCellStyle cs2 = wb.createCellStyle();
	HSSFCellStyle cs3 = wb.createCellStyle();
	HSSFFont f = wb.createFont();
	HSSFFont f2 = wb.createFont();

	f.setFontHeightInPoints((short) 12);
	f.setColor((short) 0xA);
	f.setBold(true);
	f2.setFontHeightInPoints((short) 10);
	f2.setColor((short) 0xf);
	f2.setBold(true);
	cs.setFont(f);
	cs.setDataFormat(HSSFDataFormat.getBuiltinFormat("($#,##0_);[Red]($#,##0)"));
	cs2.setBorderBottom(BorderStyle.THIN);
	cs2.setFillPattern(FillPatternType.SOLID_FOREGROUND);
	cs2.setFillForegroundColor((short) 0xA);
	cs2.setFont(f2);
	wb.setSheetName(0, "HSSF Test");
	int rownum;
	for (rownum = 0; rownum < 300; rownum++) {
	HSSFRow r = s.createRow(rownum);
	if ((rownum % 2) == 0) {
	r.setHeight((short) 0x249);
	}

	for (int cellnum = 0; cellnum < 50; cellnum += 2) {
	HSSFCell c = r.createCell(cellnum);
	c.setCellValue((rownum * 10000.0) + cellnum
	+ (rownum / 1000.0) + (cellnum / 10000.0));
	if ((rownum % 2) == 0) {
	c.setCellStyle(cs);
	}
	c = r.createCell(cellnum + 1);
	c.setCellValue(new HSSFRichTextString("TEST"));
	// 50 characters divided by 1/20th of a point
	s.setColumnWidth(cellnum + 1, (int) (50 * 8 / 0.05));
	if ((rownum % 2) == 0) {
	c.setCellStyle(cs2);
	}
	}
	}

	// draw a thick black border on the row at the bottom using BLANKS
	rownum++;
	rownum++;
	HSSFRow r = s.createRow(rownum);
	cs3.setBorderBottom(BorderStyle.THICK);
	for (int cellnum = 0; cellnum < 50; cellnum++) {
	HSSFCell c = r.createCell(cellnum);
	c.setCellStyle(cs3);
	}
	s.addMergedRegion(new CellRangeAddress(0, 3, 0, 3));
	s.addMergedRegion(new CellRangeAddress(100, 110, 100, 110));

	// end draw thick black border
	// create a sheet, set its title then delete it
	wb.createSheet();
	wb.setSheetName(1, "DeletedSheet");
	wb.removeSheetAt(1);

	// end deleted sheet
	wb.write(out);
	}
	}

	/**
	* Method main
	*
	* Given 1 argument takes that as the filename, inputs it and dumps the
	* cell values/types out to sys.out.<br>
	*
	* given 2 arguments where the second argument is the word "write" and the
	* first is the filename - writes out a sample (test) spreadsheet
	* see {@link HSSFReadWrite#testCreateSampleSheet(String)}.<br>
	*
	* given 2 arguments where the first is an input filename and the second
	* an output filename (not write), attempts to fully read in the
	* spreadsheet and fully write it out.<br>
	*
	* given 3 arguments where the first is an input filename and the second an
	* output filename (not write) and the third is "modify1", attempts to read in the
	* spreadsheet, deletes rows 0-24, 74-99. Changes cell at row 39, col 3 to
	* "MODIFIED CELL" then writes it out. Hence this is "modify test 1". If you
	* take the output from the write test, you'll have a valid scenario.
	*/
	public static void main(String[] args) throws Exception {
	if (args.length < 1) {
	System.err.println("At least one argument expected");
	return;
	}

	String fileName = args[0];
	if (args.length < 2) {

	try (HSSFWorkbook wb = HSSFReadWrite.readFile(fileName)) {
	System.out.println("Data dump:\n");

	for (int k = 0; k < wb.getNumberOfSheets(); k++) {
	HSSFSheet sheet = wb.getSheetAt(k);
	int rows = sheet.getPhysicalNumberOfRows();
	System.out.println("Sheet " + k + " \"" + wb.getSheetName(k) + "\" has " + rows + " row(s).");
	for (int r = 0; r < rows; r++) {
	HSSFRow row = sheet.getRow(r);
	if (row == null) {
	continue;
	}

	System.out.println("\nROW " + row.getRowNum() + " has " + row.getPhysicalNumberOfCells() + " cell(s).");
	for (int c = 0; c < row.getLastCellNum(); c++) {
	HSSFCell cell = row.getCell(c);
	String value;

	if (cell != null) {
	switch (cell.getCellType()) {

	case FORMULA:
	value = "FORMULA value=" + cell.getCellFormula();
	break;

	case NUMERIC:
	value = "NUMERIC value=" + cell.getNumericCellValue();
	break;

	case STRING:
	value = "STRING value=" + cell.getStringCellValue();
	break;

	case BLANK:
	value = "<BLANK>";
	break;

	case BOOLEAN:
	value = "BOOLEAN value-" + cell.getBooleanCellValue();
	break;

	case ERROR:
	value = "ERROR value=" + cell.getErrorCellValue();
	break;

	default:
	value = "UNKNOWN value of type " + cell.getCellType();
	}
	System.out.println("CELL col=" + cell.getColumnIndex() + " VALUE=" + value);
	}
	}
	}
	}
	}
	} else if (args.length == 2) {
	if ("write".equalsIgnoreCase(args[1])) {
	System.out.println("Write mode");
	long time = System.currentTimeMillis();
	HSSFReadWrite.testCreateSampleSheet(fileName);

	System.out.println("" + (System.currentTimeMillis() - time) + " ms generation time");
	} else {
	System.out.println("readwrite test");
	try (HSSFWorkbook wb = HSSFReadWrite.readFile(fileName);
	FileOutputStream stream = new FileOutputStream(args[1])) {
	wb.write(stream);
	}
	}
	} else if (args.length == 3 && "modify1".equalsIgnoreCase(args[2])) {
	// delete row 0-24, row 74 - 99 && change cell 3 on row 39 to string "MODIFIED CELL!!"

	try (HSSFWorkbook wb = HSSFReadWrite.readFile(fileName);
	FileOutputStream stream = new FileOutputStream(args[1])) {
	HSSFSheet sheet = wb.getSheetAt(0);

	for (int k = 0; k < 25; k++) {
	HSSFRow row = sheet.getRow(k);
	sheet.removeRow(row);
	}
	for (int k = 74; k < 100; k++) {
	HSSFRow row = sheet.getRow(k);
	sheet.removeRow(row);
	}
	HSSFRow row = sheet.getRow(39);
	HSSFCell cell = row.getCell(3);
	cell.setCellValue("MODIFIED CELL!!!!!");

	wb.write(stream);
	}
	}
	}
	}