~drizzle-trunk/drizzle/development

/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-

 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:

 *

 *  Copyright (C) 2008 Sun Microsystems

 *

 *  This program is free software; you can redistribute it and/or modify

 *  it under the terms of the GNU General Public License as published by

 *  the Free Software Foundation; version 2 of the License.

 *

 *  This program is distributed in the hope that it will be useful,

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 *  GNU General Public License for more details.

 *

 *  You should have received a copy of the GNU General Public License

 *  along with this program; if not, write to the Free Software

 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 */

#include <drizzled/server_includes.h>

#include CSTDINT_H

#include <drizzled/function/str/set_collation.h>

#include <drizzled/error.h>

String *Item_func_set_collation::val_str(String *str)

{

  assert(fixed == 1);

  str=args[0]->val_str(str);

  if ((null_value=args[0]->null_value))

    return 0;

  str->set_charset(collation.collation);

  return str;

}

void Item_func_set_collation::fix_length_and_dec()

{

  const CHARSET_INFO *set_collation;

  const char *colname;

  String tmp, *str= args[1]->val_str(&tmp);

  colname= str->c_ptr();

  if (colname == binary_keyword)

    set_collation= get_charset_by_csname(args[0]->collation.collation->csname, MY_CS_BINSORT);

  else

  {

    if (!(set_collation= get_charset_by_name(colname)))

    {

      my_error(ER_UNKNOWN_COLLATION, MYF(0), colname);

      return;

    }

  }

  if (!set_collation ||

      !my_charset_same(args[0]->collation.collation,set_collation))

  {

    my_error(ER_COLLATION_CHARSET_MISMATCH, MYF(0),

             colname, args[0]->collation.collation->csname);

    return;

  }

  collation.set(set_collation, DERIVATION_EXPLICIT);

  max_length= args[0]->max_length;

}

bool Item_func_set_collation::eq(const Item *item, bool binary_cmp) const

{

  /* Assume we don't have rtti */

  if (this == item)

    return 1;

  if (item->type() != FUNC_ITEM)

    return 0;

  Item_func *item_func=(Item_func*) item;

  if (arg_count != item_func->arg_count ||

      functype() != item_func->functype())

    return 0;

  Item_func_set_collation *item_func_sc=(Item_func_set_collation*) item;

  if (collation.collation != item_func_sc->collation.collation)

    return 0;

  for (uint32_t i=0; i < arg_count ; i++)

    if (!args[i]->eq(item_func_sc->args[i], binary_cmp))

      return 0;

  return 1;

}

void Item_func_set_collation::print(String *str, enum_query_type query_type)

{

  str->append('(');

  args[0]->print(str, query_type);

  str->append(STRING_WITH_LEN(" collate "));

  assert(args[1]->basic_const_item() &&

              args[1]->type() == Item::STRING_ITEM);

  args[1]->str_value.print(str);

  str->append(')');

}