---

threshold.c

Go to the documentation of this file.

//
// Copyright Onur G. Guleryuz
//
//
#include <math.h>

int hard_threshold(float **matrix,int Ni,int Nj,float threshold)

{
        int i,j,cnt=0;

        for(i=0;i<Ni;i++)
                for(j=0;j<Nj;j++) {
                        if(fabs(matrix[i][j])<=threshold) {
                                matrix[i][j]=0;
                                cnt++;
                        }
                }
        return(cnt);
}

void hard_threshold_which_nz(float **matrix,int Ni,int Nj,float threshold,int **nz_arr)

{
        int i,j,cnt=0;

        for(i=0;i<Ni;i++)
                for(j=0;j<Nj;j++) {

                        if(fabs(matrix[i][j])<=threshold) {
                                matrix[i][j]=0;
                                nz_arr[i][j]=0;
                                cnt++;
                        }
                        else
                                nz_arr[i][j]=1;
                }
}

---