1

#include <stdlib.h>

     2

     3

#include <png.h>

     4

     5

#include "common.h"

     8

#include "render_png.h"

     9

    10

    11

static void _render_png_write(png_structp png_ptr, png_bytep data, png_size_t length) {

    12

    struct render_png *ctx = png_get_io_ptr(png_ptr);

    13

    14

    if (ctx->io_write_fn)

    15

        if (ctx->io_write_fn(data, length, ctx->cb_arg)) {

    16

            // error, doesn't return

    17

            png_error(png_ptr, "_render_png_write: io_write_fn");

    18

        }

    19

}

    20

    21

static void _render_png_flush(png_structp png_ptr) {

    22

    struct render_png *ctx = png_get_io_ptr(png_ptr);

    23

    24

    if (ctx->io_flush_fn)

    25

        if (ctx->io_flush_fn(ctx->cb_arg)) {

    26

            // error, doesn't return

    27

            png_error(png_ptr, "_render_png_flush: io_flush_fn");

    28

        }

    29

}

    30

    31

static void _render_png_free (struct render_png *ctx) {

    36

}

    37

    42

    48

    51

    63

    // store some info from the struct render

    64

    ctx->io_write_fn = render->io_write_fn;

    65

    ctx->io_flush_fn = render->io_flush_fn;

    66

    ctx->cb_arg = render->cb_arg;

    67

    68

    // libpng initialization

    69

    if (!(ctx->png_ptr = png_create_write_struct (PNG_LIBPNG_VER_STRING, NULL, NULL, NULL)))

    70

        ERROR("png_create_write_struct");

    71

    72

    if (!(ctx->info_ptr = png_create_info_struct(ctx->png_ptr)))

    73

        ERROR("png_create_info_struct");

    74

    75

    // libpng error handling

    76

    if (setjmp(png_jmpbuf(ctx->png_ptr)))

    77

        ERROR("libpng");

    78

    79

    if (render->io_stream) {

    80

        // use normal libpng I/O

    81

        // XXX: who fcloses this?

    82

        png_init_io(ctx->png_ptr, render->io_stream);

    83

    } else {

    84

        // setup our custom I/O callbacks

    85

        png_set_write_fn(ctx->png_ptr, ctx, &_render_png_write, &_render_png_flush);

    86

    }

    87

    88

    // some PNG metadata

    89

    png_set_IHDR(ctx->png_ptr, ctx->info_ptr, render->img_w, render->img_h, 8, PNG_COLOR_TYPE_GRAY, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

    90

    91

    // write out the PNG header

    92

    png_write_info(ctx->png_ptr, ctx->info_ptr);

   115

   116

    // success

   117

    return ctx;

   118

   119

error:

   120

    _render_png_free(ctx);

   121

    return NULL;

   122

}

   123

   124

int render_png_row (struct render_png *ctx, unsigned char *rowbuf) {

   125

    // libpng error handling

   126

    if (setjmp(png_jmpbuf(ctx->png_ptr)))

   127

        ERROR("libpng");

   128

   129

    // write it in

   130

    png_write_row(ctx->png_ptr, rowbuf);

   131

   132

    // success

   133

    return 0;

   134

   135

error:

   136

    // don't free it here, our user needs to do that

   137

    return -1;

   138

}

   139

   140

int render_png_done (struct render_png *ctx) {

   141

    // libpng error handling

   142

    if (setjmp(png_jmpbuf(ctx->png_ptr)))

   143

        ERROR("libpng");

   144

   145

    // write end

   146

    png_write_end(ctx->png_ptr, ctx->info_ptr);

   147

   148

    // free everything

   149

    _render_png_free(ctx);

   150

   151

    // success

   152

    return 0;

   153

   154

error:

   155

    _render_png_free(ctx);

   156

    return -1;

   157

}

   158

   159

int render_png_abort (struct render_png *ctx) {

   160

    // libpng error handling

   161

    if (setjmp(png_jmpbuf(ctx->png_ptr)))

   162

        ERROR("libpng");

   163

   164

    // just free it

   165

    _render_png_free(ctx);

   166

   167

    // success

   168

    return 0;

   169

   170

error:

   171

    free(ctx);

   172

    return -1;

   173

}

   174