// ==================================================================
// Author: Jean-Michel Richer
// Date: Aug 2020
// Last modified: November 2020
// Purpose: Use of MPI::BCast 
// The master process creates its own local data and they are send
// to other process (of rank not equal to 0)
// ==================================================================
#include <unistd.h>
#include <iostream>
#include <cstdlib>
#include <unistd.h> // for sleep
using namespace std;
#include "ezmpi.h"
using namespace ez::mpi;

/**
 * run master and slaves
 */
void run(int argc, char *argv[]) {
	
	Process p(argc, argv);
	

	int global_data_size = 100;
	int *global_data = NULL; 
	
	global_data = new int [ global_data_size ];
	for (int i=0; i<global_data_size; ++i) {
		global_data[i] = 0;
	}
		 
	
	if (p.is_master()) {
	
		// master process will send data to others
		for (int i=0; i<global_data_size; ++i) global_data[i] =  i+1;
		
		p << "global array = [";
		for (int i=0; i<global_data_size; ++i) p << global_data[i] << " ";
		p << "]" << endl;
	}
	
	// Master sends data to all others
	//p.synchronize();
	p.broadcast( global_data, global_data_size );
	
	
	// Each process reports the data it has received			
	p << "local copy array=[";
	for (int i=0; i<global_data_size; ++i) {
		p << global_data[i] << " ";
	}
	p << "]" << endl;

	sleep(1);
   	
   	p.logs(cout);
}

// ==================================================================
// C++ version
// ==================================================================
int main(int argc, char ** argv) {
	// call a function that contains code for master and slave
	// in order to avoid problems of initialization and finalization
	run(argc, argv);
	
	exit(EXIT_SUCCESS);
}