Now also pickling/unpicking species.speciesCounter in restart file.

Should have done this from the beginning.
ReactionMechanismGenerator · Nov 12, 2009 · fe837a2 · fe837a2
1 parent 0585978
commit fe837a2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/source/rmg/main.py b/source/rmg/main.py
@@ -101,7 +101,7 @@ def execute(inputFile, options):
 		logging.info('Loading previous restart file...')
 		f = open(os.path.join(settings.outputDirectory,'restart.pkl'), 'rb')
 		species.speciesList = cPickle.load(f)
-		species.speciesCounter = len(species.speciesList)
+		species.speciesCounter = cPickle.load(f)
 		reaction.reactionList = cPickle.load(f)
 		reactionModel = cPickle.load(f)
 		reactionSystems = cPickle.load(f)
@@ -151,6 +151,7 @@ def execute(inputFile, options):
 		logging.info('Saving restart file...')
 		f = open(os.path.join(settings.outputDirectory,'restart.pkl'), 'wb')
 		cPickle.dump(species.speciesList, f)
+		cPickle.dump(species.speciesCounter, f)
 		cPickle.dump(reaction.reactionList, f)
 		cPickle.dump(reactionModel, f)
 		cPickle.dump(reactionSystems, f)