Commit 9e599160abed4e9dbfcd0aae6ebecf9850a1f76b
1 parent
586739ec
Exists in
master
and in
68 other branches
ENH: Removed list of list from dicom_grouper
Showing
1 changed file
with
11 additions
and
10 deletions
Show diff stats
invesalius/reader/dicom_grouper.py
| ... | ... | @@ -136,9 +136,9 @@ class DicomGroups: |
| 136 | 136 | information.acquisition.serie_number, information.image.orientation_label) |
| 137 | 137 | |
| 138 | 138 | if (key in groups_dcm.keys()): |
| 139 | - groups_dcm[key][0].append(information) | |
| 139 | + groups_dcm[key].append(information) | |
| 140 | 140 | else: |
| 141 | - groups_dcm[key] = [[information]] | |
| 141 | + groups_dcm[key] = [information] | |
| 142 | 142 | |
| 143 | 143 | self.groups_dcm = groups_dcm |
| 144 | 144 | |
| ... | ... | @@ -191,19 +191,19 @@ class DicomGroups: |
| 191 | 191 | #only adds value. Otherwise this key create in the |
| 192 | 192 | #dictionary and add value |
| 193 | 193 | if (key in groups_dcm.keys()): |
| 194 | - groups_dcm[key][0].append(information) | |
| 194 | + groups_dcm[key].append(information) | |
| 195 | 195 | else: |
| 196 | - groups_dcm[key] = [[information]] | |
| 196 | + groups_dcm[key] = [information] | |
| 197 | 197 | else: |
| 198 | 198 | cont_series = cont_series + 1 |
| 199 | - groups_dcm[key] = [[information]] | |
| 199 | + groups_dcm[key] = [information] | |
| 200 | 200 | |
| 201 | 201 | else: |
| 202 | 202 | |
| 203 | 203 | if (cont_series in groups_dcm.keys()): |
| 204 | 204 | groups_dcm[key].append(information) |
| 205 | 205 | else: |
| 206 | - groups_dcm[key] = [[information]] | |
| 206 | + groups_dcm[key] = [information] | |
| 207 | 207 | |
| 208 | 208 | cont_series = cont_series + 1 |
| 209 | 209 | |
| ... | ... | @@ -229,12 +229,12 @@ class DicomGroups: |
| 229 | 229 | key = groups_dcm.keys()[n] |
| 230 | 230 | |
| 231 | 231 | #Number of slices in the series |
| 232 | - size_list = len(groups_dcm[key][0]) | |
| 232 | + size_list = len(groups_dcm[key]) | |
| 233 | 233 | |
| 234 | 234 | for y in xrange(size_list): |
| 235 | 235 | |
| 236 | 236 | #Slices walks in the series |
| 237 | - information = groups_dcm[key][0][y] | |
| 237 | + information = groups_dcm[key][y] | |
| 238 | 238 | |
| 239 | 239 | #Generate new key to dictionary |
| 240 | 240 | image_pos = information.image.position |
| ... | ... | @@ -298,11 +298,12 @@ class DicomGroups: |
| 298 | 298 | #the number of previously existing number is |
| 299 | 299 | #greater or equal then the group keeps up, |
| 300 | 300 | #but maintains the same group of positions. |
| 301 | - #if len(self.groups_dcm.keys()) > len(groups_dcm_.keys()): | |
| 302 | - self.groups_dcm = groups_dcm_ | |
| 301 | + if len(self.groups_dcm.keys()) > len(groups_dcm_.keys()): | |
| 302 | + self.groups_dcm = groups_dcm_ | |
| 303 | 303 | |
| 304 | 304 | for j in xrange(len(self.groups_dcm.keys())): |
| 305 | 305 | key = self.groups_dcm.keys()[j] |
| 306 | + #print self.groups_dcm[key] | |
| 306 | 307 | self.groups_dcm[key].sort(key=lambda x: x.image.number) |
| 307 | 308 | |
| 308 | 309 | ... | ... |