/* for pmi2_job_info_t definition */
#include "setup.h"

/* allocate resources to track PMIX_Ring state */
int pmix_ring_init(const pmi2_job_info_t* job, char*** env);

/* free resources allocated to track PMIX_Ring state */
int pmix_ring_finalize();

/* given a global rank in stepd/srun tree for message received
 * from one of our stepd children, compute corresponding child index */
int pmix_ring_id_by_rank(int rank);

/* ring_out messages come in from our parent nodes,
 * we process this and send ring_out messages to each of our children:
 *   count - starting rank for our leftmost application process
 *   left  - left value for leftmost application process in our subtree
 *   right - right value for rightmost application process in our subtree */
int pmix_ring_out(int count, char* left, char* right);

/* we get a ring_in message from each child (stepd and application tasks),
 * once we've gotten a message from each child, we send a ring_in message
 * to our parent
 *   ring_id - index of child (all app procs first, followed by stepds)
 *   count   - count value from child
 *   left    - left value from child
 *   right   - right value from child
 *
 * upon receiving ring_in messages from all children, sends message to
 * parent consisting of:
 *   count = sum of counts from all children
 *   left  = left value from leftmost child that specified a left value
 *   right = right value from rightmost child that specified a right value */
int pmix_ring_in(int ring_id, int count, char* left, char* right);