refactor: Modify the display and interaction related to worklist.

src/forms/select/WorklistTableModel.cpp

@@ -0,0 +1,193 @@
+#include "WorklistTableModel.h"
+
+#include "event/EventCenter.h"
+
+WorklistTableModel::WorklistTableModel(QObject* aParent)
+    : QAbstractItemModel(aParent)
+    , mPatientList()
+    , mHeader(QStringList()<< tr("ID") << tr("Name") << tr("Birth Date") << tr("Gender") << tr("Add Date"))
+{
+    connect(EventCenter::Default(), &EventCenter::ReloadLanguage, this, &WorklistTableModel::reloadLanguage);
+}
+
+WorklistTableModel::~WorklistTableModel()
+{
+    qDeleteAll(mPatientList);
+}
+
+QVariant WorklistTableModel::headerData(int aSection, Qt::Orientation aOrientation, int aRole) const
+{
+    if(aOrientation == Qt::Horizontal && aRole == Qt::DisplayRole)
+    {
+        if(aSection < mHeader.size())
+        {
+            return mHeader[aSection];
+        }
+    }
+    return QVariant();
+}
+
+int WorklistTableModel::rowCount(const QModelIndex &aParent) const
+{
+    if (!aParent.isValid())
+    {
+        return mPatientList.count();
+    }
+
+    AbstractPatientInfomation* item = static_cast<AbstractPatientInfomation*>(aParent.internalPointer());
+    if(item->getType() != AbstractPatientInfomation::PatientType)
+    {
+        return 0;
+    }
+    PatientInformation* patientItem = static_cast<PatientInformation*>(item);
+    return patientItem->mAccessionList.count();
+}
+
+int WorklistTableModel::columnCount(const QModelIndex &aParent) const
+{
+    return mHeader.size();
+}
+
+QModelIndex WorklistTableModel::index(int aRow, int aColumn, const QModelIndex &aParent) const
+{
+    if (!hasIndex(aRow, aColumn, aParent))
+    {
+        return QModelIndex();
+    }
+    if (!aParent.isValid())
+    {
+        return createIndex(aRow, aColumn, mPatientList[aRow]);
+    }
+    AbstractPatientInfomation* item = static_cast<AbstractPatientInfomation*>(aParent.internalPointer());
+
+    if(item->getType() == AbstractPatientInfomation::PatientType)
+    {
+        PatientInformation* patientItem = static_cast<PatientInformation*>(item);
+        return createIndex(aRow, aColumn, patientItem->mAccessionList[aRow]);
+    }
+    return QModelIndex();
+}
+
+QVariant WorklistTableModel::data(const QModelIndex &aIndex, int aRole) const
+{
+    if (aRole != Qt::DisplayRole)
+    {
+        return QVariant();
+    }
+    if (!aIndex.isValid())
+    {
+        return QVariant();
+    }
+
+    AbstractPatientInfomation* item = static_cast<AbstractPatientInfomation*>(aIndex.internalPointer());
+    if(item->getType() == AbstractPatientInfomation::PatientType)
+    {
+        PatientInformation* patientItem = dynamic_cast<PatientInformation*>(item);
+        switch (aIndex.column())
+        {
+        case 0:return patientItem->ID;
+        case 1:return patientItem->Name;
+        case 2:return patientItem->BirthDate;
+        case 3:return patientItem->Sex == "F" ? tr("Female") : (patientItem->Sex == "M" ? tr("Male") : tr("Other"));
+        case 4:return patientItem->AddDate;
+        default:return QVariant();
+        }
+    }
+
+    if(item->getType() == AbstractPatientInfomation::AccessionType)
+    {
+        AccessionInformation* accessionItem = static_cast<AccessionInformation*>(item);
+        switch (aIndex.column())
+        {
+        case 1:return accessionItem->mAccessionNumber;
+        case 2:return accessionItem->getProtocolText();
+        case 3:return accessionItem->mScheduledStartDate;
+        default:return QVariant();
+        }
+    }
+    return QVariant();
+}
+
+QModelIndex WorklistTableModel::parent(const QModelIndex &aChild) const
+{
+    if (!aChild.isValid())
+    {
+        return QModelIndex();
+    }
+    AbstractPatientInfomation *item = static_cast<AbstractPatientInfomation*>(aChild.internalPointer());
+    if(item->getType() != AbstractPatientInfomation::AccessionType)
+    {
+        return QModelIndex();
+    }
+    AccessionInformation* accessionItem = static_cast<AccessionInformation*>(item);
+    int row = mPatientList.indexOf(accessionItem->mPatient);
+    //parent column must be 0
+    return createIndex(row, 0, accessionItem->mPatient);
+}
+
+void WorklistTableModel::reloadLanguage()
+{
+    beginResetModel();
+    mHeader = QStringList()<< tr("ID") << tr("Name") << tr("Birth Date") << tr("Gender") << tr("Add Date");
+    emit headerDataChanged(Qt::Horizontal, 0, mHeader.count() - 1);
+    for(int i=0; i<mPatientList.size(); ++i)
+    {
+        mPatientList[i]->reloadHeaderLanguage();
+    }
+    endResetModel();
+}
+
+void WorklistTableModel::instertPatients(const QList<PatientInformation*>& aPatientList)
+{
+    beginResetModel();
+    qDeleteAll(mPatientList);
+    mPatientList.clear();
+    mPatientList = aPatientList;
+    endResetModel();
+//    PatientInformation* p1 = new PatientInformation("PAT001","Sun","F","1998-08-21","2022-03-03 19:25:23");
+//    AccessionInformation* a = new AccessionInformation("a1",ScanLeft,"a3", p1, p1);
+//    AccessionInformation* b = new AccessionInformation("b1",ScanLeft,"b3", p1, p1);
+//    p1->setAccessionList(QList<AccessionInformation*>()<<a<<b);
+//    mPatientList << p1;
+
+//    PatientInformation* p2 = new PatientInformation("PAT002","wang","F","1998-08-21","2022-03-03 19:25:23");
+//    AccessionInformation* c = new AccessionInformation("c1",ScanNone,"c3", p2, p2);
+//    AccessionInformation* d = new AccessionInformation("c1",ScanLeft,"c3", p2, p2);
+//    p2->setAccessionList(QList<AccessionInformation*>()<<c<<d);
+//    mPatientList << p2 ;
+
+//    PatientInformation* p;
+//    AccessionInformation* ac;
+//    p = new PatientInformation("PAT003","jed","F","1998-08-21","2022-03-03 19:25:23");
+//    ac = new AccessionInformation("left",ScanLeft,"1998-08-21", p, p);
+//    p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//    ac = new AccessionInformation("right",ScanRight,"1998-08-21", p, p);
+//    p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//    mPatientList << p;
+
+//    for(int i=0; i<20; ++i)
+//    {
+//        p = new PatientInformation("PAT005","abc","F","1998-08-21","2022-03-03 19:25:23");
+//        ac = new AccessionInformation("a1",ScanNone,"a3", p, p);
+//        p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//        ac = new AccessionInformation("a1",ScanLeft,"a3", p, p);
+//        p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//        ac = new AccessionInformation("a1",ScanRight,"a3", p, p);
+//        p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//        ac = new AccessionInformation("a1",ScanRight,"a3", p, p);
+//        p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//        ac = new AccessionInformation("a1",ScanNone,"a3", p, p);
+//        p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//        ac = new AccessionInformation("a1",ScanNone,"a3", p, p);
+//        p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//        ac = new AccessionInformation("a1",ScanLeftRight,"a3", p, p);
+//        p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//        ac = new AccessionInformation("a1",ScanLeftRight,"a3", p, p);
+//        p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//        ac = new AccessionInformation("a1",ScanNone,"a3", p, p);
+//        p->setAccessionList(QList<AccessionInformation*>()<<ac);
+//        mPatientList << p;
+//    }
+
+
+}