<commit_before>/*
 * Copyright (C) 2015 <NAME> (<EMAIL>)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package guru.nidi.graphviz.engine;

import com.kitfox.svg.SVGDiagram;
import com.kitfox.svg.SVGException;
import com.kitfox.svg.SVGUniverse;
import org.apache.batik.transcoder.TranscoderException;
import org.apache.batik.transcoder.TranscoderInput;
import org.apache.batik.transcoder.TranscoderOutput;
import org.apache.batik.transcoder.TranscodingHints;
import org.apache.batik.transcoder.image.ImageTranscoder;
import org.apache.batik.transcoder.image.PNGTranscoder;

import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.io.StringReader;
import java.net.URI;
import java.util.function.Consumer;

public enum Rasterizer {
    SALAMANDER {
        @Override
        public BufferedImage render(Graphviz graphviz, Consumer<Graphics2D> graphicsConfigurer, String svg) {
            final SVGDiagram diagram = createDiagram(svg);
            double scaleX = graphviz.scale;
            double scaleY = graphviz.scale;
            if (graphviz.width != 0 || graphviz.height != 0) {
                scaleX = graphviz.scale * graphviz.width / diagram.getWidth();
                scaleY = graphviz.scale * graphviz.height / diagram.getHeight();
                if (scaleX == 0) {
                    scaleX = scaleY;
                }
                if (scaleY == 0) {
                    scaleY = scaleX;
                }
            }
            final BufferedImage image = new BufferedImage((int) Math.ceil(scaleX * diagram.getWidth()), (int) Math.ceil(scaleY * diagram.getHeight()), BufferedImage.TYPE_INT_ARGB);
            final Graphics2D graphics = image.createGraphics();
            configGraphics(graphics);
            if (graphicsConfigurer != null) {
                graphicsConfigurer.accept(graphics);
            }
            graphics.scale(scaleX, scaleY);
            renderDiagram(diagram, graphics);
            return image;
        }

        private SVGDiagram createDiagram(String svg) {
            final SVGUniverse universe = new SVGUniverse();
            final URI uri = universe.loadSVG(new StringReader(svg), "//graph/");
            final SVGDiagram diagram = universe.getDiagram(uri);
            diagram.setIgnoringClipHeuristic(true);
            return diagram;
        }

        private void renderDiagram(SVGDiagram diagram, Graphics2D graphics) {
            try {
                diagram.render(graphics);
            } catch (SVGException e) {
                throw new GraphvizException("Problem rendering SVG", e);
            }
        }

        private void configGraphics(Graphics2D graphics) {
            graphics.setRenderingHint(RenderingHints.KEY_ALPHA_INTERPOLATION, RenderingHints.VALUE_ALPHA_INTERPOLATION_QUALITY);
            graphics.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
            graphics.setRenderingHint(RenderingHints.KEY_COLOR_RENDERING, RenderingHints.VALUE_COLOR_RENDER_QUALITY);
            graphics.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BICUBIC);
            graphics.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);
            graphics.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
        }
    },
    BATIK {
        @Override
        public BufferedImage render(Graphviz graphviz, Consumer<Graphics2D> graphicsConfigurer, String svg) {
            final BufferedImage[] image = new BufferedImage[1];
            final TranscoderInput in = new TranscoderInput(new StringReader(svg));
            try {
                final TranscoderOutput out = new TranscoderOutput(new OutputStream() {
                    @Override
                    public void write(int b) throws IOException {
                    }
                });
                final PNGTranscoder t = new PNGTranscoder() {
                    @Override
                    public BufferedImage createImage(int width, int height) {
                        return image[0] = super.createImage(width, height);
                    }
                };
                final TranscodingHints hints = new TranscodingHints(t.getTranscodingHints());
                if (graphviz.width != 0) {
                    hints.put(ImageTranscoder.KEY_WIDTH, graphviz.scale * graphviz.width);
                }
                if (graphviz.height != 0) {
                    hints.put(ImageTranscoder.KEY_HEIGHT, graphviz.scale * graphviz.height);
                }
                t.setTranscodingHints(hints);<commit_msg>size in floats for batik<commit_after>/*
 * Copyright (C) 2015 <NAME> (<EMAIL>)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package guru.nidi.graphviz.engine;

import com.kitfox.svg.SVGDiagram;
import com.kitfox.svg.SVGException;
import com.kitfox.svg.SVGUniverse;
import org.apache.batik.transcoder.TranscoderException;
import org.apache.batik.transcoder.TranscoderInput;
import org.apache.batik.transcoder.TranscoderOutput;
import org.apache.batik.transcoder.TranscodingHints;
import org.apache.batik.transcoder.image.ImageTranscoder;
import org.apache.batik.transcoder.image.PNGTranscoder;

import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.io.StringReader;
import java.net.URI;
import java.util.function.Consumer;

public enum Rasterizer {
    SALAMANDER {
        @Override
        public BufferedImage render(Graphviz graphviz, Consumer<Graphics2D> graphicsConfigurer, String svg) {
            final SVGDiagram diagram = createDiagram(svg);
            double scaleX = graphviz.scale;
            double scaleY = graphviz.scale;
            if (graphviz.width != 0 || graphviz.height != 0) {
                scaleX = graphviz.scale * graphviz.width / diagram.getWidth();
                scaleY = graphviz.scale * graphviz.height / diagram.getHeight();
                if (scaleX == 0) {
                    scaleX = scaleY;
                }
                if (scaleY == 0) {
                    scaleY = scaleX;
                }
            }
            final BufferedImage image = new BufferedImage((int) Math.ceil(scaleX * diagram.getWidth()), (int) Math.ceil(scaleY * diagram.getHeight()), BufferedImage.TYPE_INT_ARGB);
            final Graphics2D graphics = image.createGraphics();
            configGraphics(graphics);
            if (graphicsConfigurer != null) {
                graphicsConfigurer.accept(graphics);
            }
            graphics.scale(scaleX, scaleY);
            renderDiagram(diagram, graphics);
            return image;
        }

        private SVGDiagram createDiagram(String svg) {
            final SVGUniverse universe = new SVGUniverse();
            final URI uri = universe.loadSVG(new StringReader(svg), "//graph/");
            final SVGDiagram diagram = universe.getDiagram(uri);
            diagram.setIgnoringClipHeuristic(true);
            return diagram;
        }

        private void renderDiagram(SVGDiagram diagram, Graphics2D graphics) {
            try {
                diagram.render(graphics);
            } catch (SVGException e) {
                throw new GraphvizException("Problem rendering SVG", e);
            }
        }

        private void configGraphics(Graphics2D graphics) {
            graphics.setRenderingHint(RenderingHints.KEY_ALPHA_INTERPOLATION, RenderingHints.VALUE_ALPHA_INTERPOLATION_QUALITY);
            graphics.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
            graphics.setRenderingHint(RenderingHints.KEY_COLOR_RENDERING, RenderingHints.VALUE_COLOR_RENDER_QUALITY);
            graphics.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BICUBIC);
            graphics.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);
            graphics.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
        }
    },
    BATIK {
        @Override
        public BufferedImage render(Graphviz graphviz, Consumer<Graphics2D> graphicsConfigurer, String svg) {
            final BufferedImage[] image = new BufferedImage[1];
            final TranscoderInput in = new TranscoderInput(new StringReader(svg));
            try {
                final TranscoderOutput out = new TranscoderOutput(new OutputStream() {
                    @Override
                    public void write(int b) throws IOException {
                    }
                });
                final PNGTranscoder t = new PNGTranscoder() {
                    @Override
                    public BufferedImage createImage(int width, int height) {
                        return image[0] = super.createImage(width, height);
                    }
                };
                final TranscodingHints hints = new TranscodingHints(t.getTranscodingHints());
                if (graphviz.width != 0) {
                    hints.put(ImageTranscoder.KEY_WIDTH, (float) graphviz.scale * graphviz.width);
                }
                if (graphviz.height != 0) {
                    hints.put(ImageTranscoder.KEY_HEIGHT, (float) graphviz.scale * graphviz.height);
                }
                t.setTranscodingHints(hints);
{'fraction_non_alphanumeric': 0.08480565371024736, 'fraction_numerical': 0.004011078216025213, 'mean_word_length': 2.390093881515053, 'pattern_counts': {'":': 0, '<': 11, '<?xml version=': 0, '>': 11, 'https://': 0, 'lorem ipsum': 0, 'www.': 2, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '1213789', 'n_tokens_mistral': 2985, 'n_tokens_neox': 2861, 'n_words': 633}