org.docx4j.openpackaging.parts.DrawingML.DiagramDataUnflatten Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of docx4j-core Show documentation
Show all versions of docx4j-core Show documentation
docx4j is a library which helps you to work with the Office Open
XML file format as used in docx
documents, pptx presentations, and xlsx spreadsheets.
/*
* 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>();
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> textFormats;
public List> 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 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() createListItemsForChildren(CTPt parent) {
List childListItemList
= new ArrayList();
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 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();
}
}