creaImageIOSQLiteTreeHandler.h

Go to the documentation of this file.

#ifndef __creaImageIOSQLiteTreeHandler_h_INCLUDED__
#define __creaImageIOSQLiteTreeHandler_h_INCLUDED__

#include <creaImageIOTreeHandler.h>
#include <creaImageIOTree.h>

class CppSQLite3DB;

namespace creaImageIO
{

  
  //=======================================================================
  class SQLiteTreeHandler : virtual public TreeHandler
  {
  public:
    //====================================================================
    SQLiteTreeHandler(const std::string& filename);
    virtual ~SQLiteTreeHandler();
    //====================================================================

    //====================================================================
    const std::string& GetFileName() const { return mFileName; }
    //====================================================================
 
    //====================================================================
    // QUERY METHODS
    virtual bool IsReadable() { return true; }
    virtual bool IsWritable() { return true; }
    //====================================================================


    //====================================================================
    // INITIALIZATION / FINALIZATION
    //====================================================================

    //====================================================================
    // Default mode is read only 
    // If IsWritable and writable==true then opens in read/write mode
    virtual bool Open(bool writable = false);
    virtual bool Close();
    // Default mode is read only 
    // If IsWritable and writable==true then opens in read/write mode
    virtual bool Create(bool writable = false);
    virtual bool Destroy();
        virtual void BeginTransaction();
        virtual void EndTransaction();
    //====================================================================


    //====================================================================
    // READ METHODS
    //====================================================================


    //====================================================================
    // REM : The Tree itself is a Node and asking for its number of 
    //       children returns the number of children of level 1.
    virtual unsigned int GetNumberOfChildren(tree::Node* n);
    //====================================================================

        //====================================================================
        virtual void GetAttribute(std::string levelDescriptor,
                                                                           std::string searchParam, 
                                                                           std::string searchVal, 
                                                                           std::string key, 
                                                                           std::string& result);
    //====================================================================


    //====================================================================
    // is reached.
    // If parent == NULL or parent == tree then starts with the 'children' of 
    // the tree itself.
    // Returns the total number of children loaded.
    virtual int LoadChildren(tree::Node* parent, int maxlevel);
    //====================================================================

    //====================================================================
    // WITHOUT altering the source, e.g. the database
    virtual void UnLoad(tree::Node* n);
        
        //====================================================================
    virtual void GetTopLevelNodeId(const std::string& searchParam, 
                                                                                        const std::string& searchValue, 
                                                                                        std::string& parent_id);

    //====================================================================
    // WRITE METHODS : WORK ONLY IN WRITE MODE
    //====================================================================
    // returns the Level in the tree where the branch was connected 
    // (-1 for error, 0 for top level, etc. ) 
    // Of course the branch is loaded on exit
    virtual int AddBranch( const AttributeMapType& attr );
    // Removes the node and its descendants 
    virtual bool Remove(tree::Node*);
    // Sets an attribute of a Node
    virtual bool SetAttribute(tree::Node*, 
                              const std::string& key,
                              const std::string& value);
        // Sets an attribute
    virtual void SetAttribute(const std::string& levelDescriptor, 
                              const std::string& key,
                              const std::string& value,
                                  const std::string& searchParam, 
                                  const std::string& searchVal);
        //Deletes the tuple that matches the parameters given
        virtual void DeleteTuple(std::string levelDescriptor, std::string key, std::string value);
        //Deletes the entries that match the parameters given
        virtual void RemoveEntries(const std::string i_table, 
                const std::string i_attribute, 
                const std::string i_operand, 
                const std::string i_val);
 
    //====================================================================
        void getAllAttributes(std::string i_filename, std::map<std::string, std::string> &i_results);
        
        void GetAttributes(std::string name, std::string i_id, std::string i_value, tree::LevelDescriptor::AttributeDescriptorListType i_attr, std::vector<std::string> &i_results);
        
        void GetUpLevelNodeId(int level, const std::string& searchParam, const std::string& searchValue, std::string& parent_id);

  protected:
    //======================================================================
    bool DBOpen();
    bool DBImportTreeDescription();
    //======================================================================
    //======================================================================
    // Creation
    bool DBCreate();
    void SQLAppendAttributesDefinition(int level, std::string& s);
    //======================================================================

    //======================================================================

    // provided must be grafted 
    tree::Node* DBGetParent( const AttributeMapType& attr);
    //======================================================================

    //======================================================================

    // Can recurse to numberoflevels levels
    // \return The total number of Node loaded (may be at different levels)
    int DBLoadChildren( tree::Node* parent, int numberoflevels = 1);
    //======================================================================

    //======================================================================

    // of node n
    void SQLAppendAttributesValues(tree::Node* n, std::string& s);
    //======================================================================

    //======================================================================

    void DBGraftToParent( tree::Node* parent, const AttributeMapType& attr);
    //======================================================================
    //======================================================================

    bool DBSetAttribute(tree::Node*, 
                        const std::string& key,
                        const std::string& value);
    //======================================================================
    //======================================================================
        void DBSetAttribute(const std::string& levelDescriptor, 
                              const std::string& key,
                              const std::string& value,
                                  const std::string& searchParam, 
                                  const std::string& searchVal);
    //======================================================================
    //======================================================================
        
    void DBInsert(tree::Node* n);
    //======================================================================

        
        //======================================================================

        void DBDelete(std::string levelDescriptor, std::string key, std::string value);
    //======================================================================

        //======================================================================

    void DBRecursiveRemoveNode(tree::Node* node);
        void DBRecursiveRemoveNode(int level, std::string parentId);
 
        //======================================================================

  
  private:
    CppSQLite3DB* mDB;
    std::string mFileName;
    bool mWritable;
    void SetWritable(bool w) { mWritable = w; }
    bool GetWritable() const { return mWritable; }
        bool mIsAdding;
        const std::string convert(const std::string &i_word);

   
  };
  // EO class SQLiteTreeHandler
  //=======================================================================


} // EO namespace creaImageIO

// EOF
#endif