PngTool.h

Go to the documentation of this file.

#ifndef  __JBXL_CPP_PNG_TOOl_H_
#define  __JBXL_CPP_PNG_TOOl_H_
 
#include "Gdata.h"
#include "xtools.h"
 
/*
#ifndef HAVE_PNG_H
#ifndef DISABLE_PNG
#define DISABLE_PNG
#endif
#endif
*/
 
#ifdef  DISABLE_PNG
#undef  ENABLE_PNG
#endif
 
 
#ifdef ENABLE_PNG
 
#include <png.h>
 
/*
#ifdef WIN32
#if defined(_DEBUG)
#pragma  comment(lib, "libpng16d.lib")
#else
#pragma  comment(lib, "libpng16.lib")
#endif
#endif
*/
 
/*
type
#define PNG_COLOR_TYPE_GRAY 0
#define PNG_COLOR_TYPE_PALETTE    (PNG_COLOR_MASK_COLOR | PNG_COLOR_MASK_PALETTE)
#define PNG_COLOR_TYPE_RGB        (PNG_COLOR_MASK_COLOR)
#define PNG_COLOR_TYPE_RGB_ALPHA  (PNG_COLOR_MASK_COLOR | PNG_COLOR_MASK_ALPHA)
#define PNG_COLOR_TYPE_GRAY_ALPHA (PNG_COLOR_MASK_ALPHA)
// aliases 
#define PNG_COLOR_TYPE_RGBA  PNG_COLOR_TYPE_RGB_ALPHA
#define PNG_COLOR_TYPE_GA    PNG_COLOR_TYPE_GRAY_ALPHA
*/
 
 
#ifndef PNG_SIGNATURE_SIZE
#define  PNG_SIGNATURE_SIZE  8
#endif
 
 
//
namespace jbxl {
 
 
 
class PNGImage 
{
public:
    int     xs;
    int     ys;
    int     col;
    int     length;
    int     state;
 
    uByte   type;
    uByte*  gp;         // Bitmap
 
private:
    png_structp strct;
    png_infop   info;
 
public:
    PNGImage(void)  { init();}
    virtual ~PNGImage(void) {}
 
    void    init(void);                 
    bool    isNull(void);               
    void    clear(void);                
    void    fill(uByte v=(uByte)0);     
    void    free(void);                 
 
    uByte&  point(int x, int y, int c)   { return gp[col*(y*xs + x) + c];}
    void    getm(int x, int y, int c);
    void    setzero(int x, int y, int c) { getm(x, y, c); clear();}
 
    void    readData(FILE* fp);
    int     writeData(FILE* fp);
};
 
 
 
 
PNGImage    readPNGFile (const char* fname);
PNGImage    readPNGData (FILE* fp);
int         writePNGFile(const char* fname, PNGImage* png);
int         writePNGData(FILE* fp, PNGImage* png);
 
//int        isPNGHeader(Buffer buf);
 
// template <typename T>  MSGraph<T> PNGImage2MSGraph<T>(PNGImage  png)
// template <typename T>  PNGImage  MSGraph2PNGImage(MSGraph<T> vp)
 
 
template <typename T>  MSGraph<T> PNGImage2MSGraph(PNGImage png)
{
    MSGraph<T> vp;
 
    if (png.isNull()) {
        vp.state = JBXL_GRAPH_NODATA_ERROR;
        return vp;
    }
 
    vp.set(png.xs, png.ys, png.col);
    if (vp.isNull()) {
        vp.state = JBXL_GRAPH_MEMORY_ERROR;
        return vp;
    }
    //
    if      (png.col==4) vp.color = GRAPH_COLOR_BGRA;
    else if (png.col==3) vp.color = GRAPH_COLOR_BGR;
    else if (png.col==2) vp.color = GRAPH_COLOR_GA;
    else if (png.col==1) vp.color = GRAPH_COLOR_GRAY;
    else {
        vp.free();
        vp.init();
        vp.state = JBXL_GRAPH_IVDCOLOR_ERROR;
        return vp;
    }
 
    for (int k=0; k<png.col; k++) {
        int zp = k*png.xs*png.ys;
        for (int j=0; j<png.ys; j++) {
            int yp = zp + j*png.xs;
            for (int i=0; i<png.xs; i++) {
                vp.gp[yp + i] = (T)png.point(i, png.ys-1-j, k);
            }
        }
    }
 
    return vp;
}
 
 
template <typename T>  PNGImage  MSGraph2PNGImage(MSGraph<T> vp)
{
    PNGImage png;
    png.init();
 
    if (vp.isNull()) {
        png.state = JBXL_GRAPH_NODATA_ERROR;
        return png;
    }
 
    png.setzero(vp.xs, vp.ys, vp.zs);
    if (png.isNull()) {
        png.init();
        png.state = JBXL_GRAPH_MEMORY_ERROR;
        return png;
    }
 
    if (vp.color==GRAPH_COLOR_UNKNOWN) {
        if      (vp.zs==1) vp.color = GRAPH_COLOR_GRAY;
        else if (vp.zs==2) vp.color = GRAPH_COLOR_GA;
        else if (vp.zs==3) vp.color = GRAPH_COLOR_RGB;
        else if (vp.zs==4) vp.color = GRAPH_COLOR_RGBA;
    }
    if      (vp.color==GRAPH_COLOR_GRAY) png.type = PNG_COLOR_TYPE_GRAY;
    else if (vp.color==GRAPH_COLOR_GA)   png.type = PNG_COLOR_TYPE_GA;
    else if (vp.color==GRAPH_COLOR_RGB)  png.type = PNG_COLOR_TYPE_RGB;
    else if (vp.color==GRAPH_COLOR_RGBA) png.type = PNG_COLOR_TYPE_RGBA;
    else {
        png.free();
        png.init();
        png.state = JBXL_GRAPH_IVDCOLOR_ERROR;
        return png;
    }
 
    for (int k=0; k<png.col; k++) {
       int zp = k*png.xs*png.ys;
        for (int j=0; j<png.ys; j++) {
            int yp = zp + j*png.xs;
            for (int i=0; i<png.xs; i++) {
                png.point(i, j, k) = (uByte)vp.gp[yp + i];
            }
        }
    }
 
    png.state = JBXL_NORMAL;
    return png;
}
 
 
}       // namespace
 
 
#endif  // ENABLE_PNG
 
#endif  // __JBXL_CPP_PNG_TOOl_H_