Piles

/* ========================================================================= */
/* ------------------------------------------------------------------------- *//*

 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 Please read COPYING and README files in root folder
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
/* ------------------------------------------------------------------------- */
/* ========================================================================= */
#ifndef DBMODELPRIVATE_H
#define DBMODELPRIVATE_H
//
//
//
//
/*  INCLUDES    ------------------------------------------------------------ */

#include <QAbstractTableModel>
#include <QList>

#include <dbmodel/dbmodel-config.h>
#include <dbmodel/dbmodelcol.h>
#include <dbmodel/dbmodeltbl.h>
#include <dbstruct/dbstruct.h>
#include <dbstruct/dbtaew.h>

/*  INCLUDES    ============================================================ */
//
//
//
//
/*  DEFINITIONS    --------------------------------------------------------- */

#if 1
#    define DBMODEL_DEBUGM printf
#else
#    define DBMODEL_DEBUGM black_hole
#endif

#if 0
#    define DBMODEL_TRACE_ENTRY printf("DBMODEL ENTRY %s in %s[%d]\n", __func__, __FILE__, __LINE__)
#else
#    define DBMODEL_TRACE_ENTRY
#endif

#if 0
#    define DBMODEL_TRACE_EXIT printf("DBMODEL EXIT %s in %s[%d]\n", __func__, __FILE__, __LINE__)
#else
#    define DBMODEL_TRACE_EXIT
#endif

#ifndef TMP_A
#   define TMP_A(__s__) __s__.toLatin1 ().constData ()
#endif

static inline void black_hole (...)
{}

/*  DEFINITIONS    ========================================================= */
//
//
//
//
/*  CLASS    --------------------------------------------------------------- */

class DbModelPrivate : public QAbstractTableModel {
    Q_OBJECT

    //
    //
    //
    //
    /*  DEFINITIONS    ----------------------------------------------------- */

    /*  DEFINITIONS    ===================================================== */
    //
    //
    //
    //
    /*  DATA    ------------------------------------------------------------ */

private:

    DbStruct * db_; 
    QList<DbModelCol> mapping_; 
    QList<DbModelTbl> tables_; 
    int row_highlite_; 
    int col_highlite_; 
    void * user_data_; 
    /*  DATA    ============================================================ */
    //
    //
    //
    //
    /*  FUNCTIONS    ------------------------------------------------------- */

public:

    DbModelPrivate(
            DbStruct * db,
            DbTaew * meta,
            DbModel * parent);

    DbModelPrivate (
            DbStruct * db,
            int component,
            DbModel * parent);

    Q_DISABLE_COPY(DbModelPrivate)

    
    virtual ~DbModelPrivate();

    DbModel *
    parentDbModel () const;

    bool
    isValid() const {
        return  (db_ != NULL) &&
                (tables_.count() > 0) &&
                tables_.at (0).isValid();
    }

    void
    setMeta (
            DbTaew * meta);

    void
    setMeta (
            DbStruct * database,
            DbTaew * meta);


    inline DbTaew *
    takeMeta () {
        if (tables_.count() == 0)
            return NULL;
        DbTaew * result = tables_.first().metadata();
        tables_.first().setMetadata (NULL);
        terminateMeta ();
        col_highlite_ = -1;
        row_highlite_ = -1;
        return result;
    }

    inline DbTaew *
    metaTaew () const {
        if (tables_.count() == 0)
            return NULL;
        return tables_.at (0).metadata();
    }

    inline QSqlTableModel *
    mainModel () const {
        if (tables_.count() == 0)
            return NULL;
        return tables_.at (0).sqlModel ();
    }

    bool
    selectMe ();

    int
    rowCount () const;

    int
    columnCount () const;

    inline DbStruct *
        database () const {

        return db_;
    }

    void
    setDatabase (
        DbStruct * value);

    inline DbStruct *
    takeDatabase () {
        DbStruct * result = db_;
        db_ = NULL;
        setMeta (NULL);
        col_highlite_ = -1;
        row_highlite_ = -1;
        return result;
    }

    const DbModelCol &
    columnData (
        int index) const {

        return mapping_.at (index);
    }

    const DbModelTbl &
    tableData (
        int table_index) const {

        return tables_.at (table_index);
    }

    bool
    setFilter (
            const QString & filter,
            int table_index = 0);

    bool
    setFilter (
            const QString & filter,
            const QString & table);

    bool
    setOrder (
            int column,
            Qt::SortOrder order,
            int table_index = 0);

    bool
    setOrder (
            int column,
            Qt::SortOrder order,
            const QString & table);

    int
    findTable (
            const QString & table) const;

    QSqlRecord
    record (
            int row) const;

    void
    reloadHeaders ();


    /*  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  */

public:

    Qt::ItemFlags
    flags (
        const QModelIndex &index) const;

    QVariant
    data (
        const QModelIndex &idx,
        int role = Qt::DisplayRole) const;

    QVariant
    headerData (
        int section,
        Qt::Orientation orientation,
        int role = Qt::DisplayRole) const;

    bool
    setHeaderData (
        int section,
        Qt::Orientation orientation,
        const QVariant & value,
        int role = Qt::EditRole);

    bool
    setData (
        const QModelIndex &index,
        const QVariant &value,
        int role = Qt::EditRole);

    bool
    validateIndex (
            const QModelIndex &idx) const;

    void
    sort (
        int column,
        Qt::SortOrder order = Qt::AscendingOrder) {

        setOrder (column, order);
    }

    static QVariant
    formattedData (
            const DbColumn &colorig,
            const QVariant &original_value);

    bool
    removeRows (
        int row,
        int count,
        const QModelIndex &parent);


protected:

    int
    rowCount (
        const QModelIndex &parent) const;

    int
    columnCount (
        const QModelIndex &parent) const;


    /*/! Prevent the user from using this method - use `setMeta()` instead.
    virtual void
    setTable (
            const QString & value) {
        QSqlRelationalTableModel::setTable (value);
    } */

    /*  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  */


    /*  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  */

public:

    bool
    setCurrentMarker (
        int row,
        int column);

    inline int
    getMarkerRow () const {
        return row_highlite_;
    }

    inline int
    getMarkerCol () const {
        return col_highlite_;
    }

    inline bool
    hasMarkerCell () const {
        return (col_highlite_ > -1) && (row_highlite_ > -1);
    }

    /*  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  */


    bool
    setColumnCallback (
            int table_index,
            int column_index,
            DbColumn::Callback value,
            void *user_data);

    DbColumn::Callback
    columnCallback (
            int table_index,
            int column_index);

    void *
    columnCallbackData() {
        return user_data_;
    }

private:

    void
    terminateMeta ();

    bool
    loadMeta (
        DbTaew * meta);

    const DbModelTbl &
    table (
        const QString &name);

    void
    clearTables ();

    void
    addForeignKeyColumn (
        const DbColumn & col,
        int & col_idx);



    /*  FUNCTIONS    ======================================================= */
    //
    //
    //
    //
}; /* class DbModelPrivate */

/*  CLASS    =============================================================== */
//
//
//
//


#endif // DBMODELPRIVATE_H
/* ------------------------------------------------------------------------- */
/* ========================================================================= */