org.apache.poi.sl.draw.geom.PresetGeometries Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of poi Show documentation
Apache POI - Java API To Access Microsoft Format Files
There is a newer version: 5.2.5
/*
 *  ====================================================================
 *    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.sl.draw.geom;

import java.io.InputStream;
import java.util.LinkedHashMap;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamConstants;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.transform.stream.StreamSource;

import org.apache.poi.sl.draw.binding.CTCustomGeometry2D;
import org.apache.poi.util.POILogFactory;
import org.apache.poi.util.POILogger;
import org.apache.poi.util.XMLHelper;

/**
 *
 */
public class PresetGeometries extends LinkedHashMap {
    private final static POILogger LOG = POILogFactory.getLogger(PresetGeometries.class);
    private final static String BINDING_PACKAGE = "org.apache.poi.sl.draw.binding";

    private static class SingletonHelper {
        private static JAXBContext JAXB_CONTEXT;
        static {
            try {
                JAXB_CONTEXT = JAXBContext.newInstance(BINDING_PACKAGE);
            } catch (JAXBException e) {
                throw new RuntimeException(e);
            }
        }
    }


    protected static PresetGeometries _inst;

    protected PresetGeometries(){}

    @SuppressWarnings("unused")
    public void init(InputStream is) throws XMLStreamException, JAXBException {
        XMLInputFactory staxFactory = XMLHelper.newXMLInputFactory();
        XMLStreamReader streamReader = staxFactory.createXMLStreamReader(new StreamSource(is));
        try {
            // ignore StartElement:
            streamReader.nextTag();

            // JAXB:
            JAXBContext jaxbContext = SingletonHelper.JAXB_CONTEXT;
            Unmarshaller unmarshaller = jaxbContext.createUnmarshaller();

            long cntElem = 0;
            while (streamReader.hasNext() && streamReader.nextTag() == XMLStreamConstants.START_ELEMENT) {
                String name = streamReader.getLocalName();
                JAXBElement el = unmarshaller.unmarshal(streamReader, CTCustomGeometry2D.class);
                CTCustomGeometry2D cus = el.getValue();
                cntElem++;

                if (containsKey(name)) {
                    LOG.log(POILogger.WARN, "Duplicate definition of " + name);
                }
                put(name, new CustomGeometry(cus));
            }
        } finally {
            streamReader.close();
        }
    }

    /**
     * Convert a single CustomGeometry object, i.e. from xmlbeans
     */
    public static CustomGeometry convertCustomGeometry(XMLStreamReader staxReader) {
        try {
            JAXBContext jaxbContext = SingletonHelper.JAXB_CONTEXT;
            Unmarshaller unmarshaller = jaxbContext.createUnmarshaller();
            JAXBElement el = unmarshaller.unmarshal(staxReader, CTCustomGeometry2D.class);
            return new CustomGeometry(el.getValue());
        } catch (JAXBException e) {
            LOG.log(POILogger.ERROR, "Unable to parse single custom geometry", e);
            return null;
        }
    }

    public static synchronized PresetGeometries getInstance(){
        if(_inst == null) {
            // use a local object first to not assign a partly constructed object
            // in case of failure
            PresetGeometries lInst = new PresetGeometries();
            try {
                try (InputStream is = PresetGeometries.class.
                        getResourceAsStream("presetShapeDefinitions.xml")) {
                    lInst.init(is);
                }
            } catch (Exception e){
                throw new RuntimeException(e);
            }
            _inst = lInst;
        }

        return _inst;
    }
}