added cluster.split command

[mothur.git] / alignmentdb.cpp

diff --git a/alignmentdb.cpp b/alignmentdb.cpp
index 364ef12588223c0300eccd767aac6f17e6630aa1..9b2977dd91d11604b990860f1b85299aff50b147 100644 (file)
--- a/alignmentdb.cpp
+++ b/alignmentdb.cpp
@@ -25,12 +25,14 @@ AlignmentDB::AlignmentDB(string fastaFileName, string s, int kmerSize, float gap
                m->mothurOut("Reading in the " + fastaFileName + " template sequences...\t");   cout.flush();\r
                \r
                #ifdef USE_MPI  \r
-                       int pid;\r
+                       int pid, processors;\r
                        vector<long> positions;\r
                \r
                        MPI_Status status; \r
                        MPI_File inMPI;\r
                        MPI_Comm_rank(MPI_COMM_WORLD, &pid); //find out who we are\r
+                       MPI_Comm_size(MPI_COMM_WORLD, &processors);\r
+                       int tag = 2001;\r
        \r
                        char inFileName[1024];\r
                        strcpy(inFileName, fastaFileName.c_str());\r
@@ -41,12 +43,14 @@ AlignmentDB::AlignmentDB(string fastaFileName, string s, int kmerSize, float gap
                                positions = setFilePosFasta(fastaFileName, numSeqs); //fills MPIPos, returns numSeqs\r
 \r
                                //send file positions to all processes\r
-                               MPI_Bcast(&numSeqs, 1, MPI_INT, 0, MPI_COMM_WORLD);  //send numSeqs\r
-                               MPI_Bcast(&positions[0], (numSeqs+1), MPI_LONG, 0, MPI_COMM_WORLD); //send file pos     \r
+                               for(int i = 1; i < processors; i++) { \r
+                                       MPI_Send(&numSeqs, 1, MPI_INT, i, tag, MPI_COMM_WORLD);\r
+                                       MPI_Send(&positions[0], (numSeqs+1), MPI_LONG, i, tag, MPI_COMM_WORLD);\r
+                               }\r
                        }else{\r
-                               MPI_Bcast(&numSeqs, 1, MPI_INT, 0, MPI_COMM_WORLD); //get numSeqs\r
+                               MPI_Recv(&numSeqs, 1, MPI_INT, 0, tag, MPI_COMM_WORLD, &status);\r
                                positions.resize(numSeqs+1);\r
-                               MPI_Bcast(&positions[0], (numSeqs+1), MPI_LONG, 0, MPI_COMM_WORLD); //get file positions\r
+                               MPI_Recv(&positions[0], (numSeqs+1), MPI_LONG, 0, tag, MPI_COMM_WORLD, &status);\r
                        }\r
                \r
                        //read file \r
@@ -73,7 +77,9 @@ AlignmentDB::AlignmentDB(string fastaFileName, string s, int kmerSize, float gap
                                        if (temp.getUnaligned().length() > longest)  { longest = temp.getUnaligned().length()+1; }\r
                                }\r
                        }\r
-               \r
+                       \r
+                       MPI_Barrier(MPI_COMM_WORLD); //make everyone wait - just in case\r
+                       \r
                        MPI_File_close(&inMPI);\r
                \r
        #else\r