xref: /aosp_15_r20/external/eigen/test/rand.cpp (revision bf2c37156dfe67e5dfebd6d394bad8b2ab5804d4)
1*bf2c3715SXin Li // This file is part of Eigen, a lightweight C++ template library
2*bf2c3715SXin Li // for linear algebra.
3*bf2c3715SXin Li //
4*bf2c3715SXin Li // Copyright (C) 2015 Gael Guennebaud <[email protected]>
5*bf2c3715SXin Li //
6*bf2c3715SXin Li // This Source Code Form is subject to the terms of the Mozilla
7*bf2c3715SXin Li // Public License v. 2.0. If a copy of the MPL was not distributed
8*bf2c3715SXin Li // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
9*bf2c3715SXin Li 
10*bf2c3715SXin Li #include "main.h"
11*bf2c3715SXin Li 
12*bf2c3715SXin Li typedef long long int64;
13*bf2c3715SXin Li 
check_in_range(Scalar x,Scalar y)14*bf2c3715SXin Li template<typename Scalar> Scalar check_in_range(Scalar x, Scalar y)
15*bf2c3715SXin Li {
16*bf2c3715SXin Li   Scalar r = internal::random<Scalar>(x,y);
17*bf2c3715SXin Li   VERIFY(r>=x);
18*bf2c3715SXin Li   if(y>=x)
19*bf2c3715SXin Li   {
20*bf2c3715SXin Li     VERIFY(r<=y);
21*bf2c3715SXin Li   }
22*bf2c3715SXin Li   return r;
23*bf2c3715SXin Li }
24*bf2c3715SXin Li 
check_all_in_range(Scalar x,Scalar y)25*bf2c3715SXin Li template<typename Scalar> void check_all_in_range(Scalar x, Scalar y)
26*bf2c3715SXin Li {
27*bf2c3715SXin Li   Array<int,1,Dynamic> mask(y-x+1);
28*bf2c3715SXin Li   mask.fill(0);
29*bf2c3715SXin Li   long n = (y-x+1)*32;
30*bf2c3715SXin Li   for(long k=0; k<n; ++k)
31*bf2c3715SXin Li   {
32*bf2c3715SXin Li     mask( check_in_range(x,y)-x )++;
33*bf2c3715SXin Li   }
34*bf2c3715SXin Li   for(Index i=0; i<mask.size(); ++i)
35*bf2c3715SXin Li     if(mask(i)==0)
36*bf2c3715SXin Li       std::cout << "WARNING: value " << x+i << " not reached." << std::endl;
37*bf2c3715SXin Li   VERIFY( (mask>0).all() );
38*bf2c3715SXin Li }
39*bf2c3715SXin Li 
check_histogram(Scalar x,Scalar y,int bins)40*bf2c3715SXin Li template<typename Scalar> void check_histogram(Scalar x, Scalar y, int bins)
41*bf2c3715SXin Li {
42*bf2c3715SXin Li   Array<int,1,Dynamic> hist(bins);
43*bf2c3715SXin Li   hist.fill(0);
44*bf2c3715SXin Li   int f = 100000;
45*bf2c3715SXin Li   int n = bins*f;
46*bf2c3715SXin Li   int64 range = int64(y)-int64(x);
47*bf2c3715SXin Li   int divisor = int((range+1)/bins);
48*bf2c3715SXin Li   assert(((range+1)%bins)==0);
49*bf2c3715SXin Li   for(int k=0; k<n; ++k)
50*bf2c3715SXin Li   {
51*bf2c3715SXin Li     Scalar r = check_in_range(x,y);
52*bf2c3715SXin Li     hist( int((int64(r)-int64(x))/divisor) )++;
53*bf2c3715SXin Li   }
54*bf2c3715SXin Li   VERIFY( (((hist.cast<double>()/double(f))-1.0).abs()<0.03).all() );
55*bf2c3715SXin Li }
56*bf2c3715SXin Li 
EIGEN_DECLARE_TEST(rand)57*bf2c3715SXin Li EIGEN_DECLARE_TEST(rand)
58*bf2c3715SXin Li {
59*bf2c3715SXin Li   long long_ref = NumTraits<long>::highest()/10;
60*bf2c3715SXin Li   signed char char_offset = (std::min)(g_repeat,64);
61*bf2c3715SXin Li   signed char short_offset = (std::min)(g_repeat,16000);
62*bf2c3715SXin Li 
63*bf2c3715SXin Li   for(int i = 0; i < g_repeat*10000; i++) {
64*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<float>(10,11));
65*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<float>(1.24234523,1.24234523));
66*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<float>(-1,1));
67*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<float>(-1432.2352,-1432.2352));
68*bf2c3715SXin Li 
69*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<double>(10,11));
70*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<double>(1.24234523,1.24234523));
71*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<double>(-1,1));
72*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<double>(-1432.2352,-1432.2352));
73*bf2c3715SXin Li 
74*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<int>(0,-1));
75*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<short>(0,-1));
76*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<long>(0,-1));
77*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<int>(-673456,673456));
78*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<int>(-RAND_MAX+10,RAND_MAX-10));
79*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<short>(-24345,24345));
80*bf2c3715SXin Li     CALL_SUBTEST(check_in_range<long>(-long_ref,long_ref));
81*bf2c3715SXin Li   }
82*bf2c3715SXin Li 
83*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<signed char>(11,11));
84*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<signed char>(11,11+char_offset));
85*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<signed char>(-5,5));
86*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<signed char>(-11-char_offset,-11));
87*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<signed char>(-126,-126+char_offset));
88*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<signed char>(126-char_offset,126));
89*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<signed char>(-126,126));
90*bf2c3715SXin Li 
91*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<short>(11,11));
92*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<short>(11,11+short_offset));
93*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<short>(-5,5));
94*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<short>(-11-short_offset,-11));
95*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<short>(-24345,-24345+short_offset));
96*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<short>(24345,24345+short_offset));
97*bf2c3715SXin Li 
98*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<int>(11,11));
99*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<int>(11,11+g_repeat));
100*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<int>(-5,5));
101*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<int>(-11-g_repeat,-11));
102*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<int>(-673456,-673456+g_repeat));
103*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<int>(673456,673456+g_repeat));
104*bf2c3715SXin Li 
105*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<long>(11,11));
106*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<long>(11,11+g_repeat));
107*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<long>(-5,5));
108*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<long>(-11-g_repeat,-11));
109*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<long>(-long_ref,-long_ref+g_repeat));
110*bf2c3715SXin Li   CALL_SUBTEST(check_all_in_range<long>( long_ref, long_ref+g_repeat));
111*bf2c3715SXin Li 
112*bf2c3715SXin Li   CALL_SUBTEST(check_histogram<int>(-5,5,11));
113*bf2c3715SXin Li   int bins = 100;
114*bf2c3715SXin Li   CALL_SUBTEST(check_histogram<int>(-3333,-3333+bins*(3333/bins)-1,bins));
115*bf2c3715SXin Li   bins = 1000;
116*bf2c3715SXin Li   CALL_SUBTEST(check_histogram<int>(-RAND_MAX+10,-RAND_MAX+10+bins*(RAND_MAX/bins)-1,bins));
117*bf2c3715SXin Li   CALL_SUBTEST(check_histogram<int>(-RAND_MAX+10,-int64(RAND_MAX)+10+bins*(2*int64(RAND_MAX)/bins)-1,bins));
118*bf2c3715SXin Li }
119