//===========================================================================
// pnggl2d.hpp : PNG framebuffer
//===========================================================================

#ifndef _PNGGL2D_
#define _PNGGL2D_

#include <fstream>

#include <RGB.h>
#include <pngencoder.h>
#include <bitmapimage.h>
using namespace std ;
using namespace Colosseum ;

class pngGL2d
{
 public:

  BitmapImage FB;
  int NumPixels;            // DIMENSIONS
  UBYTE1 fgR, fgG, fgB;     // CURRENT FOREGROUND COLOR
  UBYTE1 bgR, bgG, bgB;     // CURRENT BACKGROUND COLOR

  // THE CONSTRUCTOR: REQUIRES THE DESIRED DIMENSIONS OF THE FRAMEBUFFER
  pngGL2d(int width=0, int height=0)
  {
    NumPixels = width*height;       // STORE NUMPIXELS IN FB
    if (NumPixels != 0) {
      FB.setSize(-1, 24, width, height);
      SetBackground(0,0,0);           // DEFAULT BACKGROUND COLOR (BLACK)
      SetForeground(1,1,1);           // DEFAULT FOREGROUND COLOR (WHITE)
      ClearColor();
    }
  };

  // THE DESTRUCTOR: FREE THE FRAMEBUFFER
  ~pngGL2d()
  {
  };

  // CLEAR THE FRAMEBUFFER WITH THE CURRENT BACKGROUND COLOR
  void ClearColor()
  {
    for (int i=0; i<FB.getHeight(); i++)
    { 
      UBYTE1 *p = FB[i]; // a row
      for (int j=0; j<FB.getWidth(); j++) {
        p[(3*j)+2]=bgR; p[(3*j)+1]=bgG; p[3*j]=bgB;
      }
    }
  }

  void SetSize(int w, int h)
  {
    FB.setSize(-1,24,w,h);
    NumPixels = w*h;
  }
  void GetSize(int *w, int *h) const
  {
    *w = FB.getWidth();
    *h = FB.getHeight();
  }

  // SET CURRENT FOREGROUND AND BACKGROUND COLORS (normalized values in [0,1])
  void SetBackground(float r, float g, float b)
    { bgR=(UBYTE1)(r*255); bgG=(UBYTE1)(g*255); bgB=(UBYTE1)(b*255); }
  void SetForeground(float r, float g, float b)
    { fgR=(UBYTE1)(r*255); fgG=(UBYTE1)(g*255); fgB=(UBYTE1)(b*255); }

  // PIXEL INDEX GET/SET ROUTINES
  int GetPixelIndex(int x, int y) 
  { return( y * FB.getWidth() + x ); }

  void SetPixelIndex(int i)
  { 
    //assert 0;
    // UBYTE1 *p = FB[i];
   // p[1]=fgR; p[2]=fgG; p[3]=fgB;
  }

  // SET PIXEL ROUTINE: SETS THE PIXEL AT (X,Y) TO THE CURRENT FOREGROUND COLOR
  void SetPixel(int x, int y)
  {
    FB[y][x*3+2] = fgR;
    FB[y][x*3+1] = fgG;
    FB[y][x*3] = fgB;
  }
  void SetPixel(int x, int y, float r, float g, float b)
  {
    SetForeground(r, g, b);
    SetPixel(x, y);
  }
  void GetPixel(int x, int y, float *r, float *g, float *b)
  {
    const float f = 1.0f/255.0f;
    *r = float(FB[y][x*3+2])*f;
    *g = float(FB[y][x*3+1])*f;
    *b = float(FB[y][x*3])*f;
  }

  // WRITES THE FRAMEBUFFER TO A PNG FILE
  void Write(const char* FileName)
  {
    PngEncoder encoder;
    encoder.writeImageFile(FileName, FB);
  };
};

#endif