int VibesTreeModel::rowCount(const QModelIndex &parent) const
{
    if (parent.column() > 0)
        return 0;
/*
    const void *parentItem;
    if (!parent.isValid())
        parentItem = figures;
    else
        parentItem = parent.internalPointer();

    if (parentItem == figures)
    {
        return figures->size();
    }
    */

    // root node
    if (!parent.isValid())
    {
        return figures->size();
    }
    // figure node
    if (parent.internalPointer() == 0)
    {
        Figure2D * fig = figures->values().at(parent.row());
        return fig->scene()->namedItems().size();
    }

    return 0;
}
Esempio n. 2
0
void RenderScene::mousePressEvent(QGraphicsSceneMouseEvent *event)  {
   if (_isDrawing) {

       drawingFigure->addPoint(event->scenePos());

       if (drawingFigure->isFinished()) {

           figures.push_back(drawingFigure);
           drawingFigure->draw(*this);
           endDrawing();
       }
    }
    else {
       startDrawing();
       drawingFigure = getDrawingFigure(figureName, event->scenePos());
       if (!drawingFigure) {
           endDrawing();
           return;
       }
       drawingFigure->setBorderColor(new QColor(lineColor));
       Figure2D* df = dynamic_cast<Figure2D*>(drawingFigure);
       if (withFill && df)
           df->setFillColor(new QColor(fillColor));
    }
}
Esempio n. 3
0
void VibesWindow::exportCurrentFigureGraphics()
{
    // Get current selected item in tree view
    QModelIndex selectId = ui->treeView->currentIndex();
    // If no selection, return
    if (!selectId.isValid())
        return;
    // If the selected item is a figure, export it
    Figure2D * pfig = static_cast<Figure2D*> (selectId.internalPointer());
    if (figures.values().contains(pfig))
    {
        pfig->exportGraphics();
    }
}
Esempio n. 4
0
bool
VibesWindow::processMessage(const QByteArray &msg_data)
{
    QJsonParseError parse_error;
    QJsonDocument doc = QJsonDocument::fromJson(msg_data, &parse_error);
    // Check parsing error
    if (parse_error.error != QJsonParseError::NoError)
    {
        /// \todo Notify or error in input data
        return false;
    }
    // Check if our document is an object
    if (!doc.isObject())
    {
        return false;
    }

    // Process message
    QJsonObject msg = doc.object();
    // Action is a mandatory field
    if (!msg.contains("action"))
    {
        return false;
    }

    QString action = msg["action"].toString();
    QString fig_name = msg["figure"].toString();

    // Get pointer to target figure
    Figure2D * fig = 0;
    if (figures.contains(fig_name))
        fig = figures[fig_name];

    // Close a figure
    if (action == "close")
    {
        // Figure has to exist to be closed
        if (!fig)
            return false;
        // Remove from the list of figures an delete
        figures.remove(fig_name);
        delete fig;
        static_cast<VibesTreeModel*> (ui->treeView->model())->forceUpdate();
    }
        // Create a new figure
    else if (action == "new")
    {
        // Create a new figure (previous with the same name will be destroyed)
        fig = newFigure(fig_name);
        static_cast<VibesTreeModel*> (ui->treeView->model())->forceUpdate();
    }
        // Clear the contents of a figure
    else if (action == "clear")
    {
        // Figure has to exist
        if (!fig)
            return false;
        // Clears the scene
        fig->scene()->clear();
        /// \todo Remove named objects references if needed
    }
        // Sets the view
    else if (action == "view")
    {
        // Figure has to exist
        if (!fig)
            return false;
        // Set the view rectangle to a box
        if (msg["box"].isArray())
        {
            const QJsonArray box = msg["box"].toArray();
            if (box.size() < 4) return false;
            double lb_x = box[0].toDouble();
            double ub_x = box[1].toDouble();
            double lb_y = box[2].toDouble();
            double ub_y = box[3].toDouble();
            fig->fitInView(lb_x, lb_y, ub_x - lb_x, ub_y - lb_y);
        }
            // Auto-set the view rectangle
        else if (msg["box"].toString() == "auto")
        {
            fig->fitInView(fig->scene()->sceneRect());
        }
    }
        // Export to a graphical file
    else if (action == "export")
    {
        // Figure has to exist
        if (!fig)
            return false;
        // Exports to given filename (if not defined, shows a save dialog)
        fig->exportGraphics(msg["file"].toString());
    }
        // Draw a shape
    else if (action == "draw")
    {
        if (!fig) // Create a new figure if it does not exist
            fig = newFigure(fig_name);

        QColor fill_color;
        QColor edge_color;

        QGraphicsItem * item = 0;

        if (msg.contains("shape"))
        {
            QJsonObject shape = msg.value("shape").toObject();
            // Get shape color (or default if not specified)
            const QBrush & brush = brushes[shape.value("color").toString()];
            if (shape.contains("type"))
            {
                QString type = shape["type"].toString();

                if (type == "box")
                {
                    QJsonArray bounds = shape["bounds"].toArray();
                    if (bounds.size() >= 4)
                    {
                        double lb_x = bounds[0].toDouble();
                        double ub_x = bounds[1].toDouble();
                        double lb_y = bounds[2].toDouble();
                        double ub_y = bounds[3].toDouble();

                        item = fig->scene()->addRect(lb_x, lb_y, ub_x - lb_x, ub_y - lb_y, defaultPen, brush);
                    }
                }
                else if (type == "boxes")
                {
                    QJsonArray boundsX_lb = shape["boundsX_lb"].toArray();
                    QJsonArray boundsX_ub = shape["boundsX_ub"].toArray();
                    QJsonArray boundsY_lb = shape["boundsY_lb"].toArray();
                    QJsonArray boundsY_ub = shape["boundsY_ub"].toArray();

                    if (boundsX_lb.size() == boundsX_ub.size() &&
                            boundsX_ub.size() == boundsY_lb.size() &&
                            boundsY_lb.size() == boundsY_ub.size())
                    {
                        bool colors = shape.contains("colors");
                        bool enoughColors = false;
                        if (colors)
                            enoughColors = shape["colors"].toArray().size() == boundsX_lb.size();
                        for (int i = 0; i < boundsX_lb.size(); i++)
                        {
                            double lb_x = boundsX_lb[i].toDouble();
                            double ub_x = boundsX_ub[i].toDouble();
                            double lb_y = boundsY_lb[i].toDouble();
                            double ub_y = boundsY_ub[i].toDouble();

                            item = fig->scene()->addRect(lb_x, lb_y, ub_x - lb_x, ub_y - lb_y, defaultPen, brush);
                        }
                    }
                }
                else if (type == "ellipse")
                {
                    QJsonArray center = shape["center"].toArray();
                    if (center.size() >= 2)
                    {
                        double x = center[0].toDouble();
                        double y = center[1].toDouble();
                        double wx, wy, angle;
                        if (shape.contains("axis") && shape.contains("orientation"))
                        {
                            QJsonArray axis = shape["axis"].toArray();
                            wx = axis[0].toDouble();
                            wy = axis[1].toDouble();
                            angle = shape.value("orientation").toDouble();
                        }
                        else if (shape.contains("covariance"))
                        {
                            double sxx, sxy, syy, s, eval1, eval2, det, trace, rightTerm;
                            double evect1[2], evect2[2];
                            s = shape.contains("sigma") ? shape["sigma"].toDouble() : 5;
                            QJsonArray covariance = shape["covariance"].toArray();
                            sxx = covariance[0].toDouble();
                            sxy = covariance[1].toDouble();
                            syy = covariance[3].toDouble();
                            if (sxy == 0)
                            {
                                eval1 = sxx;
                                eval2 = syy;
                                evect1[0] = 1;
                                evect1[1] = 0;
                                evect2[0] = 0;
                                evect2[1] = 1;
                            }
                            else
                            {
                                det = sxx * syy - pow(sxy, 2);
                                trace = sxx + syy;
                                rightTerm = sqrt(pow(sxx + syy, 2) / 4 - det);
                                eval1 = trace / 2 + rightTerm;
                                eval2 = trace / 2 - rightTerm;

                                evect1[0] = evect2[0] = 1; // We set the x-component of the eigenvectors to 1
                                evect1[1] = (eval1 - sxy - sxx) / (sxy + syy - eval1);
                                evect2[1] = (eval2 - sxy - sxx) / (sxy + syy - eval2);
                            }
                            // (evect1; evect2) give us the rotation matrix
                            // s*sqrt(eval1) s*sqrt(eval2) give us the main axis-sizes of the ellipse

                            angle = (evect1[0] != evect1[0]) || (evect1[1] != evect1[1]) ? (atan2(evect2[1], evect2[0])*180 * M_1_PI - 90) : atan2(evect1[1], evect1[0])*180 * M_1_PI;
                            wx = s * sqrt(eval1);
                            wy = s * sqrt(eval2);
                        }
                        else // should not be here
                            return false;
                        item = fig->scene()->addEllipse(-wx, -wy, 2 * wx, 2 * wy, defaultPen, brush);
                        item->setRotation(angle);
                        item->setPos(x, y);
                    }
                }
                else if (type == "point")
                {

                }
                else if (type == "points")
                {

                }
            }
        }
    }
        // Unknown action
    else
    {
        return false;
    }
    return true;
}