pngwio.c
上传用户:sesekoo
上传日期:2020-07-18
资源大小:21543k
文件大小:8k
源码类别:

界面编程

开发平台:

Visual C++

  1. /* pngwio.c - functions for data output
  2.  *
  3.  * Last changed in libpng 1.2.30 [August 15, 2008]
  4.  * For conditions of distribution and use, see copyright notice in png.h
  5.  * Copyright (c) 1998-2008 Glenn Randers-Pehrson
  6.  * (Version 0.96 Copyright (c) 1996, 1997 Andreas Dilger)
  7.  * (Version 0.88 Copyright (c) 1995, 1996 Guy Eric Schalnat, Group 42, Inc.)
  8.  *
  9.  * This file provides a location for all output.  Users who need
  10.  * special handling are expected to write functions that have the same
  11.  * arguments as these and perform similar functions, but that possibly
  12.  * use different output methods.  Note that you shouldn't change these
  13.  * functions, but rather write replacement functions and then change
  14.  * them at run time with png_set_write_fn(...).
  15.  */
  16. #define PNG_INTERNAL
  17. #include "png.h"
  18. #ifdef PNG_WRITE_SUPPORTED
  19. /* Write the data to whatever output you are using.  The default routine
  20.    writes to a file pointer.  Note that this routine sometimes gets called
  21.    with very small lengths, so you should implement some kind of simple
  22.    buffering if you are using unbuffered writes.  This should never be asked
  23.    to write more than 64K on a 16 bit machine.  */
  24. void /* PRIVATE */
  25. png_write_data(png_structp png_ptr, png_bytep data, png_size_t length)
  26. {
  27.    if (png_ptr->write_data_fn != NULL )
  28.       (*(png_ptr->write_data_fn))(png_ptr, data, length);
  29.    else
  30.       png_error(png_ptr, "Call to NULL write function");
  31. }
  32. #if !defined(PNG_NO_STDIO)
  33. /* This is the function that does the actual writing of data.  If you are
  34.    not writing to a standard C stream, you should create a replacement
  35.    write_data function and use it at run time with png_set_write_fn(), rather
  36.    than changing the library. */
  37. #ifndef USE_FAR_KEYWORD
  38. void PNGAPI
  39. png_default_write_data(png_structp png_ptr, png_bytep data, png_size_t length)
  40. {
  41.    png_uint_32 check;
  42.    if (png_ptr == NULL) return;
  43. #if defined(_WIN32_WCE)
  44.    if ( !WriteFile((HANDLE)(png_ptr->io_ptr), data, length, &check, NULL) )
  45.       check = 0;
  46. #else
  47.    check = (png_uint_32)fwrite(data, 1, length, (png_FILE_p)(png_ptr->io_ptr));
  48. #endif
  49.    if (check != length)
  50.       png_error(png_ptr, "Write Error");
  51. }
  52. #else
  53. /* this is the model-independent version. Since the standard I/O library
  54.    can't handle far buffers in the medium and small models, we have to copy
  55.    the data.
  56. */
  57. #define NEAR_BUF_SIZE 1024
  58. #define MIN(a,b) (a <= b ? a : b)
  59. void PNGAPI
  60. png_default_write_data(png_structp png_ptr, png_bytep data, png_size_t length)
  61. {
  62.    png_uint_32 check;
  63.    png_byte *near_data;  /* Needs to be "png_byte *" instead of "png_bytep" */
  64.    png_FILE_p io_ptr;
  65.    if (png_ptr == NULL) return;
  66.    /* Check if data really is near. If so, use usual code. */
  67.    near_data = (png_byte *)CVT_PTR_NOCHECK(data);
  68.    io_ptr = (png_FILE_p)CVT_PTR(png_ptr->io_ptr);
  69.    if ((png_bytep)near_data == data)
  70.    {
  71. #if defined(_WIN32_WCE)
  72.       if ( !WriteFile(io_ptr, near_data, length, &check, NULL) )
  73.          check = 0;
  74. #else
  75.       check = fwrite(near_data, 1, length, io_ptr);
  76. #endif
  77.    }
  78.    else
  79.    {
  80.       png_byte buf[NEAR_BUF_SIZE];
  81.       png_size_t written, remaining, err;
  82.       check = 0;
  83.       remaining = length;
  84.       do
  85.       {
  86.          written = MIN(NEAR_BUF_SIZE, remaining);
  87.          png_memcpy(buf, data, written); /* copy far buffer to near buffer */
  88. #if defined(_WIN32_WCE)
  89.          if ( !WriteFile(io_ptr, buf, written, &err, NULL) )
  90.             err = 0;
  91. #else
  92.          err = fwrite(buf, 1, written, io_ptr);
  93. #endif
  94.          if (err != written)
  95.             break;
  96.          else
  97.             check += err;
  98.          data += written;
  99.          remaining -= written;
  100.       }
  101.       while (remaining != 0);
  102.    }
  103.    if (check != length)
  104.       png_error(png_ptr, "Write Error");
  105. }
  106. #endif
  107. #endif
  108. /* This function is called to output any data pending writing (normally
  109.    to disk).  After png_flush is called, there should be no data pending
  110.    writing in any buffers. */
  111. #if defined(PNG_WRITE_FLUSH_SUPPORTED)
  112. void /* PRIVATE */
  113. png_flush(png_structp png_ptr)
  114. {
  115.    if (png_ptr->output_flush_fn != NULL)
  116.       (*(png_ptr->output_flush_fn))(png_ptr);
  117. }
  118. #if !defined(PNG_NO_STDIO)
  119. void PNGAPI
  120. png_default_flush(png_structp png_ptr)
  121. {
  122. #if !defined(_WIN32_WCE)
  123.    png_FILE_p io_ptr;
  124. #endif
  125.    if (png_ptr == NULL) return;
  126. #if !defined(_WIN32_WCE)
  127.    io_ptr = (png_FILE_p)CVT_PTR((png_ptr->io_ptr));
  128.    if (io_ptr != NULL)
  129.       fflush(io_ptr);
  130. #endif
  131. }
  132. #endif
  133. #endif
  134. /* This function allows the application to supply new output functions for
  135.    libpng if standard C streams aren't being used.
  136.    This function takes as its arguments:
  137.    png_ptr       - pointer to a png output data structure
  138.    io_ptr        - pointer to user supplied structure containing info about
  139.                    the output functions.  May be NULL.
  140.    write_data_fn - pointer to a new output function that takes as its
  141.                    arguments a pointer to a png_struct, a pointer to
  142.                    data to be written, and a 32-bit unsigned int that is
  143.                    the number of bytes to be written.  The new write
  144.                    function should call png_error(png_ptr, "Error msg")
  145.                    to exit and output any fatal error messages.
  146.    flush_data_fn - pointer to a new flush function that takes as its
  147.                    arguments a pointer to a png_struct.  After a call to
  148.                    the flush function, there should be no data in any buffers
  149.                    or pending transmission.  If the output method doesn't do
  150.                    any buffering of ouput, a function prototype must still be
  151.                    supplied although it doesn't have to do anything.  If
  152.                    PNG_WRITE_FLUSH_SUPPORTED is not defined at libpng compile
  153.                    time, output_flush_fn will be ignored, although it must be
  154.                    supplied for compatibility. */
  155. void PNGAPI
  156. png_set_write_fn(png_structp png_ptr, png_voidp io_ptr,
  157.    png_rw_ptr write_data_fn, png_flush_ptr output_flush_fn)
  158. {
  159.    if (png_ptr == NULL) return;
  160.    png_ptr->io_ptr = io_ptr;
  161. #if !defined(PNG_NO_STDIO)
  162.    if (write_data_fn != NULL)
  163.       png_ptr->write_data_fn = write_data_fn;
  164.    else
  165.       png_ptr->write_data_fn = png_default_write_data;
  166. #else
  167.    png_ptr->write_data_fn = write_data_fn;
  168. #endif
  169. #if defined(PNG_WRITE_FLUSH_SUPPORTED)
  170. #if !defined(PNG_NO_STDIO)
  171.    if (output_flush_fn != NULL)
  172.       png_ptr->output_flush_fn = output_flush_fn;
  173.    else
  174.       png_ptr->output_flush_fn = png_default_flush;
  175. #else
  176.    png_ptr->output_flush_fn = output_flush_fn;
  177. #endif
  178. #endif /* PNG_WRITE_FLUSH_SUPPORTED */
  179.    /* It is an error to read while writing a png file */
  180.    if (png_ptr->read_data_fn != NULL)
  181.    {
  182.       png_ptr->read_data_fn = NULL;
  183.       png_warning(png_ptr,
  184.          "Attempted to set both read_data_fn and write_data_fn in");
  185.       png_warning(png_ptr,
  186.          "the same structure.  Resetting read_data_fn to NULL.");
  187.    }
  188. }
  189. #if defined(USE_FAR_KEYWORD)
  190. #if defined(_MSC_VER)
  191. void *png_far_to_near(png_structp png_ptr, png_voidp ptr, int check)
  192. {
  193.    void *near_ptr;
  194.    void FAR *far_ptr;
  195.    FP_OFF(near_ptr) = FP_OFF(ptr);
  196.    far_ptr = (void FAR *)near_ptr;
  197.    if (check != 0)
  198.       if (FP_SEG(ptr) != FP_SEG(far_ptr))
  199.          png_error(png_ptr, "segment lost in conversion");
  200.    return(near_ptr);
  201. }
  202. #  else
  203. void *png_far_to_near(png_structp png_ptr, png_voidp ptr, int check)
  204. {
  205.    void *near_ptr;
  206.    void FAR *far_ptr;
  207.    near_ptr = (void FAR *)ptr;
  208.    far_ptr = (void FAR *)near_ptr;
  209.    if (check != 0)
  210.       if (far_ptr != ptr)
  211.          png_error(png_ptr, "segment lost in conversion");
  212.    return(near_ptr);
  213. }
  214. #   endif
  215. #   endif
  216. #endif /* PNG_WRITE_SUPPORTED */