anj1 / thinplatesplines.jl Goto Github PK

julia> start_pts = [0 0 0; 0 0 1; 0 1 0; 1 0 0]
4×3 Matrix{Int64}:
 0  0  0
 0  0  1
 0  1  0
 1  0  0

julia> end_pts = [-0.7 -0.7 0; 0 0 1; 0 1 0; 1 0 0]
4×3 Matrix{Float64}:
 -0.7  -0.7  0.0
  0.0   0.0  1.0
  0.0   1.0  0.0
  1.0   0.0  0.0

julia> tps = tps_solve(start_pts, end_pts, 1.0)
ThinPlateSpline(1.0, [0 0 0; 0 0 1; 0 1 0; 1 0 0], [1.0 -0.7 -0.7 0.0; 1.0 0.0 0.0 1.0; 1.0 0.0 1.0 0.0; 1.0 1.0 0.0 0.0], [0.0 0.0 0.0 0.0; 0.0 0.0 0.6931471805599455 0.6931471805599455; 0.0 0.6931471805599455 0.0 0.6931471805599455; 0.0 0.6931471805599455 0.6931471805599455 0.0], [0.9999999999999999 -0.7 -0.6999999999999996 -4.143642362094349e-17; 2.3532914146181465e-16 1.6999999999999997 0.6999999999999996 2.3373269225929986e-16; 1.447594855068065e-16 0.6999999999999998 1.699999999999999 -6.798699777552591e-17; 1.7663540160192472e-16 0.7000000000000001 0.7 1.0], [0.0 0.0 0.0 0.0; 0.0 0.0 0.0 0.0; 0.0 0.0 0.0 0.0; 0.0 0.0 0.0 0.0])

julia> tps_deform(Float64[0 0 0], tps)
1×2 Matrix{Float64}:
 -0.7  -0.7

import ThinPlateSplines: my_norm, is_zero, tps_basis

function tps_deform_old(x2::AbstractArray,tps::ThinPlateSpline) where T<:Any
	x1,d,c=tps.x1,tps.d,tps.c
	d==[] && throw(ArgumentError("Affine component not available; run tps_solve with compute_affine=true."))

	# deform
	y2 = zeros(eltype(x2),size(x2,1),size(x2,2)+1)
	for i = 1 : size(x2,1)
		z = x2[i:i,:]
		defc = zeros(eltype(x2),1,size(x2,2)+1)
		for j = 1 : size(x1,1)
			n = my_norm(z - x1[j:j,:])
			defc += tps_basis(n)*c[j:j,:]
		end
		y2[i:i,:] = cat(dims=2, 1.0, z)*d + defc
	end

	y2[:,2:end]
end

julia> tps_deform_old(Float64[0 0 0], tps)
1×3 Matrix{Float64}:
 -0.7  -0.7  -4.14364e-17