/*//////////////////////////////////////////////////////////////////////////////
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
////////////////////////////////////////////////////////////////////////////////

  Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
  All rights reserved

  Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >

  ------------------------------------------------------------------
  This library is free software;  you  can  redistribute  it  and/or
  modify it  under the  terms  of  the  GNU  Lesser  General  Public
  License as published  by  the  Free  Software  Foundation;  either
  version 3.0 of the License, or (at your option) any later version.

  This library is  distributed in  the hope that it will  be useful,
  but  WITHOUT ANY WARRANTY;  without  even  the implied warranty of
  MERCHANTABILITY  or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  Lesser General Public License for more details.

  You should have received a copy of  the GNU Lesser General  Public
  License along with this library.

//////////////////////////////////////////////////////////////////////////////*/

#include <vtkVersion.h>
#include <vtkPolyData.h>
#include <vtkPolyDataWriter.h>
#include <vtkXMLPolyDataWriter.h>
#include <vtkSmartPointer.h>

#include "vtkPolydata.h"

namespace uLib {
namespace Vtk {

void Polydata::SaveToFile(const char *vtk_file)
{
    vtkSmartPointer<vtkPolyDataWriter> writer =
      vtkSmartPointer<vtkPolyDataWriter>::New();
    writer->SetFileName(vtk_file);
    vtkPolyData * data = GetPolyData();
    if(data) {
#       if VTK_MAJOR_VERSION <= 5
        writer->SetInputConnection(data->GetProducerPort());
#       else
        writer->SetInputData(data);
#       endif
        writer->Update();
    }
}

void Polydata::SaveToXMLFile(const char *vtp_file)
{
    vtkSmartPointer<vtkXMLPolyDataWriter> writer =
      vtkSmartPointer<vtkXMLPolyDataWriter>::New();
    writer->SetFileName(vtp_file);
    vtkPolyData * data = GetPolyData();
    if(data) {
#       if VTK_MAJOR_VERSION <= 5
        writer->SetInputConnection(data->GetProducerPort());
#       else
        writer->SetInputData(data);
#       endif
        writer->Update();
    }
}

} // Vtk
} // uLib