From 71af784a726aebe474e45a68a181ef08087716db Mon Sep 17 00:00:00 2001
From: Carl-Martin Pfeiler <carl-martin.pfeiler@asc.tuwien.ac.at>
Date: Tue, 27 Nov 2018 10:07:33 +0100
Subject: [PATCH] I.SaveMagnetization() extension always .vtk forced

---
 _tools/vtk/saveMagnetization.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/_tools/vtk/saveMagnetization.py b/_tools/vtk/saveMagnetization.py
index 2e62e76..e12b941 100644
--- a/_tools/vtk/saveMagnetization.py
+++ b/_tools/vtk/saveMagnetization.py
@@ -10,7 +10,11 @@ def SaveMagnetization(filename, mesh, m, label="m"):
   content += Mesh2VTK(mesh)
   content += Mag2VTK(mesh, m, label)
   
-  output = open(filename + ".vtk", "w")
+  ext = ".vtk"
+  if(filename[-4:] == ext):
+    ext = ""
+
+  output = open(filename + ext, "w")
   output.write(content)
   output.close()
 
-- 
GitLab