mirror of
https://github.com/lucaspalomodevelop/meshlab.git
synced 2026-03-18 18:44:39 +00:00
8e49109c5a
All and only the official plugins now are in the meshplugins folder. Old stuff moved to the unsupported folder. New stuff moved here from the experimental folder.
131 lines
3.3 KiB
C++
131 lines
3.3 KiB
C++
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include <math.h>
|
|
|
|
#include <iostream>
|
|
#include <QImage> /*debug*/
|
|
#include "mutual.h"
|
|
|
|
using namespace std;
|
|
MutualInfo::MutualInfo(unsigned int _nbins, int _bweight, bool _use_background):
|
|
bweight(_bweight), use_background(_use_background),
|
|
histo2D(NULL), histoA(NULL), histoB(NULL) {
|
|
|
|
setBins(_nbins);
|
|
}
|
|
|
|
MutualInfo::~MutualInfo() {
|
|
delete []histo2D;
|
|
delete []histoA;
|
|
delete []histoB;
|
|
}
|
|
|
|
void MutualInfo::setBins(unsigned int _nbins) {
|
|
nbins = _nbins;
|
|
assert(!(nbins & (nbins-1)));
|
|
|
|
if(histo2D) delete []histo2D;
|
|
if(histoA) delete []histoA;
|
|
if(histoB) delete []histoB;
|
|
histo2D = new unsigned int[nbins*nbins];
|
|
histoA = new unsigned int[nbins];
|
|
histoB = new unsigned int[nbins];
|
|
}
|
|
|
|
double MutualInfo::info(int width, int height,
|
|
unsigned char *target, unsigned char *render,
|
|
int startx, int endx,
|
|
int starty, int endy) {
|
|
histogram(width, height, target, render, startx, endx, starty, endy);
|
|
|
|
memset(histoA, 0, nbins*sizeof(int));
|
|
memset(histoB, 0, nbins*sizeof(int));
|
|
double n = 0.0;
|
|
|
|
int i = 0;
|
|
for(unsigned int y = 0; y < nbins; y++) {
|
|
unsigned int &b = histoB[y];
|
|
for(unsigned int x = 0; x < nbins; x++) {
|
|
int ab = histo2D[i++];
|
|
histoA[x] += ab;
|
|
b += ab;
|
|
// if(ab != 0) cout << "1";
|
|
// else cout << "0";
|
|
}
|
|
// cout << endl;
|
|
n += b;
|
|
}
|
|
//cout << endl;
|
|
double ILOG2 = 1/log(2.0);
|
|
//assert(n > 0);
|
|
if(n == 0) n = 1;
|
|
double m = 0.0;
|
|
for(unsigned int y = 0; y < nbins; y++) {
|
|
double b = histoB[y];
|
|
if(b == 0) continue;
|
|
for(unsigned int x = 0; x < nbins; x++) {
|
|
double ab = histo2D[x + nbins*y];
|
|
if(ab == 0) continue;
|
|
double a = histoA[x];
|
|
m += ab * log((n*ab)/(a*b))*ILOG2;
|
|
}
|
|
}
|
|
m /= n;
|
|
|
|
#ifdef USE_MUTUAL
|
|
for(int x = 0; x < bins; x++) {
|
|
double a = histoA[x]/n;
|
|
double b = histoB[x]/n;
|
|
if(a > 0)
|
|
m -= a*log(a);
|
|
if(b > 0) m -= b*log(b);
|
|
}
|
|
#endif
|
|
return m;
|
|
}
|
|
|
|
void MutualInfo::histogram(int width, int height,
|
|
unsigned char *target, unsigned char *render,
|
|
int startx, int endx,
|
|
int starty, int endy) {
|
|
if(endx == 0) endx = width;
|
|
if(endy == 0) endy = height;
|
|
memset(histo2D, 0, nbins*nbins*sizeof(int));
|
|
int side = 256/nbins;
|
|
assert(!(side & (side-1)));
|
|
|
|
//k is the log2 of side
|
|
int k = 0;
|
|
while ( side>>=1 ) { ++k; }
|
|
|
|
int bins = nbins;
|
|
int s = 0;
|
|
while ( bins>>=1) { ++s; }
|
|
|
|
for(int y = starty; y < endy; y++) {
|
|
int offset = width*y + startx;
|
|
for(int x = startx; x < endx; x++, offset++) {
|
|
unsigned char a = target[offset]>>k; //instead of /side;
|
|
unsigned char b = render[offset]>>k; //instead of /side;
|
|
histo2D[a + (b<<s)] += 2;//bweight; //instead of nbins*s
|
|
}
|
|
}
|
|
//weight of background is divided.
|
|
//background is when b = 0 -> first row of histo2D
|
|
if(bweight != 0) {
|
|
for(unsigned int i = 0; i < nbins; i++)
|
|
histo2D[i] /= bweight;
|
|
} else
|
|
memset(histo2D, 0, nbins*sizeof(int));
|
|
|
|
/*unsigned char buffer[nbins*nbins*4];
|
|
for(int i = 0; i < nbins*nbins; i++) {
|
|
buffer[i*4] = buffer[i*4+1] = buffer[i*4+2] = (unsigned char)(10*log(histo2D[i]));
|
|
buffer[i*4+3] = 0;
|
|
}
|
|
QImage im(buffer, nbins, nbins, QImage::Format_ARGB32);
|
|
im.save("prova.jpg");*/
|
|
}
|
|
|