Adding back PUGH::enable_all_storage which got accidently removed some time ago.

Also a bit more error checking.


git-svn-id: http://svn.cactuscode.org/arrangements/CactusPUGH/PUGH/trunk@314 b61c5cb5-eaca-4651-9a7a-d64986f99364

1 files changed, 15 insertions, 1 deletions

diff --git a/src/GHExtension.c b/src/GHExtension.c
index 0708603..92f70dc 100644
--- a/src/GHExtension.c
+++ b/src/GHExtension.c
@@ -158,6 +158,7 @@ void *PUGH_SetupGH(tFleshConfig *config,
 
     if (groupsize) free(groupsize);
     if (groupghostsize) free(groupghostsize);
+
   }
 
   free(nsize);
@@ -263,7 +264,7 @@ static void PUGH_InitGHVariables (cGH *GH)
             ((pGA ***)(mypGH->variables))[var][level]->data; 
           break;
         default:
-          CCTK_WARN(1,"Unknown group type in PUGH_InitGHVariables");
+          CCTK_WARN(1,"PUGH_InitGHVariables: Unknown group type in PUGH_InitGHVariables");
       }
     }
   }
@@ -285,8 +286,21 @@ static void PUGH_InitGHVariables (cGH *GH)
 @@*/
 int PUGH_InitGH(cGH *GH)
 {
+  DECLARE_CCTK_PARAMETERS
+
+  int i;
+
   PUGH_InitGHBasics (GH);
 
+  /* Enable storage for all variables if required */
+  if (enable_all_storage)
+  {
+    for (i=0;i<CCTK_NumGroups();i++)
+    {
+      CCTK_EnableGroupStorage(GH,CCTK_GroupName(i));
+    }
+  }
+
   return 0;
 }