summaryrefslogtreecommitdiff
path: root/FEAST/MIToolbox/MIToolboxMex.c
diff options
context:
space:
mode:
Diffstat (limited to 'FEAST/MIToolbox/MIToolboxMex.c')
-rw-r--r--FEAST/MIToolbox/MIToolboxMex.c494
1 files changed, 0 insertions, 494 deletions
diff --git a/FEAST/MIToolbox/MIToolboxMex.c b/FEAST/MIToolbox/MIToolboxMex.c
deleted file mode 100644
index e27009f..0000000
--- a/FEAST/MIToolbox/MIToolboxMex.c
+++ /dev/null
@@ -1,494 +0,0 @@
-/*******************************************************************************
-**
-** MIToolboxMex.cpp
-** is the MATLAB entry point for the MIToolbox functions when called from
-** a MATLAB/OCTAVE script.
-**
-** Copyright 2010 Adam Pocock, The University Of Manchester
-** www.cs.manchester.ac.uk
-**
-** This file is part of MIToolbox.
-**
-** MIToolbox is free software: you can redistribute it and/or modify
-** it under the terms of the GNU Lesser General Public License as published by
-** the Free Software Foundation, either version 3 of the License, or
-** (at your option) any later version.
-**
-** MIToolbox is distributed in the hope that it will be useful,
-** but WITHOUT ANY WARRANTY; without even the implied warranty of
-** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-** GNU Lesser General Public License for more details.
-**
-** You should have received a copy of the GNU Lesser General Public License
-** along with MIToolbox. If not, see <http://www.gnu.org/licenses/>.
-**
-*******************************************************************************/
-#include "MIToolbox.h"
-#include "ArrayOperations.h"
-#include "CalculateProbability.h"
-#include "Entropy.h"
-#include "MutualInformation.h"
-
-/*******************************************************************************
-**entry point for the mex call
-**nlhs - number of outputs
-**plhs - pointer to array of outputs
-**nrhs - number of inputs
-**prhs - pointer to array of inputs
-*******************************************************************************/
-void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
-{
- /*****************************************************************************
- ** this function takes a flag and a variable number of arguments
- ** depending on the value of the flag and returns either a construct
- ** containing probability estimates, a merged vector or a double value
- ** representing an entropy or mutual information
- *****************************************************************************/
-
- int flag, i, numberOfSamples, checkSamples, thirdCheckSamples, numberOfFeatures, checkFeatures, thirdCheckFeatures;
- int numArities, errorTest;
- double *dataVector, *condVector, *targetVector, *firstVector, *secondVector, *output, *numStates;
- double *matrix, *mergedVector, *arities;
- int *outputIntVector, *intArities;
-
- double *jointOutput, *numJointStates, *firstOutput, *numFirstStates, *secondOutput, *numSecondStates;
-
- ProbabilityState state;
- JointProbabilityState jointState;
-
- /*if (nlhs != 1)
- {
- printf("Incorrect number of output arguments\n");
- }//if not 1 output
- */
- if (nrhs == 2)
- {
- /*printf("Must be H(X), calculateProbability(X), merge(X), normaliseArray(X)\n");*/
- }
- else if (nrhs == 3)
- {
- /*printf("Must be H(XY), H(X|Y), calculateJointProbability(XY), I(X;Y)\n");*/
- }
- else if (nrhs == 4)
- {
- /*printf("Must be I(X;Y|Z)\n");*/
- }
- else
- {
- printf("Incorrect number of arguments, format is MIToolbox(\"FLAG\",varargin)\n");
- }
-
- /* number to function map
- ** 1 = calculateProbability
- ** 2 = calculateJointProbability
- ** 3 = mergeArrays
- ** 4 = H(X)
- ** 5 = H(XY)
- ** 6 = H(X|Y)
- ** 7 = I(X;Y)
- ** 8 = I(X;Y|Z)
- ** 9 = normaliseArray
- */
-
- flag = *mxGetPr(prhs[0]);
-
- switch (flag)
- {
- case 1:
- {
- /*
- **calculateProbability
- */
- numberOfSamples = mxGetM(prhs[1]);
- dataVector = (double *) mxGetPr(prhs[1]);
-
- /*ProbabilityState calculateProbability(double *dataVector, int vectorLength);*/
- state = calculateProbability(dataVector,numberOfSamples);
-
- plhs[0] = mxCreateDoubleMatrix(state.numStates,1,mxREAL);
- plhs[1] = mxCreateDoubleMatrix(1,1,mxREAL);
- output = (double *)mxGetPr(plhs[0]);
- numStates = (double *) mxGetPr(plhs[1]);
-
- *numStates = state.numStates;
-
- for (i = 0; i < state.numStates; i++)
- {
- output[i] = state.probabilityVector[i];
- }
-
- break;
- }/*case 1 - calculateProbability*/
- case 2:
- {
- /*
- **calculateJointProbability
- */
- numberOfSamples = mxGetM(prhs[1]);
- firstVector = (double *) mxGetPr(prhs[1]);
- secondVector = (double *) mxGetPr(prhs[2]);
-
- /*JointProbabilityState calculateJointProbability(double *firstVector, double *secondVector int vectorLength);*/
- jointState = calculateJointProbability(firstVector,secondVector,numberOfSamples);
-
- plhs[0] = mxCreateDoubleMatrix(jointState.numJointStates,1,mxREAL);
- plhs[1] = mxCreateDoubleMatrix(1,1,mxREAL);
- plhs[2] = mxCreateDoubleMatrix(jointState.numFirstStates,1,mxREAL);
- plhs[3] = mxCreateDoubleMatrix(1,1,mxREAL);
- plhs[4] = mxCreateDoubleMatrix(jointState.numSecondStates,1,mxREAL);
- plhs[5] = mxCreateDoubleMatrix(1,1,mxREAL);
- jointOutput = (double *)mxGetPr(plhs[0]);
- numJointStates = (double *) mxGetPr(plhs[1]);
- firstOutput = (double *)mxGetPr(plhs[2]);
- numFirstStates = (double *) mxGetPr(plhs[3]);
- secondOutput = (double *)mxGetPr(plhs[4]);
- numSecondStates = (double *) mxGetPr(plhs[5]);
-
- *numJointStates = jointState.numJointStates;
- *numFirstStates = jointState.numFirstStates;
- *numSecondStates = jointState.numSecondStates;
-
- for (i = 0; i < jointState.numJointStates; i++)
- {
- jointOutput[i] = jointState.jointProbabilityVector[i];
- }
- for (i = 0; i < jointState.numFirstStates; i++)
- {
- firstOutput[i] = jointState.firstProbabilityVector[i];
- }
- for (i = 0; i < jointState.numSecondStates; i++)
- {
- secondOutput[i] = jointState.secondProbabilityVector[i];
- }
-
- break;
- }/*case 2 - calculateJointProbability */
- case 3:
- {
- /*
- **mergeArrays
- */
- numberOfSamples = mxGetM(prhs[1]);
- numberOfFeatures = mxGetN(prhs[1]);
-
- numArities = 0;
- if (nrhs > 2)
- {
- numArities = mxGetN(prhs[2]);
- /*printf("arities = %d, features = %d, samples = %d\n",numArities,numberOfFeatures,numberOfSamples);*/
- }
-
- plhs[0] = mxCreateDoubleMatrix(0,0,mxREAL);
-
- if (numArities == 0)
- {
- /*
- **no arities therefore compress output
- */
- if ((numberOfFeatures > 0) && (numberOfSamples > 0))
- {
- matrix = (double *) mxGetPr(prhs[1]);
- mergedVector = (double *) mxCalloc(numberOfSamples,sizeof(double));
-
- plhs[0] = mxCreateDoubleMatrix(numberOfSamples,1,mxREAL);
- output = (double *)mxGetPr(plhs[0]);
-
- /*int mergeMultipleArrays(double *inputMatrix, double *outputVector, int matrixWidth, int vectorLength)*/
- mergeMultipleArrays(matrix, mergedVector, numberOfFeatures, numberOfSamples);
- for (i = 0; i < numberOfSamples; i++)
- {
- output[i] = mergedVector[i];
- }
-
- mxFree(mergedVector);
- mergedVector = NULL;
- }
- }
- else if (numArities == numberOfFeatures)
- {
- if ((numberOfFeatures > 0) && (numberOfSamples > 0))
- {
-
- matrix = (double *) mxGetPr(prhs[1]);
- mergedVector = (double *) mxCalloc(numberOfSamples,sizeof(double));
-
- arities = (double *) mxGetPr(prhs[2]);
- intArities = (int *) mxCalloc(numberOfFeatures,sizeof(int));
- for (i = 0; i < numArities; i++)
- {
- intArities[i] = (int) floor(arities[i]);
- }
-
- /*int mergeMultipleArrays(double *inputMatrix, double *outputVector, int matrixWidth, int *arities, int vectorLength);*/
- errorTest = mergeMultipleArraysArities(matrix, mergedVector, numberOfFeatures, intArities, numberOfSamples);
-
- if (errorTest != -1)
- {
- plhs[0] = mxCreateDoubleMatrix(numberOfSamples,1,mxREAL);
- output = (double *)mxGetPr(plhs[0]);
- for (i = 0; i < numberOfSamples; i++)
- {
- output[i] = mergedVector[i];
- }
- }
- else
- {
- printf("Incorrect arities supplied. More states in data than specified\n");
- }
-
- mxFree(mergedVector);
- mergedVector = NULL;
- }
- }
- else
- {
- printf("Number of arities does not match number of features, arities should be a row vector\n");
- }
-
- break;
- }/*case 3 - mergeArrays*/
- case 4:
- {
- /*
- **H(X)
- */
- numberOfSamples = mxGetM(prhs[1]);
- numberOfFeatures = mxGetN(prhs[1]);
-
- dataVector = (double *) mxGetPr(prhs[1]);
-
- plhs[0] = mxCreateDoubleMatrix(1,1,mxREAL);
- output = (double *)mxGetPr(plhs[0]);
-
- if (numberOfFeatures == 1)
- {
- /*double calculateEntropy(double *dataVector, int vectorLength);*/
- *output = calculateEntropy(dataVector,numberOfSamples);
- }
- else
- {
- printf("No columns in input\n");
- *output = -1.0;
- }
-
- break;
- }/*case 4 - H(X)*/
- case 5:
- {
- /*
- **H(XY)
- */
- numberOfSamples = mxGetM(prhs[1]);
- checkSamples = mxGetM(prhs[2]);
-
- numberOfFeatures = mxGetN(prhs[1]);
- checkFeatures = mxGetN(prhs[2]);
-
- firstVector = mxGetPr(prhs[1]);
- secondVector = mxGetPr(prhs[2]);
-
- plhs[0] = mxCreateDoubleMatrix(1,1,mxREAL);
- output = (double *)mxGetPr(plhs[0]);
-
-
- if ((numberOfFeatures == 1) && (checkFeatures == 1))
- {
- if ((numberOfSamples == 0) && (checkSamples == 0))
- {
- *output = 0.0;
- }
- else if (numberOfSamples == 0)
- {
- *output = calculateEntropy(secondVector,numberOfSamples);
- }
- else if (checkSamples == 0)
- {
- *output = calculateEntropy(firstVector,numberOfSamples);
- }
- else if (numberOfSamples == checkSamples)
- {
- /*double calculateJointEntropy(double *firstVector, double *secondVector, int vectorLength);*/
- *output = calculateJointEntropy(firstVector,secondVector,numberOfSamples);
- }
- else
- {
- printf("Vector lengths do not match, they must be the same length\n");
- *output = -1.0;
- }
- }
- else
- {
- printf("No columns in input\n");
- *output = -1.0;
- }
-
- break;
- }/*case 5 - H(XY)*/
- case 6:
- {
- /*
- **H(X|Y)
- */
- numberOfSamples = mxGetM(prhs[1]);
- checkSamples = mxGetM(prhs[2]);
-
- numberOfFeatures = mxGetN(prhs[1]);
- checkFeatures = mxGetN(prhs[2]);
-
- dataVector = mxGetPr(prhs[1]);
- condVector = mxGetPr(prhs[2]);
-
- plhs[0] = mxCreateDoubleMatrix(1,1,mxREAL);
- output = (double *)mxGetPr(plhs[0]);
-
- if ((numberOfFeatures == 1) && (checkFeatures == 1))
- {
- if (numberOfSamples == 0)
- {
- *output = 0.0;
- }
- else if (checkSamples == 0)
- {
- *output = calculateEntropy(dataVector,numberOfSamples);
- }
- else if (numberOfSamples == checkSamples)
- {
- /*double calculateConditionalEntropy(double *dataVector, double *condVector, int vectorLength);*/
- *output = calculateConditionalEntropy(dataVector,condVector,numberOfSamples);
- }
- else
- {
- printf("Vector lengths do not match, they must be the same length\n");
- *output = -1.0;
- }
- }
- else
- {
- printf("No columns in input\n");
- *output = -1.0;
- }
- break;
- }/*case 6 - H(X|Y)*/
- case 7:
- {
- /*
- **I(X;Y)
- */
- numberOfSamples = mxGetM(prhs[1]);
- checkSamples = mxGetM(prhs[2]);
-
- numberOfFeatures = mxGetN(prhs[1]);
- checkFeatures = mxGetN(prhs[2]);
-
- firstVector = mxGetPr(prhs[1]);
- secondVector = mxGetPr(prhs[2]);
-
- plhs[0] = mxCreateDoubleMatrix(1,1,mxREAL);
- output = (double *)mxGetPr(plhs[0]);
-
- if ((numberOfFeatures == 1) && (checkFeatures == 1))
- {
- if ((numberOfSamples == 0) || (checkSamples == 0))
- {
- *output = 0.0;
- }
- else if (numberOfSamples == checkSamples)
- {
- /*double calculateMutualInformation(double *firstVector, double *secondVector, int vectorLength);*/
- *output = calculateMutualInformation(firstVector,secondVector,numberOfSamples);
- }
- else
- {
- printf("Vector lengths do not match, they must be the same length\n");
- *output = -1.0;
- }
- }
- else
- {
- printf("No columns in input\n");
- *output = -1.0;
- }
- break;
- }/*case 7 - I(X;Y)*/
- case 8:
- {
- /*
- **I(X;Y|Z)
- */
- numberOfSamples = mxGetM(prhs[1]);
- checkSamples = mxGetM(prhs[2]);
- thirdCheckSamples = mxGetM(prhs[3]);
-
- numberOfFeatures = mxGetN(prhs[1]);
- checkFeatures = mxGetN(prhs[2]);
- thirdCheckFeatures = mxGetN(prhs[3]);
-
- firstVector = mxGetPr(prhs[1]);
- targetVector = mxGetPr(prhs[2]);
- condVector = mxGetPr(prhs[3]);
-
- plhs[0] = mxCreateDoubleMatrix(1,1,mxREAL);
- output = (double *)mxGetPr(plhs[0]);
-
- if ((numberOfFeatures == 1) && (checkFeatures == 1))
- {
- if ((numberOfSamples == 0) || (checkSamples == 0))
- {
- *output = 0.0;
- }
- else if ((thirdCheckSamples == 0) || (thirdCheckFeatures != 1))
- {
- *output = calculateMutualInformation(firstVector,targetVector,numberOfSamples);
- }
- else if ((numberOfSamples == checkSamples) && (numberOfSamples == thirdCheckSamples))
- {
- /*double calculateConditionalMutualInformation(double *firstVector, double *targetVector, double *condVector, int vectorLength);*/
- *output = calculateConditionalMutualInformation(firstVector,targetVector,condVector,numberOfSamples);
- }
- else
- {
- printf("Vector lengths do not match, they must be the same length\n");
- *output = -1.0;
- }
- }
- else
- {
- printf("No columns in input\n");
- *output = -1.0;
- }
- break;
- }/*case 8 - I(X;Y|Z)*/
- case 9:
- {
- /*
- **normaliseArray
- */
- numberOfSamples = mxGetM(prhs[1]);
- dataVector = (double *) mxGetPr(prhs[1]);
-
- outputIntVector = (int *) mxCalloc(numberOfSamples,sizeof(int));
-
- plhs[0] = mxCreateDoubleMatrix(numberOfSamples,1,mxREAL);
- plhs[1] = mxCreateDoubleMatrix(1,1,mxREAL);
- output = (double *)mxGetPr(plhs[0]);
- numStates = (double *) mxGetPr(plhs[1]);
-
- /*int normaliseArray(double *inputVector, int *outputVector, int vectorLength);*/
- *numStates = normaliseArray(dataVector, outputIntVector, numberOfSamples);
-
- for (i = 0; i < numberOfSamples; i++)
- {
- output[i] = outputIntVector[i];
- }
-
- break;
- }/*case 9 - normaliseArray*/
- default:
- {
- printf("Unrecognised flag\n");
- break;
- }/*default*/
- }/*switch(flag)*/
-
- return;
-}/*mexFunction()*/
generated by cgit v1.2.3 (git 2.39.1) at 2025-07-02 01:00:07 +0000