Cosmetics.
[clueless-kmeans.git] / misc.h
1 /*
2  *  clueless-kmean is a variant of k-mean which enforces balanced
3  *  distribution of classes in every cluster
4  *
5  *  Copyright (c) 2013 Idiap Research Institute, http://www.idiap.ch/
6  *  Written by Francois Fleuret <francois.fleuret@idiap.ch>
7  *
8  *  This file is part of clueless-kmean.
9  *
10  *  clueless-kmean is free software: you can redistribute it and/or
11  *  modify it under the terms of the GNU General Public License
12  *  version 3 as published by the Free Software Foundation.
13  *
14  *  clueless-kmean is distributed in the hope that it will be useful,
15  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
16  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  *  General Public License for more details.
18  *
19  *  You should have received a copy of the GNU General Public License
20  *  along with selector.  If not, see <http://www.gnu.org/licenses/>.
21  *
22  */
23
24 #ifndef MISC_H
25 #define MISC_H
26
27 #include <iostream>
28 #include <cmath>
29 #include <fstream>
30 #include <cfloat>
31 #include <stdlib.h>
32 #include <string.h>
33
34 using namespace std;
35
36 typedef double scalar_t;
37 // typedef float scalar_t;
38
39 const int buffer_size = 1024;
40
41 using namespace std;
42
43 #ifdef DEBUG
44 #define ASSERT(x) if(!(x)) { \
45   std::cerr << "ASSERT FAILED IN " << __FILE__ << ":" << __LINE__ << endl; \
46   abort(); \
47 }
48 #else
49 #define ASSERT(x)
50 #endif
51
52 template<class T>
53 T smooth_min(T x, T y) {
54   T z = exp(x - y);
55   return 0.5 * (x + y - (x - y)/(1 + 1/z) - (y - x)/(1 + z));
56 }
57
58 template <class T>
59 void write_var(ostream *os, const T *x) { os->write((char *) x, sizeof(T)); }
60
61 template <class T>
62 void read_var(istream *is, T *x) { is->read((char *) x, sizeof(T)); }
63
64 template <class T>
65 void grow(int *nb_max, int nb, T** current, int factor) {
66   ASSERT(*nb_max > 0);
67   if(nb == *nb_max) {
68     T *tmp = new T[*nb_max * factor];
69     memcpy(tmp, *current, *nb_max * sizeof(T));
70     delete[] *current;
71     *current = tmp;
72     *nb_max *= factor;
73   }
74 }
75
76 template <class T>
77 inline T sq(T x) {
78   return x * x;
79 }
80
81 inline scalar_t log2(scalar_t x) {
82   return log(x)/log(2.0);
83 }
84
85 inline scalar_t xi(scalar_t x) {
86   if(x <= 0.0) return 0.0;
87   else         return - x * log(x)/log(2.0);
88 }
89
90 scalar_t discrete_entropy(int *n, int nb);
91
92 char *basename(char *name);
93
94 char *next_word(char *buffer, char *r, int buffer_size);
95
96 void random_permutation(int *val, int nb);
97 void tag_subset(bool *val, int nb_total, int nb_to_tag);
98
99 struct Couple {
100   int index;
101   scalar_t value;
102 };
103
104 int compare_couple(const void *a, const void *b);
105
106 #endif