def_logcosh1.cpp

本程序将独立分量分析技术和数字水印技术有机地结合在一起

#include <malloc.h>
#include <stdio.h>
#include <math.h>

void mat_transpose (float *, int , int , float *);
void mat_mult (float *, int , int , float *, int , int , float *);
/* Def_logcosh1-w修正子程序
 *该子程序实现对分离矩阵的修正
 */
void Def_logcosh1 (float *w_init, int e, float *data, int f, int p, float step, float *w_final)
{	
/* Function that carries out Deflation ICA using an logcosh approximation to the neg. entropy function */
	
	float *mat1, *mat2, *mat3, *mat4, *mat5, *mat6;
	int i, j;
	float Beta,numSamples,myy,sum;
	myy = step;
	
	if (e != f) {
		printf ("error in Def_logcosh_JM, dims dont match");
	}
	else {
		mat1 = (float*)malloc(1 * p* sizeof(float));
		mat2 = (float*)malloc(e * p* sizeof(float));
		mat3 = (float*)malloc(1 * e* sizeof(float));
		mat4 = (float*)malloc(1 * e* sizeof(float));
        mat5 = (float*)malloc(1 * p* sizeof(float));
		mat6 = (float*)malloc(1 * 1* sizeof(float));


		mat_mult (w_init, 1, e, data, e, p, mat1);
		
		
		for (i = 0; i < p; i++) {
			mat1[i] = (float)tanh (mat1[i]);
		}			
		
		mat_transpose (data, e, p, mat2);
        
		/*对行取均值，做E{}运算
		for (i = 0; i < e; i++) {
			for (j = 0; j < p; j++) {
				mat2[i * p + j] = (mat2[i * p + j]) / p;
			}
		}
		*/
		mat_mult (mat1, 1, p, mat2, p, e, mat3);

		for (i = 0; i < p; i++) {
			mat5[i] = ( (1 - (mat1[i]) * (mat1[i])));
		}
		
		sum = 0;
		for (j = 0; j < p; j++) {
			sum += mat5[j];
		}
        
        mat_transpose (w_init, 1, e, mat1);
		mat_mult (mat3, 1, e, mat1, e, 1, mat6);
		Beta=*mat6;
		numSamples = sum - Beta;


		for (i = 0; i < e; i++) {
			mat4[i] = (w_init[i]) * Beta;
		}			
		for (i = 0; i < e; i++) {
			w_final[i] = w_init[i]-myy*(mat3[i] - mat4[i])/numSamples;
		}		
			
		
	}
}