quantization.h 3.93 KB
Newer Older
Pierre Kraemer's avatar
Pierre Kraemer committed
1 2 3
/*******************************************************************************
* CGoGN: Combinatorial and Geometric modeling with Generic N-dimensional Maps  *
* version 0.1                                                                  *
4
* Copyright (C) 2009-2012, IGG Team, LSIIT, University of Strasbourg           *
Pierre Kraemer's avatar
Pierre Kraemer committed
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
*                                                                              *
* This library 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 2.1 of the License, or (at your     *
* option) any later version.                                                   *
*                                                                              *
* This library 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 this library; if not, write to the Free Software Foundation,      *
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.           *
*                                                                              *
20
* Web site: http://cgogn.unistra.fr/                                           *
Pierre Kraemer's avatar
Pierre Kraemer committed
21 22 23 24 25 26 27
* Contact information: cgogn@unistra.fr                                        *
*                                                                              *
*******************************************************************************/

#ifndef __QUANTIZATION_H__
#define __QUANTIZATION_H__

Sylvain Thery's avatar
Sylvain Thery committed
28 29 30
#include <vector>
#include <list>

Pierre Kraemer's avatar
Pierre Kraemer committed
31 32 33 34 35 36 37 38 39 40
#define epsSplitVector 0.00001f

#define epsilonDistortion 0.0001f

#include <vector>
#include <math.h>

namespace CGoGN
{

Sylvain Thery's avatar
Sylvain Thery committed
41
namespace Utils
Pierre Kraemer's avatar
Pierre Kraemer committed
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
{

template <typename VEC>
struct CodeVector
{
	VEC v ;
	unsigned int regionNbVectors ;
	VEC regionVectorsSum ;
	float regionDistortion ;

	bool operator<(CodeVector<VEC>& c)
	{
		return regionDistortion > c.regionDistortion ;
	}
} ;

template <typename VEC>
class Quantization
{
	typedef typename std::list<CodeVector<VEC> >::iterator CodeVectorID ;

private:
	const std::vector<VEC>& sourceVectors ; // source vectors
	std::vector<CodeVectorID> associatedCodeVectors ; // for each source vector, id of its associated codeVector
	std::list<CodeVector<VEC> > codeVectors ; // codebook
	unsigned int nbCodeVectors ; // size of codebook

	VEC meanSourceVector ;
	float distortion ;
	float discreteEntropy, differentialEntropy ;
	float determinantSigma, traceSigma ;

	void computeMeanSourceVector() ;
	CodeVectorID nearestNeighbour(int v) ; // for the sourceVector of the given index, search the id of the nearest codeVector
	void algoLloydMax() ; // Lloyd Iteration

public:
	Quantization(const std::vector<VEC>& source) ;

//	void scalarQuantization(unsigned int nbCodeVectors, std::vector<VEC>& result) ;

	void vectorQuantizationInit() ;
	void vectorQuantizationNbRegions(unsigned int nbCodeVectors, std::vector<VEC>& result) ;
	void vectorQuantizationDistortion(float distortionGoal, std::vector<VEC>& result) ;

	unsigned int getNbCodeVectors() { return nbCodeVectors ; }

	// only available after a quantization
	float getDiscreteEntropy() { return discreteEntropy ; }
	// available immediately after object construction
	float getDifferentialEntropy() { return differentialEntropy ; }
	float getDeterminantSigma() { return determinantSigma ; }
	float getTraceSigma() { return traceSigma ; }

private:
	void computeDiscreteEntropy() ;
	void computeDifferentialEntropy() ;
	float covariance(int x, int y) ;
	float determinant(float* matrice) ;
} ;


Sylvain Thery's avatar
Sylvain Thery committed
104
} //namespace Utils
Pierre Kraemer's avatar
Pierre Kraemer committed
105 106 107 108 109 110

} //namespace CGoGN

#include "quantization.hpp"

#endif