Renumber grains when one is deleted

This is a great fix for the fact that hexrd expects the grains to be sequentially numbered. We might be able to skip this in the future if the requirement is removed from hexrd, but this works fine for now. Signed-off-by: Patrick Avery <[email protected]>
HEXRD · Aug 25, 2022 · a9ef8b9 · a9ef8b9
1 parent b659f52
commit a9ef8b9
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/hexrd/ui/indexing/grains_table_model.py b/hexrd/ui/indexing/grains_table_model.py
@@ -79,8 +79,18 @@ def delete_grains(self, grain_ids):
             any_modified = True
 
         if any_modified:
+            self.renumber_grains()
             self.grains_table_modified.emit()
 
+    def renumber_grains(self):
+        sorted_indices = np.argsort(self.full_grains_table[:, 0])
+        print(f'Renumbering grains from 0 to {len(sorted_indices) - 1}...')
+
+        for i, ind in enumerate(sorted_indices):
+            self.full_grains_table[ind, 0] = i
+
+        self.regenerate_grains_table()
+
     def remove_rows(self, rows):
         for row in rows:
             self.beginRemoveRows(QModelIndex(), row, row)