Thankfully the gear box failure was just down to a loose pipe on top of the box.
The tdi is certainly a different bell housing but I dont know about the input shaft. Having several V8 autos and only one Tdi, when I take the box out I'll let you know mate.
Hopefully it will be a while before the box comes out!!!!! :D