BNDC.h

/*
  score/BNDC.{h,c} : BNDC network score function.
  Copyright (C) 2018-2024 Yoshinori Tamada <tamada A T ytlab.jp>
  All rights reserved.
 
  THE FEATURES OF ECv CALCULATION ARE PROTECTED BY PATENT. 
  KYOTO UNIVERSITY HOLDS THE COMMERCIAL RIGHTS TO THIS SOFTWARE.
  DO NOT COPY, MODIFY OR REDISTRIBUTE WITHOUT PERMISSION.
 
  See LICENSE.txt for details of the licensing agreement.
*/
 
#ifndef __YTLIB_BNDC_H
#define __YTLIB_BNDC_H
 
#include "BNRC.h"
 
#include "util/ytData.h"
#include "util/ytKeyValues.h"
#include "net/ytNode.h"
#include "net/ytEdge.h"
#include "net/ytNetwork.h"
 
/* In mega bytes */
//#define BNDC_DEFAULT_MAX_MEM 500
#define BNDC_DEFAULT_MAX_MEM 1000
#define BNDC_DEF   {"BNDC",\
   BNDC_init,\
   BNDC_score,\
   BNDC_reinit,\
   Score_default_cache,\
   BNDC_edgeProp,\
   BNDC_nodeProp,\
   BNDC_finalize,\
   BNDC_partialResidual,\
   BNDC_calcEdgeContrib,\
   BNDC_setEdgeProp,\
   BNDC_setNodeProp,\
   BNDC_status,\
   BNDC_setParam,\
  }
 
typedef struct {
  int * Njmr; /* Njmr[r + m * R]: the number of samples whose value is 'r' with
                 respect to the parent pattern 'm'. */
  int * Njm; /* Njm[m] = Sum_r Njmr[r + m * R]: */
  int * pattern; /* pattern[i] represent the parent pattern id of sample 'i' */
  int * work; /* work area */
  int total; /* The number of patterns stored in work area */
} BDE_Data;
 
void * BNDC_init(ytData * data, ytKeyValues * args);
void BNDC_reinit(void * buff, ytData * data);
double BNDC_score(void * buff, const int j, const int * parents, const int q);
void BNDC_edgeProp(void * buff, const int j, const int * parents, const int q,
                     const int k, ytEdge * edge);
void BNDC_nodeProp(void * buff, int j, ytNode * node);
void BNDC_finalize(void * buff);
void BNDC_partialResidual(void * buff, const int j, const int * parents, const int q, 
                            FILE ** fp, double * ll);
void BNDC_setEdgeProp(void * buff, const int j, const int * parents, const int q,
                        const int k, const ytEdge * edge);
void BNDC_setNodeProp(void * buff, const int j, const ytNode * node);
void BNDC_calcEdgeContrib(void * buff, ytNetwork * network, int j, const int * parents, int q,
                          int mode, double * ar);
void BNDC_status(void * buff);
int BNDC_debug(int argc, char * argv[]);
void BNDC_setParam(void * buff, const char * key, ytObject * value);
 
int BDE_count(const BNRC_Data * D, int j, int n,
              const int * parents, int q, int R, BDE_Data * B);
 
void BDE_Data_finalize(BDE_Data * B);
 
 
#endif /* __YTLIB_BNDC_H */