0-based material indices

This commit is contained in:
Martin Diehl 2020-10-28 09:31:55 +01:00
parent c3a413e023
commit 5f03e8cf8f
2 changed files with 4 additions and 3 deletions

View File

@ -417,7 +417,7 @@ class Geom:
Number of periods per unit cell. Defaults to 1.
materials : (int, int), optional
Material IDs. Defaults to (1,2).
Notes
-----
The following triply-periodic minimal surfaces are implemented:
@ -687,10 +687,10 @@ class Geom:
def renumber(self):
"""Renumber sorted material indices to 1,...,N."""
"""Renumber sorted material indices to 0,...,N-1."""
renumbered = np.empty(self.grid,dtype=self.material.dtype)
for i, oldID in enumerate(np.unique(self.material)):
renumbered = np.where(self.material == oldID, i+1, renumbered)
renumbered = np.where(self.material == oldID, i, renumbered)
return Geom(material = renumbered,
size = self.size,

View File

@ -176,6 +176,7 @@ class TestGeom:
material = default.material.copy()
for m in np.unique(material):
material[material==m] = material.max() + np.random.randint(1,30)
default.material -= 1
modified = Geom(material,
default.size,
default.origin)