discreteDST.h

00001 /* MLPACK 0.2
00002  *
00003  * Copyright (c) 2008, 2009 Alexander Gray,
00004  *                          Garry Boyer,
00005  *                          Ryan Riegel,
00006  *                          Nikolaos Vasiloglou,
00007  *                          Dongryeol Lee,
00008  *                          Chip Mappus, 
00009  *                          Nishant Mehta,
00010  *                          Hua Ouyang,
00011  *                          Parikshit Ram,
00012  *                          Long Tran,
00013  *                          Wee Chin Wong
00014  *
00015  * Copyright (c) 2008, 2009 Georgia Institute of Technology
00016  *
00017  * This program is free software; you can redistribute it and/or
00018  * modify it under the terms of the GNU General Public License as
00019  * published by the Free Software Foundation; either version 2 of the
00020  * License, or (at your option) any later version.
00021  *
00022  * This program is distributed in the hope that it will be useful, but
00023  * WITHOUT ANY WARRANTY; without even the implied warranty of
00024  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00025  * General Public License for more details.
00026  *
00027  * You should have received a copy of the GNU General Public License
00028  * along with this program; if not, write to the Free Software
00029  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
00030  * 02110-1301, USA.
00031  */
00032 #ifndef FASTLIB_DISCRETE_DISTRIBUTION_H
00033 #define FASTLIB_DISCRETE_DISTRIBUTION_H
00034 #include "fastlib/fastlib.h"
00035 class DiscreteDST {
00036   Vector p;
00037   Vector ACC_p;
00038  public:
00039   void Init(int N = 2);
00040   void generate(int* v);
00041   double get(int i) { return p[i]; }
00042   void set(const Vector& p_) { p.CopyValues(p_); }
00043 
00044   void start_accumulate() { ACC_p.SetZero(); }
00045   void accumulate(int i, double v) { ACC_p[i]+=v; }
00046   void end_accumulate();
00047 };
00048 #endif