mitkTubeGraphEdge.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)
Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef _MITK_TubeGraphEdge_H
#define _MITK_TubeGraphEdge_H

#include <MitkTubeGraphExports.h>

#include "mitkCircularProfileTubeElement.h"
#include "mitkTubeElement.h"
#include "mitkTubeGraphVertex.h"

namespace mitk
{
  class MITKTUBEGRAPH_EXPORT TubeGraphEdge
  {
  public:
    TubeGraphEdge();
    virtual ~TubeGraphEdge();

    void SetElementVector(std::vector<TubeElement *> elementVector);

    std::vector<TubeElement *> GetElementVector();

    unsigned int GetNumberOfElements() const;

    void AddTubeElement(TubeElement *element);

    void AddTubeElementAt(unsigned int position, TubeElement *element);

    TubeElement *GetTubeElement(unsigned int position);

    float GetEdgeLength(TubeGraphVertex &source, TubeGraphVertex &target);

    float GetEdgeAverageDiameter(TubeGraphVertex &source, TubeGraphVertex &target);

    bool operator==(const TubeGraphEdge &right) const;

  private:
    std::vector<TubeElement *> m_ElementVector;

  }; // class

} // namespace
#endif