Package org.apache.poi.hssf.usermodel

Source Code of org.apache.poi.hssf.usermodel.TestHSSFFormulaEvaluator$EvalCountListener

/* ====================================================================
   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.hssf.usermodel;

import junit.framework.AssertionFailedError;
import junit.framework.TestCase;

import org.apache.poi.hssf.HSSFTestDataSamples;
import org.apache.poi.hssf.record.NameRecord;
import org.apache.poi.hssf.record.formula.eval.NumberEval;
import org.apache.poi.hssf.record.formula.eval.ValueEval;
import org.apache.poi.ss.formula.EvaluationCell;
import org.apache.poi.ss.formula.EvaluationListener;
import org.apache.poi.ss.formula.WorkbookEvaluator;
import org.apache.poi.ss.formula.WorkbookEvaluatorTestHelper;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellValue;
/**
*
* @author Josh Micich
*/
public final class TestHSSFFormulaEvaluator extends TestCase {

  /**
   * Test that the HSSFFormulaEvaluator can evaluate simple named ranges
   *  (single cells and rectangular areas)
   */
  public void testEvaluateSimple() {
    HSSFWorkbook wb = HSSFTestDataSamples.openSampleWorkbook("testNames.xls");
    HSSFSheet sheet = wb.getSheetAt(0);
    HSSFCell cell = sheet.getRow(8).getCell(0);
    HSSFFormulaEvaluator fe = new HSSFFormulaEvaluator(wb);
    CellValue cv = fe.evaluate(cell);
    assertEquals(HSSFCell.CELL_TYPE_NUMERIC, cv.getCellType());
    assertEquals(3.72, cv.getNumberValue(), 0.0);
  }

  public void testFullColumnRefs() {
    HSSFWorkbook wb = new HSSFWorkbook();
    HSSFSheet sheet = wb.createSheet("Sheet1");
    HSSFRow row = sheet.createRow(0);
    HSSFCell cell0 = row.createCell(0);
    cell0.setCellFormula("sum(D:D)");
    HSSFCell cell1 = row.createCell(1);
    cell1.setCellFormula("sum(D:E)");

    // some values in column D
    setValue(sheet, 1, 3, 5.0);
    setValue(sheet, 2, 3, 6.0);
    setValue(sheet, 5, 3, 7.0);
    setValue(sheet, 50, 3, 8.0);

    // some values in column E
    setValue(sheet, 1, 4, 9.0);
    setValue(sheet, 2, 4, 10.0);
    setValue(sheet, 30000, 4, 11.0);

    // some other values
    setValue(sheet, 1, 2, 100.0);
    setValue(sheet, 2, 5, 100.0);
    setValue(sheet, 3, 6, 100.0);


    HSSFFormulaEvaluator fe = new HSSFFormulaEvaluator(wb);
    assertEquals(26.0, fe.evaluate(cell0).getNumberValue(), 0.0);
    assertEquals(56.0, fe.evaluate(cell1).getNumberValue(), 0.0);
  }

  private static void setValue(HSSFSheet sheet, int rowIndex, int colIndex, double value) {
    HSSFRow row = sheet.getRow(rowIndex);
    if (row == null) {
      row = sheet.createRow(rowIndex);
    }
    row.createCell(colIndex).setCellValue(value);
  }

  /**
   * {@link HSSFFormulaEvaluator#evaluate(org.apache.poi.ss.usermodel.Cell)} should behave the same whether the cell
   * is <code>null</code> or blank.
   */
  public void testEvaluateBlank() {
    HSSFWorkbook wb = new HSSFWorkbook();
    HSSFFormulaEvaluator fe = new HSSFFormulaEvaluator(wb);
    assertNull(fe.evaluate(null));
    HSSFSheet sheet = wb.createSheet("Sheet1");
    HSSFCell cell = sheet.createRow(0).createCell(0);
    assertNull(fe.evaluate(cell));
  }

  /**
   * Test for bug due to attempt to convert a cached formula error result to a boolean
   */
  public void testUpdateCachedFormulaResultFromErrorToNumber_bug46479() {

    HSSFWorkbook wb = new HSSFWorkbook();
    HSSFSheet sheet = wb.createSheet("Sheet1");
    HSSFRow row = sheet.createRow(0);
    HSSFCell cellA1 = row.createCell(0);
    HSSFCell cellB1 = row.createCell(1);
    cellB1.setCellFormula("A1+1");
    HSSFFormulaEvaluator fe = new HSSFFormulaEvaluator(wb);

    cellA1.setCellErrorValue((byte)HSSFErrorConstants.ERROR_NAME);
    fe.evaluateFormulaCell(cellB1);

    cellA1.setCellValue(2.5);
    fe.notifyUpdateCell(cellA1);
    try {
      fe.evaluateInCell(cellB1);
    } catch (IllegalStateException e) {
      if (e.getMessage().equals("Cannot get a numeric value from a error formula cell")) {
        throw new AssertionFailedError("Identified bug 46479a");
      }
    }
    assertEquals(3.5, cellB1.getNumericCellValue(), 0.0);
  }

  /**
   * When evaluating defined names, POI has to decide whether it is capable.  Currently
   * (May2009) POI only supports simple cell and area refs.<br/>
   * The sample spreadsheet (bugzilla attachment 23508) had a name flagged as 'complex'
   * which contained a simple area ref.  It is not clear what the 'complex' flag is used
   * for but POI should look elsewhere to decide whether it can evaluate the name.
   */
  public void testDefinedNameWithComplexFlag_bug47048() {
    // Mock up a spreadsheet to match the critical details of the sample
    HSSFWorkbook wb = new HSSFWorkbook();
    HSSFSheet sheet = wb.createSheet("Input");
    HSSFName definedName = wb.createName();
    definedName.setNameName("Is_Multicar_Vehicle");
    definedName.setRefersToFormula("Input!$B$17:$G$17");

    // Set up some data and the formula
    HSSFRow row17 = sheet.createRow(16);
    row17.createCell(0).setCellValue(25.0);
    row17.createCell(1).setCellValue(1.33);
    row17.createCell(2).setCellValue(4.0);

    HSSFRow row = sheet.createRow(0);
    HSSFCell cellA1 = row.createCell(0);
    cellA1.setCellFormula("SUM(Is_Multicar_Vehicle)");

    // Set the complex flag - POI doesn't usually manipulate this flag
    NameRecord nameRec = TestHSSFName.getNameRecord(definedName);
    nameRec.setOptionFlag((short)0x10); // 0x10 -> complex

    HSSFFormulaEvaluator hsf = new HSSFFormulaEvaluator(wb);
    CellValue value;
    try {
      value = hsf.evaluate(cellA1);
    } catch (RuntimeException e) {
      if (e.getMessage().equals("Don't now how to evalate name 'Is_Multicar_Vehicle'")) {
        throw new AssertionFailedError("Identified bug 47048a");
      }
      throw e;
    }

    assertEquals(Cell.CELL_TYPE_NUMERIC, value.getCellType());
    assertEquals(5.33, value.getNumberValue(), 0.0);
  }
  private static final class EvalCountListener extends EvaluationListener {
    private int _evalCount;
    public EvalCountListener() {
      _evalCount = 0;
    }
    public void onStartEvaluate(EvaluationCell cell, ICacheEntry entry) {
      _evalCount++;
    }
    public int getEvalCount() {
      return _evalCount;
    }
  }

  /**
   * The HSSFFormula evaluator performance benefits greatly from caching of intermediate cell values
   */
  public void testShortCircuitIfEvaluation() {

    // Set up a simple IF() formula that has measurable evaluation cost for its operands.
    HSSFWorkbook wb = new HSSFWorkbook();
    HSSFSheet sheet = wb.createSheet("Sheet1");
    HSSFRow row = sheet.createRow(0);
    HSSFCell cellA1 = row.createCell(0);
    cellA1.setCellFormula("if(B1,C1,D1+E1+F1)");
    // populate cells B1..F1 with simple formulas instead of plain values so we can use
    // EvaluationListener to check which parts of the first formula get evaluated
    for (int i=1; i<6; i++) {
      // formulas are just literal constants "1".."5"
      row.createCell(i).setCellFormula(String.valueOf(i));
    }

    EvalCountListener evalListener = new EvalCountListener();
    WorkbookEvaluator evaluator = WorkbookEvaluatorTestHelper.createEvaluator(wb, evalListener);
    ValueEval ve = evaluator.evaluate(HSSFEvaluationTestHelper.wrapCell(cellA1));
    int evalCount = evalListener.getEvalCount();
    if (evalCount == 6) {
      // Without short-circuit-if evaluation, evaluating cell 'A1' takes 3 extra evaluations (for D1,E1,F1)
      throw new AssertionFailedError("Identifed bug 48195 - Formula evaluator should short-circuit IF() calculations.");
    }
    assertEquals(3, evalCount);
    assertEquals(2.0, ((NumberEval)ve).getNumberValue(), 0D);
  }
}
TOP

Related Classes of org.apache.poi.hssf.usermodel.TestHSSFFormulaEvaluator$EvalCountListener

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.