Treatment planning and diagnosis