added 7

2020-03-16 19:56:07 -06:00
parent 1d67b1ff76
commit 4141bf4195
40 changed files with 3308 additions and 0 deletions
--- a/HW7/main.cpp
+++ b/HW7/main.cpp
@@ -0,0 +1,130 @@
+#include<iostream>
+#include<mpi.h>
+#include<stdlib.h>
+#include<unistd.h>
+#include<time.h>
+#include<queue>
+#define MCW MPI_COMM_WORLD
+#define ANY MPI_ANY_SOURCE
+#define IGNORE MPI_STATUS_IGNORE
+#define BLACK 0
+#define WHITE 1
+#define JOB 1
+#define TOKEN 2
+#define CREATED 3
+#define COMPLETED 4
+
+using namespace std;
+
+int main(int argc, char **argv) {
+    int rank, size;
+    int jobFlag, tokenFlag;
+    int tasksCreated = 0;
+    int tasksCompleted = 0;
+    int taskLimit;
+    int tokenColor = WHITE;
+    int myColor = WHITE;
+    int newTask;
+    bool hasToken = false;
+    queue<int> myTasks;
+    MPI_Status status;
+    MPI_Request request;
+
+    MPI_Init(&argc, &argv);
+    MPI_Comm_rank(MCW, &rank);
+    MPI_Comm_size(MCW, &size);
+
+    srand(time(NULL)+rank);
+    taskLimit = rand() % 1025 + 1024; //Set the limit to some number between 1024 and 2048
+
+    if(rank == 0) {
+        hasToken = true;
+        int job = rand() % 3 + 1;
+        tasksCreated++;
+        MPI_Send(&job, 1, MPI_INT, rand() % 4, JOB, MCW); //Send out the initial job
+    }
+
+    while(1) {
+        MPI_Probe(ANY,JOB,MCW,&status);
+        MPI_Recv(&newTask,1,MPI_INT,ANY,JOB,MCW,&status);
+        myTasks.push(newTask);
+        //MPI_IProbe(ANY, JOB, MCW, &jobFlag, &status); //Check to see if a job is available
+//        if(jobFlag) {
+//            MPI_Irecv(&newTask, 1, MPI_INT, ANY, JOB, MCW, &request);
+//            myTasks.push(newTask);
+//        }
+        if(newTask == -1) { //If the stop signal is received, send my info to process 0 and break out of the loop
+            MPI_Send(&tasksCreated, 1, MPI_INT, 0, CREATED, MCW);
+            MPI_Send(&tasksCompleted, 1, MPI_INT, 0, COMPLETED, MCW);
+            break;
+        }
+        if(myTasks.size() > 16) { //If my task queue is greater than size 16, send off 2 tasks
+            for(int i = 0; i < 2; i++) {
+                int task = myTasks.front();
+                myTasks.pop();
+                int dest = rand() % 4;
+                while(dest == rank) {dest = rand() % 4;}
+                if(dest < rank) {myColor = BLACK;}
+                MPI_Send(&task, 1, MPI_INT, dest, JOB, MCW);
+            }
+        }
+        if(!myTasks.empty()) { //Perform a task
+            int num = myTasks.front();
+            myTasks.pop();
+            sleep(static_cast<double>(num)/5);
+            tasksCompleted++;
+        }
+        if(tasksCreated < taskLimit) { //Create more tasks
+            for(int i = 0; i < rand() % 3 + 1; i++) {
+                myTasks.push(rand() % 3 + 1);
+                tasksCreated++;
+            }
+        }
+        if(myTasks.empty()) { //Check the token information
+            if(hasToken) {
+                cout<<"Rank: "<<rank<<", token: "<<(tokenColor==BLACK ? "Black" : "White")<<endl;
+                int dest = rank + 1 < size ? rank + 1 : 0;
+                if(rank == 0) {tokenColor = WHITE;}
+                MPI_Send(&tokenColor, 1, MPI_INT, dest, TOKEN, MCW);
+                hasToken = false;
+                myColor = WHITE;
+            }
+            else {
+                int src = rank - 1 >= 0 ? rank - 1 : size - 1;
+                MPI_Iprobe(src, TOKEN, MCW, &tokenFlag, &status);
+                if(tokenFlag) {
+                    MPI_Irecv(&tokenColor, 1, MPI_INT, src, TOKEN, MCW, &request);
+                    hasToken = true;
+                    if(rank == 0 && tokenColor == WHITE) {
+                        int globalStop = -1;
+                        for(int i = 1; i < size; i++) {
+                            MPI_Send(&globalStop, 1, MPI_INT, i, JOB, MCW);
+                        }
+                        break;
+                    }
+                    if(myColor == BLACK) {
+                        tokenColor = BLACK;
+                    }
+                }
+            }
+        }
+    }
+
+    if(rank == 0) { //Print out the task information
+        int created, completed;
+        cout<<"Rank 0 created "<<tasksCreated<<" tasks and completed "<<tasksCompleted<<" tasks"<<endl;
+        for(int i = 1; i < size; i++) {
+            MPI_Recv(&created, 1, MPI_INT, i, CREATED, MCW, IGNORE);
+            MPI_Recv(&completed, 1, MPI_INT, i, COMPLETED, MCW, IGNORE);
+            cout<<"Rank "<<i<<" created "<<created<<" tasks and completed "<<completed<<" tasks"<<endl;
+            tasksCreated += created;
+            tasksCompleted += completed;
+        }
+        cout<<"Tasks Created: "<<tasksCreated<<endl;
+        cout<<"Tasks Completed: "<<tasksCompleted<<endl;
+    }
+
+
+    MPI_Finalize();
+    return 0;
+}