[pdf_extract]Add support for parsing pin directions

Many datasheets present a table with the pin number, pin name and pin
direction columns. This patch add support for the pin direction parsing.

Author: martonmiklos

src/pdf_extract/datasheet.cpp

@@ -284,6 +284,29 @@ QRectF Datasheet::toGlobalPos(const QRectF &rect, Poppler::Page *page, int pageN
     return rect;
 }
 
+// use lowercase letters in when extending the map!
+QMap <QString, Pin::ElectricalType> Datasheet::_pinDirectionToElectricalPinTypeMap = {
+    {QStringLiteral("nc"), Pin::NotConnected},
+    {QStringLiteral("n/c"), Pin::NotConnected},
+    {QStringLiteral("oe"), Pin::OpenEmitter},
+    {QStringLiteral("open-emitter"), Pin::OpenEmitter},
+    {QStringLiteral("oc"), Pin::OpenCollector},
+    {QStringLiteral("open-collector"), Pin::OpenCollector},
+    {QStringLiteral("od"), Pin::OpenCollector},
+    {QStringLiteral("open-drain"), Pin::OpenCollector},
+    {QStringLiteral("tris"), Pin::Tristate},
+    {QStringLiteral("sup"), Pin::PowerIn},
+    {QStringLiteral("p"), Pin::Passive},
+    {QStringLiteral("pas"), Pin::Passive},
+    {QStringLiteral("passive"), Pin::Passive},
+    {QStringLiteral("io"), Pin::Bidir},
+    {QStringLiteral("i/o"), Pin::Bidir},
+    {QStringLiteral("i"), Pin::Input},
+    {QStringLiteral("in"), Pin::Input},
+    {QStringLiteral("o"), Pin::Output},
+    {QStringLiteral("out"), Pin::Output},
+};
+
 QList<DatasheetPin *> Datasheet::extractPins(int numPage)
 {
     QList<DatasheetPin *> pins;
@@ -426,7 +449,7 @@ QList<DatasheetPin *> Datasheet::extractPins(int numPage)
         DatasheetPin *pin = new DatasheetPin();
         qreal dist = 999999999999;
         QPointF center = number->pos.center();
-        DatasheetBox *assocLabel = NULL;
+        DatasheetBox *pinNameLabel = nullptr;
 
         for (int i = 0; i < _labels.count(); i++)
         {
@@ -445,27 +468,65 @@ QList<DatasheetPin *> Datasheet::extractPins(int numPage)
             if (newdist < dist)
             {
                 dist = newdist;
-                assocLabel = label;
+                pinNameLabel = label;
             }
         }
 
-        if (assocLabel)
+        if (pinNameLabel)
         {
-            assocLabel->associated = true;
+            pinNameLabel->associated = true;
             number->associated = true;
 
             pin->pin = number->text.toInt();
             pin->numPos = number->pos;
             pin->numberBox = number;
 
-            pin->name = assocLabel->text;
+            pin->name = pinNameLabel->text;
             pin->name.remove(QRegExp("\\([0-9]+\\)"));
             pin->name.remove(QRegExp(" +"));
-            pin->nameBox = assocLabel;
+            pin->nameBox = pinNameLabel;
 
             pin->page = number->page;
-            pin->pos = number->pos.united(assocLabel->pos);
+            pin->pos = number->pos.united(pinNameLabel->pos);
+
+            dist = 999999999999;
+            center = pinNameLabel->pos.center();
+            DatasheetBox *directionLabel = nullptr;
+            // loop over the labels to see if there is a pin direction label
+            // in the following columns
+            for (DatasheetBox *label : _labels) {
+                if (label == pinNameLabel)
+                    continue;
+
+                if (label->page != number->page)
+                    continue;
+
+                // look for the closest text to the pin name
+                qreal newdist = label->distanceToPoint(center);
+                if (newdist < dist)
+                {
+                    dist = newdist;
+
+                    if (!DatasheetBox::isAlign(*label, *number)
+                                         || !DatasheetBox::isAlign(*label, *pinNameLabel))
+                                        continue;
+
+                    if (!_pinDirectionToElectricalPinTypeMap.contains(label->text.toLower()))
+                        continue;
 
+                    if (label->associated)
+                        continue;
+
+                    if (label == number)
+                        continue;
+                    directionLabel = label;
+                }
+            }
+
+            if (directionLabel) {
+                directionLabel->associated = true;
+                pin->electricalType = _pinDirectionToElectricalPinTypeMap.value(directionLabel->text.toLower());
+            }
             pins.push_back(pin);
         }
     }

src/pdf_extract/datasheet.h

@@ -23,6 +23,7 @@
 
 #include <QObject>
 #include <QImage>
+#include <QMap>
 
 #include "datasheetpackage.h"
 #include "datasheetbox.h"
@@ -84,6 +85,7 @@ class DATASHEET_EXTRACTOR_EXPORT Datasheet : public QObject
     QList<DatasheetPackage *> _packages;
     bool _debug;
     bool _force;
+    static QMap <QString, Pin::ElectricalType> _pinDirectionToElectricalPinTypeMap;
 };
 
 #endif  // DATASHEET_H

src/pdf_extract/datasheetpackage.cpp

@@ -43,6 +43,7 @@ Component *DatasheetPackage::toComponent() const
     {
         Pin *pin = new Pin();
         pin->setName(dpin->name);
+        pin->setElectricalType(dpin->electricalType);
         pin->setPadName(QString::number(dpin->pin));
         comp->addPin(pin);
     }

src/pdf_extract/datasheetpin.h

@@ -22,6 +22,7 @@
 #include "pdf_extract_common.h"
 
 #include "datasheetbox.h"
+#include "../kicad/model/pin.h"
 
 #include <QString>
 #include <QRectF>
@@ -39,6 +40,7 @@ class DATASHEET_EXTRACTOR_EXPORT DatasheetPin
     QRectF pos;
     int page;
     QRectF numPos;
+    Pin::ElectricalType electricalType = Pin::Input;
 
     DatasheetBox *numberBox;
     DatasheetBox *nameBox;