Examples with PngMetadata ar.com.hjg.pngj.chunks.PngMetadata used on opensource projects

Example 1 with PngMetadata

use of ar.com.hjg.pngj.chunks.PngMetadata in project imageio-ext by geosolutions-it.

the class PNGWriter method writePNG.

public RenderedImage writePNG(RenderedImage image, OutputStream outStream, float quality, FilterType filterType, Map<String, String> text) throws Exception {
    // compute the compression level similarly to what the Clib code does
    int level = Math.round(9 * (1f - quality));
    // get the optimal scanline provider for this image
    RenderedImage original = image;
    ScanlineProvider scanlines = ScanlineProviderFactory.getProvider(image);
    if (scanlines == null) {
        throw new IllegalArgumentException("Could not find a scanline extractor for " + original);
    }
    // encode using the PNGJ library and the GeoServer own scanline providers
    ColorModel colorModel = image.getColorModel();
    boolean indexed = colorModel instanceof IndexColorModel;
    ImageInfo ii = getImageInfo(image, scanlines, colorModel, indexed);
    PngWriter pw = new PngWriter(outStream, ii);
    pw.setShouldCloseStream(false);
    try {
        pw.setCompLevel(level);
        pw.setFilterType(filterType);
        ChunksListForWrite chunkList = pw.getChunksList();
        PngMetadata metadata = pw.getMetadata();
        if (indexed) {
            IndexColorModel icm = (IndexColorModel) colorModel;
            PngChunkPLTE palette = metadata.createPLTEChunk();
            int ncolors = icm.getMapSize();
            palette.setNentries(ncolors);
            for (int i = 0; i < ncolors; i++) {
                final int red = icm.getRed(i);
                final int green = icm.getGreen(i);
                final int blue = icm.getBlue(i);
                palette.setEntry(i, red, green, blue);
            }
            if (icm.hasAlpha()) {
                PngChunkTRNS transparent = new PngChunkTRNS(ii);
                int[] alpha = new int[ncolors];
                for (int i = 0; i < ncolors; i++) {
                    final int a = icm.getAlpha(i);
                    alpha[i] = a;
                }
                transparent.setPalletteAlpha(alpha);
                chunkList.queue(transparent);
            }
        }
        if (text != null && !text.isEmpty()) {
            Iterator<Entry<String, String>> entrySetIterator = text.entrySet().iterator();
            while (entrySetIterator.hasNext()) {
                Entry<String, String> entrySet = entrySetIterator.next();
                metadata.setText(entrySet.getKey(), entrySet.getValue(), true, false);
            }
        }
        // write out the actual image lines
        for (int row = 0; row < image.getHeight(); row++) {
            pw.writeRow(scanlines);
        }
        pw.end();
    } catch (Exception e) {
        LOGGER.log(Level.SEVERE, "Failed to encode the PNG", e);
        throw e;
    } finally {
        pw.close();
    }
    return image;
}

Example 2 with PngMetadata

use of ar.com.hjg.pngj.chunks.PngMetadata in project imageio-ext by geosolutions-it.

the class PNGWriterTest method testTeXt.

@Test
public void testTeXt() throws Exception {
    PNGWriter writer = new PNGWriter();
    OutputStream out = null;
    File pngOut = null;
    final String title = "Title";
    final String description = "Sample Description";
    final String software = "ImageIO-Ext";
    final String author = "Me";
    try {
        // read test image
        BufferedImage read = ImageIO.read(TestData.file(this, "sample.jpeg"));
        pngOut = TestData.temp(this, "test.png", true);
        out = new FileOutputStream(pngOut);
        Map<String, String> textMetadata = new HashMap<String, String>();
        textMetadata.put("Title", title);
        textMetadata.put("Author", author);
        textMetadata.put("Software", software);
        textMetadata.put("Description", description);
        writer.writePNG(read, out, 1, FilterType.FILTER_NONE, textMetadata);
    } finally {
        if (out != null) {
            try {
                out.close();
            } catch (IOException e) {
            }
        }
    }
    BufferedImage test = ImageIO.read(pngOut);
    assertNotNull(test);
    PngReader reader = null;
    try {
        reader = new PngReader(pngOut);
        reader.readSkippingAllRows();
        PngMetadata metadata = reader.getMetadata();
        assertNotNull(metadata);
        assertEquals(title, metadata.getTxtForKey("Title"));
        assertEquals(description, metadata.getTxtForKey("Description"));
        assertEquals(author, metadata.getTxtForKey("Author"));
        assertEquals(software, metadata.getTxtForKey("Software"));
    } finally {
        if (reader != null) {
            reader.close();
        }
    }
}