/* * MATLAB*P visulaiztion routines.  * Created by Jack Holloway, Adnan Sulejmanpasic, & Oskar Bruening * Spring 2002 final project 6.338 */#include "prec.h"#include "PPServer.h"#include "PPPackage.h"#include "PPLogger.h"#include "PPArg.h"#include "enum.h"#include "PPUtils.h"#include <iostream>#include <mpi.h>#include "spyrender.h"#include "surfrender.h"#include "meshrender.h"#include "myblacs.h"#ifdef PPSTATIC #include "PPInit.h"#endif//#define DEBUG_PPVISUAL//#define VERBOSEextern "C" {  void ppsize(PPServer &, PPArgList &, PPArgList &);  void ppspy(PPServer &, PPArgList &, PPArgList &);  void ppsurf(PPServer &, PPArgList &, PPArgList &);  void ppmesh(PPServer &, PPArgList &, PPArgList &);}int ppspy_getData(PPServer &, PPMatrixID, float *&, int, 		  int, int, int, int, int);void ppspy_dumpFile(int, int, int, int, DistLayout *, double *, int);int pp3d_getData(PPServer &, PPMatrixID, float *&, int, 		 int, double, double, double, bool);void pp3d(PPServer &, PPArgList &, PPArgList &, bool);/* gets the size of a matrix and returns it as a normal integer b/c   I couldn't figure out how to do it using the layoot stuff*/extern "C" voidppsize(PPServer &inServer, PPArgList &inArgs, PPArgList &outArgs){    if (inArgs.length() != 1) {    outArgs.addError(BADINPUTARGS, "Number of input args should be 1.");    return;  }      PPMatrixID in = (PPMatrixID) *(inArgs[0]); // input maxtrix  if (!(inServer.mMan)->isValidMatrix(in)) {    outArgs.addError(BADINPUTARGS, "Invalid Matrix in arg 0");    return;  }    outArgs.addNoError();  if( (inServer.sCon)->isRoot())    {      int gRows = (inServer.mMan)->getGRows(in);//global rows      int gCols = (inServer.mMan)->getGCols(in);//global columns      outArgs.add(gRows);      outArgs.add(gCols);      return;    }}extern "C" voidppspy(PPServer &inServer, PPArgList &inArgs, PPArgList &outArgs){  int ret_code, image_w, image_h, sr, sc, er, ec;  int gRows, gCols;  float *pixel_buffer;  pixel_buffer = NULL;  //pixel_buffer = new float[image_w*image_h];  // Adnan generates this, doesn't delete it in his destructor.  // input checking  if (inArgs.length() != 7) {    outArgs.addError(BADINPUTARGS, "Number of input args should be 7.");    return;  }  PPMatrixID in = (PPMatrixID) *(inArgs[0]); // input maxtrix  if (!(inServer.mMan)->isValidMatrix(in)) {    outArgs.addError(BADINPUTARGS, "Invalid Matrix in arg 0");    return;  }  gRows = (inServer.mMan)->getGRows(in);//global rows  gCols = (inServer.mMan)->getGCols(in);//global cols  image_h = *(inArgs[1]);  image_w = *(inArgs[2]);  sr = *(inArgs[3]);  sc = *(inArgs[4]);  er = *(inArgs[5]);  er = er + sr - 1;  ec = *(inArgs[6]);  ec = ec + sc - 1;  /*  if(ec > gCols)    ec = gCols;  if(er > gRows)    er = gRows;  */#ifdef DEBUG_PPVISUAL  if( (inServer.sCon)->isRoot())    {      cout<<"gCols = "<<gCols<<endl;      cout<<"gRows = "<<gRows<<endl;      cout<<"image height = *(inArgs[1])="<<image_h<<endl;      cout<<"image width = *(inArgs[2])="<<image_w<<endl;      cout<<"sr = *(inArgs[3])="<<sr<<endl;      cout<<"sc = *(inArgs[4])="<<sc<<endl;      cout<<"er = *(inArgs[5])="<<er<<endl;      cout<<"ec = *(inArgs[6])="<<ec<<endl;    }#endif  if (image_w < 1) {    outArgs.addError(BADINPUTARGS, "Integer expected in arg 1(image HEIGHT)");    return;  }  if (image_h < 1) {    outArgs.addError(BADINPUTARGS, "Integer expected in arg 2(image WIDTH)");    return;  }  if (sr < 0) {    outArgs.addError(BADINPUTARGS, "Integer expected in arg 3(zoom start row)");    return;  }  if (sc < 0) {    outArgs.addError(BADINPUTARGS, "Integer expected in arg 4(zoom start column)");    return;  }  if (er < 1) {    outArgs.addError(BADINPUTARGS, "Integer expected in arg 5(zoom rect HEIGHT)");    return;  }  if (ec < 1) {    outArgs.addError(BADINPUTARGS, "Integer expected in arg 6(zoom rect WIDTH)");    return;  }    int myid = (inServer.sCon)->processorID();  /* Make the important call */  ret_code = ppspy_getData(inServer, in, pixel_buffer, image_w, image_h, 			   sr, sc, er,ec);      if(ret_code == -1)    {      outArgs.addError(BADINPUTARGS, "Data not distributed correctly.");      return;    }  else if(ret_code == -2)    {      outArgs.addError(BADINPUTARGS, "Error finding data distribution on node.");      return;    }  else if(ret_code == 0)    {      // should be all set      outArgs.addNoError();      if( (inServer.sCon)->isRoot())	{	  // return the pixel buffer as a column vector (rgb values for each pixel)	  // messed up.	  if(pixel_buffer != NULL) {	    //cout<<"Root node here: I'm sending the pixel buffer to MATLAB.\n";	    outArgs.add((float *)pixel_buffer,image_w*image_h*3,1);	  }	  else	    cerr<<"This is BAD NEWS: Pixel buffer NULL.\n";	  	}          }}int ppspy_getData(PPServer &inServer, PPMatrixID in, 		  float *&pixel_buffer, int imgwidth, int imgheight,		  int sr, int sc, int er, int ec) {  // get some useful information  int nprocs = (inServer.sCon)->numProcessors();  int lRows = (inServer.mMan)->getLRows(in);//local rows  int lCols = (inServer.mMan)->getLCols(in);//local columns  int lSize = (inServer.mMan)->getLSize(in);//local size  int gRows = (inServer.mMan)->getGRows(in);//global rows  int gCols = (inServer.mMan)->getGCols(in);//global columns  int gSize = (inServer.mMan)->getGSize(in);//local size  int MB = (inServer.mMan)->getMB(in);  int NB = (inServer.mMan)->getNB(in);  int myid = (inServer.sCon)->processorID();  PPDistType dist = (inServer.mMan)->getDist(in);  DistLayout *data_layout;  int mystartrow, myendrow, mystartcolumn, myendcolumn;    // Get info about the context we are using    data_layout = (inServer.mMan)->getDistWindows(in,nprocs, myid);  if(data_layout == NULL) // something bad happened    return -2;  /*mystartrow = (inServer.mMan)->getStartRow(in,nprocs);    mystartcolumn = (inServer.mMan)->getStartCol(in,nprocs);        myendrow = mystartrow + lRows - 1;    myendcolumn = mystartcolumn + lCols - 1;  */        // data real  if ((inServer.mMan)->isReal(in)) {        ELTYPE *raw = (ELTYPE *) (inServer.mMan)->getData(in);    #ifdef DEBUG_PPVISUAL    ppspy_dumpFile(gRows, gCols, lRows, lCols, data_layout, raw, myid);#ifdef VERBOSE    cout<<"------- ppspy: node "<<myid<<" ---------\nlRows: "<<lRows	<<" lCols: "<<lCols<<"\ngRows: "<<gRows	<<" gCols: "<<gCols<<endl;#endif#endif        // send stuff to Adnan's code...    // He'll need the current proc's rank,     // it's beginning and ending columns and    // rows, the actual data array, the global size (column * row)    #ifdef _RENDER_H_        SpyRender ppspyRenderer = SpyRender(imgwidth, imgheight, 					myid, raw,					sr, sc, 					er, ec,					data_layout);        // need to get the root node to write the     // pixel buffer into the int *pixel_buffer buffer.    pixel_buffer = ppspyRenderer.draw();    if(((inServer.sCon)->isRoot()) && (pixel_buffer == NULL)) {      cerr<<"BAD NEWS, son, this here pixel buffer ain't nothin' but NULL.\n";    }            #endif  }  // data complex  /*    else {        CELTYPE *raw = (CELTYPE *) (inServer.mMan)->getData(in);        // we need this transposed!  because thus sath FFTW    for (int j=0; j<lCols; j++) {    for (int i=0; i<lRows; i++) {    data[i*lCols+j].re = raw[j*lRows+i].real();    data[i*lCols+j].im = raw[j*lRows+i].imag();    }    }    }  */    delete data_layout;  return 0;}void ppspy_dumpFile(int gRows, int gCols, int lRows, int lCols, DistLayout *data_layout, double *raw, int rank){  ofstream out;  char file_name[80];  int rows, cols, offset, r1, c1, r2, c2, roffset, coffset;  sprintf(file_name, "debug_matrix_%i.dat", rank);  out.open(file_name);  out<<"%Info for processor "<<rank<<endl;  for(int block = 0; block < data_layout->blocks; block ++)    {      r1 = data_layout->block_data[block*6 + 0];      r2 = data_layout->block_data[block*6 + 2];      c1 = data_layout->block_data[block*6 + 1];      c2 = data_layout->block_data[block*6 + 3];      rows = r2 - r1 + 1;      cols = c2 - c1 + 1;      roffset = data_layout->block_data[block*6 + 4];      coffset = data_layout->block_data[block*6 + 5];            if(rows > 1 && cols > 1)	{	  cout<<"processor "<<rank<<" dumping data for it's "	      <<block<<" block: ("<<r1<<", "<<c1<<") -> ("	      <<r2<<", "<<c2<<") :: offset = "<<offset<<".\n";	  out<<"%Block "<<block<<" from ("<<r1<<", "<<c1<<") to ("	     <<r2<<", "<<c2<<")"<<endl;	  for(int j=0; j<cols; j++)	    {	      for(int i = 0; i<rows; i++) 		{		  out<<raw[(j+coffset)*lCols+(i+roffset)]<<" ";		}	      out<<endl;	    }	}      else	cout<<"processor "<<rank<<" skipping dump of it's "<<block<<" block.\n";          }  out.close();}extern "C" voidppmesh(PPServer &inServer, PPArgList &inArgs, PPArgList &outArgs){  //cout<<"::::::::::::::::::::::::ppmesh\n";  pp3d(inServer, inArgs, outArgs, false);}/* entry for matlab*p surface command */extern "C" voidppsurf(PPServer &inServer, PPArgList &inArgs, PPArgList &outArgs){  //cout<<"::::::::::::::::::::::::ppsurf\n";  pp3d(inServer, inArgs, outArgs, true);}voidpp3d(PPServer &inServer, PPArgList &inArgs, PPArgList &outArgs, bool surf){  int ret_code, image_w, image_h;  double theta, phi, zoom;  int gRows, gCols;  float *pixel_buffer;  pixel_buffer = NULL;  if (inArgs.length() != 6) {    outArgs.addError(BADINPUTARGS, "Number of input args should be 6.");    return;  }  PPMatrixID in = (PPMatrixID) *(inArgs[0]); // input maxtrix  if (!(inServer.mMan)->isValidMatrix(in)) {    outArgs.addError(BADINPUTARGS, "Invalid Matrix in arg 0");    return;  }  gRows = (inServer.mMan)->getGRows(in);//global rows  gCols = (inServer.mMan)->getGCols(in);//global rows  image_w = *(inArgs[1]);  image_h = *(inArgs[2]);  theta = *(inArgs[3]);  phi = *(inArgs[4]);  zoom = *(inArgs[5]);#ifdef DEBUG_PPVISUAL  if( (inServer.sCon)->isRoot())    {      cout<<"image width = *(inArgs[1])="<<image_w<<endl;      cout<<"image height = *(inArgs[2])="<<image_h<<endl;      cout<<"theta = *(inArgs[3])="<<theta<<endl;      cout<<"phi = *(inArgs[4])="<<phi<<endl;      cout<<"zoom = *(inArgs[5])="<<zoom<<endl;    }#endif  if (image_w < 1) {    outArgs.addError(BADINPUTARGS, "Integer expected in arg 1(IMAGE WIDTH)");    return;  }  if (image_h < 1) {    outArgs.addError(BADINPUTARGS, "Integer expected in arg 2(IMAGE HEIGHT)");    return;  }    if (zoom < 0) {    outArgs.addError(BADINPUTARGS, "POSITIVE integer expected in arg 5(ZOOM)");    return;  }  /* Make the important call */  if(surf)    ret_code = pp3d_getData(inServer, in, pixel_buffer, image_w, image_h, 			    theta, phi, zoom, true);  else    ret_code = pp3d_getData(inServer, in, pixel_buffer, image_w, image_h, 			    theta, phi, zoom, false);  // the last argument is tells us that we need to be doing a surface  // rendering.  if(ret_code == -1)    {      outArgs.addError(BADINPUTARGS, "Data not distributed correctly.");      return;    }  else if(ret_code == -2)    {      outArgs.addError(BADINPUTARGS, "Error finding data distribution on node.");      return;    }  else if(ret_code == 0)    {      // should be all set      outArgs.addNoError();      if( (inServer.sCon)->isRoot())	{	  if(pixel_buffer != NULL) {	    //cout<<"Matlab is getting : "<<pixel_buffer<<endl;	    outArgs.add((float *)pixel_buffer,image_w*image_h*3,1);	  }	  else	    cerr<<"This is BAD NEWS: Pixel buffer NULL.\n";	  	}          }}int pp3d_getData(PPServer &inServer, PPMatrixID in, 		  float *&pixel_buffer, int imgheight, int imgwidth,		  double theta, double phi, double zoom, bool surf) {  int nprocs = (inServer.sCon)->numProcessors();  int lRows = (inServer.mMan)->getLRows(in);//local rows  int lCols = (inServer.mMan)->getLCols(in);//local columns  int lSize = (inServer.mMan)->getLSize(in);//local size  int gRows = (inServer.mMan)->getGRows(in);//global rows  int gCols = (inServer.mMan)->getGCols(in);//global columns  int gSize = (inServer.mMan)->getGSize(in);//local size  int MB = (inServer.mMan)->getMB(in);  int NB = (inServer.mMan)->getNB(in);  int myid = (inServer.sCon)->processorID();  PPDistType dist = (inServer.mMan)->getDist(in);  DistLayout *data_layout;  int mystartrow, myendrow, mystartcolumn, myendcolumn;    data_layout = (inServer.mMan)->getDistWindows(in,nprocs, myid);  if(data_layout == NULL) // something bad happened    return -2;  if ((inServer.mMan)->isReal(in)) {        ELTYPE *raw = (ELTYPE *) (inServer.mMan)->getData(in);    #ifdef DEBUG_PPVISUAL    ppspy_dumpFile(gRows, gCols, lRows, lCols, data_layout, raw, myid);#ifdef VERBOSE    cout<<"------- ppspy: node "<<myid<<" ---------\nlRows: "<<lRows	<<" lCols: "<<lCols<<"\ngRows: "<<gRows	<<" gCols: "<<gCols<<endl;#endif#endif    #ifdef _SURF_RENDER_H_#ifdef _MESH_RENDER_H_    if(surf) {      SurfRender ppsurfRenderer = SurfRender(imgwidth, imgheight, 					     myid, raw, 					     gRows, gCols,					     data_layout, 					     theta, phi, zoom);            pixel_buffer = ppsurfRenderer.draw();    }    else {      MeshRender ppmeshRenderer = MeshRender(imgwidth, imgheight, 					     myid, raw, 					     gRows, gCols,					     data_layout, 					     theta, phi, zoom);            pixel_buffer = ppmeshRenderer.draw();    }    if(((inServer.sCon)->isRoot()) && (pixel_buffer == NULL)) {      cerr<<"BAD NEWS, son, this here pixel buffer ain't nothin' but NULL.\n";    }#endif#endif  }  delete data_layout;  return 0;}///////////////////////////////////// INITIALIZATION///////////////////////////////////#ifndef PPSTATICextern "C" PPError ppinitialize(PPPackage *pack, PPLogger *Log) {#elseextern "C" PPError PPVisual_initialize(PPPackage *pack, PPLogger *Log) {#endif    pack->addFunc("ppspy",&ppspy);  pack->addFunc("ppsurf",&ppsurf);  pack->addFunc("ppmesh",&ppmesh);  pack->addFunc("ppsize",&ppsize);    return NOERR;}