com.itextpdf.kernel.pdf.PdfResources Maven / Gradle / Ivy
The newest version!
/*
This file is part of the iText (R) project.
Copyright (c) 1998-2024 Apryse Group NV
Authors: Apryse Software.
This program is offered under a commercial and under the AGPL license.
For commercial licensing, contact us at https://itextpdf.com/sales. For AGPL licensing, see below.
AGPL licensing:
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see {
private static final String F = "F";
private static final String Im = "Im";
private static final String Fm = "Fm";
private static final String Gs = "Gs";
private static final String Pr = "Pr";
private static final String Cs = "Cs";
private static final String P = "P";
private static final String Sh = "Sh";
private Map resourceToName = new HashMap<>();
private ResourceNameGenerator fontNamesGen = new ResourceNameGenerator(PdfName.Font, F);
private ResourceNameGenerator imageNamesGen = new ResourceNameGenerator(PdfName.XObject, Im);
private ResourceNameGenerator formNamesGen = new ResourceNameGenerator(PdfName.XObject, Fm);
private ResourceNameGenerator egsNamesGen = new ResourceNameGenerator(PdfName.ExtGState, Gs);
private ResourceNameGenerator propNamesGen = new ResourceNameGenerator(PdfName.Properties, Pr);
private ResourceNameGenerator csNamesGen = new ResourceNameGenerator(PdfName.ColorSpace, Cs);
private ResourceNameGenerator patternNamesGen = new ResourceNameGenerator(PdfName.Pattern, P);
private ResourceNameGenerator shadingNamesGen = new ResourceNameGenerator(PdfName.Shading, Sh);
private boolean readOnly = false;
private boolean isModified = false;
/**
* Creates new instance from given dictionary.
*
* @param pdfObject the {@link PdfDictionary} object from which the resource object will be created.
*/
public PdfResources(PdfDictionary pdfObject) {
super(pdfObject);
buildResources(pdfObject);
}
/**
* Creates new instance from empty dictionary.
*/
public PdfResources() {
this(new PdfDictionary());
}
/**
* Adds font to resources and registers PdfFont in the document for further flushing.
*
* @param pdfDocument a {@link PdfDocument} instance to which the font is added for further flushing
* @param font a {@link PdfFont} instance to be added
* @return added font resource name.
*/
public PdfName addFont(PdfDocument pdfDocument, PdfFont font) {
pdfDocument.addFont(font);
return addResource(font, fontNamesGen);
}
/**
* Adds {@link PdfImageXObject} object to the resources.
*
* @param image the {@link PdfImageXObject} to add.
* @return added image resource name.
*/
public PdfName addImage(PdfImageXObject image) {
return addResource(image, imageNamesGen);
}
/**
* Adds {@link PdfStream} to the resources as image.
*
* @param image the {@link PdfStream} to add.
* @return added image resources name.
*/
public PdfName addImage(PdfStream image) {
return addResource(image, imageNamesGen);
}
public PdfImageXObject getImage(PdfName name) {
PdfStream image = getResource(PdfName.XObject).getAsStream(name);
return image != null && PdfName.Image.equals(image.getAsName(PdfName.Subtype)) ? new PdfImageXObject(image) : null;
}
/**
* Adds {@link PdfFormXObject} object to the resources.
*
* @param form the {@link PdfFormXObject} to add.
* @return added form resource name.
*/
public PdfName addForm(PdfFormXObject form) {
return addResource(form, formNamesGen);
}
/**
* Adds {@link PdfStream} to the resources as form.
*
* @param form the {@link PdfStream} to add.
* @return added form resources name.
*/
public PdfName addForm(PdfStream form) {
return addResource(form, formNamesGen);
}
/**
* Adds the given Form XObject to the current instance of {@link PdfResources}.
*
* @param form Form XObject.
* @param name Preferred name for the given Form XObject.
* @return the {@link PdfName} of the newly added resource
*/
public PdfName addForm(PdfFormXObject form, PdfName name) {
if (getResourceNames(PdfName.XObject).contains(name)) {
name = addResource(form, formNamesGen);
} else {
addResource(form.getPdfObject(), PdfName.XObject, name);
}
return name;
}
public PdfFormXObject getForm(PdfName name) {
PdfStream form = getResource(PdfName.XObject).getAsStream(name);
return form != null && PdfName.Form.equals(form.getAsName(PdfName.Subtype)) ? new PdfFormXObject(form) : null;
}
/**
* Adds {@link PdfExtGState} object to the resources.
*
* @param extGState the {@link PdfExtGState} to add.
* @return added graphics state parameter dictionary resource name.
*/
public PdfName addExtGState(PdfExtGState extGState) {
return addResource(extGState, egsNamesGen);
}
/**
* Adds {@link PdfDictionary} to the resources as graphics state parameter dictionary.
*
* @param extGState the {@link PdfDictionary} to add.
* @return added graphics state parameter dictionary resources name.
*/
public PdfName addExtGState(PdfDictionary extGState) {
return addResource(extGState, egsNamesGen);
}
public PdfExtGState getPdfExtGState(PdfName name) {
PdfDictionary dic = getResource(PdfName.ExtGState).getAsDictionary(name);
return dic != null ? new PdfExtGState(dic) : null;
}
/**
* Adds {@link PdfDictionary} to the resources as properties list.
*
* @param properties the {@link PdfDictionary} to add.
* @return added properties list resources name.
*/
public PdfName addProperties(PdfDictionary properties) {
return addResource(properties, propNamesGen);
}
public PdfObject getProperties(PdfName name) {
return getResourceObject(PdfName.Properties, name);
}
/**
* Adds {@link PdfColorSpace} object to the resources.
*
* @param cs the {@link PdfColorSpace} to add.
* @return added color space resource name.
*/
public PdfName addColorSpace(PdfColorSpace cs) {
return addResource(cs, csNamesGen);
}
/**
* Adds {@link PdfObject} to the resources as color space.
*
* @param colorSpace the {@link PdfObject} to add.
* @return added color space resources name.
*/
public PdfName addColorSpace(PdfObject colorSpace) {
return addResource(colorSpace, csNamesGen);
}
public PdfColorSpace getColorSpace(PdfName name) {
PdfObject colorSpace = getResourceObject(PdfName.ColorSpace, name);
return colorSpace != null ? PdfColorSpace.makeColorSpace(colorSpace) : null;
}
/**
* Adds {@link PdfPattern} object to the resources.
*
* @param pattern the {@link PdfPattern} to add.
* @return added pattern resource name.
*/
public PdfName addPattern(PdfPattern pattern) {
return addResource(pattern, patternNamesGen);
}
/**
* Adds {@link PdfDictionary} to the resources as pattern.
*
* @param pattern the {@link PdfDictionary} to add.
* @return added pattern resources name.
*/
public PdfName addPattern(PdfDictionary pattern) {
return addResource(pattern, patternNamesGen);
}
public PdfPattern getPattern(PdfName name) {
PdfObject pattern = getResourceObject(PdfName.Pattern, name);
return pattern instanceof PdfDictionary ? PdfPattern.getPatternInstance((PdfDictionary) pattern) : null;
}
/**
* Adds {@link AbstractPdfShading} object to the resources.
*
* @param shading the {@link AbstractPdfShading} to add.
* @return added shading resource name.
*/
public PdfName addShading(AbstractPdfShading shading) {
return addResource(shading, shadingNamesGen);
}
/**
* Adds {@link PdfDictionary} to the resources as shading dictionary.
*
* @param shading the {@link PdfDictionary} to add.
* @return added shading dictionary resources name.
*/
public PdfName addShading(PdfDictionary shading) {
return addResource(shading, shadingNamesGen);
}
public AbstractPdfShading getShading(PdfName name) {
PdfObject shading = getResourceObject(PdfName.Shading, name);
return shading instanceof PdfDictionary ? AbstractPdfShading.makeShading((PdfDictionary) shading) : null;
}
protected boolean isReadOnly() {
return readOnly;
}
protected void setReadOnly(boolean readOnly) {
this.readOnly = readOnly;
}
protected boolean isModified() {
return isModified;
}
/**
* {@inheritDoc}
*/
@Override
public PdfObjectWrapper setModified() {
this.isModified = true;
return super.setModified();
}
/**
* Sets the value of default Gray Color Space (see ISO-320001 Paragraph 8.6.5.6).
*
* @param defaultCs the color space to set.
*/
public void setDefaultGray(PdfColorSpace defaultCs) {
addResource(defaultCs.getPdfObject(), PdfName.ColorSpace, PdfName.DefaultGray);
}
/**
* Sets the value of default RGB Color Space (see ISO-320001 Paragraph 8.6.5.6).
*
* @param defaultCs the color space to set.
*/
public void setDefaultRgb(PdfColorSpace defaultCs) {
addResource(defaultCs.getPdfObject(), PdfName.ColorSpace, PdfName.DefaultRGB);
}
/**
* Sets the value of default CMYK Color Space (see ISO-320001 Paragraph 8.6.5.6).
*
* @param defaultCs the color space to set.
*/
public void setDefaultCmyk(PdfColorSpace defaultCs) {
addResource(defaultCs.getPdfObject(), PdfName.ColorSpace, PdfName.DefaultCMYK);
}
/**
* Gets the mapped resource name of the {@link PdfObject} under the given wrapper.
*
*
* Note: if the name for the object won't be found, then the name of object's Indirect Reference will be searched.
*
* @param resource the wrapper of the {@link PdfObject}, for which the name will be searched.
* @param the type of the underlined {@link PdfObject} in wrapper.
* @return the mapped resource name or {@code null} if object isn't added to resources.
*/
public PdfName getResourceName(PdfObjectWrapper resource) {
return getResourceName(resource.getPdfObject());
}
/**
* Gets the mapped resource name of the given {@link PdfObject}.
*
*
* Note: if the name for the object won't be found, then the name of object's Indirect Reference will be searched.
*
* @param resource the object, for which the name will be searched.
* @return the mapped resource name or {@code null} if object isn't added to resources.
*/
public PdfName getResourceName(PdfObject resource) {
PdfName resName = resourceToName.get(resource);
if (resName == null)
resName = resourceToName.get(resource.getIndirectReference());
return resName;
}
/**
* Gets the names of all the added resources.
*
* @return the name of all the added resources.
*/
public Set getResourceNames() {
Set names = new TreeSet<>();
for (PdfName resType : getPdfObject().keySet()) {
names.addAll(getResourceNames(resType));
}
return names;
}
/**
* Gets the array of predefined procedure set names (see ISO-320001 Paragraph 14.2).
* Deprecated in PDF 2.0.
*
* @return the array of predefined procedure set names.
*/
public PdfArray getProcSet() {
return getPdfObject().getAsArray(PdfName.ProcSet);
}
/**
* Sets the array of predefined procedure set names (see ISO-320001 Paragraph 14.2).
* Deprecated in PDF 2.0.
*
* @param array the array of predefined procedure set names to be set.
*/
public void setProcSet(PdfArray array) {
getPdfObject().put(PdfName.ProcSet, array);
}
/**
* Gets the names of all resources of specified type.
*
* @param resType the resource type. Should be {@link PdfName#ColorSpace}, {@link PdfName#ExtGState},
* {@link PdfName#Pattern}, {@link PdfName#Shading}, {@link PdfName#XObject}, {@link PdfName#Font}.
* @return set of resources name of corresponding type. May be empty.
* Will be empty in case of incorrect resource type.
*/
public Set getResourceNames(PdfName resType) {
PdfDictionary resourceCategory = getPdfObject().getAsDictionary(resType);
return resourceCategory == null ? Collections.emptySet() : resourceCategory.keySet();
}
/**
* Get the {@link PdfDictionary} object that that contain resources of specified type.
*
* @param resType the resource type. Should be {@link PdfName#ColorSpace}, {@link PdfName#ExtGState},
* {@link PdfName#Pattern}, {@link PdfName#Shading}, {@link PdfName#XObject}, {@link PdfName#Font}.
* @return the {@link PdfDictionary} object containing all resources of specified type,
* or {@code null} in case of incorrect resource type.
*/
public PdfDictionary getResource(PdfName resType) {
return getPdfObject().getAsDictionary(resType);
}
/**
* Get the {@link PdfObject} object with specified type and name.
*
* @param resType the resource type. Should be {@link PdfName#ColorSpace}, {@link PdfName#ExtGState},
* {@link PdfName#Pattern}, {@link PdfName#Shading}, {@link PdfName#XObject}, {@link PdfName#Font}.
* @param resName the name of the resource object.
* @return the {@link PdfObject} with specified name in the resources of specified type or {@code null}
* in case of incorrect type or missing resource with such name.
*/
public PdfObject getResourceObject(PdfName resType, PdfName resName) {
PdfDictionary resource = getResource(resType);
if (resource != null) {
return resource.get(resName);
}
return null;
}
@Override
protected boolean isWrappedObjectMustBeIndirect() {
return false;
}
PdfName addResource(PdfObjectWrapper resource, ResourceNameGenerator nameGen) {
return addResource(resource.getPdfObject(), nameGen);
}
protected void addResource(PdfObject resource, PdfName resType, PdfName resName) {
if (resType.equals(PdfName.XObject)) {
checkAndResolveCircularReferences(resource);
}
if (readOnly) {
setPdfObject(getPdfObject().clone(Collections.emptyList()));
buildResources(getPdfObject());
isModified = true;
readOnly = false;
}
if (getPdfObject().containsKey(resType) && getPdfObject().getAsDictionary(resType).containsKey(resName)) {
return;
}
resourceToName.put(resource, resName);
PdfDictionary resourceCategory = getPdfObject().getAsDictionary(resType);
if (resourceCategory == null) {
getPdfObject().put(resType, resourceCategory = new PdfDictionary());
} else {
resourceCategory.setModified();
}
resourceCategory.put(resName, resource);
setModified();
}
PdfName addResource(PdfObject resource, ResourceNameGenerator nameGen) {
PdfName resName = getResourceName(resource);
if (resName == null) {
resName = nameGen.generate(this);
addResource(resource, nameGen.getResourceType(), resName);
}
return resName;
}
protected void buildResources(PdfDictionary dictionary) {
for (PdfName resourceType : dictionary.keySet()) {
if (getPdfObject().get(resourceType) == null) {
getPdfObject().put(resourceType, new PdfDictionary());
}
PdfDictionary resources = dictionary.getAsDictionary(resourceType);
if (resources == null) {
continue;
}
for (PdfName resourceName : resources.keySet()) {
PdfObject resource = resources.get(resourceName, false);
resourceToName.put(resource, resourceName);
}
}
}
private void checkAndResolveCircularReferences(PdfObject pdfObject) {
// Consider the situation when an XObject references the resources of the first page.
// We add this XObject to the first page, there is no need to resolve any circular references
// and then we flush this object and try to add it to the second page.
// Now there are circular references and we cannot resolve them because the object is flushed
// and we cannot get resources.
// On the other hand, this situation may occur any time when object is already flushed and we
// try to add it to resources and it seems difficult to overcome this without keeping /Resources key value.
if (pdfObject instanceof PdfDictionary && !pdfObject.isFlushed()) {
PdfDictionary pdfXObject = (PdfDictionary) pdfObject;
PdfObject pdfXObjectResources = pdfXObject.get(PdfName.Resources);
if (pdfXObjectResources != null && pdfXObjectResources.getIndirectReference() != null) {
if (pdfXObjectResources.getIndirectReference().equals(getPdfObject().getIndirectReference())) {
PdfObject cloneResources = getPdfObject().clone();
cloneResources.makeIndirect(getPdfObject().getIndirectReference().getDocument());
pdfXObject.put(PdfName.Resources, cloneResources.getIndirectReference());
}
}
}
}
/**
* Represents a resource name generator. The generator takes into account
* the names of already existing resources thus providing us a unique name.
* The name consists of the following parts: prefix (literal) and number.
*/
static class ResourceNameGenerator {
private PdfName resourceType;
private int counter;
private String prefix;
/**
* Constructs an instance of {@link ResourceNameGenerator} class.
*
* @param resourceType Type of resource. Should be {@link PdfName#ColorSpace}, {@link PdfName#ExtGState},
* {@link PdfName#Pattern}, {@link PdfName#Shading}, {@link PdfName#XObject}, {@link PdfName#Font}.
* @param prefix Prefix used for generating names.
* @param seed Seed for the value which is appended to the number each time
* new name is generated.
*/
public ResourceNameGenerator(PdfName resourceType, String prefix, int seed) {
this.prefix = prefix;
this.resourceType = resourceType;
this.counter = seed;
}
/**
* Constructs an instance of {@link ResourceNameGenerator} class.
*
* @param resourceType Type of resource. Should be {@link PdfName#ColorSpace}, {@link PdfName#ExtGState},
* {@link PdfName#Pattern}, {@link PdfName#Shading}, {@link PdfName#XObject}, {@link PdfName#Font}.
* @param prefix Prefix used for generating names.
*/
public ResourceNameGenerator(PdfName resourceType, String prefix) {
this(resourceType, prefix, 1);
}
/**
* Gets the resource type of generator.
*
* @return Type of resource. May be {@link PdfName#ColorSpace}, {@link PdfName#ExtGState},
* {@link PdfName#Pattern}, {@link PdfName#Shading}, {@link PdfName#XObject}, {@link PdfName#Font}.
*/
public PdfName getResourceType() {
return resourceType;
}
/**
* Generates new (unique) resource name.
*
* @param resources the {@link PdfResources} object for which name will be generated.
* @return new (unique) resource name.
*/
public PdfName generate(PdfResources resources) {
PdfName newName = new PdfName(prefix + counter++);
PdfDictionary r = resources.getPdfObject();
if (r.containsKey(resourceType)) {
while (r.getAsDictionary(resourceType).containsKey(newName)) {
newName = new PdfName(prefix + counter++);
}
}
return newName;
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy