From 8753ec544ca184c23d89c06896fee12fe7f6daca Mon Sep 17 00:00:00 2001
From: Stephan Hilb <stephan@ecshi.net>
Date: Fri, 9 Jul 2021 19:51:17 +0200
Subject: [PATCH] fix inpainting operator

---
 src/mesh.jl | 2 +-
 src/run.jl  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/mesh.jl b/src/mesh.jl
index b5da461..da5326f 100644
--- a/src/mesh.jl
+++ b/src/mesh.jl
@@ -56,7 +56,7 @@ function vtk_mesh(filename, mesh::Mesh)
 end
 
 "convenience function for saving to vtk"
-function save(filename, mesh::Mesh, fs...)
+function save(filename::String, mesh::Mesh, fs...)
     vtk = vtk_mesh(filename, mesh)
     for f in fs
 	f.space.mesh == mesh ||
diff --git a/src/run.jl b/src/run.jl
index 5e22ef9..2f46eb8 100644
--- a/src/run.jl
+++ b/src/run.jl
@@ -201,7 +201,7 @@ function inpaint(img, imgmask; name, params...)
     Vg = FeSpace(mesh, P1(), (1,))
     mask = FeFunction(Vg, name="mask")
 
-    T(tdata, u) = iszero(tdata) ? zero(u) : u
+    T(tdata, u) = isone(tdata[begin]) ? u : zero(u)
     S(u, nablau) = u
 
     ctx = L1L2TVContext(name, mesh, m; T, tdata = mask, S, params...)
-- 
GitLab