// matrix2xpm converts a matrix of 0's and 1's to a black & white xpm file
// filename is an extention 
#ifndef SCREEN_H
#define SCREEN_H

#include <iostream.h>
#include <stdlib.h>
#include <iomanip.h>
#include <fstream.h>
#include <string.h>


class Screen {
 private:
   int *pic;
   int imgwidth, imgheight;
   float xmin,ymin;
   float xsize,ysize;

   int xval(float x);      // Calc X Coordinate in pic
   int yval(float y);      // Calc Y coordinate in pic

 public:
   Screen(int width, int height);              // Constructor
   ~Screen();                     // Destructor
   void matrix2xpm(const char *filename);   // Makes a XPM file with name filename
   void setrect(float, float, float, float);   // sets the rectangle that the picture holds 
   bool plotxy(float x,float y);   //Assigns the value 1 to x,y 
   void clear();                   //Set the array to zeros
};

Screen::Screen(int width, int height){
  imgwidth=width; imgheight=height;
  pic=new int [imgwidth*imgheight];
  clear();
}

Screen::~Screen(){
  delete [] pic;
}

void Screen::matrix2xpm(const char *filename){
  ofstream out(filename);
  char filename2[30];
  strcpy(filename2,filename);

  if (!out){            // on file open error
    cerr << "File could not be opened for output\n";
    exit(1);
  }

  for (int i=0; i<strlen(filename2); i++ ){   // filename2 is filename with . replaced with _ 
    if (filename2[i]=='.')
      filename2[i]='_';
  }
  out << "/* XPM */\n";
  out << "static char * " << filename2 << "[] = {\n";
  out << "\"" << imgwidth << " " << imgheight << " 2 1\",\n";
  out << "\".	c Black\",\n";
  out << "\"+	c White";

  for (int y=0; y<imgheight; y++){
    out << "\",\n\"";
    for (int x=0; x<imgwidth; x++){
      if (pic[x+y*imgwidth]==1)
        out << ".";
      else
        out << "+";
    }
  }
  out << "\"};";
  out.close();
  return;
}

void Screen::setrect(float x, float y, float xs, float ys){   
// sets the rectangle that the picture holds 
  xmin=x;
  ymin=y;
  xsize=xs;
  ysize=ys;
}

int Screen::xval(float x){      // Calc X Coordinate in pic
return int( (x-xmin)*imgwidth/xsize );
}

int Screen::yval(float y){      // Calc Y coordinate in pic
return int( (y-ymin)*imgheight/ysize );
}

bool Screen::plotxy(float x,float y){   //Assigns the value 1 to x,y
  int xc,yc;
  xc=xval(x);
  yc=yval(y);
  if ((xc>=0)&&(yc>=0)&&(xc<imgwidth)&&(yc<imgheight)){
    pic[xc+yc*imgwidth]=1;
    return true;
  } else {
    return false;
  }
}

void Screen::clear(){                   //Set the array to zeros
  for (int i=0; i<imgwidth*imgheight;i++)
    pic[i]=0;
}

#endif