pin_tree_model.cpp

Go to the documentation of this file.

#include "gui/selection_details_widget/gate_details_widget/pin_tree_model.h"
 
#include "gui/gui_globals.h"
#include "hal_core/netlist/gate.h"
#include "hal_core/netlist/gate_library/gate_type.h"
#include "hal_core/netlist/net.h"
#include "hal_core/utilities/enums.h"
 
#include <QDebug>
#include <QVector>
 
namespace hal
{
 
    PinTreeItem::PinTreeItem(const std::string &pinName, QString pinDirection, QString pinType, QString netName)
        :mPinName(pinName), mPinDirection(pinDirection), mPinType(pinType), mNetName(netName)
    {;}
 
    PinTreeItem::PinTreeItem()
    {;}
 
    QVariant PinTreeItem::getData(int index) const
    {
        switch (index)
        {
        case 0: {
            QVariant qvPinName = QVariant(QString::fromStdString(mPinName));
            return qvPinName;
            break;}
        case 1: {
            QVariant qvPinDirection  = QVariant(mPinDirection);
            return qvPinDirection;
            break;}
        case 2: {
            QVariant qvPinType = QVariant(mPinType);
            return qvPinType;
            break;}
        case 3: {
            QVariant qvNetName = QVariant(mNetName);
            return qvNetName;
            break;}
        }
        return QVariant();
    }
 
    void PinTreeItem::setData(QList<QVariant> data)
    {
        mPinName = data[0].toString().toStdString();
        mPinDirection = data[1].toString();
        mPinType = data[2].toString();
        mNetName = data[3].toString();
    }
 
    void PinTreeItem::setDataAtIndex(int index, QVariant &data)
    {
        switch (index)
        {
        case 0: {
            mPinName = data.toString().toStdString();
            break;}
        case 1: {
            mPinDirection = data.toString();
            break;}
        case 2: {
            mPinType = data.toString();
            break;}
        case 3: {
            mNetName = data.toString();
            break;}
        }
 
 
    }
 
    void PinTreeItem::appendData(QVariant data) {}
 
    int PinTreeItem::getColumnCount() const
    {
        return 4;
    }
 
    GatePinsTreeModel::GatePinsTreeModel(QObject* parent) : BaseTreeModel(parent)
    {
        setHeaderLabels(QStringList() << "Name"
                                      << "Direction"
                                      << "Type"
                                      << "Connected Net");
 
        //added to store a list of (multiple) net ids in a given treeitem (perhaps dont do this
        //at all, handle it in the view? (since the gate-id and pin name is accessable, the nets can be evaluated there
        qRegisterMetaType<QList<int>>();
    }
 
    GatePinsTreeModel::~GatePinsTreeModel()
    {
        delete mRootItem;
    }
 
    void GatePinsTreeModel::clear()
    {
        BaseTreeModel::clear();
        mPinGroupToTreeItem.clear();
        mGateId = -1;
    }
 
    void GatePinsTreeModel::setGate(Gate* g)
    {
        clear();
        mGateId = g->get_id();
 
        beginResetModel();
        GateType* gateType = g->get_type();
        for (auto pin : gateType->get_pins())
        {
            PinTreeItem* pinItem = new PinTreeItem();
            //get all infos for that pin
            const std::string& grouping = pin->get_group().first->get_name();
            PinDirection direction      = pin->get_direction();
            QString pinDirection        = QString::fromStdString(enum_to_string(direction));
            QString pinType             = QString::fromStdString(enum_to_string(pin->get_type()));
 
            //evaluate netname (in case of inout multiple possible nets), method depends on pindirection (kind of ugly switch)
            QString netName = "";
            QList<u32> netIDs;
            switch (direction)
            {
                case PinDirection::input:
                    if (g->get_fan_in_net(pin))
                    {
                        netName = QString::fromStdString(g->get_fan_in_net(pin)->get_name());
                        netIDs.append(g->get_fan_in_net(pin)->get_id());
                    }
                    break;
                    //netName = (g->get_fan_in_net(pin)) ? QString::fromStdString(g->get_fan_in_net(pin)->get_name()) : "" ; break;
                case PinDirection::output:
                    if (g->get_fan_out_net(pin))
                    {
                        netName = QString::fromStdString(g->get_fan_out_net(pin)->get_name());
                        netIDs.append(g->get_fan_out_net(pin)->get_id());
                    }
                    break;
                    //netName = (g->get_fan_out_net(pin)) ? QString::fromStdString(g->get_fan_out_net(pin)->get_name()) : "" ; break;
                case PinDirection::inout:    //must take input and output net into account
                    if (g->get_fan_in_net(pin))
                    {
                        netName += QString::fromStdString(g->get_fan_in_net(pin)->get_name());
                        netIDs.append(g->get_fan_in_net(pin)->get_id());
                    }
                    if (g->get_fan_out_net(pin))
                    {
                        if (!netName.isEmpty())
                            netName += " / ";    //add / when there is a input net to seperate it from the output net
                        netName += QString::fromStdString(g->get_fan_out_net(pin)->get_name());
                        netIDs.append(g->get_fan_out_net(pin)->get_id());
                    }
                    break;
                default:
                    break;    //none and internal, dont know how to handle internal (whatever an internal pin is)
            }
 
            pinItem->setData(QList<QVariant>() << QString::fromStdString(pin->get_name()) << pinDirection << pinType << netName);
            pinItem->setType(PinTreeItem::Pin);
            pinItem->setNetIds(netIDs);
            if (!grouping.empty())
            {
                PinTreeItem* pingroupItem = dynamic_cast<PinTreeItem*>(mPinGroupToTreeItem.value(grouping, nullptr));    //since its a map, its okay
                if (!pingroupItem)
                {
                    //assume all items in the same grouping habe the same direction and type, so the grouping-item has also these types
                    pingroupItem = new PinTreeItem(grouping, pinDirection, pinType, "");
                    pingroupItem->setType(PinTreeItem::Group);
                    mRootItem->appendChild(pingroupItem);
                    mPinGroupToTreeItem.insert(grouping, pingroupItem);
                }
                pingroupItem->appendChild(pinItem);
            }
            else
                mRootItem->appendChild(pinItem);
        }
        endResetModel();
    }
 
    int GatePinsTreeModel::getCurrentGateID()
    {
        return mGateId;
    }
 
    int GatePinsTreeModel::getNumberOfDisplayedPins()
    {
        Gate* g = gNetlist->get_gate_by_id(mGateId);
        if (!g)
            return 0;
 
        return g->get_type()->get_pins().size();
    }
 
 
 
}    // namespace hal