1

#include <stdlib.h>

     2

     3

#include <png.h>

     4

     5

#include "common.h"

     6

#include "render_internal.h"

     7

#include "render_png.h"

     8

     9

struct render_png {

    10

    // libpng handles

    11

    png_structp png_ptr;

    12

    png_infop info_ptr;

    13

    18

};

    19

    21

    struct render_png *ctx = png_get_io_ptr(png_ptr);

    22

    25

            // error, doesn't return

    26

            png_error(png_ptr, "_render_png_write: io_write_fn");

    27

        }

    28

}

    29

    31

    struct render_png *ctx = png_get_io_ptr(png_ptr);

    32

    35

            // error, doesn't return

    36

            png_error(png_ptr, "_render_png_flush: io_flush_fn");

    37

        }

    38

}

    39

    41

    if (ctx)

    42

        png_destroy_write_struct(&ctx->png_ptr, &ctx->info_ptr);

    43

    44

    free(ctx);

    45

}

    46

    47

    48

struct render_png *render_png_init (struct render *render) {

    49

    50

    struct render_png *ctx = NULL;

    51

    52

    // calloc the render_png

    53

    if (!(ctx = calloc(1, sizeof(struct render_png))))

    54

        ERROR("calloc");

    55

    60

    61

    // libpng initialization

    62

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

    63

        ERROR("png_create_write_struct");

    64

    65

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

    66

        ERROR("png_create_info_struct");

    67

    68

    // libpng error handling

    69

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

    70

        ERROR("libpng");

    71

    72

    if (render->io_stream) {

    73

        // use normal libpng I/O

    74

        // XXX: who fcloses this?

    75

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

    76

    } else {

    77

        // setup our custom I/O callbacks

    78

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

    79

    }

    80

    81

    // some PNG metadata

    82

    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);

    83

    84

    // write out the PNG header

    85

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

    86

    87

    // success

    88

    return ctx;

    89

    90

error:

    91

    _render_png_free(ctx);

    92

    return NULL;

    93

}

    94

    95

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

    96

    // libpng error handling

    97

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

    98

        ERROR("libpng");

    99

   100

    // write it in

   101

    png_write_row(ctx->png_ptr, rowbuf);

   102

   103

    // success

   104

    return 0;

   105

   106

error:

   107

    _render_png_free(ctx);

   108

    return -1;

   109

}

   110

   111

int render_png_done (struct render_png *ctx) {

   112

    // libpng error handling

   113

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

   114

        ERROR("libpng");

   115

   116

    // write end

   117

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

   121

   122

    // success

   123

    return 0;

   124

   125

error:

   126

    _render_png_free(ctx);

   127

    return -1;

   128

}

   129

   130

int render_png_abort (struct render_png *ctx) {

   131

    // libpng error handling

   132

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

   133

        ERROR("libpng");

   136

    _render_png_free(ctx);

   137

   138

    // success

   139

    return 0;

   140

   141

error:

   142

    return -1;

   143

}

   144