← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/function/math/rand.cc

Reverted 1103

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/function/math/rand.cc
 
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
 
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
 
3
 *
 
4
 *  Copyright (C) 2008 Sun Microsystems
 
5
 *
 
6
 *  This program is free software; you can redistribute it and/or modify
 
7
 *  it under the terms of the GNU General Public License as published by
 
8
 *  the Free Software Foundation; version 2 of the License.
 
9
 *
 
10
 *  This program is distributed in the hope that it will be useful,
 
11
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
 *  GNU General Public License for more details.
 
14
 *
 
15
 *  You should have received a copy of the GNU General Public License
 
16
 *  along with this program; if not, write to the Free Software
 
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
18
 */
 
19
 
 
20
#include <drizzled/server_includes.h>
 
21
#include CSTDINT_H
 
22
#include <drizzled/function/math/rand.h>
 
23
#include <drizzled/session.h>
 
24
 
 
25
void Item_func_rand::seed_random(Item *arg)
 
26
{
 
27
  /*
 
28
    TODO: do not do reinit 'rand' for every execute of PS/SP if
 
29
    args[0] is a constant.
 
30
  */
 
31
  uint64_t tmp= (uint64_t) arg->val_int();
 
32
  _seed_random_int(tmp * 0x10001L + 55555555L, tmp * 0x10000001L);
 
33
}
 
34
 
 
35
void Item_func_rand::_seed_random_int(uint64_t new_seed1, uint64_t new_seed2)
 
36
{
 
37
  max_value= 0x3FFFFFFFL;
 
38
  max_value_dbl=(double) max_value;
 
39
  seed1= new_seed1 % max_value;
 
40
  seed2= new_seed2 % max_value;
 
41
}
 
42
 
 
43
bool Item_func_rand::fix_fields(Session *session,Item **ref)
 
44
{
 
45
  if (Item_real_func::fix_fields(session, ref))
 
46
    return true;
 
47
 
 
48
  used_tables_cache|= RAND_TABLE_BIT;
 
49
  if (arg_count)
 
50
  {                                     // Only use argument once in query
 
51
    /*
 
52
      No need to send a Rand log event if seed was given eg: RAND(seed),
 
53
      as it will be replicated in the query as such.
 
54
    */
 
55
    if (args[0]->const_item())
 
56
      seed_random(args[0]);
 
57
  }
 
58
  else
 
59
  {
 
60
    uint64_t tmp= sql_rnd();
 
61
    _seed_random_int(tmp + (uint64_t) ref, tmp + (uint64_t) session->thread_id);
 
62
  }
 
63
 
 
64
  return false;
 
65
}
 
66
 
 
67
void Item_func_rand::update_used_tables()
 
68
{
 
69
  Item_real_func::update_used_tables();
 
70
  used_tables_cache|= RAND_TABLE_BIT;
 
71
}
 
72
 
 
73
double Item_func_rand::val_real()
 
74
{
 
75
  assert(fixed == 1);
 
76
  if (arg_count && !args[0]->const_item())
 
77
    seed_random(args[0]);
 
78
 
 
79
  seed1= (seed1 * 3 + seed2) % max_value;
 
80
  seed2= (seed1 + seed2 + 33) % max_value;
 
81
  return (((double) seed1) / max_value_dbl);
 
82
}