parser.h

Go to the documentation of this file.

 
#ifndef PARSER_H
#define PARSER_H
 
#include <QThread>
#include <QHash>
#include <QStringList>
#include <QPointF>
#include <QObject>
#include <QMultiMap>
#include <QDebug>
#include <QLoggingCategory>
#include <memory>
#include "graph/io/graph_parse_sink.h"
 
class QXmlStreamReader;
class QXmlStreamAttributes;
 
// WS14 L3: shared across parser.cpp and all parser_*.cpp files (they all include this header) --
// one category for the whole parsing subsystem rather than one per format, since a user debugging
// a load issue wants "parser debug" on regardless of which format it turns out to be.
Q_DECLARE_LOGGING_CATEGORY(lcParser)
 

struct Actor
{
    QString key;
    int value;
};

class CompareActors
{
public:
    bool operator()(Actor &t1, Actor &t2)
    {
        if (t1.value == t2.value)
            return t1.key > t2.key;
        return t1.value > t2.value; // minimum priority
        // Returns true if t2.value smaller than t1.value
    }
};

class Parser : public QObject
{
    Q_OBJECT
 
private:
    bool validateAndInitialize(const QByteArray &rawData, const QString &delimiter, const bool &sm_has_labels, QStringList &nodeLabels);
    void resetCounters();
    bool doParseAdjacency(QTextStream &ts, const QString &delimiter, const QStringList &nodeLabels);
    void createNodeWithDefaults(int nodeIndex, const QString &label);
    bool createEdgesForRow(const QStringList &currentRow, int rowIndex);
    bool containsReservedKeywords(const QString &str) const;

    struct ParseConfig
    {
        QString fileName;
        QString codecName;
 
        int fileFormat;
        QString delim;
        int sm_mode;
        bool sm_has_labels;
 
        int initNodeSize;
        QString initNodeColor;
        QString initNodeShape;
        QString initNodeNumberColor;
        int initNodeNumberSize;
        QString initNodeLabelColor;
        int initNodeLabelSize;
 
        QString initEdgeColor;
 
        int gwWidth;
        int gwHeight;
    };
 
    SocNetV::IO::IGraphParseSink *m_parseSink = nullptr;
    std::unique_ptr<SocNetV::IO::IGraphParseSink> m_ownedParseSink;
    QHash<QString, int> nodeHash;
    QHash<QString, QString> keyFor, keyName, keyType, keyDefaultValue;
    QHash<QString, QString> edgesMissingNodesHash;
    QStringList edgeMissingNodesList, edgeMissingNodesListData, relationsList;
    QMultiMap<int, int> firstModeMultiMap, secondModeMultiMap;
    QXmlStreamReader *xml;
    QString fileDirPath;
    QString m_textCodecName;
    QString networkName;
    QString initNodeColor, initNodeShape, initNodeCustomIcon;
    QString initNodeNumberColor, initNodeLabelColor;
    QString initEdgeColor, initEdgeLabel, delimiter;
    QString errorMessage;
    QString nodeColor, edgeColor, edgeType, nodeShape, nodeLabel, edgeLabel;
    QString nodeIconPath;
    QString nodeNumberColor, nodeLabelColor;
    QHash<QString, QString> nodeCustomAttributes, initNodeCustomAttributes;
    QHash<QString, QString> edgeCustomAttributes;
    QString key_id, key_value, key_name, key_what, key_type;
    QString node_id, edge_id, edge_source, edge_target, edge_weight, edge_directed;
    int gwWidth, gwHeight;
    int totalLinks, totalNodes, fileFormat, two_sm_mode, edgeDirType;
    int initNodeSize, initNodeNumberSize, nodeNumberSize, initNodeLabelSize;
    int nodeLabelSize, source, target, nodeSize;
    qreal initEdgeWeight, edgeWeight, arrowSize;
    qreal bez_p1_x, bez_p1_y, bez_p2_x, bez_p2_y;
    bool fileLoaded, missingNode;
    bool arrows, bezier, conv_OK;
    bool bool_key, bool_node, bool_edge, fileContainsNodeColors;
    bool fileContainsNodeCoords, fileContainsLinkColors;
    bool fileContainsLinkLabels;
    double randX, randY;
 
public:
    Parser();
    ~Parser();
    void setParseSink(SocNetV::IO::IGraphParseSink *sink);
    void setOwnedParseSink(std::unique_ptr<SocNetV::IO::IGraphParseSink> sink);
    void load(const QString &fileName, const QString &codecName, const int &defNodeSize,
              const QString &defNodeColor, const QString &defNodeShape, const QString &defNodeNumberColor,
              const int &defNodeNumberSize, const QString &defNodeLabelColor, const int &defNodeLabelSize,
              const QString &defEdgeColor, const int &canvasWidth, const int &canvasHeight, const int &format,
              const QString &delim = QString(), const int &sm_mode = 1, const bool &sm_has_labels = false);
 
    bool parseAsPajek(const QByteArray &rawData);
    bool parseAsAdjacency(const QByteArray &rawData, const ParseConfig &cfg, const QString &delimiter);
    bool parseAsDot(const QByteArray &rawData);
    QString preprocessDotContent(const QString &dotContent);
    bool parseAsGraphML(const QByteArray &rawData);
    bool parseAsGML(const QByteArray &rawData);
    bool parseAsDL(const QByteArray &rawData);
    bool parseAsEdgeListSimple(const QByteArray &rawData, const QString &delimiter);
    bool parseAsEdgeListWeighted(const QByteArray &rawData, const QString &delimiter);
    bool parseAsTwoModeSociomatrix(const QByteArray &rawData);
 
    bool readDLKeywords(QStringList &strList,
                        int &N,
                        int &NM,
                        int &NR,
                        int &NC,
                        bool &fullmatrixFormat,
                        bool &edgelist1Format,
                        bool &diagonalPresent);
 
    void readDotProperties(QString str, qreal &nValue, QString &label,
                           QString &shape, QString &color, QString &fontName,
                           QString &fontColor,
                           QHash<QString, QString> &customAttrs);
 
    bool readGraphML(QXmlStreamReader &);
    void readGraphMLElementGraph(QXmlStreamReader &);
    void readGraphMLElementNode(QXmlStreamReader &);
    void endGraphMLElementNode(QXmlStreamReader &);
    void readGraphMLElementEdge(QXmlStreamAttributes &);
    void endGraphMLElementEdge(QXmlStreamReader &);
    void readGraphMLElementData(QXmlStreamReader &);
    void readGraphMLElementUnknown(QXmlStreamReader &);
    void readGraphMLElementKey(QXmlStreamAttributes &);
    void readGraphMLElementDefaultValue(QXmlStreamReader &);
    void readGraphMLElementNodeGraphics(QXmlStreamReader &);
    void readGraphMLElementEdgeGraphics(QXmlStreamReader &);
    void createMissingNodeEdges();
 
    bool isComment(QString str);
    void createRandomNodes(const int &fixedNum = 1, const QString &label = QString(),
                           const int &newNodes = 1);
 
    static QString normalizeQuotedIdentifier(const QString &s);
 
signals:
 
    void finished(QString);
};
 
#endif