IO.h

#ifndef _IO_SPGEMM_H
#define _IO_SPGEMM_H

#include "Triple.h"
#include "CSC.h"

#define READBUFFER (512 * 1024 * 1024)  // in MB

template <typename IT, typename NT>
int ReadBinary(string filename, CSC<IT,NT> & csc)
{
    FILE * f = fopen(filename.c_str(), "r");
    if(!f)
    {
        cerr << "Problem reading binary input file" << filename << endl;
        return -1;
    }
    IT m,n,nnz;
    fread(&m, sizeof(IT), 1, f);
    fread(&n, sizeof(IT), 1, f);
    fread(&nnz, sizeof(IT), 1, f);
    
    if (m <= 0 || n <= 0 || nnz <= 0)
    {
        cerr << "Problem with matrix size in binary input file" << filename << endl;
        return -1;
    }
    double start = omp_get_wtime( );
    cout << "Reading matrix with dimensions: "<< m << "-by-" << n <<" having "<< nnz << " nonzeros" << endl;
    
    IT * rowindices = new IT[nnz];
    IT * colindices = new IT[nnz];
    NT * vals = new NT[nnz];
    
    size_t rows = fread(rowindices, sizeof(IT), nnz, f);
    size_t cols = fread(colindices, sizeof(IT), nnz, f);
    size_t nums = fread(vals, sizeof(NT), nnz, f);
    
    if(rows != nnz || cols != nnz || nums != nnz)
    {
        cerr << "Problem with FREAD, aborting... " << endl;
        return -1;
    }
    double end = omp_get_wtime( );
    // printf("start = %.16g\nend = %.16g\ndiff = %.16g\n", start, end, end - start);
    printf("Converting matrix data from binary to COO fromat takes %.16g seconds.\n", end - start);

    fclose(f);
    
    csc = *(new CSC<IT,NT>(rowindices, colindices, vals , nnz, m, n));
    
    delete [] rowindices;
    delete [] colindices;
    delete [] vals;
    return 1;
}

template <typename IT, typename NT>
int ReadASCII(string filename, CSC<IT,NT> & csc, IT cluster_sz = 0)
{
    bool isSymmetric = false;
    double start = omp_get_wtime( );
    ifstream infile(filename.c_str());
    char line[256];
    char c = infile.get();
    while(c == '%')
    {
        infile.getline(line,256);
        if (strstr(line, "symmetric")) {
            isSymmetric = true;
            cout << "Matrix is symmetric" << endl;
        }
        c = infile.get();
    }
    infile.unget();

    infile.getline(line,256);
    IT m,n,nnz;
    if (typeid(IT) == typeid(int)) {
        sscanf(line, "%d %d %d", &m, &n, &nnz);
    }
    else if (typeid(IT) == typeid(long long int)) {
        sscanf(line, "%ld %ld %ld", &m, &n, &nnz);
    }
    else {
        sscanf(line, "%ld %ld %ld", &m, &n, &nnz);
    }

//    cout << "Reading matrix with dimensions: "<< m << "-by-" << n <<" having "<< nnz << " nonzeros" << endl;
    
    if (isSymmetric) {
        nnz *= 2;
    }

//    cout << "come so far..." << endl;
    Triple<IT,NT> * triples = new Triple<IT,NT>[nnz];
//    cout << "mem allocation dome..." << endl;
    if (infile.is_open())
    {
        IT cnz = 0;	// current number of nonzeros
        while (! infile.eof() && cnz < nnz)
        {
            infile.getline(line,256);
            char *ch = line;
//            cout << "line: " << line << endl;
            triples[cnz].row = (IT)(atoi(ch));
//            ch = strchr(ch, ' ');
//            ch++;
            ch = strpbrk(ch, " \t");
            while (*ch == ' ' || *ch == '\t') ++ch;
            triples[cnz].col = (IT)(atoi(ch));
//            ch = strchr(ch, ' ');
            ch = strpbrk(ch, " \t");
            if (ch != NULL) {
//                ch++;
                while (*ch == ' ' || *ch == '\t') ++ch;
                /* Read third word (value data)*/
//                triples[cnz].val = (NT)(atoi(ch));
//                triples[cnz].val = (NT)(atof(ch));
                triples[cnz].val = (NT)(atof(ch) + 1.0);
//                ch = strchr(ch, ' ');
            }
            else {
                triples[cnz].val = 1.0;
            }
            // infile >> triples[cnz].row >> triples[cnz].col >> triples[cnz].val;	// row-col-value
            triples[cnz].row--;
            triples[cnz].col--;
            if (isSymmetric) {
                if (triples[cnz].col != triples[cnz].row) {
                    cnz++;
                    triples[cnz].col = triples[cnz - 1].row;
                    triples[cnz].row = triples[cnz - 1].col;
                    triples[cnz].val = triples[cnz - 1].val;
                }
                else {
                    nnz--;
                }
            }
            ++cnz;
//            if(cnz % 1000000 == 0) cout << "read " << (cnz / 1000000) << "M entries" << endl;
        }
//        IT rid, cid;
//        NT v;
//        while (infile >> rid >> cid >> v)
//        {
//            triples[cnz].row = rid;
//            triples[cnz].col = cid;
//            triples[cnz].val = v;
//
//            triples[cnz].row--;
//            triples[cnz].col--;
//
//            if (isSymmetric) {
//                if (triples[cnz].col != triples[cnz].row) {
//                    cnz++;
//                    triples[cnz].col = triples[cnz - 1].row;
//                    triples[cnz].row = triples[cnz - 1].col;
//                    triples[cnz].val = triples[cnz - 1].val;
//                }
//                else {
//                    nnz--;
//                }
//            }
//            ++cnz;
////            if(cnz % 1000000 == 0) cout << "read " << (cnz / 1000000) << "M entries" << endl;
//        }
//        if(cnz != nnz) cout << "nnz: " << nnz << ", cnz: " << cnz << endl;
        assert(cnz == nnz);
    }
    
    double end = omp_get_wtime( );
    // printf("start = %.16g\nend = %.16g\ndiff = %.16g\n", start, end, end - start);
    printf("Converting matrix data from ASCII to COO format: %.16g seconds\n", end - start);
    printf("Input Matrix: Rows = %d, Columns= %d, nnz = %d, cluster_sz = %d\n", m, n, nnz, cluster_sz);

//    cout << "cluster_sz: " << cluster_sz << endl;
    cout << "Converting to csc ... " << endl << endl;
    if(cluster_sz != 0) {
      if(m%cluster_sz != 0) {
        m = ((m / cluster_sz) + 1) * cluster_sz;
        n = m;
      }
    }
    csc= *(new CSC<IT,NT>(triples, nnz, m, n));
    csc.totalcols = n;
    delete [] triples;
    return 1;
}

#endif