mirror of
https://github.com/Stirling-Tools/Stirling-PDF.git
synced 2025-06-06 18:30:57 +00:00
114 lines
4.8 KiB
Java
114 lines
4.8 KiB
Java
package stirling.software.SPDF.controller.api;
|
|
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.IOException;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
|
|
import org.apache.pdfbox.multipdf.LayerUtility;
|
|
import org.apache.pdfbox.pdmodel.PDDocument;
|
|
import org.apache.pdfbox.pdmodel.PDPage;
|
|
import org.apache.pdfbox.pdmodel.PDPageContentStream;
|
|
import org.apache.pdfbox.pdmodel.common.PDRectangle;
|
|
import org.apache.pdfbox.pdmodel.graphics.form.PDFormXObject;
|
|
import org.apache.pdfbox.util.Matrix;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.web.bind.annotation.PostMapping;
|
|
import org.springframework.web.bind.annotation.RequestParam;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import io.swagger.v3.oas.annotations.Operation;
|
|
import io.swagger.v3.oas.annotations.Parameter;
|
|
import io.swagger.v3.oas.annotations.media.Schema;
|
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
|
import stirling.software.SPDF.utils.WebResponseUtils;
|
|
@RestController
|
|
@Tag(name = "General", description = "General APIs")
|
|
public class ScalePagesController {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(ScalePagesController.class);
|
|
|
|
@PostMapping(value = "/scale-pages", consumes = "multipart/form-data")
|
|
@Operation(summary = "Change the size of a PDF page/document", description = "This operation takes an input PDF file and the size to scale the pages to in the output PDF file. Input:PDF Output:PDF Type:SISO")
|
|
public ResponseEntity<byte[]> scalePages(
|
|
@Parameter(description = "The input PDF file", required = true) @RequestParam("fileInput") MultipartFile file,
|
|
@Parameter(description = "The scale of pages in the output PDF. Acceptable values are A0-A10, B0-B9, LETTER, TABLOID, LEDGER, LEGAL, EXECUTIVE.", required = true, schema = @Schema(type = "string", allowableValues = {
|
|
"A0", "A1", "A2", "A3", "A4", "A5", "A6", "A7", "A8", "A9", "A10", "B0", "B1", "B2", "B3", "B4",
|
|
"B5", "B6", "B7", "B8", "B9", "LETTER", "TABLOID", "LEDGER", "LEGAL",
|
|
"EXECUTIVE" })) @RequestParam("pageSize") String targetPDRectangle,
|
|
@Parameter(description = "The scale of the content on the pages of the output PDF. Acceptable values are floats.", required = true, schema = @Schema(type = "integer")) @RequestParam("scaleFactor") float scaleFactor)
|
|
throws IOException {
|
|
|
|
Map<String, PDRectangle> sizeMap = new HashMap<>();
|
|
// Add A0 - A10
|
|
sizeMap.put("A0", PDRectangle.A0);
|
|
sizeMap.put("A1", PDRectangle.A1);
|
|
sizeMap.put("A2", PDRectangle.A2);
|
|
sizeMap.put("A3", PDRectangle.A3);
|
|
sizeMap.put("A4", PDRectangle.A4);
|
|
sizeMap.put("A5", PDRectangle.A5);
|
|
sizeMap.put("A6", PDRectangle.A6);
|
|
|
|
// Add other sizes
|
|
sizeMap.put("LETTER", PDRectangle.LETTER);
|
|
sizeMap.put("LEGAL", PDRectangle.LEGAL);
|
|
|
|
if (!sizeMap.containsKey(targetPDRectangle)) {
|
|
throw new IllegalArgumentException(
|
|
"Invalid PDRectangle. It must be one of the following: A0, A1, A2, A3, A4, A5, A6, A7, A8, A9, A10");
|
|
}
|
|
|
|
PDRectangle targetSize = sizeMap.get(targetPDRectangle);
|
|
|
|
PDDocument sourceDocument = PDDocument.load(file.getBytes());
|
|
PDDocument outputDocument = new PDDocument();
|
|
|
|
int totalPages = sourceDocument.getNumberOfPages();
|
|
for (int i = 0; i < totalPages; i++) {
|
|
PDPage sourcePage = sourceDocument.getPage(i);
|
|
PDRectangle sourceSize = sourcePage.getMediaBox();
|
|
|
|
float scaleWidth = targetSize.getWidth() / sourceSize.getWidth();
|
|
float scaleHeight = targetSize.getHeight() / sourceSize.getHeight();
|
|
float scale = Math.min(scaleWidth, scaleHeight) * scaleFactor;
|
|
|
|
PDPage newPage = new PDPage(targetSize);
|
|
outputDocument.addPage(newPage);
|
|
|
|
PDPageContentStream contentStream = new PDPageContentStream(outputDocument, newPage, PDPageContentStream.AppendMode.APPEND, true);
|
|
|
|
float x = (targetSize.getWidth() - sourceSize.getWidth() * scale) / 2;
|
|
float y = (targetSize.getHeight() - sourceSize.getHeight() * scale) / 2;
|
|
|
|
contentStream.saveGraphicsState();
|
|
contentStream.transform(Matrix.getTranslateInstance(x, y));
|
|
contentStream.transform(Matrix.getScaleInstance(scale, scale));
|
|
|
|
LayerUtility layerUtility = new LayerUtility(outputDocument);
|
|
PDFormXObject form = layerUtility.importPageAsForm(sourceDocument, i);
|
|
contentStream.drawForm(form);
|
|
|
|
contentStream.restoreGraphicsState();
|
|
contentStream.close();
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
outputDocument.save(baos);
|
|
outputDocument.close();
|
|
sourceDocument.close();
|
|
|
|
|
|
return WebResponseUtils.bytesToWebResponse(baos.toByteArray(),
|
|
file.getOriginalFilename().replaceFirst("[.][^.]+$", "") + "_scaled.pdf");
|
|
}
|
|
|
|
|
|
}
|