git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1406496 13f79535-47bb-0310-9956-ffa450edef68tags/3.10-beta1
@@ -0,0 +1,152 @@ | |||
/* | |||
* ==================================================================== | |||
* 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.xslf.usermodel; | |||
import org.apache.poi.POIXMLDocumentPart; | |||
import org.apache.poi.ss.util.CellRangeAddress; | |||
import org.apache.poi.ss.util.CellReference; | |||
import org.apache.poi.xssf.usermodel.XSSFRow; | |||
import org.apache.poi.xssf.usermodel.XSSFSheet; | |||
import org.apache.poi.xssf.usermodel.XSSFWorkbook; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTAxDataSource; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTChart; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTNumData; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTNumDataSource; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTNumVal; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTPieChart; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTPieSer; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTPlotArea; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTSerTx; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTStrData; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.CTStrVal; | |||
import java.io.BufferedReader; | |||
import java.io.FileInputStream; | |||
import java.io.FileOutputStream; | |||
import java.io.FileReader; | |||
import java.io.OutputStream; | |||
/** | |||
* Build a pie chart from a template pptx | |||
* | |||
* @author Yegor Kozlov | |||
*/ | |||
public class PieChartDemo { | |||
private static void usage(){ | |||
System.out.println("Usage: PieChartDemo <pie-chart-template.pptx> <pie-chart-data.txt>"); | |||
System.out.println(" pie-chart-template.pptx template with a pie chart"); | |||
System.out.println(" pie-chart-data.txt the model to set. First line is chart title, " + | |||
"then go pairs {axis-label value}"); | |||
} | |||
public static void main(String[] args) throws Exception { | |||
if(args.length < 2) { | |||
usage(); | |||
return; | |||
} | |||
BufferedReader modelReader = new BufferedReader(new FileReader(args[1])); | |||
String chartTitle = modelReader.readLine(); // first line is chart title | |||
XMLSlideShow pptx = new XMLSlideShow(new FileInputStream(args[0])); | |||
XSLFSlide slide = pptx.getSlides()[0]; | |||
// find chart in the slide | |||
XSLFChart chart = null; | |||
for(POIXMLDocumentPart part : slide.getRelations()){ | |||
if(part instanceof XSLFChart){ | |||
chart = (XSLFChart) part; | |||
break; | |||
} | |||
} | |||
if(chart == null) throw new IllegalStateException("chart not found in the template"); | |||
// embedded Excel workbook that holds the chart data | |||
POIXMLDocumentPart xlsPart = chart.getRelations().get(0); | |||
XSSFWorkbook wb = new XSSFWorkbook(); | |||
XSSFSheet sheet = wb.createSheet(); | |||
CTChart ctChart = chart.getCTChart(); | |||
CTPlotArea plotArea = ctChart.getPlotArea(); | |||
CTPieChart pieChart = plotArea.getPieChartArray(0); | |||
//Pie Chart Series | |||
CTPieSer ser = pieChart.getSerArray(0); | |||
// Series Text | |||
CTSerTx tx = ser.getTx(); | |||
tx.getStrRef().getStrCache().getPtArray(0).setV(chartTitle); | |||
sheet.createRow(0).createCell(1).setCellValue(chartTitle); | |||
String titleRef = new CellReference(sheet.getSheetName(), 0, 1, true, true).formatAsString(); | |||
tx.getStrRef().setF(titleRef); | |||
// Category Axis Data | |||
CTAxDataSource cat = ser.getCat(); | |||
CTStrData strData = cat.getStrRef().getStrCache(); | |||
// Values | |||
CTNumDataSource val = ser.getVal(); | |||
CTNumData numData = val.getNumRef().getNumCache(); | |||
strData.setPtArray(null); // unset old axis text | |||
numData.setPtArray(null); // unset old values | |||
// set model | |||
int idx = 0; | |||
int rownum = 1; | |||
String ln; | |||
while((ln = modelReader.readLine()) != null){ | |||
String[] vals = ln.split("\\s+"); | |||
CTNumVal numVal = numData.addNewPt(); | |||
numVal.setIdx(idx); | |||
numVal.setV(vals[1]); | |||
CTStrVal sVal = strData.addNewPt(); | |||
sVal.setIdx(idx); | |||
sVal.setV(vals[0]); | |||
idx++; | |||
XSSFRow row = sheet.createRow(rownum++); | |||
row.createCell(0).setCellValue(vals[0]); | |||
row.createCell(1).setCellValue(Double.valueOf(vals[1])); | |||
} | |||
numData.getPtCount().setVal(idx); | |||
strData.getPtCount().setVal(idx); | |||
String numDataRange = new CellRangeAddress(1, rownum-1, 1, 1).formatAsString(sheet.getSheetName(), true); | |||
val.getNumRef().setF(numDataRange); | |||
String axisDataRange = new CellRangeAddress(1, rownum-1, 0, 0).formatAsString(sheet.getSheetName(), true); | |||
cat.getStrRef().setF(axisDataRange); | |||
// updated the embedded workbook with the data | |||
OutputStream xlsOut = xlsPart.getPackagePart().getOutputStream(); | |||
wb.write(xlsOut); | |||
xlsOut.close(); | |||
// save the result | |||
FileOutputStream out = new FileOutputStream("pie-chart-demo-output.pptx"); | |||
pptx.write(out); | |||
out.close(); | |||
} | |||
} |
@@ -0,0 +1,4 @@ | |||
My Chart | |||
First 1.0 | |||
Second 3.0 | |||
Third 4.0 |
@@ -0,0 +1,134 @@ | |||
/* | |||
* ==================================================================== | |||
* 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.xslf.usermodel; | |||
import junit.framework.TestCase; | |||
import org.apache.poi.POIXMLDocumentPart; | |||
import org.apache.poi.ss.util.CellRangeAddress; | |||
import org.apache.poi.ss.util.CellReference; | |||
import org.apache.poi.xslf.XSLFTestDataSamples; | |||
import org.apache.poi.xssf.usermodel.XSSFRow; | |||
import org.apache.poi.xssf.usermodel.XSSFSheet; | |||
import org.apache.poi.xssf.usermodel.XSSFWorkbook; | |||
import org.openxmlformats.schemas.drawingml.x2006.chart.*; | |||
import org.openxmlformats.schemas.drawingml.x2006.main.*; | |||
import org.openxmlformats.schemas.presentationml.x2006.main.CTConnector; | |||
import java.awt.*; | |||
import java.io.*; | |||
import java.util.LinkedHashMap; | |||
import java.util.Map; | |||
/** | |||
* @author Yegor Kozlov | |||
*/ | |||
public class TestXSLFChart extends TestCase { | |||
/** | |||
* a modified version from POI-examples | |||
*/ | |||
public void testFillChartTemplate() throws Exception { | |||
String chartTitle = "Apache POI"; // first line is chart title | |||
XMLSlideShow pptx = XSLFTestDataSamples.openSampleDocument("pie-chart.pptx"); | |||
XSLFSlide slide = pptx.getSlides()[0]; | |||
// find chart in the slide | |||
XSLFChart chart = null; | |||
for(POIXMLDocumentPart part : slide.getRelations()){ | |||
if(part instanceof XSLFChart){ | |||
chart = (XSLFChart) part; | |||
break; | |||
} | |||
} | |||
if(chart == null) throw new IllegalStateException("chart not found in the template"); | |||
// embedded Excel workbook that holds the chart data | |||
POIXMLDocumentPart xlsPart = chart.getRelations().get(0); | |||
XSSFWorkbook wb = new XSSFWorkbook(); | |||
XSSFSheet sheet = wb.createSheet(); | |||
CTChart ctChart = chart.getCTChart(); | |||
CTPlotArea plotArea = ctChart.getPlotArea(); | |||
CTPieChart pieChart = plotArea.getPieChartArray(0); | |||
//Pie Chart Series | |||
CTPieSer ser = pieChart.getSerArray(0); | |||
// Series Text | |||
CTSerTx tx = ser.getTx(); | |||
tx.getStrRef().getStrCache().getPtArray(0).setV(chartTitle); | |||
sheet.createRow(0).createCell(1).setCellValue(chartTitle); | |||
String titleRef = new CellReference(sheet.getSheetName(), 0, 1, true, true).formatAsString(); | |||
tx.getStrRef().setF(titleRef); | |||
// Category Axis Data | |||
CTAxDataSource cat = ser.getCat(); | |||
CTStrData strData = cat.getStrRef().getStrCache(); | |||
// Values | |||
CTNumDataSource valSrc = ser.getVal(); | |||
CTNumData numData = valSrc.getNumRef().getNumCache(); | |||
strData.setPtArray(null); // unset old axis text | |||
numData.setPtArray(null); // unset old values | |||
Map<String, Double> pieModel = new LinkedHashMap<String, Double>(); | |||
pieModel.put("First", 1.0); | |||
pieModel.put("Second", 3.0); | |||
pieModel.put("Third", 4.0); | |||
// set model | |||
int idx = 0; | |||
int rownum = 1; | |||
for(String key : pieModel.keySet()){ | |||
double val = pieModel.get(key); | |||
CTNumVal numVal = numData.addNewPt(); | |||
numVal.setIdx(idx); | |||
numVal.setV("" + val); | |||
CTStrVal sVal = strData.addNewPt(); | |||
sVal.setIdx(idx); | |||
sVal.setV(key); | |||
idx++; | |||
XSSFRow row = sheet.createRow(rownum++); | |||
row.createCell(0).setCellValue(key); | |||
row.createCell(1).setCellValue(val); | |||
} | |||
numData.getPtCount().setVal(idx); | |||
strData.getPtCount().setVal(idx); | |||
String numDataRange = new CellRangeAddress(1, rownum-1, 1, 1).formatAsString(sheet.getSheetName(), true); | |||
valSrc.getNumRef().setF(numDataRange); | |||
String axisDataRange = new CellRangeAddress(1, rownum-1, 0, 0).formatAsString(sheet.getSheetName(), true); | |||
cat.getStrRef().setF(axisDataRange); | |||
// updated the embedded workbook with the data | |||
OutputStream xlsOut = xlsPart.getPackagePart().getOutputStream(); | |||
wb.write(xlsOut); | |||
xlsOut.close(); | |||
} | |||
} |
@@ -17,9 +17,10 @@ | |||
package org.apache.poi.xslf.usermodel; | |||
import junit.framework.TestCase; | |||
import org.openxmlformats.schemas.drawingml.x2006.main.STLineEndLength; | |||
import org.openxmlformats.schemas.drawingml.x2006.main.STLineEndType; | |||
import org.openxmlformats.schemas.drawingml.x2006.main.STLineEndWidth; | |||
import org.openxmlformats.schemas.drawingml.x2006.main.*; | |||
import org.openxmlformats.schemas.presentationml.x2006.main.CTConnector; | |||
import java.awt.*; | |||
/** | |||
* @author Yegor Kozlov | |||
@@ -106,4 +107,37 @@ public class TestXSLFConnectorShape extends TestCase { | |||
} | |||
public void testAddConnector(){ | |||
XMLSlideShow pptx = new XMLSlideShow(); | |||
XSLFSlide slide = pptx.createSlide(); | |||
XSLFAutoShape rect1 = slide.createAutoShape(); | |||
rect1.setShapeType(XSLFShapeType.RECT); | |||
rect1.setAnchor(new Rectangle(100, 100, 100, 100)); | |||
rect1.setFillColor(Color.blue); | |||
XSLFAutoShape rect2 = slide.createAutoShape(); | |||
rect2.setShapeType(XSLFShapeType.RECT); | |||
rect2.setAnchor(new Rectangle(300, 300, 100, 100)); | |||
rect2.setFillColor(Color.red); | |||
XSLFConnectorShape connector1 = slide.createConnector(); | |||
connector1.setAnchor(new Rectangle(200, 150, 100, 200)); | |||
CTConnector ctConnector = (CTConnector)connector1.getXmlObject(); | |||
ctConnector.getSpPr().getPrstGeom().setPrst(STShapeType.BENT_CONNECTOR_3); | |||
CTNonVisualConnectorProperties cx = ctConnector.getNvCxnSpPr().getCNvCxnSpPr(); | |||
// connection start | |||
CTConnection stCxn = cx.addNewStCxn(); | |||
stCxn.setId(rect1.getShapeId()); | |||
// side of the rectangle to attach the connector: left=1, bottom=2,right=3, top=4 | |||
stCxn.setIdx(2); | |||
CTConnection end = cx.addNewEndCxn(); | |||
end.setId(rect2.getShapeId()); | |||
// side of the rectangle to attach the connector: left=1, bottom=2,right=3, top=4 | |||
end.setIdx(3); | |||
} | |||
} |