main page namespaces classes files
CPSP home

src/cpsp/HCoreDatabasePGSQL.cc

Go to the documentation of this file.

/*
 *  Main authors:
 *     Martin Mann http://www.bioinf.uni-freiburg.de/~mmann/
 *
 *  Contributing authors:
 *     Sebastian Will http://www.bioinf.uni-freiburg.de/~will/
 *
 *  Copyright:
 *     Martin Mann, 2007
 *
 *  This file is part of the CPSP-tools package:
 *     http://www.bioinf.uni-freiburg.de/sw/cpsp/
 *
 *  See the file "LICENSE" for information on usage and
 *  redistribution of this file, and for a
 *     DISCLAIMER OF ALL WARRANTIES.
 *
 */

/* Processed by ecpg (4.0.1) */
/* These include files are added by the preprocessor */
#include <ecpgtype.h>
#include <ecpglib.h>
#include <ecpgerrno.h>
#include <sqlca.h>
/* End of automatic include section */
#line 1 "HCoreDatabasePGSQL.pgc"
#include "cpsp/HCoreDatabasePGSQL.hh"
#include <biu/assertbiu.hh>
#include <biu/LatticeDescriptorCUB.hh>
#include <biu/LatticeDescriptorFCC.hh>

namespace cpsp
{
    
    const std::string HCoreDatabasePGSQL::DEF_DBNAME = "bitest";
    const std::string HCoreDatabasePGSQL::DEF_USERNAME = "mmann";
    const std::string HCoreDatabasePGSQL::DEF_PASSWORD = "R00mO14";

    HCoreDatabasePGSQL::HCoreDatabasePGSQL() :
        connected(false), cursorOpen(false) 
    {
    }
    
    HCoreDatabasePGSQL::HCoreDatabasePGSQL( const std::string& hostName, 
                            const std::string& dbName,
                            const std::string& user, 
                            const std::string& password) :
        connected(false), cursorOpen(false) 
    {
        connect(hostName, dbName, user, password);
    }
    
    HCoreDatabasePGSQL::~HCoreDatabasePGSQL()
    {
        disconnect();
    }


    bool HCoreDatabasePGSQL::connect(   const std::string& hostName, 
                            const std::string& dbName,
                            const std::string& user, 
                            const std::string& password)
    {
    #ifndef NDEBUG
        /* exec sql whenever sqlerror  sqlprint ; */

#line 39 "HCoreDatabasePGSQL.pgc"

    #endif
        if (connected)
            disconnect();
            
            // SQL-HOST-variablen erstellen
            std::string dbAtHost = dbName+"@"+hostName;
        /* exec sql begin declare section */
                 
                 
                 
        
#line 47 "HCoreDatabasePGSQL.pgc"
 const char * dbAtHostName  = dbAtHost . c_str () ;
 
#line 48 "HCoreDatabasePGSQL.pgc"
 const char * userName  = user . c_str () ;
 
#line 49 "HCoreDatabasePGSQL.pgc"
 const char * passw  = password . c_str () ;
/* exec sql end declare section */
#line 50 "HCoreDatabasePGSQL.pgc"

        
            // connect to server
        { ECPGconnect(__LINE__, 0, dbAtHostName , userName , passw , NULL, 0); 
#line 53 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 53 "HCoreDatabasePGSQL.pgc"

        
        connected = sqlca.sqlcode == ECPG_NO_ERROR;
            
            // autoconnect = on
        if (connected) {
            { ECPGsetcommit(__LINE__, "on", NULL);
#line 59 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 59 "HCoreDatabasePGSQL.pgc"
    
        }       
    #ifndef NDEBUG
        assertbiu(sqlca.sqlcode != -402, "Connection to database '"+(std::string(dbAtHostName))+"' failed.");
    #endif
        
        return connected;
    }

    bool HCoreDatabasePGSQL::disconnect() {
        bool retVal = true;
        if (cursorOpen) {
            { ECPGdo(__LINE__, 0, 1, NULL, "close tableCursor", ECPGt_EOIT, ECPGt_EORT);
#line 71 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 71 "HCoreDatabasePGSQL.pgc"

            cursorOpen = false;
        }
        if (connected) {
            { ECPGdisconnect(__LINE__, "CURRENT");
#line 75 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 75 "HCoreDatabasePGSQL.pgc"

            retVal &= sqlca.sqlcode == ECPG_NO_ERROR;
        }
        return retVal;
    }
    
    void HCoreDatabasePGSQL::closeCursor() {
        if (cursorOpen) {       // close old connection if still open
            { ECPGdo(__LINE__, 0, 1, NULL, "close tableCursor", ECPGt_EOIT, ECPGt_EORT);
#line 83 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 83 "HCoreDatabasePGSQL.pgc"

            cursorOpen = false;
            { ECPGtrans(__LINE__, NULL, "commit");
#line 85 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 85 "HCoreDatabasePGSQL.pgc"

        }
    }


    bool HCoreDatabasePGSQL::initCoreAccess(const biu::LatticeDescriptor& latDescr, const unsigned int size, 
            const unsigned int minContacts, const unsigned int maxContacts)
    {
        closeCursor();          // close current cursor if neccessary
        if (!connected) {       // if no db connection no select possible
            return false;
        }
        
    // (1) init data
        this->coreSize = size;
        this->minHH = minContacts;
        this->maxHH = maxContacts;
        
            // create SQL-HOST-variables
            
        /* exec sql begin declare section */
                
        
#line 105 "HCoreDatabasePGSQL.pgc"
 const char * s_latName  = latDescr . getName () . c_str () ;
 
#line 106 "HCoreDatabasePGSQL.pgc"
 const int  s_size  = coreSize ;
 
#line 107 "HCoreDatabasePGSQL.pgc"
 const int  s_minCont  = minHH ;
 
#line 108 "HCoreDatabasePGSQL.pgc"
 const int  s_maxCont  = maxHH ;
/* exec sql end declare section */
#line 109 "HCoreDatabasePGSQL.pgc"


    // (2) select starten
        { ECPGtrans(__LINE__, NULL, "begin transaction ");
#line 112 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 112 "HCoreDatabasePGSQL.pgc"

        /* declare tableCursor  cursor  for select  core_id , contacts , size  from core_stat where size =  ? and lat_id in ( select  lat_id  from lattice where name =  ?   ) and contacts >=  ? and contacts <=  ?   order by contacts desc , core_id */
#line 120 "HCoreDatabasePGSQL.pgc"

            // daten auslesen
        { ECPGdo(__LINE__, 0, 1, NULL, "declare tableCursor  cursor  for select  core_id , contacts , size  from core_stat where size =  ? and lat_id in ( select  lat_id  from lattice where name =  ?   ) and contacts >=  ? and contacts <=  ?   order by contacts desc , core_id", 
    ECPGt_int,&(s_size),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, 
    ECPGt_char,&(s_latName),(long)0,(long)1,(1)*sizeof(char), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, 
    ECPGt_int,&(s_minCont),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, 
    ECPGt_int,&(s_maxCont),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, ECPGt_EOIT, ECPGt_EORT);
#line 122 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 122 "HCoreDatabasePGSQL.pgc"


        cursorOpen = sqlca.sqlcode == ECPG_NO_ERROR;
        
        return cursorOpen;
    }
    
    bool HCoreDatabasePGSQL::getNextCore(HCore& toFill) {
        

        toFill.reset(INT_MAX, INT_MAX); // invalid dummy core
        if (!connected || !cursorOpen)
            return false;
            
    // (1) naechste kern ID holen
            // SQL-HOST-variablen erstellen
        /* exec sql begin declare section */
             
             
             
        
#line 139 "HCoreDatabasePGSQL.pgc"
 int  s_core_id   ;
 
#line 140 "HCoreDatabasePGSQL.pgc"
 int  s_contacts   ;
 
#line 141 "HCoreDatabasePGSQL.pgc"
 int  s_size   ;
/* exec sql end declare section */
#line 142 "HCoreDatabasePGSQL.pgc"

        
                // naechste core_id holen
        { ECPGdo(__LINE__, 0, 1, NULL, "fetch next in tableCursor", ECPGt_EOIT, 
    ECPGt_int,&(s_core_id),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, 
    ECPGt_int,&(s_contacts),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, 
    ECPGt_int,&(s_size),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, ECPGt_EORT);
#line 145 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 145 "HCoreDatabasePGSQL.pgc"

            // wenn tabellenende erreicht oder anderer fehler abbrechen
        if (sqlca.sqlcode != 0 || s_contacts < (int)minHH) {    // fehlerfall oder untere schranke unterschritten
            closeCursor();          // cursor schliessen
            return false;
        }
        
    // (2) kern reset
        toFill.reset((unsigned int)s_contacts, (unsigned int)s_size);
        
    // (3) kernpositionen holen
        /* exec sql begin declare section */
               
        
#line 157 "HCoreDatabasePGSQL.pgc"
 int  s_x   ,  s_y   ,  s_z   ;
/* exec sql end declare section */
#line 158 "HCoreDatabasePGSQL.pgc"

            // neuer temporaerer cursor
        /* declare pointCursor  cursor  for select  x , y , z  from core_point where core_id =  ?   */
#line 161 "HCoreDatabasePGSQL.pgc"
        // daten auslesen
        { ECPGdo(__LINE__, 0, 1, NULL, "declare pointCursor  cursor  for select  x , y , z  from core_point where core_id =  ?  ", 
    ECPGt_int,&(s_core_id),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, ECPGt_EOIT, ECPGt_EORT);
#line 162 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 162 "HCoreDatabasePGSQL.pgc"

        if (sqlca.sqlcode == 0) {
            { ECPGdo(__LINE__, 0, 1, NULL, "fetch next in pointCursor", ECPGt_EOIT, 
    ECPGt_int,&(s_x),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, 
    ECPGt_int,&(s_y),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, 
    ECPGt_int,&(s_z),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, ECPGt_EORT);
#line 164 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 164 "HCoreDatabasePGSQL.pgc"

            while (sqlca.sqlcode == 0) {
                toFill.addPoint(HCore::CorePoint(s_x, s_y, s_z));
                { ECPGdo(__LINE__, 0, 1, NULL, "fetch next in pointCursor", ECPGt_EOIT, 
    ECPGt_int,&(s_x),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, 
    ECPGt_int,&(s_y),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, 
    ECPGt_int,&(s_z),(long)1,(long)1,sizeof(int), 
    ECPGt_NO_INDICATOR, NULL , 0L, 0L, 0L, ECPGt_EORT);
#line 167 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 167 "HCoreDatabasePGSQL.pgc"

            }
            { ECPGdo(__LINE__, 0, 1, NULL, "close pointCursor", ECPGt_EOIT, ECPGt_EORT);
#line 169 "HCoreDatabasePGSQL.pgc"

if (sqlca.sqlcode < 0) sqlprint();}
#line 169 "HCoreDatabasePGSQL.pgc"

        }
    
        return toFill.isValid();
    }


} // namespace cpsp