UDBase.h

/*
 * Copyright (C) 2025 Yury Bobylev <bobilev_yury@mail.ru>
 *
 * 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 3.
 *
 * 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, see <https://www.gnu.org/licenses/>.
 */
#ifndef UDBASE_H
#define UDBASE_H

#include <UDBElement.h>
#include <filesystem>
#include <functional>
#include <omp.h>
#include <vector>

class UDBase
{
public:
  UDBase();

  virtual ~UDBase();

  UDBase(const UDBase &other);

  UDBase(UDBase &&other);

  UDBase &
  operator=(const UDBase &other);

  UDBase &
  operator=(UDBase &&other);

  UDBase &
  operator+=(const UDBase &other);

  void
  writeToBuffer(std::vector<char> &result);

  void
  readFromBuffer(const std::vector<char> &buf, const size_t &offset = 0,
                 const size_t &bytes_to_read = 0);
  void
  writeToFile(const std::filesystem::path &f_path);

  void
  readFromFile(const std::filesystem::path &f_path);

  std::vector<UDBElement> *
  getRawBase() const;

  void
  addElement(const UDBElement &el);

  void
  addElements(const std::vector<UDBElement> &elements);

  virtual void
  removeElement(const UDBElement &el);

  virtual void
  removeElements(std::function<bool(const UDBElement &)> predicate);

  virtual void
  replaceElement(const UDBElement &old_el, const UDBElement &new_el);

  void
  clearBase();

  void
  shrinkToFit();

  virtual UDBase
  searchElement(std::function<bool(const UDBElement &)> search_function);

  virtual UDBase
  searchElement(
      std::function<void(const UDBElement &, UDBase &)> search_function);

  virtual std::vector<UDBElement>
  searchElementV(std::function<bool(const UDBElement &)> search_function);

  virtual void
  sortBase(std::function<bool(const UDBElement &, const UDBElement &)>
               sort_function);

  size_t
  baseSize() const;

  static UDBase
  convertToBase(const std::vector<UDBElement> &elements);

  size_t
  calculateWriteSize();

  std::vector<UDBase>
  splitBase(const size_t &bytes_limit,
            const size_t &minimum_elements = size_t(1));

protected:
  void
  readFromBufferInner(const std::vector<char> &buf, const size_t &offset,
                      const size_t &bytes_to_read);

  void
  writeToBufferInner(std::vector<char> &result);

  std::vector<UDBElement> base;

  std::string base_version = "UDB 1.0";
};

#endif // UDBASE_H