Skip to content
Snippets Groups Projects
petrack.cpp 149 KiB
Newer Older
  • Learn to ignore specific revisions
  • d.kilic's avatar
    d.kilic committed
                }
    
                    debout << "Warning: error while reading showOnlyVisible list from input line!" << std::endl;
    
    d.kilic's avatar
    d.kilic committed
            }
    
            return onlyVisible; // in anzeige wird ab 1 gezaehlt, in datenstruktur ab 0
        }
        return QSet<int>();
    }
    
    /**
     * @brief Checks which pedestrians/trajectories are selected for evaluation.
     *
     * If "only for selected" is checked, then only selected people (@see
     * Petrack::getPedestrianUserSelection()) are going to be tracked, all people
     * otherwise.
     *
     * @return all trajectories which should be evaluated; empty when all should be evaluated
     */
    QSet<int> Petrack::getPedestriansToTrack()
    {
    
        if(mControlWidget->trackOnlySelected->checkState() == Qt::Checked)
    
            return getPedestrianUserSelection();
    
    d.kilic's avatar
    d.kilic committed
    }
    
    
    void Petrack::addManualTrackPointOnlyVisible(const QPointF &pos)
    
    d.kilic's avatar
    d.kilic committed
    {
    
        int pers = addOrMoveManualTrackPoint(pos) + 1;
        if(pers == 0)
    
            pers = mTracker->size() + 1;
    
    d.kilic's avatar
    d.kilic committed
        pers = mControlWidget->trackShowOnlyNr->maximum();
        mControlWidget->trackShowOnlyNr->setValue(pers);
        mControlWidget->trackShowOnly->setChecked(true);
    }
    
    void Petrack::updateControlWidget()
    {
        mControlWidget->trackNumberAll->setText(QString("%1").arg(mTracker->size()));
    
        mControlWidget->trackShowOnlyNr->setMaximum(MAX(mTracker->size(), 1));
    
    d.kilic's avatar
    d.kilic committed
        mControlWidget->trackNumberVisible->setText(QString("%1").arg(mTracker->visible(mAnimation->getCurrentFrameNum())));
    }
    
    void Petrack::splitTrackPerson(QPointF pos)
    {
    
        mTracker->splitPersonAt((Vec2F) pos, mAnimation->getCurrentFrameNum(), getPedestrianUserSelection());
    
    d.kilic's avatar
    d.kilic committed
        updateControlWidget();
    }
    
    
     * @brief Lets the user add or move a TrackPoint manually
     *
     * There is an check inside addPoint which inhibits adding a point,
     * if only selected trajectories are visualized, since one wouldn't
     * see the newly added TrackPoint.
    
     *
     * @param pos pixel position of mouse on image
    
     * @return index of person whose point was moved; -1 if failed or new trajectory is started
    
    int Petrack::addOrMoveManualTrackPoint(const QPointF &pos)
    
    d.kilic's avatar
    d.kilic committed
    {
    
        int        pers = -1;
    
        TrackPoint tP(Vec2F{pos}, 110); // 110 is higher than 100 (max. quality) and gets clamped to 100 after insertion
        // allows replacemet of every point (check for better quality always passes)
    
        mTracker->addPoint(
            tP, mAnimation->getCurrentFrameNum(), getPedestrianUserSelection(), mReco.getRecoMethod(), &pers);
    
        updateControlWidget();
        return pers;
    
    d.kilic's avatar
    d.kilic committed
    }
    
    // direction zeigt an, ob bis zum aktuellen (-1), ab dem aktuellen (1) oder ganzer trackpath (0)
    // loeschen von Trackpoints einer Trajektorie
    
    void Petrack::deleteTrackPoint(QPointF pos, int direction) // const QPoint &pos
    
    d.kilic's avatar
    d.kilic committed
    {
    
        mTracker->delPoint((Vec2F) pos, direction, mAnimation->getCurrentFrameNum(), getPedestrianUserSelection());
    
    d.kilic's avatar
    d.kilic committed
        updateControlWidget();
    }
    void Petrack::editTrackPersonComment(QPointF pos)
    {
    
        mTracker->editTrackPersonComment((Vec2F) pos, mAnimation->getCurrentFrameNum(), getPedestrianUserSelection());
    
    d.kilic's avatar
    d.kilic committed
        updateControlWidget();
    }
    void Petrack::setTrackPersonHeight(QPointF pos)
    {
    
        mTracker->setTrackPersonHeight((Vec2F) pos, mAnimation->getCurrentFrameNum(), getPedestrianUserSelection());
    
    d.kilic's avatar
    d.kilic committed
        updateControlWidget();
    }
    void Petrack::resetTrackPersonHeight(QPointF pos)
    {
    
        mTracker->resetTrackPersonHeight((Vec2F) pos, mAnimation->getCurrentFrameNum(), getPedestrianUserSelection());
    
    d.kilic's avatar
    d.kilic committed
        updateControlWidget();
    }
    
    // direction zeigt an, ob bis zum aktuellen (-1), ab dem aktuellen (1) oder ganzer trackpath (0)
    // loeschen von Trackpoints aller Trajektorien
    
    void Petrack::deleteTrackPointAll(int direction) // const QPoint &pos
    
    d.kilic's avatar
    d.kilic committed
    {
        mTracker->delPointAll(direction, mAnimation->getCurrentFrameNum());
        updateControlWidget();
    }
    
    void Petrack::deleteTrackPointROI()
    {
        mTracker->delPointROI();
        updateControlWidget();
        mScene->update();
    }
    
    void Petrack::deleteTrackPointInsideROI()
    {
        mTracker->delPointInsideROI();
        updateControlWidget();
        mScene->update();
    }
    void Petrack::updateSourceInOutFrames()
    {
        mPlayerWidget->setFrameInNum(mAnimation->getSourceInFrameNum());
    
        mPlayerWidget->setFrameOutNum(mAnimation->getSourceOutFrameNum());
    
    d.kilic's avatar
    d.kilic committed
    }
    
    // delta gibt menge an Umdrehungen und richtung an
    void Petrack::skipToFrameWheel(int delta)
    {
    
        mPlayerWidget->skipToFrame(mPlayerWidget->getPos() + delta);
    
    d.kilic's avatar
    d.kilic committed
    }
    
    void Petrack::setPeTrackVersion(const std::string &petrackVersion)
    
    {
        mPetrackVersion = QString::fromStdString(petrackVersion);
    }
    
    void Petrack::setGitInformation(
    
        const std::string &gitCommitID,
        const std::string &gitCommitDate,
        const std::string &gitCommitBranch)
    
        mGitCommitID     = QString::fromStdString(gitCommitID);
        mGitCommitDate   = QString::fromStdString(gitCommitDate);
    
        mGitCommitBranch = QString::fromStdString(gitCommitBranch);
    }
    
    
    void Petrack::setCompileInformation(
        const std::string &compileTimeStamp,
        const std::string &compilerID,
        const std::string &compilerVersion)
    
        mCompileDate     = QString::fromStdString(compileTimeStamp);
        mCompilerID      = QString::fromStdString(compilerID);
    
        mCompilerVersion = QString::fromStdString(compilerVersion);
    }
    
    
    #include "moc_petrack.cpp"