← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/function/cast/boolean.cc

  • Committer: Brian Aker
  • Date: 2010-12-24 21:27:54 UTC
  • mto: (2035.1.1 clean)
  • mto: This revision was merged to the branch mainline in revision 2037.
  • Revision ID: brian@tangent.org-20101224212754-85xvjullymvhibr9
Merge in cast() for BOOLEAN.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/function/cast/boolean.cc
 
1
/* - mode: c++ c-basic-offset: 2; indent-tabs-mode: nil; -*-
 
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
 
3
 *
 
4
 *  Copyright (C) 2010 Brian Aker
 
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; either version 2 of the License, or
 
9
 *  (at your option) any later version.
 
10
 *
 
11
 *  This program is distributed in the hope that it will be useful,
 
12
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 
13
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
14
 *  GNU General Public License for more details.
 
15
 *
 
16
 *  You should have received a copy of the GNU General Public License
 
17
 *  along with this program; if not, write to the Free Software
 
18
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
19
 */
 
20
 
 
21
#include "config.h"
 
22
 
 
23
#include "drizzled/function/cast/boolean.h"
 
24
#include "drizzled/error.h"
 
25
 
 
26
namespace drizzled {
 
27
namespace function {
 
28
namespace cast {
 
29
 
 
30
void Boolean::print(String *str, enum_query_type query_type)
 
31
{
 
32
  str->append(STRING_WITH_LEN("cast("));
 
33
  args[0]->print(str, query_type);
 
34
  str->append(STRING_WITH_LEN(" as boolean)"));
 
35
}
 
36
 
 
37
drizzled::String *Boolean::val_str(drizzled::String *value)
 
38
{
 
39
  switch (args[0]->result_type())
 
40
  {
 
41
  case STRING_RESULT:
 
42
    {
 
43
      drizzled::String _res, *res;
 
44
 
 
45
      if (not (res= args[0]->val_str(&_res)))
 
46
      { 
 
47
        null_value= true; 
 
48
 
 
49
        break;
 
50
      }
 
51
      null_value= false; 
 
52
 
 
53
      if (res->length() == 1)
 
54
      {
 
55
        switch (res->ptr()[0])
 
56
        {
 
57
        case 'y': case 'Y':
 
58
        case 't': case 'T': // PG compatibility
 
59
          return evaluate(true, value);
 
60
 
 
61
        case 'n': case 'N':
 
62
        case 'f': case 'F': // PG compatibility
 
63
          return evaluate(false, value);
 
64
 
 
65
        default:
 
66
          break;
 
67
        }
 
68
      }
 
69
      else if ((res->length() == 5) and (strcasecmp(res->ptr(), "FALSE") == 0))
 
70
      {
 
71
        return evaluate(false, value);
 
72
      }
 
73
      if ((res->length() == 4) and (strcasecmp(res->ptr(), "TRUE") == 0))
 
74
      {
 
75
        return evaluate(true, value);
 
76
      }
 
77
      else if ((res->length() == 5) and (strcasecmp(res->ptr(), "FALSE") == 0))
 
78
      {
 
79
        return evaluate(false, value);
 
80
      }
 
81
      else if ((res->length() == 3) and (strcasecmp(res->ptr(), "YES") == 0))
 
82
      {
 
83
        return evaluate(true, value);
 
84
      }
 
85
      else if ((res->length() == 2) and (strcasecmp(res->ptr(), "NO") == 0))
 
86
      {
 
87
        return evaluate(false, value);
 
88
      }
 
89
 
 
90
      my_error(ER_INVALID_CAST_TO_BOOLEAN, MYF(0), res->ptr());
 
91
      return evaluate(false, value);
 
92
    }
 
93
 
 
94
  case REAL_RESULT:
 
95
  case ROW_RESULT:
 
96
  case DECIMAL_RESULT:
 
97
  case INT_RESULT:
 
98
    {
 
99
      bool tmp= args[0]->val_bool();
 
100
      null_value=args[0]->null_value;
 
101
 
 
102
      return evaluate(tmp, value);
 
103
    }
 
104
  }
 
105
 
 
106
  // We should never reach this
 
107
  return evaluate(false, value);
 
108
}
 
109
 
 
110
String *Boolean::evaluate(const bool &result, String *val_buffer)
 
111
{
 
112
  const CHARSET_INFO * const cs= &my_charset_bin;
 
113
  uint32_t mlength= (5) * cs->mbmaxlen;
 
114
 
 
115
  val_buffer->alloc(mlength);
 
116
  char *buffer=(char*) val_buffer->ptr();
 
117
 
 
118
  if (result)
 
119
  {
 
120
    memcpy(buffer, "TRUE", 4);
 
121
    val_buffer->length(4);
 
122
  }
 
123
  else
 
124
  {
 
125
    memcpy(buffer, "FALSE", 5);
 
126
    val_buffer->length(5);
 
127
  }
 
128
 
 
129
  return val_buffer;
 
130
}
 
131
 
 
132
#if 0
 
133
bool Boolean::val_bool()
 
134
{
 
135
  bool tmp= false;
 
136
 
 
137
  switch (args[0]->result_type())
 
138
  {
 
139
  case STRING_RESULT:
 
140
    {
 
141
      drizzled::String _res, *res;
 
142
 
 
143
      if (not (res= args[0]->val_str(&_res)))
 
144
      { 
 
145
        null_value= true; 
 
146
 
 
147
        break;
 
148
      }
 
149
      null_value= false; 
 
150
 
 
151
      if (res->length() == 1)
 
152
      {
 
153
        switch (res->ptr()[0])
 
154
        {
 
155
        case 'y': case 'Y':
 
156
        case 't': case 'T': // PG compatibility
 
157
          return true;
 
158
 
 
159
        case 'n': case 'N':
 
160
        case 'f': case 'F': // PG compatibility
 
161
          return false;
 
162
 
 
163
        default:
 
164
          break;
 
165
        }
 
166
      }
 
167
      else if ((res->length() == 5) and (strcasecmp(res->ptr(), "FALSE") == 0))
 
168
      {
 
169
        return false;
 
170
      }
 
171
      if ((res->length() == 4) and (strcasecmp(res->ptr(), "TRUE") == 0))
 
172
      {
 
173
        return true;
 
174
      }
 
175
      else if ((res->length() == 5) and (strcasecmp(res->ptr(), "FALSE") == 0))
 
176
      {
 
177
        return false;
 
178
      }
 
179
      else if ((res->length() == 3) and (strcasecmp(res->ptr(), "YES") == 0))
 
180
      {
 
181
        return true;
 
182
      }
 
183
      else if ((res->length() == 2) and (strcasecmp(res->ptr(), "NO") == 0))
 
184
      {
 
185
        return false;
 
186
      }
 
187
 
 
188
      break;
 
189
    }
 
190
 
 
191
  default:
 
192
    tmp= args[0]->val_bool();
 
193
    null_value=args[0]->null_value;
 
194
  }
 
195
 
 
196
  return tmp;
 
197
}
 
198
#endif
 
199
 
 
200
 
 
201
} // namespace cast
 
202
} // namespace function
 
203
} // namespace drizzled