ReadWriteMtx.cpp

Go to the documentation of this file.

#include <mpi.h>
#include <sys/time.h>
#include <iostream>
#include <functional>
#include <algorithm>
#include <vector>
#include <sstream>
#include "CombBLAS/CombBLAS.h"
 
using namespace std;
using namespace combblas;
 
 
 
int main(int argc, char* argv[])
{
    int nprocs, myrank;
    MPI_Init(&argc, &argv);
    MPI_Comm_size(MPI_COMM_WORLD,&nprocs);
    MPI_Comm_rank(MPI_COMM_WORLD,&myrank);
 
    if(argc < 4)
    {
        if(myrank == 0)
        {
            cout << "Usage: ./ReadWriteMtx <InputMTX> <OutputMTX> <0|1> <permute>" << endl;
            cout << "<InputMTX> can have reverse edges missing or with different values, but the reader will choose the edge with maximum value when that happens (common in dna/protein sequence search)" << endl;
            cout << "<OutputMTX> is just the name of the output after such incompatibilities in MTX file are removed and all bidirectional edges now have the same value" << endl;
            cout << "<0|1>: zero or one indexed (for inputs)" << endl;
            cout << "<permute> randomly permute the matrix (default: 0 - meaning false)" << endl;
            
        }
        MPI_Finalize();
        return -1;
    }
    {
        string Aname(argv[1]);
        string Bname(argv[2]);
        bool permute = 0;
    bool index = static_cast<bool>(atoi(argv[3]));
 
    if(myrank ==0)
    {
        cout << "Reading file " << Aname << " that is " << index << " indexed" << endl;  
    }
        if(argc == 5)
        {
            permute = static_cast<bool>(atoi(argv[4]));
            if(myrank == 0)
            {
                if(permute)
                    cout << "Randomly permuting" << endl;
            }
        }
 
    
        SpParMat<int64_t, double, SpDCCols<int64_t,double>> A;
        
        A.ParallelReadMM(Aname, index, maximum<double>());
        if(permute)
        {
            FullyDistVec<int64_t,int64_t> perm;    // get a different permutation
            perm.iota(A.getnrow(), 0);
            perm.RandPerm();
            A(perm, perm, true);    // in-place permute to save memory
        }
            
        A.ParallelWriteMM(Bname,true);
    }
    MPI_Finalize();
    return 0;
}