sql_handler.h

Go to the documentation of this file.

//Copyright 2010 Thomas A Caswell
//tcaswell@uchicago.edu
//http://jfi.uchicago.edu/~tcaswell
//
//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; either version 3 of the License, or (at
//your option) any later version.
//
//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 <http://www.gnu.org/licenses>.

#ifndef SQL_HANDLER
#define SQL_HANDLER

#include<string>

// forward definition of sqlite3 class do get around include the header for sqlite3 in
// a lot of my code.
struct sqlite3;


namespace utilities
{
class Md_store;

typedef enum F_TYPE{
  F_NOFUNCTION    = 0,
  F_IDEN          = 1,
  F_IDEN_AVG      = 2,
  F_GOFR          = 3,
  F_GOFR3D        = 4,
  F_GOFR_BY_PLANE = 5,
  F_LINK3D        = 6,
  F_MSD           = 7,
  F_PHI6          = 8,
  F_TRACK_STATS   = 9,
  F_TRACKING      = 10,
  F_VANHOVE       = 11,
  F_MSD_SWEEP     = 12
}F_TYPE;

  

std::string ftype_to_str(F_TYPE f);

  
class SQL_handler
{
public:
  SQL_handler();
  ~SQL_handler();
  
  
  void open_connection(const std::string& db_name);
 
  void close_connection();
  
  int start_comp(int dset_key,
                 int & comp_key,
                 F_TYPE f_type
                 );
  
  void add_mdata(const Md_store & md_store);
  
  void commit();
  void rollback();


  void make_test_db(std::string fname = "");

  void get_comp_mdata(int comp_key,Md_store & md_store,std::string table_name);
  
protected:
  void iden_md_fun( const  Md_store& md_store);
  void tracking_md_fun( const Md_store & md_store);
  void msd_md_fun( const Md_store & md_store);
  void gofr_md_fun( const Md_store & md_store);
  void vanHove_md_fun( const Md_store & md_store);
  void gofr_by_plane_md_fun(const  Md_store & md_store);
  void msd_sweep_md_fun( const Md_store & md_store);
  void phi6_md_fun( const Md_store & md_store);

  
private:
  sqlite3 * db_;
  
  bool conn_open_;
  bool trans_open_;
  F_TYPE trans_type_;

  int comp_key_;
  
};

}


#endif