mitkTrackvis.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 _TRACKVIS
#define _TRACKVIS

#include <mitkCommon.h>
#include <mitkFiberBundle.h>
#include <vtkSmartPointer.h>
#include <vtkPolyData.h>
#include <vtkCellArray.h>
#include <vtkPoints.h>
#include <vtkPolyLine.h>
#include <itkSize.h>

using namespace std;

// Structure to hold metadata of a TrackVis file
// ---------------------------------------------
struct TrackVis_header
{
    char                id_string[6];
    short int           dim[3];
    float               voxel_size[3];
    float               origin[3];
    short int           n_scalars;
    char                scalar_name[10][20];
    short int           n_properties;
    char                property_name[10][20];
    char                reserved[508];
    char                voxel_order[4];
    char                pad2[4];
    float               image_orientation_patient[6];
    char                pad1[2];
    unsigned char       invert_x;
    unsigned char       invert_y;
    unsigned char       invert_z;
    unsigned char       swap_xy;
    unsigned char       swap_yz;
    unsigned char       swap_zx;
    int                 n_count;
    int                 version;
    int                 hdr_size;
};

// Class to handle TrackVis files.
// -------------------------------
class MITKFIBERTRACKING_EXPORT TrackVisFiberReader
{
private:
    string              m_Filename;
    FILE*               m_FilePointer;

public:
    TrackVis_header     m_Header;

    short   create(string m_Filename, const mitk::FiberBundle* fib);
    short   open( string m_Filename );
    short   read( mitk::FiberBundle* fib );
    short   append(const mitk::FiberBundle* fib );
    void    writeHdr();
    void    updateTotal( int totFibers );
    void    close();
    bool    IsTransformValid();

    TrackVisFiberReader();
    ~TrackVisFiberReader();
};

#endif