Skip to content

Commit 3736a3d

Browse files
committed
Added a .index attr to ARCSpecies
Used by T3 for migrating away from RMG Species() and into ARCSpecies()
1 parent 72b90ae commit 3736a3d

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

arc/species/species.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -287,6 +287,7 @@ class ARCSpecies(object):
287287
multi_species: (str): The multi-species set this species belongs to. Used for running a set of species
288288
simultaneously in a single ESS input file.
289289
symmetry_number (int): The external symmetry number of the species, calculated from mol_list.
290+
index (int): An optional index for the species.
290291
"""
291292

292293
def __init__(self,
@@ -366,6 +367,7 @@ def __init__(self,
366367
self.project_directory = project_directory
367368
self.label = label
368369
self.symmetry_number = None
370+
self.index = None
369371

370372
if species_dict is not None:
371373
# Reading from a dictionary (it's possible that the dict contains only a 'yml_path' argument, check first)
@@ -678,6 +680,8 @@ def as_dict(self,
678680
species_dict['arkane_file'] = self.arkane_file
679681
if not self.consider_all_diastereomers:
680682
species_dict['consider_all_diastereomers'] = self.consider_all_diastereomers
683+
if self.index is not None:
684+
species_dict['index'] = self.index
681685
if self.is_ts:
682686
if len(self.ts_guesses):
683687
species_dict['ts_guesses'] = [tsg.as_dict() for tsg in self.ts_guesses]
@@ -827,6 +831,7 @@ def from_dict(self, species_dict):
827831
self.ts_conf_spawned = species_dict['ts_conf_spawned'] if 'ts_conf_spawned' in species_dict \
828832
else False if self.is_ts else None
829833
self.adjlist = species_dict['adjlist'] if 'adjlist' in species_dict else None
834+
self.index = species_dict['index'] if 'index' in species_dict else None
830835
if self.is_ts:
831836
self.ts_number = species_dict['ts_number'] if 'ts_number' in species_dict else None
832837
self.ts_guesses_exhausted = species_dict['ts_guesses_exhausted'] if 'ts_guesses_exhausted' in species_dict else False

0 commit comments

Comments
 (0)