mitkTubeGraphEdge.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

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

#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