// version 1.0.0  14Feb2018
// source file for MyMatrix class
#include <stdio.h>
#include <new>
#include "mymatrix.h"
#include "stplugin.h"

#define M(i,j)  *(mat+(i)*c+j) 

// Constructor
//   creates matrix and and initializes elements to zero 
// Notes: matrices are long vectors with row-major storage
//    The i,j element of an r x c matrix is 
//    the (i-1)*r + (j-1) element of the of the vector
//    under C-style zero-base indexing
//
MyMatrix::MyMatrix(ST_int rows, ST_int cols) : r(rows), c(cols) 
{
    ST_int i, j, TotalSize ;

    TotalSize = r * c ;
    try {
        mat = new ST_double[TotalSize];
        for(i=0; i<r; i++) {
            for(j=0; j<c; j++) {
                M(i,j) = 0.0 ; 
            }
        }
        rc  = 0 ;
    }
    catch (std::bad_alloc) {
        rc = 909 ;
    }
}

// Destructor deallocates matrix
MyMatrix::~MyMatrix() 
{
    if (mat) {
        delete [] mat ;
    }
}

// Divide each element by val
void MyMatrix::DivideByScalar(ST_double val) 
{
    ST_int  i, j ;

    for(i=0; i<r; i++) {
        for(j=0; j<c; j++) {
            M(i,j) /= val ;  
        }
    }
}

// Copy matrix object values to Stata matrix smname
ST_retcode MyMatrix::CopyCtoStataMatrix(char *smname) 
{
    ST_int      i, j; 
    ST_retcode  rc_st ;
    ST_double   val ;
    char        msg[80] ;

    rc_st = 0 ;
    for(i=0; i<r; i++) {
        for(j=0; j<c; j++) {
            val = GetValue(i, j) ;
            rc_st = SF_mat_store(smname, (i+1) , (j+1), val ) ;
            if(rc_st>0) {
                sprintf(msg, "cannot access Stata matrix %s\n", smname) ;
                SF_error(msg) ;
                return(rc_st) ;
            }
        }
    }
    return(rc_st) ;
}

#undef M