From ac86133781d7cd50964579f79522a4e3f8c3f339 Mon Sep 17 00:00:00 2001
From: Calvin Morrison <mutantturkey@gmail.com>
Date: Wed, 19 Feb 2014 16:02:52 -0500
Subject: add a safe alloc function so our program doesn't segfault on memory
 shortage

---
 FEAST/MIToolbox/MutualInformation.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'FEAST/MIToolbox/MutualInformation.c')

diff --git a/FEAST/MIToolbox/MutualInformation.c b/FEAST/MIToolbox/MutualInformation.c
index 0fb4766..d7d10b6 100644
--- a/FEAST/MIToolbox/MutualInformation.c
+++ b/FEAST/MIToolbox/MutualInformation.c
@@ -35,6 +35,7 @@
 #include "CalculateProbability.h"
 #include "Entropy.h"
 #include "MutualInformation.h"
+#include "util.h"
 
 double calculateMutualInformation(double *dataVector, double *targetVector, int vectorLength)
 {
@@ -76,7 +77,7 @@ double calculateConditionalMutualInformation(double *dataVector, double *targetV
 {
   double mutualInformation = 0.0;
   double firstCondition, secondCondition;
-  double *mergedVector = (double *) CALLOC_FUNC(vectorLength,sizeof(double));
+  double *mergedVector = safe_calloc(vectorLength,sizeof(double));
   
   mergeArrays(targetVector,conditionVector,mergedVector,vectorLength);
   
-- 
cgit v1.2.3