mirror of
https://github.com/MariaDB/server.git
synced 2025-10-30 18:36:12 +01:00
105 lines
3.3 KiB
C++
105 lines
3.3 KiB
C++
#ifndef ITEM_VECTORFUNC_INCLUDED
|
|
#define ITEM_VECTORFUNC_INCLUDED
|
|
|
|
/* Copyright (C) 2023, MariaDB
|
|
|
|
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 Street, Fifth Floor, Boston, MA 02110-1335 USA */
|
|
|
|
/* This file defines all vector functions */
|
|
#include <my_global.h>
|
|
#include "item.h"
|
|
#include "lex_string.h"
|
|
#include "item_func.h"
|
|
|
|
class Item_func_vec_distance: public Item_real_func
|
|
{
|
|
Item_field *get_field_arg() const
|
|
{
|
|
if (args[0]->real_item()->type() == Item::FIELD_ITEM && args[1]->const_item())
|
|
return (Item_field*)(args[0]->real_item());
|
|
if (args[1]->real_item()->type() == Item::FIELD_ITEM && args[0]->const_item())
|
|
return (Item_field*)(args[1])->real_item();
|
|
return NULL;
|
|
}
|
|
bool check_arguments() const override
|
|
{
|
|
return check_argument_types_or_binary(NULL, 0, arg_count);
|
|
}
|
|
double (*calc_distance)(float *v1, float *v2, size_t v_len);
|
|
|
|
public:
|
|
enum distance_kind { EUCLIDEAN, COSINE, AUTO } kind;
|
|
Item_func_vec_distance(THD *thd, Item *a, Item *b, distance_kind kind);
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name[3]= {
|
|
{ STRING_WITH_LEN("VEC_DISTANCE_EUCLIDEAN") },
|
|
{ STRING_WITH_LEN("VEC_DISTANCE_COSINE") },
|
|
{ STRING_WITH_LEN("VEC_DISTANCE") }
|
|
};
|
|
return name[kind];
|
|
}
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
double val_real() override;
|
|
Item *get_const_arg() const
|
|
{
|
|
if (args[0]->real_item()->type() == Item::FIELD_ITEM && args[1]->const_item())
|
|
return args[1];
|
|
if (args[1]->real_item()->type() == Item::FIELD_ITEM && args[0]->const_item())
|
|
return args[0];
|
|
return NULL;
|
|
}
|
|
key_map part_of_sortkey() const override;
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_vec_distance>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_vec_totext: public Item_str_ascii_checksum_func
|
|
{
|
|
bool check_arguments() const override
|
|
{
|
|
return check_argument_types_or_binary(NULL, 0, arg_count);
|
|
}
|
|
|
|
public:
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
Item_func_vec_totext(THD *thd, Item *a);
|
|
String *val_str_ascii(String *buf) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= { STRING_WITH_LEN("VEC_ToText") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_vec_totext>(thd, this); }
|
|
};
|
|
|
|
|
|
class Item_func_vec_fromtext: public Item_str_func
|
|
{
|
|
String tmp_js;
|
|
public:
|
|
bool fix_length_and_dec(THD *thd) override;
|
|
Item_func_vec_fromtext(THD *thd, Item *a);
|
|
String *val_str(String *buf) override;
|
|
LEX_CSTRING func_name_cstring() const override
|
|
{
|
|
static LEX_CSTRING name= { STRING_WITH_LEN("VEC_FromText") };
|
|
return name;
|
|
}
|
|
Item *do_get_copy(THD *thd) const override
|
|
{ return get_item_copy<Item_func_vec_fromtext>(thd, this); }
|
|
};
|
|
#endif
|