BN-Inference/cluster.h at master · alexvk/BN-Inference · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
/********************************************************************
$RCSfile: cluster.h,v $
$Author: alexvk $
$Revision: 1.1 $
$Date: 1997/10/15 02:54:41 $
********************************************************************/

#ifndef CLUSTER_INCLUDED
#define CLUSTER_INCLUDED

#include <stdlib.h>
#include "utils.h"

typedef enum {EDGE_NO_MSG, EDGE_MSG_U2D, EDGE_MSG_D2U} STATUS;

/* a cluster node with probability distribution */
typedef struct {
  int numNodes;
  int *nodes;
  long *odometer;
  long stateSpaceSize;
  VECTOR *probDistr;
  int numNghb;
  int *edges;

  /* optfact specific */
  int inclNode;

  /* parallel implementation specific */
  long pars, pare;

  /* MR specific */
  int numProbs;

} CLUSTER;

/* an edge with precomputed summation indices */
typedef struct {
  int numNodes;
  int *nodes;
  long msgLen;
  int unode;
  long uindicesLen;
  long *uindices;
  int dnode;
  long dindicesLen;
  long *dindices;
  STATUS status;

  /* parallel implementation specific */
  long pars, pare;
  int numToWait;
  int masterID;

  /* MR specific */
  int numOrigProbs;
  int* origProbs;

} EDGE;

#endif /* CLUSTER_INCLUDED */