cxRepContainer.h

Go to the documentation of this file.

/*=========================================================================
This file is part of CustusX, an Image Guided Therapy Application.
                 
Copyright (c) SINTEF Department of Medical Technology.
All rights reserved.
                 
CustusX is released under a BSD 3-Clause license.
                 
See Lisence.txt (https://github.com/SINTEFMedtek/CustusX/blob/master/License.txt) for details.
=========================================================================*/

#ifndef CXREPCONTAINER_H
#define CXREPCONTAINER_H

#include "cxResourceVisualizationExport.h"

#include <QObject>
#include <map>
#include <vector>
#include "cxForwardDeclarations.h"
#include "cxTool.h"

namespace cx
{
typedef boost::shared_ptr<class Rep> RepPtr;

typedef std::map<QString, RepPtr> RepMap;
typedef std::map<QString, VolumetricBaseRepPtr> VolumetricRepMap;

class cxResourceVisualization_EXPORT RepContainer
{
public:
        RepContainer(std::vector<RepPtr> reps) : mReps(reps) {}

        template<class REP>
        boost::shared_ptr<REP> findFirst(ToolPtr tool)
        {
                for (unsigned i = 0; i < mReps.size(); ++i)
                {
                        boost::shared_ptr<REP> rep = boost::dynamic_pointer_cast<REP>(mReps[i]);
                        if (rep && rep->hasTool(tool))
                        {
                                return rep;
                        }
                }
                return boost::shared_ptr<REP>();
        }

        template<class REP>
        boost::shared_ptr<REP> findFirst(DataPtr data)
        {
                for (unsigned i = 0; i < mReps.size(); ++i)
                {
                        boost::shared_ptr<REP> rep = boost::dynamic_pointer_cast<REP>(mReps[i]);
                        if (rep && rep->hasData(data))
                                return rep;
                }
                return boost::shared_ptr<REP>();
        }

        template<class REP>
        boost::shared_ptr<REP> findFirst()
        {
                for (unsigned i = 0; i < mReps.size(); ++i)
                {
                        boost::shared_ptr<REP> rep = boost::dynamic_pointer_cast<REP>(mReps[i]);
                        if (rep)
                                return rep;
                }
                return boost::shared_ptr<REP>();
        }

        template<class REP>
        static std::vector<boost::shared_ptr<REP> > findReps(std::vector<RepPtr> reps)
        {
                std::vector<boost::shared_ptr<REP> > retval;
                for (unsigned i = 0; i < reps.size(); ++i)
                {
                        boost::shared_ptr<REP> rep = boost::dynamic_pointer_cast<REP>(reps[i]);
                        if (rep)
                                retval.push_back(rep);
                }
                return retval;
        }

        template<class REP>
        boost::shared_ptr<REP> findManualToolRep()
        {
                for (unsigned i = 0; i < mReps.size(); ++i)
                {
                        boost::shared_ptr<REP> rep = boost::dynamic_pointer_cast<REP>(mReps[i]);
                        if (rep && rep->getTool()->hasType(Tool::TOOL_MANUAL))
                        {
                                return rep;
                        }
                }
                return boost::shared_ptr<REP>();
        }

private:
        std::vector<RepPtr> mReps;
};


} //namespace cx

#endif // CXREPCONTAINER_H