dumpKernelMatrix.cpp

Go to the documentation of this file.

#include <sys/types.h>
#include <sys/mman.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <math.h>
#include <errno.h>
#include <fstream>
#include <Common/Exception.h>
#include <Common/Util.h>

using namespace std;
using namespace Util;


int main(int argc, char *argv[]) {
  try {
    Exception::Assert( argc==3, "Usage: %s <matrix id> <dump file>\n", argv[0]);
    char *matrix_id = argv[1];
    char *out_file = argv[2];
    
    // open shared memory
    int mdesc = shm_open(matrix_id,O_RDONLY,0);
    Exception::Assert(mdesc>=0, "Error while opening shared memory: %s", strerror(errno));

    // get size
    struct stat infos;
    int ret = fstat(mdesc,&infos);
    Exception::Assert(ret==0, "Error while retrieving size of shared memory: %s", strerror(errno));

    int tot_space = infos.st_size;
    uint size = (uint)((-1. + sqrt(1.+8.*(double)(tot_space/sizeof(float))))/2.);
    cout << "Total allocated space: " << tot_space << endl;
    cout << "Matrix rows: " << size << endl;

    // map to memory
    float* matrix = (float*)mmap(NULL,tot_space,PROT_READ,MAP_SHARED,mdesc,0);
    Exception::Assert( matrix!=(float*)-1, "Error while mapping shared memory: %s", strerror(errno));
    
    // get matrix
    float* buffer = new float[size*size];
    cout << "Get matrix: " << flush;
    for(uint i=0; i<size; i++) 
      for(uint l=0; l<=i; l++ ) 
        buffer[i*size+l] = buffer[l*size+i] = matrix[l+i*(i+1)/2];      
    
    // write matrix
    ofstream ofs(out_file); Exception::Assert(ofs, "Error while opening file %s", out_file);

    for(uint i=0; i<size; i++) {
      outProgress(i,100);
      for(uint l=0; l<size; l++) {
        ofs << buffer[i*size+l];
        if( l<size-1 )
          ofs << " ";
      }
      ofs << endl;
    }
    endProgress(size);

    ofs.close();

    // unmap memory
    ret = munmap(matrix,tot_space);
    Exception::Assert( ret==0, "Error while unmapping shared memory: %s", strerror(errno));
  }
  catch(Exception *e) {
    cerr << e->GetMessage() << endl;
    return 1;
  }

  return 0;
}

---