#include "mpi.h"
#include "numa.h"
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>

#define LEN 1000

int main(int argc,char ** argv) {
	MPI_Init(NULL,NULL);
	int rank,size;
	MPI_Comm_rank (MPI_COMM_WORLD, &rank);
	MPI_Comm_size (MPI_COMM_WORLD, &size);
	char* name = (char*) malloc(LEN);
	gethostname(name,LEN);
	const char* chdb_rank = getenv("CHDB_RANK");
	char * arg = (argc>=1)?argv[1]:"";
	
	int num_cpus=numa_num_task_cpus();
        int num_cpus_ttl = numa_num_configured_cpus();
	char * alloc_cpus = (char *) calloc(sizeof(char),num_cpus_ttl+1);
	int i=0; for (i=0;i<num_cpus_ttl;++i) {
	    alloc_cpus[i] = '.';
	    if (numa_bitmask_isbitset(numa_all_cpus_ptr,i)) alloc_cpus[i] = 'X';
        }
	printf("hello from the slave nb %3s, running on %s (parameter %8s), I got mpi rank %3i/%3i and I am allowed to run on %3i/%3i cpus: %s\n",chdb_rank,name,arg,rank,size,num_cpus,num_cpus_ttl,alloc_cpus);
	MPI_Finalize();
}