Coverage for suppy\utils\_calc_dvh.py: 18%

1import numpy as np 

2 

3 

4def calc_dvh(dose, idxs: list): 

5 """""" 

6 

7 d_min = np.min(dose) 

8 d_max = np.max(dose) 

9 n = 1000 

10 dose_grid = np.linspace(0, 1.05 * d_max, n) 

11 dvh = [] 

12 for struct in idxs: 

13 n_struct = dose[struct].shape[0] 

14 dvh.append((dose[struct, None] >= dose_grid).sum(axis=0) / n_struct) 

15 return dvh, dose_grid