#!/usr/bin/env python
# coding: utf-8

# In[ ]:


# Install dependencies for this example
# Note: This does not include itkwidgets, itself
import sys
get_ipython().system('{sys.executable} -m pip install --upgrade --pre itk')


# In[1]:


from urllib.request import urlretrieve
import os

import itk

from itkwidgets import view


# In[2]:


# Example data from ITKTubeTK: https://github.com/KitwareMedical/ITKTubeTK

tubes_file = "Normal071-VascularNetwork.tre"
if not os.path.exists(tubes_file):
    url = 'https://data.kitware.com/api/v1/item/5e1b3727af2e2eed35f5c48a/download'
    urlretrieve(url, tubes_file)
    
mra_file = "Normal071-MRA.mha"
if not os.path.exists(mra_file):
    url = 'https://data.kitware.com/api/v1/item/5e1b3721af2e2eed35f5c482/download'
    urlretrieve(url, mra_file)


# In[3]:


PixelType = itk.F
Dimension = 3
    
# Read tre file
tubesReader = itk.SpatialObjectReader[Dimension].New()
tubesReader.SetFileName(tubes_file)
tubesReader.Update()
tubes = tubesReader.GetGroup()

# Read mra image
mra = itk.imread(mra_file, PixelType)


# In[6]:


view(image=mra, point_sets=tubes)