main page namespaces classes files
CPSP home

src/cpsp/HCoreDatabasePGSQL.hh

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.
 *
 */

#ifndef HCOREDATABASEPGSQL_HH_
#define HCOREDATABASEPGSQL_HH_


#include "cpsp/HCoreDatabase.hh"
#include <string>

namespace cpsp
{
    class HCoreDatabasePGSQL : public HCoreDatabase
    {
    private:
        static const std::string DEF_DBNAME, DEF_USERNAME, DEF_PASSWORD;
        
    protected:
        
        unsigned int coreSize;          
        unsigned int minHH, maxHH;      

        bool connected;                 
        bool cursorOpen;                
        
        void closeCursor();             

    public:
        HCoreDatabasePGSQL();
        HCoreDatabasePGSQL( const std::string& hostName, 
                            const std::string& dbName = DEF_DBNAME,
                            const std::string& user = DEF_USERNAME, 
                            const std::string& password = DEF_PASSWORD);
        virtual ~HCoreDatabasePGSQL();

        virtual bool initCoreAccess(    const biu::LatticeDescriptor& latDescr, 
                                        const unsigned int size, 
                                        const unsigned int minContacts=0, 
                                        const unsigned int maxContacts=INT_MAX);
                                        
        virtual bool getNextCore(HCore& toFill);
        
        virtual unsigned int getActCoreSize() { return coreSize; }
        
        virtual unsigned int getActMinHHcontacts() { return minHH; }
        
        virtual void setActMinHHcontacts(const unsigned int minContacts) { 
            minHH = minContacts; 
        }
        
        virtual unsigned int getActMaxHHcontacts() { return maxHH; }
        
        bool connect(const std::string& dbAtServer, 
                    const std::string& dbName = DEF_DBNAME,
                    const std::string& user = DEF_USERNAME, 
                    const std::string& password = DEF_PASSWORD);
                    
        bool isConnected() const { return connected; }
        
        bool disconnect();

    };

} // namespace cpsp

#endif /*HCOREDATABASEPGSQL_HH_*/