/*
* Copyright 2010, Plutext Pty Ltd.
*
* This file is part of docx4j.
docx4j is licensed 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.docx4j.openpackaging.parts.DrawingML;
import java.util.ArrayList;
import java.util.List;
import java.util.Map.Entry;
import javax.xml.bind.JAXBElement;
import org.docx4j.XmlUtils;
import org.docx4j.dml.CTBlip;
import org.docx4j.dml.CTTextBody;
import org.docx4j.dml.CTTextParagraph;
import org.docx4j.dml.diagram.CTCxn;
import org.docx4j.dml.diagram.CTCxnList;
import org.docx4j.dml.diagram.CTElemPropSet;
import org.docx4j.dml.diagram.CTPt;
import org.docx4j.dml.diagram.CTPtList;
import org.docx4j.dml.diagram.STPtType;
import org.docx4j.openpackaging.contenttype.ContentTypes;
import org.docx4j.openpackaging.packages.OpcPackage;
import org.docx4j.openpackaging.parts.Part;
import org.docx4j.openpackaging.parts.PartName;
import org.docx4j.openpackaging.parts.WordprocessingML.BinaryPartAbstractImage;
import org.opendope.SmartArt.dataHierarchy.SibTransBody;
import org.opendope.SmartArt.dataHierarchy.SmartArtDataHierarchy;
import org.opendope.SmartArt.dataHierarchy.SmartArtDataHierarchy.Images;
import org.opendope.SmartArt.dataHierarchy.SmartArtDataHierarchy.Images.Image;
import org.opendope.SmartArt.dataHierarchy.SmartArtDataHierarchy.Texts;
import org.opendope.SmartArt.dataHierarchy.SmartArtDataHierarchy.Texts.IdentifiedText;
import org.opendope.SmartArt.dataHierarchy.TextBody;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Convert dgm:dataModel to OpenDoPE SmartArtDataHierarchy
* format.
*
* @author jharrop
*/
public class DiagramDataUnflatten {
private static Logger log = LoggerFactory.getLogger(DiagramDataUnflatten.class);
protected static String PRESNAME_FOR_IMAGE = "image";
/* Generating SmartArtDataHierarchy xml from
* a real SmartArt drawing is an easy way for
* users to explore our format.
*
* Generating SmartArtDataHierarchy xml from
* a real SmartArt drawing, and then using it
* to generate the SmartArt drawing again, is
* a good way to test things.
*
*/
public DiagramDataUnflatten(DiagramDataPart diagramDataPart) {
// Source structures
this.diagramDataPart = diagramDataPart;
ptList = diagramDataPart.getJaxbElement().getPtLst();
cxnList = diagramDataPart.getJaxbElement().getCxnLst();
// Target structures
factory = new org.opendope.SmartArt.dataHierarchy.ObjectFactory();
texts = factory.createSmartArtDataHierarchyTexts();
images = factory.createSmartArtDataHierarchyImages();
textFormats = new ArrayList<JAXBElement<CTTextBody>>();
dmlFactory = new org.docx4j.dml.ObjectFactory();
diagramFactory = new org.docx4j.dml.diagram.ObjectFactory();
}
private org.docx4j.dml.ObjectFactory dmlFactory;
private org.docx4j.dml.diagram.ObjectFactory diagramFactory;
// Source structures
private DiagramDataPart diagramDataPart;
private CTPtList ptList;
private CTCxnList cxnList;
// Target structures
private org.opendope.SmartArt.dataHierarchy.ObjectFactory factory;
public org.opendope.SmartArt.dataHierarchy.ObjectFactory getDataHierarchyObjectFactory() {
return factory;
}
private Texts texts;
public Texts getTexts() {
return texts;
}
private Images images;
private List<JAXBElement<CTTextBody>> textFormats;
public List<JAXBElement<CTTextBody>> getTextFormats() {
return textFormats;
}
public SmartArtDataHierarchy convert() {
SmartArtDataHierarchy smartArtDataHierarchy
= factory.createSmartArtDataHierarchy();
CTPt docPt = ptList.getPt().get(0);
if (docPt.getPrSet()!=null
&& docPt.getPrSet().getLoTypeId()!=null) {
smartArtDataHierarchy.setLoTypeId(
docPt.getPrSet().getLoTypeId() );
} else {
log.error("Couldn't read @loTypeId");
}
org.opendope.SmartArt.dataHierarchy.List docList = factory.createList();
// dgm:pt[@type="doc" and @modelId="0"]
org.opendope.SmartArt.dataHierarchy.ListItem listItem = factory.createListItem();
docList.getListItem().add(listItem);
listItem.setId(docPt.getModelId());
listItem.setDepth(0);
processChildrenOf(docPt, listItem);
smartArtDataHierarchy.setList(docList);
smartArtDataHierarchy.setImages(images);
smartArtDataHierarchy.setTexts(texts);
return smartArtDataHierarchy;
}
public CTTextBody processText(org.opendope.SmartArt.dataHierarchy.ListItem thisListItem,
CTPt thisPoint) {
CTTextBody textBody = thisPoint.getT();
if (textBody!=null) {
TextBody tb = getDataHierarchyObjectFactory().createTextBody();
thisListItem.setTextBody(tb);
for(CTTextParagraph p : textBody.getP() ) {
if (!p.getEGTextRun().isEmpty()
&& p.getEGTextRun().get(0) instanceof org.docx4j.dml.CTRegularTextRun
) {
// TODO; assumes a single r child, which is
// all we handle. ie this model doesn't support
// multiple runs, some of which formatted
org.docx4j.dml.CTRegularTextRun run = (org.docx4j.dml.CTRegularTextRun)p.getEGTextRun().get(0);
tb.getP().add( run.getT() );
}
}
}
return textBody;
}
private void processChildrenOf(CTPt pt, org.opendope.SmartArt.dataHierarchy.ListItem listItem) {
List<org.opendope.SmartArt.dataHierarchy.ListItem> childModelIds = createListItemsForChildren(pt);
if (childModelIds.isEmpty()) return;
org.opendope.SmartArt.dataHierarchy.List list = factory.createList();
listItem.setList(list);
for (org.opendope.SmartArt.dataHierarchy.ListItem thisListItem : childModelIds) {
list.getListItem().add(thisListItem);
thisListItem.setDepth(listItem.getDepth()+1);
String modelId = thisListItem.getId();
// Find the pt
CTPt thisPoint = getPoint(modelId);
// attach its text
CTTextBody textBody = processText(thisListItem, thisPoint);
if (textFormats.size()<thisListItem.getDepth() ) {
// we don't have a template for this level yet,
// so add this one as the template for this level
if (textBody==null) {
// Add an empty one
textFormats.add(
diagramFactory.createT(
dmlFactory.createCTTextBody()));
} else {
textFormats.add(
diagramFactory.createT(textBody));
}
}
// attach its image
CTPt imgPt = getAssociatedPres(modelId, PRESNAME_FOR_IMAGE);
if (imgPt!=null
&& imgPt.getSpPr()!=null
&& imgPt.getSpPr().getBlipFill()!=null
&& imgPt.getSpPr().getBlipFill().getBlip()!=null) {
// Get the relId
CTBlip blip = imgPt.getSpPr().getBlipFill().getBlip();
if (blip.getEmbed()!=null) {
String relId = blip.getEmbed();
//Relationship r = diagramDataPart.getRelationshipsPart().getRelationshipByID(relId);
BinaryPartAbstractImage bpai = (BinaryPartAbstractImage)diagramDataPart.getRelationshipsPart().getPart(relId);
// Add it
Image image = factory.createSmartArtDataHierarchyImagesImage();
image.setContentType(bpai.getContentType());
image.setId(imgPt.getModelId());
image.setValue(bpai.getBytes());
images.getImage().add(image);
// reference
org.opendope.SmartArt.dataHierarchy.ImageRef imageRef = factory.createImageRef();
imageRef.setContentRef(imgPt.getModelId());
// Other attributes
if (imgPt.getPrSet()!=null ) {
CTElemPropSet props = imgPt.getPrSet();
if (props.getCustLinFactNeighborX()!=null) {
imageRef.setCustLinFactNeighborX(props.getCustLinFactNeighborX());
}
if (props.getCustLinFactNeighborY()!=null) {
imageRef.setCustLinFactNeighborY(props.getCustLinFactNeighborY());
}
if (props.getCustScaleX()!=null) {
imageRef.setCustScaleX(props.getCustScaleX());
}
if (props.getCustScaleY()!=null) {
imageRef.setCustScaleY(props.getCustScaleY());
}
}
thisListItem.setImageRef(imageRef);
} else if (blip.getLink()!=null) {
// TODO
// -check a linked image actually works in SmartArt.
// -our export format could be extended to support linked images
// -and/or we could fetch the image and embed it
}
}
// attach its sibTrans text (if applicable)
CTPt sibTrans = getPoint(thisListItem.getSibTransBody().getContentRef() );
// Don't clutter up our export, if it doesn't contain content
if (sibTrans.getT()!=null
&& sibTrans.getT().getP() !=null
&& sibTrans.getT().getP().get(0) !=null
&& sibTrans.getT().getP().get(0).getEGTextRun() !=null
&& !sibTrans.getT().getP().get(0).getEGTextRun().isEmpty() ) {
IdentifiedText wrapper = factory.createSmartArtDataHierarchyTextsIdentifiedText();
wrapper.setId(sibTrans.getModelId()); // = @sibTransContentRef
wrapper.setT( sibTrans.getT() );
texts.getIdentifiedText().add(wrapper);
} else {
// remove the reference
thisListItem.setSibTransBody(null);
}
// recurse
processChildrenOf(thisPoint, thisListItem);
}
}
public CTPt getPoint(String modelId) {
for (CTPt pt : ptList.getPt()) {
if (pt.getModelId().equals(modelId)) {
return pt;
}
}
return null;
}
public CTPt getAssociatedPres(String modelId, String presNamePrefix) {
for (CTPt pt : ptList.getPt()) {
if (pt.getType().equals(STPtType.PRES)
&& pt.getPrSet()!=null
&& pt.getPrSet().getPresName()!=null
&& pt.getPrSet().getPresName().startsWith(presNamePrefix)
&& pt.getPrSet().getPresAssocID().equals(modelId)) {
return pt;
}
}
return null;
}
public List<org.opendope.SmartArt.dataHierarchy.ListItem> createListItemsForChildren(CTPt parent) {
List<org.opendope.SmartArt.dataHierarchy.ListItem> childListItemList
= new ArrayList<org.opendope.SmartArt.dataHierarchy.ListItem>();
String parentId = parent.getModelId();
for (CTCxn cxn : cxnList.getCxn() ) {
if (cxn.getSrcId().equals(parentId) && !cxn.getSibTransId().equals("0") ) {
// Create a corresponding ListItem
org.opendope.SmartArt.dataHierarchy.ListItem thisListItem = factory.createListItem();
thisListItem.setId(cxn.getDestId());
// we'll manipulate this more later
SibTransBody sibTransBody = factory.createSibTransBody();
sibTransBody.setContentRef(cxn.getSibTransId());
thisListItem.setSibTransBody(sibTransBody);
childListItemList.add(thisListItem);
//childModelIds.add(cxn.getDestId() );
}
}
return childListItemList;
}
public static void main(String[] args) throws Exception {
OpcPackage pkg = OpcPackage
.load(new java.io.File(
System.getProperty("user.dir")
+ "/sample-docs/glox/extracted/apes.pptx"));
DiagramDataPart thisPart = null;
for (Entry<PartName,Part> entry : pkg.getParts().getParts().entrySet() ) {
if (entry.getValue().getContentType().equals( ContentTypes.DRAWINGML_DIAGRAM_DATA )) {
thisPart = (DiagramDataPart)entry.getValue();
break;
}
}
if (thisPart==null) {
System.out.println("No SmartArt found in this docx.");
return;
}
thisPart.setFriendlyIds(thisPart.getJaxbElement());
// System.out.println( XmlUtils.marshaltoString(thisPart.getJaxbElement(), true, true));
// What does it look like in our format?
DiagramDataUnflatten diagramDataUnflatten = new DiagramDataUnflatten(thisPart);
String exchange= XmlUtils.marshaltoString(diagramDataUnflatten.convert(), true, true);
System.out.println( exchange );
// PrintWriter out = new PrintWriter(System.getProperty("user.dir")
// + "/SmartArt/12hi.xml");
// out.println(exchange);
// out.flush();
// out.close();
}
}