BaseKeeper.h

/*
 * Copyright (C) 2024-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 BASEKEEPER_H
#define BASEKEEPER_H
 
#include <AuxFunc.h>
#include <BookBaseEntry.h>
#include <FileParseEntry.h>
#include <NotesBaseEntry.h>
#include <filesystem>
#include <functional>
#include <memory>
#include <mutex>
#include <string>
#include <vector>
 
#ifdef USE_OPENMP
#include <omp.h>
#else
#include <atomic>
#endif

class BaseKeeper
{
public:
  BaseKeeper(const std::shared_ptr<AuxFunc> &af);

  virtual ~BaseKeeper();

  void
  loadCollection(const std::string &col_name);

  std::vector<BookBaseEntry>
  searchBook(const BookBaseEntry &search,
             const double &coef_coincedence = double(0.7));

  std::vector<std::string>
  collectionAuthors();

  std::vector<BookBaseEntry>
  booksWithNotes(const std::vector<NotesBaseEntry> &notes);

  void
  stopSearch();

  void
  clearBase();

  std::vector<FileParseEntry>
  get_base_vector();

  static std::filesystem::path
  get_books_path(const std::string &collection_name,
                 const std::shared_ptr<AuxFunc> &af);

  size_t
  getBooksQuantity();

  std::function<void(const double &progr, const double &sz)> auth_show_progr;
 
private:
  FileParseEntry
  readFileEntry(const std::string &base, size_t &rb);
 
  std::vector<BookParseEntry>
  readBookEntry(const std::string &entry, size_t &rb);
 
  void
  parseBookEntry(const std::string &e, std::string &read_val, size_t &rb);
 
  enum Normalization
  {
    Author,
    Other
  };
 
  bool
  searchLineFunc(const std::string &to_search, const std::string &source,
                 const double &coef_coincidence,
                 const Normalization &variant = Normalization::Other);
 
  bool
  searchSurname(const BookBaseEntry &search,
                std::vector<BookBaseEntry> &result,
                const double &coef_coincidence);
  bool
  searchFirstName(const BookBaseEntry &search,
                  std::vector<BookBaseEntry> &result,
                  const double &coef_coincidence);
 
  bool
  searchLastName(const BookBaseEntry &search,
                 std::vector<BookBaseEntry> &result,
                 const double &coef_coincidence);
 
  void
  searchBook(const BookBaseEntry &search, std::vector<BookBaseEntry> &result,
             const double &coef_coincidence);
 
  void
  searchSeries(const BookBaseEntry &search, std::vector<BookBaseEntry> &result,
               const double &coef_coincidence);
 
  void
  searchGenre(const BookBaseEntry &search, std::vector<BookBaseEntry> &result,
              const double &coef_coincidence);
 
  bool
  exactMatchSearchFunc(const BookBaseEntry &el, const BookBaseEntry &search);
 
  void
  normalizeString(std::string &str, const Normalization &variant);
 
  std::shared_ptr<AuxFunc> af;
 
  std::vector<FileParseEntry> base;
  std::string collection_name;
  std::filesystem::path collection_path;
  size_t books_in_base = 0;
#ifdef USE_OPENMP
  omp_lock_t basemtx;
  bool cancel_search;
#else
  std::mutex basemtx;
  std::atomic<bool> cancel_search;
#endif
};
 
#endif // BASEKEEPER_H