Visualization2DBasic.h

Go to the documentation of this file.

/*
 * Copyright (C) 2012 by
 *   MetraLabs GmbH (MLAB), GERMANY
 * and
 *   Neuroinformatics and Cognitive Robotics Labs (NICR) at TU Ilmenau, GERMANY
 * All rights reserved.
 *
 * Contact: info@mira-project.org
 *
 * Commercial Usage:
 *   Licensees holding valid commercial licenses may use this file in
 *   accordance with the commercial license agreement provided with the
 *   software or, alternatively, in accordance with the terms contained in
 *   a written agreement between you and MLAB or NICR.
 *
 * GNU General Public License Usage:
 *   Alternatively, this file may be used under the terms of the GNU
 *   General Public License version 3.0 as published by the Free Software
 *   Foundation and appearing in the file LICENSE.GPL3 included in the
 *   packaging of this file. Please review the following information to
 *   ensure the GNU General Public License version 3.0 requirements will be
 *   met: http://www.gnu.org/copyleft/gpl.html.
 *   Alternatively you may (at your option) use any later version of the GNU
 *   General Public License if such license has been publicly approved by
 *   MLAB and NICR (or its successors, if any).
 *
 * IN NO EVENT SHALL "MLAB" OR "NICR" BE LIABLE TO ANY PARTY FOR DIRECT,
 * INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING OUT OF
 * THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF "MLAB" OR
 * "NICR" HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * "MLAB" AND "NICR" SPECIFICALLY DISCLAIM ANY WARRANTIES, INCLUDING,
 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE PROVIDED HEREUNDER IS
 * ON AN "AS IS" BASIS, AND "MLAB" AND "NICR" HAVE NO OBLIGATION TO
 * PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS OR MODIFICATIONS.
 */

#ifndef _MIRA_VISUALIZATION2DBASIC_H_
#define _MIRA_VISUALIZATION2DBASIC_H_

#include <QGraphicsScene>
#include <QGraphicsItem>

#include <visualization/Visualization2D.h>

#include <widgets/QtUtils.h>

namespace mira {


template <typename T>
class Visualization2DBasic : public Visualization2D
{
public:

    Visualization2DBasic(const std::string channelDisplayName = "Channel") :
        mChannelDisplayName(channelDisplayName),
        mItem(NULL) {
        mDataChannel.setDataChangedCallback(
                boost::bind(&Visualization2DBasic::dataChangedCallback, this, _1));
        mDataChannel.setChannelChangedCallback(
                boost::bind(&Visualization2DBasic::channelChangedCallback, this));
    }

    virtual ~Visualization2DBasic() {
        delete mItem;
    }

    template <typename Reflector>
    void reflect(Reflector& r) {
        MIRA_REFLECT_BASE(r, Visualization2D);
        channelProperty(r, mChannelDisplayName.c_str(), mDataChannel, "The channel to be visualized");
    }

public:

    virtual void setupScene(IVisualization2DSite* site) {
        QGraphicsScene* mgr = site->getSceneManager();
        mItem = setupScene(mgr);
    }

    virtual QGraphicsItem* setupScene(QGraphicsScene* mgr) = 0;

    virtual QGraphicsItem* getItem() {
        return mItem;
    }

public:

    void setEnabled(bool enabled) {
        Visualization2D::setEnabled(enabled);
        if (mItem != NULL)
            mItem->setVisible(enabled);
    }

public:

    virtual DataConnection getDataConnection() {
        return DataConnection(mDataChannel);
    }

protected:

    virtual void update(Duration dt) {
        if (!mDataChannel.isValid() || mDataChannel.getDataUpdateCount() == 0)
            return;
        if (mDataFrameID.empty())
            MIRA_THROW(TransformerBase::XTransform, "Data has no frame for transforming it into the views camera frame");

        RigidTransform2f d;
        try {
            d = getAuthority().template getTransform<RigidTransform2f>(
                mDataFrameID, mDataTimestamp,
                getSite()->getCameraFrame(), Time::now(),
                getSite()->getFixedFrame(), LinearInterpolatorNearestNeighbourExtrapolator());
        }
        catch(...) {    // fall back to non interpolating versions
            d = getAuthority().template getTransform<RigidTransform2f>(
                mDataFrameID, mDataTimestamp,
                getSite()->getCameraFrame(), Time::now(),
                getSite()->getFixedFrame());
        }
        QtUtils::setTransform(mItem, d);
    }

    void dataChangedCallback(ChannelRead<T> data) {
        mDataFrameID = data->frameID;
        mDataTimestamp = data->timestamp;
        dataChanged(data);
    }

    void channelChangedCallback() {
        mDataFrameID = "";
        try {
            this->setName(mDataChannel.getID());
        }
        catch (XRuntime&) {}
        channelChanged();
    }

    virtual void channelChanged() {}

    virtual void dataChanged(ChannelRead<T> data) {
        onDataChanged(data);    // only for backward compatibility
    }

    virtual void onDataChanged(ChannelRead<T> data) {}

protected:

    ChannelProperty<T> mDataChannel;
    std::string mChannelDisplayName;

    std::string mDataFrameID;
    Time mDataTimestamp;

    QGraphicsItem* mItem;
};


}

#endif