15.1.46. tablet_qt/db/ancillaryfunc.h

/*
    Copyright (C) 2012, University of Cambridge, Department of Psychiatry.
    Created by Rudolf Cardinal (rnc1001@cam.ac.uk).

    This file is part of CamCOPS.

    CamCOPS is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    CamCOPS is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with CamCOPS. If not, see <https://www.gnu.org/licenses/>.
*/

#pragma once
#include <QSqlDatabase>
#include <QSqlQuery>

#include "core/camcopsapp.h"
#include "db/databasemanager.h"
#include "db/sqlargs.h"

namespace ancillaryfunc {

// ============================================================================
// Assistance function to load multiple ancillary objects
// - Class must inherit from DatabaseObject
// - Class must have a constructor like SomeAncillary(app, db, pk)
// ============================================================================

// Load ancillary objects for a task (e.g. photos for a PhotoSequence).
template<class AncillaryType, class AncillaryPtrType>
void loadAncillary(
    QVector<AncillaryPtrType>& ancillaries,
    CamcopsApp& app,
    DatabaseManager& db,
    const QString& fk_name,
    const OrderBy& order_by,
    int parent_pk
)
{
    // Load all objects whose FK match the specified PK.
    ancillaries.clear();
    WhereConditions where;
    where.add(fk_name, parent_pk);
    AncillaryType specimen(app, db);
    SqlArgs sqlargs = specimen.fetchQuerySql(where, order_by);
    QueryResult result = db.query(sqlargs);
    int nrows = result.nRows();
    for (int row = 0; row < nrows; ++row) {
        auto raw_ptr_ancillary
            = new AncillaryType(app, db, dbconst::NONEXISTENT_PK);
        raw_ptr_ancillary->setFromQuery(result, row, true);
        AncillaryPtrType ancillary(raw_ptr_ancillary);
        ancillaries.append(ancillary);
    }
}

// Load *all* objects from a table.
template<class Type, class PtrType>
void loadAllRecords(
    QVector<PtrType>& objects,
    CamcopsApp& app,
    DatabaseManager& db,
    const OrderBy& order_by
)
{
    objects.clear();
    WhereConditions where;
    Type specimen(app, db);
    SqlArgs sqlargs = specimen.fetchQuerySql(where, order_by);
    QueryResult result = db.query(sqlargs);
    int nrows = result.nRows();
    for (int row = 0; row < nrows; ++row) {
        auto raw_ptr = new Type(app, db, dbconst::NONEXISTENT_PK);
        raw_ptr->setFromQuery(result, row, true);
        PtrType object(raw_ptr);
        objects.append(object);
    }
}


}  // namespace ancillaryfunc