org.sejda.model.parameter.ExtractPagesParameters Maven / Gradle / Ivy
/*
* Created on 25/ago/2011
* Copyright 2011 by Andrea Vacondio ([email protected]).
*
* This file is part of the Sejda source code
*
* 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 .
*/
package org.sejda.model.parameter;
import jakarta.validation.Valid;
import jakarta.validation.constraints.NotNull;
import org.apache.commons.lang3.builder.EqualsBuilder;
import org.apache.commons.lang3.builder.HashCodeBuilder;
import org.sejda.commons.collection.NullSafeSet;
import org.sejda.model.optimization.OptimizationPolicy;
import org.sejda.model.parameter.base.DiscardableOutlineTaskParameters;
import org.sejda.model.parameter.base.MultiplePdfSourceMultipleOutputParameters;
import org.sejda.model.parameter.base.OptimizableOutputTaskParameters;
import org.sejda.model.pdf.page.PageRange;
import org.sejda.model.pdf.page.PageRangeSelection;
import org.sejda.model.pdf.page.PagesSelection;
import org.sejda.model.pdf.page.PredefinedSetOfPages;
import org.sejda.model.validation.constraint.HasSelectedPages;
import org.sejda.model.validation.constraint.NotAllowed;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Set;
/**
* Parameter class for an Extract pages task. Allow to specify a predefined set of pages to extract (odd, even) or a set of page ranges but not both. Page ranges are validated to
* make sure that there is no intersection.
*
* @author Andrea Vacondio
*
*/
@HasSelectedPages
public class ExtractPagesParameters extends MultiplePdfSourceMultipleOutputParameters implements PageRangeSelection,
PagesSelection, OptimizableOutputTaskParameters, DiscardableOutlineTaskParameters {
@NotNull
private OptimizationPolicy optimizationPolicy = OptimizationPolicy.NO;
private boolean discardOutline = false;
@NotNull
@NotAllowed(disallow = { PredefinedSetOfPages.ALL_PAGES })
private PredefinedSetOfPages predefinedSetOfPages;
@Valid
private final Set pageSelection = new NullSafeSet<>();
private boolean invertSelection = false;
private boolean separateFileForEachRange = false;
/**
* Creates and empty instance where page selection can be set
*/
public ExtractPagesParameters() {
this.predefinedSetOfPages = PredefinedSetOfPages.NONE;
}
/**
* Creates an instance using a predefined set of pages to extract.
*
* @param predefinedSetOfPages
*/
public ExtractPagesParameters(PredefinedSetOfPages predefinedSetOfPages) {
this.predefinedSetOfPages = predefinedSetOfPages;
}
public void addPageRange(PageRange range) {
pageSelection.add(range);
}
public void addAllPageRanges(Collection ranges) {
pageSelection.addAll(ranges);
}
public PredefinedSetOfPages getPredefinedSetOfPages() {
return predefinedSetOfPages;
}
/**
* @return an unmodifiable view of the pageSelection
*/
@Override
public Set getPageSelection() {
return Collections.unmodifiableSet(pageSelection);
}
/**
* @param upperLimit
* the number of pages of the document (upper limit).
* @return the selected set of pages. Iteration ordering is predictable, it is the order in which elements were inserted into the {@link PageRange} set or the natural order in
* case of {@link PredefinedSetOfPages}.
* @see PagesSelection#getPages(int)
*/
@Override
public Set getPages(int upperLimit) {
Set pages = new NullSafeSet<>();
if (predefinedSetOfPages != PredefinedSetOfPages.NONE) {
pages = predefinedSetOfPages.getPages(upperLimit);
} else {
for (PageRange range : pageSelection) {
pages.addAll(range.getPages(upperLimit));
}
}
if (!invertSelection) {
return pages;
}
Set invertedPages = new NullSafeSet<>();
for (int i = 1; i <= upperLimit; i++) {
if (!pages.contains(i)) {
invertedPages.add(i);
}
}
return invertedPages;
}
/**
*
* @param upperLimit
* the number of pages of the document (upper limit).
* @return a list of sets of pages. Each set represents an extraction set that generates an output PDF file.
*/
public List> getPagesSets(int upperLimit) {
ArrayList> pages = new ArrayList<>();
if (predefinedSetOfPages == PredefinedSetOfPages.NONE && isSeparateFileForEachRange()) {
if (invertSelection) {
Set currentRange = new NullSafeSet<>();
int previous = 0;
for (int current : getPages(upperLimit)) {
if (currentRange.isEmpty() || (current == previous + 1)) {
currentRange.add(current);
previous = current;
} else {
pages.add(currentRange);
currentRange = new NullSafeSet<>();
currentRange.add(current);
previous = current;
}
}
if (!currentRange.isEmpty()) {
pages.add(currentRange);
}
} else {
for (PageRange range : pageSelection) {
pages.add(range.getPages(upperLimit));
}
}
return pages;
}
return Collections.singletonList(getPages(upperLimit));
}
@Override
public OptimizationPolicy getOptimizationPolicy() {
return optimizationPolicy;
}
@Override
public void setOptimizationPolicy(OptimizationPolicy optimizationPolicy) {
this.optimizationPolicy = optimizationPolicy;
}
@Override
public boolean discardOutline() {
return discardOutline;
}
@Override
public void discardOutline(boolean discardOutline) {
this.discardOutline = discardOutline;
}
public boolean isInvertSelection() {
return invertSelection;
}
public void setInvertSelection(boolean invertSelection) {
this.invertSelection = invertSelection;
}
/**
* @return if true the task will generate a separate PDF file for each PageRange in the parameters
*/
public boolean isSeparateFileForEachRange() {
return separateFileForEachRange;
}
public void setSeparateFileForEachRange(boolean separateFileForEachRange) {
this.separateFileForEachRange = separateFileForEachRange;
}
@Override
public int hashCode() {
return new HashCodeBuilder().appendSuper(super.hashCode()).append(optimizationPolicy).append(discardOutline)
.append(predefinedSetOfPages).append(invertSelection).append(separateFileForEachRange)
.append(pageSelection).toHashCode();
}
@Override
public boolean equals(Object other) {
if (this == other) {
return true;
}
if (!(other instanceof ExtractPagesParameters parameter)) {
return false;
}
return new EqualsBuilder().appendSuper(super.equals(other))
.append(predefinedSetOfPages, parameter.predefinedSetOfPages)
.append(optimizationPolicy, parameter.optimizationPolicy)
.append(discardOutline, parameter.discardOutline).append(pageSelection, parameter.pageSelection)
.append(invertSelection, parameter.invertSelection)
.append(separateFileForEachRange, parameter.separateFileForEachRange).isEquals();
}
}