testareaptg.java

介绍java核心技术的pio编程方法以及基本概念

        
/* ====================================================================
   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.record.formula;

import junit.framework.TestCase;

import org.apache.poi.hssf.model.FormulaParser;

/**
 * Tests for {@link AreaPtg}.
 *
 * @author Dmitriy Kumshayev
 */
public class TestAreaPtg extends TestCase
{

	AreaPtg relative;
	AreaPtg absolute;
	
	protected void setUp() throws Exception
	{
		super.setUp();
		short firstRow=5;
		short lastRow=13;
		short firstCol=7;
		short lastCol=17;
		relative = new AreaPtg(firstRow,lastRow,firstCol,lastCol,true,true,true,true);
		absolute = new AreaPtg(firstRow,lastRow,firstCol,lastCol,false,false,false,false);
	}

	public void testSetColumnsAbsolute()
	{
		resetColumns(absolute);
		validateReference(true, absolute);
	}
	public void testSetColumnsRelative()
	{
		resetColumns(relative);
		validateReference(false, relative);
	}

	private void validateReference(boolean abs, AreaPtg ref)
	{
		assertEquals("First column reference is not "+(abs?"absolute":"relative"),abs,!ref.isFirstColRelative());
		assertEquals("Last column reference is not "+(abs?"absolute":"relative"),abs,!ref.isLastColRelative());
		assertEquals("First row reference is not "+(abs?"absolute":"relative"),abs,!ref.isFirstRowRelative());
		assertEquals("Last row reference is not "+(abs?"absolute":"relative"),abs,!ref.isLastRowRelative());
	}


	public void resetColumns(AreaPtg aptg)
	{
		short fc = aptg.getFirstColumn();
		short lc = aptg.getLastColumn();
		aptg.setFirstColumn(fc);
		aptg.setLastColumn(lc);
		assertEquals(fc , aptg.getFirstColumn() );
		assertEquals(lc , aptg.getLastColumn() );
	}
	
	public void testFormulaParser()
	{
		String formula1="SUM($E$5:$E$6)";
		String expectedFormula1="SUM($F$5:$F$6)";
		String newFormula1 = shiftAllColumnsBy1(formula1);
		assertEquals("Absolute references changed", expectedFormula1, newFormula1);
		
		String formula2="SUM(E5:E6)";
		String expectedFormula2="SUM(F5:F6)";
		String newFormula2 = shiftAllColumnsBy1(formula2);
		assertEquals("Relative references changed", expectedFormula2, newFormula2);
	}
	
	private String shiftAllColumnsBy1(String  formula)
	{
		int letUsShiftColumn1By1Column=1;
		
		FormulaParser parser = new FormulaParser(formula,null);
		parser.parse();

		final Ptg[] ptgs = parser.getRPNPtg();
		for(int i=0; i<ptgs.length; i++)
		{
			Ptg ptg = ptgs[i];
			if (ptg instanceof AreaPtg )
			{
				AreaPtg aptg = (AreaPtg)ptg;
				aptg.setFirstColumn((short)(aptg.getFirstColumn()+letUsShiftColumn1By1Column));
				aptg.setLastColumn((short)(aptg.getLastColumn()+letUsShiftColumn1By1Column));
			}
		}
		String newFormula = parser.toFormulaString(ptgs);
		return newFormula;
	}
	
	

}