mirror of
https://github.com/MariaDB/server.git
synced 2025-09-18 23:25:30 +02:00
b8a7728963
The initial hard capped limit on the depth was 32. It was implemented using static arrays of relevant type and size 32. Hence, to implement unlimited depth, dynamic array on mem_root was implemented which grows by 3200 as needed. Relevant arrays were replaced with this dynamic array.
114 lines
3.5 KiB
C++
114 lines
3.5 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;
|
|
MEM_ROOT current_mem_root;
|
|
bool mem_root_inited;
|
|
json_engine_t je;
|
|
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); }
|
|
void cleanup() override
|
|
{
|
|
if (mem_root_inited)
|
|
free_root(¤t_mem_root, MYF(0));
|
|
Item_str_func::cleanup();
|
|
}
|
|
};
|
|
#endif
|