All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.docx4j.openpackaging.parts.DrawingML.DiagramDataPart Maven / Gradle / Ivy

    Go to download

    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.

    There is a newer version: 6.1.2
    Show newest version
    /*
 *  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.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.UnsupportedEncodingException;
import java.util.HashMap;
import java.util.List;
import java.util.UUID;

import javax.xml.bind.JAXBElement;

import org.apache.commons.codec.binary.Base64;
import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.docx4j.TraversalUtil;
import org.docx4j.XmlUtils;
import org.docx4j.TraversalUtil.Callback;
import org.docx4j.dml.CTTextBody;
import org.docx4j.dml.diagram.CTCxn;
import org.docx4j.dml.diagram.CTDataModel;
import org.docx4j.dml.diagram.CTElemPropSet;
import org.docx4j.dml.diagram.CTPt;
import org.docx4j.dml.diagram.STCxnType;
import org.docx4j.dml.diagram.STPtType;
import org.docx4j.dml.wordprocessingDrawing.Inline;
import org.docx4j.openpackaging.exceptions.InvalidFormatException;
import org.docx4j.openpackaging.io.SaveToZipFile;
import org.docx4j.openpackaging.packages.WordprocessingMLPackage;
import org.docx4j.openpackaging.parts.JaxbXmlPart;
import org.docx4j.openpackaging.parts.PartName;
import org.docx4j.openpackaging.parts.WordprocessingML.BinaryPartAbstractImage;
import org.docx4j.openpackaging.parts.relationships.Namespaces;
import org.docx4j.relationships.Relationship;


public final class DiagramDataPart extends JaxbDmlPart {
	
	private static Logger log = LoggerFactory.getLogger(DiagramDataPart.class);			
	
	public DiagramDataPart(PartName partName) throws InvalidFormatException {
		super(partName);
		init();
		
	}

	public DiagramDataPart() throws InvalidFormatException {
		super(new PartName("/word/diagrams/data1.xml")); 
		init();		
	}		
		
		public void init() {
		
		// Used if this Part is added to [Content_Types].xml 
		setContentType(new  org.docx4j.openpackaging.contenttype.ContentType( 
				org.docx4j.openpackaging.contenttype.ContentTypes.DRAWINGML_DIAGRAM_DATA));

		// Used when this Part is added to a rels 
		setRelationshipType(Namespaces.DRAWINGML_DIAGRAM_DATA);
	}
		
		
	/**
	 * This method ensures generated IDs are
	 * valid and (subject to that constraint)
	 * easy-ish to understand.
	 */
	public static void setFriendlyIds(Object jaxbElement) {
				
		map = new HashMap();
		
		// Go through once creating a map of IDs
		generateIdMap(jaxbElement);
		
		// Go through again, replacing
		ReplaceIds(jaxbElement);
		
	}
	
	static HashMap map;
	static int index = 0;
	protected static void generateIdMap(Object jaxbElement) {
		
		new TraversalUtil(jaxbElement,

				new Callback() {
			
					// Unfortunately, the schema says
					// a model id must be an int or a GUID.
					// Word 2007 won't open a docx which violates that.

					@Override
					public List apply(Object o) {
						
						if (o instanceof CTPt) {
							
							CTPt pt = (CTPt)o;
							
							
							String from = pt.getModelId();
							
							// Everything goes invisible in Word 2007 if you map these
							// to an int (rather than a guid),
							// and images disappear (not shown as broken)
							if (pt.getType()!=null
									&& pt.getType().equals(STPtType.PRES)) {
								map.put(from, generateNiceGuid(index));
								index++;
								return null;
							}
							
//							String to = generateNiceGuid(index);
							String to = "" + index;
							index++;								
							map.put(from, to);							
						}

						if (o instanceof CTCxn) {
							
							CTCxn cxn = (CTCxn)o;
							String from = cxn.getModelId();
//							String to = generateNiceGuid(index);
							String to = "" + index;
							index++;								
							
							map.put(from, to);							
						}
						
						return null;
					}

					@Override
					public boolean shouldTraverse(Object o) {
						return true;
					}

					// Depth first
					@Override
					public void walkJAXBElements(Object parent) {
						
						List children = getChildren(parent);
						if (children != null) {

							for (Object o : children) {

								// if its wrapped in javax.xml.bind.JAXBElement, get its
								// value
								o = XmlUtils.unwrap(o);

								this.apply(o);

								if (this.shouldTraverse(o)) {
									walkJAXBElements(o);
								}

							}
						}

					}

					@Override
					public List getChildren(Object o) {
						return TraversalUtil.getChildrenImpl(o);
					}

				}

				);
		
		
	}
	
	private static String generateNiceGuid(int index) {
		
		/* It seems that if there are more than about 100
		 * items, Word won't open a docx with friendly ids
		 * of the following form:
		 * 
			if (index<10) {
				return "{00000000-0000-0000-0000-00000000000" + index + "}";
			} else if (index<100) {
				return "{00000000-0000-0000-0000-0000000000" + index + "}";			
			} else {
				return "{00000000-0000-0000-0000-000000000" + index + "}";			
			}

		 * Speculate that this is because M$ internally
		 * uses some broken equality test.
		 * 
		 * The following works better ...
		 */
		

		if (index<10) {
			return "{0000000" + index + "-0000-0000-0000-000000000000}";
		} else if (index<100) {
			return "{000000" +  index + "-0000-0000-0000-000000000000}";			
		} else {
			return "{00000" +   index + "-0000-0000-0000-000000000000}";			
		}
		
		// This works as well...
//		return "{" + UUID.randomUUID().toString().toUpperCase() + "}";
		 
		
	}

	private static String mapGet(HashMap map, String key) {
		
		if (key==null) {
			System.out.println("Key not found!");
			return null;
		}
		
		String val = map.get(key);
		if (val==null) {
			System.out.println("No val for Key " + key);
			return key;
		}
		
		return val;
	}
	
	protected static void ReplaceIds(Object jaxbElement) {
		
		new TraversalUtil(jaxbElement,

				new Callback() {

					@Override
					public List apply(Object o) {
												
						if (o instanceof CTPt) {
							
							CTPt pt = (CTPt)o;
														
							pt.setModelId( 
									mapGet(map, pt.getModelId() ));
							
							if (pt.getPrSet()!=null) {
								CTElemPropSet pr = (CTElemPropSet)pt.getPrSet();
								if (pr.getPresAssocID()!=null) {
									pr.setPresAssocID(
											mapGet(map, pr.getPresAssocID() ));
								}
							}
							
							if (!pt.getCxnId().equals("0")) {
								pt.setCxnId( 
										mapGet(map, pt.getCxnId() ));
							}							
						}

						if (o instanceof CTCxn) {
							
							CTCxn cxn = (CTCxn)o;
							if (cxn.getModelId()!=null) {
								cxn.setModelId(mapGet(map,cxn.getModelId()));
							}
							
							cxn.setSrcId(mapGet(map, cxn.getSrcId()));
							cxn.setDestId( mapGet(map,cxn.getDestId() ));

							if (!cxn.getSibTransId().equals("0")) {
								cxn.setSibTransId( 
										mapGet(map, cxn.getSibTransId() ));
							}
							if (!cxn.getParTransId().equals("0")) {
								cxn.setParTransId( 
										mapGet(map, cxn.getParTransId() ));
							}
//							if (cxn.getPresId()!=null) {
//								cxn.setPresId( 
//										map.get(cxn.getPresId() ));
//							}							
						}
						
						return null;
					}

					@Override
					public boolean shouldTraverse(Object o) {
						return true;
					}

					// Depth first
					@Override
					public void walkJAXBElements(Object parent) {


						List children = getChildren(parent);
						if (children != null) {

							for (Object o : children) {

								// if its wrapped in javax.xml.bind.JAXBElement, get its
								// value
								o = XmlUtils.unwrap(o);

								this.apply(o);

								if (this.shouldTraverse(o)) {
									walkJAXBElements(o);
								}

							}
						}

					}

					@Override
					public List getChildren(Object o) {
						return TraversalUtil.getChildrenImpl(o);
					}

				}

				);
		
		
	}
	
	public static String addImage(DiagramDataPart ddp, String base64) {
		// No need to pass content type.
				
		log.debug("Adding image: " + base64);
		
		BinaryPartAbstractImage imagePart = null;
		try {
			// Base64 decode it
			byte[] bytes = Base64.decodeBase64( base64.getBytes("UTF8") );
			
			// Create image part and add it
			imagePart = BinaryPartAbstractImage.createImagePart(
					ddp.getPackage(), ddp, bytes);
		} catch (Exception e) {
			e.printStackTrace();
			log.error(e.getMessage(), e);

			// Can't use this image, so insert a placeholder
			log.info(".. attempting to use broken image placeholder");
			try {
				byte[] bytes = IOUtils.toByteArray(org.docx4j.utils.ResourceUtils.getResource(
						"image_broken.gif"));
				
				imagePart = BinaryPartAbstractImage.createImagePart(
						ddp.getPackage(), ddp, bytes);

				log.info(".. used broken image placeholder");
				
			} catch (Exception e1) {
				e1.printStackTrace();
				return "";
			}
		}
		
        return imagePart.getSourceRelationships().get(0).getId();
	}
	
	public static void debug(String message) {
		
		System.out.println(message);
	}
	
	public static void main(String[] args) throws Exception {

		WordprocessingMLPackage wordMLPackage = WordprocessingMLPackage
				.load(new java.io.File(
						System.getProperty("user.dir")
						+ "/SmartArt/OUT-xx.docx"));
		
		Relationship r = wordMLPackage.getMainDocumentPart().getRelationshipsPart().getRelationshipByType(Namespaces.DRAWINGML_DIAGRAM_DATA);
		
		if (r==null) {
			System.out.println("No DDP!");
			return;
		}

		DiagramDataPart thisPart = (DiagramDataPart)wordMLPackage.getMainDocumentPart().getRelationshipsPart().getPart(r);			
		
		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();

		// Check our format templates
		List> textFormats = diagramDataUnflatten.getTextFormats();
		System.out.println("Template list =============== ");
		for (JAXBElement tb : textFormats) {
			System.out.println( XmlUtils.marshaltoString(tb, true, true));				
		}
		System.out.println("============================= ");
		
		// Now fix the IDs in the drawing part to match
		// TODO: just drop this part altogether; we don't need it
		Relationship r2 = wordMLPackage.getMainDocumentPart().getRelationshipsPart().getRelationshipByType(Namespaces.DRAWINGML_DIAGRAM_DRAWING);		
		if (r2==null) {
			System.out.println("No DDrawingP!");
		} else {
			DiagramDrawingPart drawingPart = (DiagramDrawingPart)wordMLPackage.getMainDocumentPart().getRelationshipsPart().getPart(r2);
			drawingPart.setFriendlyIds(thisPart.map);			
			System.out.println( XmlUtils.marshaltoString(drawingPart.getJaxbElement(), true, true));
		}		
		
		SaveToZipFile saver = new SaveToZipFile(wordMLPackage);
		saver.save(System.getProperty("user.dir")
				+ "/SmartArt/OUT-clean.docx");
		
	}	
		
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy