libstdc++-v3/testsuite/26_numerics/random/piecewise_linear_distribution/cons/num_xbound_fun.cc

   1 // { dg-options "-std=c++0x" }
   2 //
   3 // 2008-12-03  Edward M. Smith-Rowland <3dw4rd@verizon.net>
   4 //
   5 // Copyright (C) 2008 Free Software Foundation, Inc.
   6 //
   7 // This file is part of the GNU ISO C++ Library.  This library is free
   8 // software; you can redistribute it and/or modify it under the
   9 // terms of the GNU General Public License as published by the
  10 // Free Software Foundation; either version 2, or (at your option)
  11 // any later version.
  12 //
  13 // This library is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17 //
  18 // You should have received a copy of the GNU General Public License along
  19 // with this library; see the file COPYING.  If not, write to the Free
  20 // Software Foundation, 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
  21 // USA.
  22
  23 // 26.4.8.5.3 Class template piecewise_linear_distribution [rand.dist.samp.plinear]
  24 // 26.4.2.4 Concept RandomNumberDistribution [rand.concept.dist]
  25
  26 #include <random>
  27 #include <cmath>
  28 #include <testsuite_hooks.h>
  29
  30 struct cosine_distribution
  31 {
  32     cosine_distribution(double x0, double lambda)
  33     : _M_x0(x0), _M_lambda(lambda)
  34     { }
  35
  36     double
  37     operator()(double x)
  38     {
  39       if (x - _M_x0 < -_M_lambda / 4)
  40         return 0.0;
  41       else if (x - _M_x0 > _M_lambda / 4)
  42         return 0.0;
  43       else
  44         return std::cos(2 * M_PI * (x - _M_x0) / _M_lambda);
  45     }
  46
  47 private:
  48     double _M_x0;
  49     double _M_lambda;
  50 };
  51
  52 void
  53 test01()
  54 {
  55   bool test __attribute__((unused)) = true;
  56
  57   cosine_distribution cd(1.5, 3.0);
  58   std::piecewise_linear_distribution<> u(21, -10.0, 10.0, cd);
  59   std::vector<double> interval = u.intervals();
  60   std::vector<double> density = u.densities();
  61   VERIFY( interval.size() == 22 );
  62   VERIFY( density.size() == 22 );
  63 }
  64
  65 int main()
  66 {
  67   test01();
  68   return 0;
  69 }