Commit 4e80f1ec877de787818a8daf44e5790cb74d5823
1 parent
83eaa757
Exists in
master
and in
68 other branches
ENH: Measurements operations in project
Showing
1 changed file
with
27 additions
and
6 deletions
Show diff stats
invesalius/project.py
| ... | ... | @@ -60,14 +60,14 @@ class Project(object): |
| 60 | 60 | |
| 61 | 61 | # Masks (vtkImageData) |
| 62 | 62 | self.mask_dict = {} |
| 63 | - self.last_mask_index = 0 | |
| 63 | + #self.last_mask_index = 0 | |
| 64 | 64 | |
| 65 | 65 | # Surfaces are (vtkPolyData) |
| 66 | 66 | self.surface_dict = {} |
| 67 | - self.last_surface_index = -1 | |
| 67 | + #self.last_surface_index = -1 | |
| 68 | 68 | |
| 69 | - # TODO: Future | |
| 70 | - self.measure_dict = {} | |
| 69 | + # Measurements | |
| 70 | + self.measurement_dict = {} | |
| 71 | 71 | |
| 72 | 72 | # TODO: Future ++ |
| 73 | 73 | self.annotation_dict = {} |
| ... | ... | @@ -109,7 +109,7 @@ class Project(object): |
| 109 | 109 | output |
| 110 | 110 | @ index: index of item that was inserted |
| 111 | 111 | """ |
| 112 | - self.last_mask_index = mask.index | |
| 112 | + #self.last_mask_index = mask.index | |
| 113 | 113 | index = len(self.mask_dict) |
| 114 | 114 | self.mask_dict[index] = mask |
| 115 | 115 | return index |
| ... | ... | @@ -128,7 +128,7 @@ class Project(object): |
| 128 | 128 | return self.mask_dict[index] |
| 129 | 129 | |
| 130 | 130 | def AddSurface(self, surface): |
| 131 | - self.last_surface_index = surface.index | |
| 131 | + #self.last_surface_index = surface.index | |
| 132 | 132 | index = len(self.surface_dict) |
| 133 | 133 | self.surface_dict[index] = surface |
| 134 | 134 | return index |
| ... | ... | @@ -147,6 +147,27 @@ class Project(object): |
| 147 | 147 | new_dict[i-1].index = i-1 |
| 148 | 148 | self.surface_dict = new_dict |
| 149 | 149 | |
| 150 | + | |
| 151 | + def AddMeasurement(self, measurement): | |
| 152 | + index = len(self.measurement_dict) | |
| 153 | + self.measurement_dict[index] = measurement | |
| 154 | + return index | |
| 155 | + | |
| 156 | + def ChangeMeasurement(self, measurement): | |
| 157 | + index = measurement.index | |
| 158 | + self.measurement_dict[index] = measurement | |
| 159 | + | |
| 160 | + def RemoveMeasurement(self, index): | |
| 161 | + new_dict = {} | |
| 162 | + for i in self.measurement_dict: | |
| 163 | + if i < index: | |
| 164 | + new_dict[i] = self.measurement_dict[i] | |
| 165 | + if i > index: | |
| 166 | + new_dict[i-1] = self.measurement_dict[i] | |
| 167 | + new_dict[i-1].index = i-1 | |
| 168 | + self.measurement_dict = new_dict | |
| 169 | + | |
| 170 | + | |
| 150 | 171 | def SetAcquisitionModality(self, type_=None): |
| 151 | 172 | if type_ is None: |
| 152 | 173 | type_ = self.modality | ... | ... |