gistfile1.jl

using PyCall
@pyimport vtk

function create_vtk_object(mesh::Mesh)

    n_nodes = length(mesh.nodes)

    # Add points to vtkPoints
    points = pycall(vtk.vtkPoints, PyAny)
    for node in mesh.nodes
        d_coords = node.coordinates
        points[:InsertNextPoint](d_coords[1], d_coords[2], d_coords[3])
    end

    # Add elements to vtkCellArray
    elems_vtk = pycall(vtk.vtkCellArray, PyAny)
    for element in mesh.elements
        elem_vtk = pycall(vtk.vtkTriangle, PyAny)
        for (i, vertex) in enumerate(element.vertices)
            ids = elem_vtk[:GetPointIds]()
            ids[:SetId](i-1, vertex - 1)
        end
        elems_vtk[:InsertNextCell](elem_vtk)
    end


    # Create a polydata to store everything in
    polydata = pycall(vtk.vtkPolyData, PyAny)

    # Add the points to the polydata object
    polydata[:SetPoints](points)

    # Add the elements to the polydata object
    polydata[:SetPolys](elems_vtk)

    return polydata
end


function write_vtk_file(mesh::Mesh, name::ASCIIString, write_ascii::Bool = false)
    polydata = create_vtk_object(mesh)

    writer = pycall(vtk.vtkXMLPolyDataWriter, PyAny)
    writer[:SetFileName](name)
    if vtk.VTK_MAJOR_VERSION <= 5
        writer[:SetInput](polydata)
    else
        writer[:SetInputData](polydata)
    end

    if write_ascii
        writer[:SetDataModeToAscii]()
    end
    writer[:Write]()
end