Rcpp Version 1.0.9
rgeom.h
Go to the documentation of this file.
1 // -*- mode: C++; c-indent-level: 4; c-basic-offset: 4; indent-tabs-mode: nil; -*-
2 //
3 // rgeom.h: Rcpp R/C++ interface class library --
4 //
5 // Copyright (C) 2010 - 2016 Douglas Bates, Dirk Eddelbuettel and Romain Francois
6 //
7 // This file is part of Rcpp.
8 //
9 // Rcpp is free software: you can redistribute it and/or modify it
10 // under the terms of the GNU General Public License as published by
11 // the Free Software Foundation, either version 2 of the License, or
12 // (at your option) any later version.
13 //
14 // Rcpp is distributed in the hope that it will be useful, but
15 // WITHOUT ANY WARRANTY; without even the implied warranty of
16 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 // GNU General Public License for more details.
18 //
19 // You should have received a copy of the GNU General Public License
20 // along with Rcpp. If not, see <http://www.gnu.org/licenses/>.
21 
22 #ifndef Rcpp__stats__random_rgeom_h
23 #define Rcpp__stats__random_rgeom_h
24 
25 namespace Rcpp {
26 namespace stats {
27 
28 class GeomGenerator : public ::Rcpp::Generator<double> {
29 public:
30 
31  GeomGenerator( double p ) : lambda( (1-p)/p ) {}
32 
33  inline double operator()() const {
34  return ::Rf_rpois(exp_rand() * lambda);
35  }
36 
37 private:
38  double lambda ;
39 };
40 
41 } // stats
42 } // Rcpp
43 
44 #endif
double operator()() const
Definition: rgeom.h:33
GeomGenerator(double p)
Definition: rgeom.h:31
double exp_rand(void)
Definition: Rmath.h:32
Rcpp API.
Definition: algo.h:28