1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
/**********************************************************************
* file: sqliteDB.hh
* licence: Artistic Licence, see file LICENCE.TXT or
* http://www.opensource.org/licenses/artistic-license.php
* descr.: wrapper class around the SQLite interface
* authors: Stefanie Koenig
*
*********************************************************************/
#ifndef __SQLITEDB_H__
#define __SQLITEDB_H__
#include <string>
#include <vector>
#include <sqlite3.h>
using namespace std;
//forward declarations
class SQLiteDB;
class Statement;
/* database opening flags:
* ro (read-only mode, throws and error if db does not exist, default flag)
* rw (writing only possible if db is not write protected, throws and error if db does not exist)
* crw (reading and writing, creates db if it does not exist)
*/
enum OpenMode{ro,rw,crw};
class SQLiteDB{
public:
SQLiteDB(const char* f, OpenMode m=ro) : dbfile(f), database(NULL) {
open(m);
}
~SQLiteDB(){
close();
}
void connect(const char* filename);
void close();
void exec(const char *sql);
void open(OpenMode m);
// create tables
bool tableExists(string table_name);
void createTableGenomes();
void createTableSpeciesnames();
void createTableSeqnames();
void createTableHints();
void createTableFeatureTypes();
int getSpeciesID(string species, bool clean=false, bool noInsert=false);
void deleteHints(int speciesid);
void deleteGenome(int speciesid);
void deleteSeqNames(int speciesid);
void beginTransaction();
void endTransaction();
inline int lastInsertID(){return (int)sqlite3_last_insert_rowid(database);}
inline const char* error(){return sqlite3_errmsg(database);}
inline int numChanges(){return sqlite3_changes(database);}
friend class Statement;
private:
const char* dbfile;
sqlite3 *database;
};
class Statement
{
public:
Statement(SQLiteDB* db) : stmt(NULL), database(db->database) {}
~Statement(){finalize();}
void prepare(const char *sql);
void step();
bool nextResult();
void bindInt(int idx, int x);
void bindInt64(int idx, uint64_t x);
void bindDouble(int idx, double d);
void bindText(int idx, const char* text);
inline void reset(){sqlite3_reset(stmt);}
inline void finalize(){sqlite3_finalize(stmt);}
inline int numCols(){return sqlite3_column_count(stmt);}
inline int intColumn(int colNum){return sqlite3_column_int(stmt,colNum);}
inline bool boolColumn(int colNum){return sqlite3_column_int(stmt,colNum) != 0;}
inline uint64_t int64Column(int colNum){return (uint64_t)sqlite3_column_int64(stmt,colNum);}
inline double doubleColumn(int colNum){return sqlite3_column_double(stmt,colNum);}
inline char* textColumn(int colNum){return (char*)sqlite3_column_text(stmt,colNum);}
inline const char* error(){return sqlite3_errmsg(database);}
private:
sqlite3_stmt *stmt;
sqlite3 *database;
};
#endif
|
