tkdb_table.h

/********************************************************************************
 *   Copyright (c) : Université de Lyon 1, CNRS/IN2P3, UMR5822,                 *
 *                   IP2I, F-69622 Villeurbanne Cedex, France                   *
 *                   Normandie Université, ENSICAEN, UNICAEN, CNRS/IN2P3,       *
 *                   LPC Caen, F-14000 Caen, France                             *
 *   Contibutor(s) :                                                            *
 *      Jérémie Dudouet jeremie.dudouet@cnrs.fr [2020]                          *
 *      Diego Gruyer    diego.gruyer@cnrs.fr    [2020]                          *
 *                                                                              *
 *   Licensed under the MIT License <http://opensource.org/licenses/MIT>.       *   
 *   SPDX-License-Identifier: MIT                                               *
 ********************************************************************************/
 
#ifndef tkdb_table_H
#define tkdb_table_H
 
#include <map>
#include <iostream>
#include <array>
#include <vector>
 
#include "sqlite3.h"
#include "tkstring.h"
#include "tkdb_column.h"
 
namespace tkn {
 
typedef std::map<tkstring,tkdb_column> row;
 
class tkdb_table
{
 
    /* data types in sqlite3
#define SQLITE_INTEGER  1
#define SQLITE_FLOAT    2
#define SQLITE_BLOB     4
#define SQLITE_NULL     5
#ifdef SQLITE_TEXT
# undef SQLITE_TEXT
#else
# define SQLITE_TEXT     3
#endif
#define SQLITE3_TEXT     3
    */
 
public:

    struct measure_data_struct {
        // never ever ever change the order of the fields
        double value    = 0.;
        double err      = 0.;
        double err_low  = 0.;
        double err_high = 0.;
        tkstring unit   = "";
        tkstring info   = "";
        bool filled     = false;
 
        void clear() {
            value    = 0.;
            err      = 0.;
            err_low  = 0.;
            err_high = 0.;
            unit     = "";
            info     = "";
            filled   = false;
        }
 
        void print() {
            std::cout<<"value   : "<<value<<std::endl;
            std::cout<<"unit    : "<<unit<<std::endl;
            std::cout<<"err     : "<<err<<std::endl;
            std::cout<<"err_low : "<<err_low<<std::endl;
            std::cout<<"err_high: "<<err_high<<std::endl;
            std::cout<<"info    : "<<info<<std::endl;
        }
    };
 
    using measure_columns = std::array<tkdb_column *, 6>;
 
protected:
    tkstring fName;
    sqlite3* fDataBase{}; // pointer to the sql database owning the table
    row fColumns;       // map of db_columns ('name', column)
    std::vector<tkstring> fOrderedColumnNames; // column name in insertion order
    sqlite3_stmt* fSQL_stmt{}; // sql statement to select rows and loop over them (begin() next() end())
    sqlite3_stmt* fInsertStatement{}; // reusable prepared INSERT statement
    bool fReading = false;
    tkstring fConstraint;
 
   public:
   tkdb_table(const char* _name, sqlite3 *_db);
   tkdb_table(){}
   tkdb_table(const tkdb_table &_other);
   tkdb_table &operator=(const tkdb_table &_other);
   tkdb_table(tkdb_table &&_other) noexcept;
   tkdb_table &operator=(tkdb_table &&_other) noexcept;
   virtual ~tkdb_table();
 
   tkstring get_name(){return fName;}
 
   tkdb_column &operator[](const char* _column) {
            return fColumns[_column];
         }
 
   void print(const tkstring &_opt="");
   void print_columns(const tkstring &_opt="");
   void print_rows(const char* _selection, const char *_properties="*");
 
   void begin(tkstring _condition, tkstring _selection="*"); 
   bool next(); 
   void end(); 
 
   void load();
   void add_column(const char *_colname, const char *_coltype);
   void add_constraint(const char *_constraint);
   bool has_column(const tkstring &_col_name);
   tkdb_column *get_column(const tkstring &_col_name);
   void reset_column_values();
   void push_row();
   void update_row(const tkstring &_where);
   void write_to_database();
   void push_dummy_row();
   row & get_columns(){return fColumns;}
 
   void read_measure(measure_data_struct &_struct, const tkstring &_col_base_name);
   measure_columns get_measure_columns(const tkstring &_col_base_name);
   void read_measure(measure_data_struct &_struct, const measure_columns &_columns);
 
private:
   void finalize_statements();
   bool prepare_insert_statement();
   int exec_sql(const char *_cmd);
 
#ifdef HAS_ROOT
   ClassDef(tkdb_table,0);
#endif
 
};
}
#endif