Search in sources :

Example 1 with RandomAccess

use of org.apache.pdfbox.io.RandomAccess in project pdfbox by apache.

the class COSInputStream method create.

static COSInputStream create(List<Filter> filters, COSDictionary parameters, InputStream in, ScratchFile scratchFile, DecodeOptions options) throws IOException {
    List<DecodeResult> results = new ArrayList<>();
    InputStream input = in;
    if (filters.isEmpty()) {
        input = in;
    } else {
        // apply filters
        for (int i = 0; i < filters.size(); i++) {
            if (scratchFile != null) {
                // scratch file
                final RandomAccess buffer = scratchFile.createBuffer();
                DecodeResult result = filters.get(i).decode(input, new RandomAccessOutputStream(buffer), parameters, i, options);
                results.add(result);
                input = new RandomAccessInputStream(buffer) {

                    @Override
                    public void close() throws IOException {
                        buffer.close();
                    }
                };
            } else {
                // in-memory
                ByteArrayOutputStream output = new ByteArrayOutputStream();
                DecodeResult result = filters.get(i).decode(input, output, parameters, i, options);
                results.add(result);
                input = new ByteArrayInputStream(output.toByteArray());
            }
        }
    }
    return new COSInputStream(input, results);
}
Also used : RandomAccessInputStream(org.apache.pdfbox.io.RandomAccessInputStream) FilterInputStream(java.io.FilterInputStream) ByteArrayInputStream(java.io.ByteArrayInputStream) InputStream(java.io.InputStream) ArrayList(java.util.ArrayList) RandomAccess(org.apache.pdfbox.io.RandomAccess) IOException(java.io.IOException) ByteArrayOutputStream(java.io.ByteArrayOutputStream) DecodeResult(org.apache.pdfbox.filter.DecodeResult) ByteArrayInputStream(java.io.ByteArrayInputStream) RandomAccessOutputStream(org.apache.pdfbox.io.RandomAccessOutputStream) RandomAccessInputStream(org.apache.pdfbox.io.RandomAccessInputStream)

Example 2 with RandomAccess

use of org.apache.pdfbox.io.RandomAccess in project pdfbox by apache.

the class COSOutputStream method close.

@Override
public void close() throws IOException {
    try {
        if (buffer != null) {
            try {
                // apply filters in reverse order
                for (int i = filters.size() - 1; i >= 0; i--) {
                    try (InputStream unfilteredIn = new RandomAccessInputStream(buffer)) {
                        if (i == 0) {
                            /*
                                 * The last filter to run can encode directly to the enclosed output
                                 * stream.
                                 */
                            filters.get(i).encode(unfilteredIn, out, parameters, i);
                        } else {
                            RandomAccess filteredBuffer = scratchFile.createBuffer();
                            try (OutputStream filteredOut = new RandomAccessOutputStream(filteredBuffer)) {
                                filters.get(i).encode(unfilteredIn, filteredOut, parameters, i);
                            } finally {
                                buffer.close();
                                buffer = filteredBuffer;
                            }
                        }
                    }
                }
            } finally {
                buffer.close();
                buffer = null;
            }
        }
    } finally {
        super.close();
    }
}
Also used : RandomAccessInputStream(org.apache.pdfbox.io.RandomAccessInputStream) InputStream(java.io.InputStream) OutputStream(java.io.OutputStream) RandomAccessOutputStream(org.apache.pdfbox.io.RandomAccessOutputStream) FilterOutputStream(java.io.FilterOutputStream) RandomAccess(org.apache.pdfbox.io.RandomAccess) RandomAccessOutputStream(org.apache.pdfbox.io.RandomAccessOutputStream) RandomAccessInputStream(org.apache.pdfbox.io.RandomAccessInputStream)

Aggregations

InputStream (java.io.InputStream)2 RandomAccess (org.apache.pdfbox.io.RandomAccess)2 RandomAccessInputStream (org.apache.pdfbox.io.RandomAccessInputStream)2 RandomAccessOutputStream (org.apache.pdfbox.io.RandomAccessOutputStream)2 ByteArrayInputStream (java.io.ByteArrayInputStream)1 ByteArrayOutputStream (java.io.ByteArrayOutputStream)1 FilterInputStream (java.io.FilterInputStream)1 FilterOutputStream (java.io.FilterOutputStream)1 IOException (java.io.IOException)1 OutputStream (java.io.OutputStream)1 ArrayList (java.util.ArrayList)1 DecodeResult (org.apache.pdfbox.filter.DecodeResult)1