Add support for evolving complex GFs and (real and complex) GAs.

Only works with ICN or RK2 for now - in fact this commit may break the generic RK methods temporarily.

Note the documentation isn't quite right - there's no longer a seperate function for each different type...


git-svn-id: http://svn.cactuscode.org/arrangements/CactusNumerical/MoL/trunk@12 578cdeb0-5ea1-4b81-8215-5a3b8777ee0b

1 files changed, 56 insertions, 0 deletions

diff --git a/src/ICN.c b/src/ICN.c
index cac5081..7284d00 100644
--- a/src/ICN.c
+++ b/src/ICN.c
@@ -69,12 +69,21 @@ void MoL_ICNAdd(CCTK_ARGUMENTS)
   DECLARE_CCTK_ARGUMENTS;
   DECLARE_CCTK_PARAMETERS;
   
+  cGroupDynamicData arraydata;
+  CCTK_INT groupindex, ierr;
+  CCTK_INT arraytotalsize, arraydim;
+
   CCTK_INT index, var;
   CCTK_INT totalsize;
   CCTK_REAL *OldVar;
   CCTK_REAL *UpdateVar;
   CCTK_REAL *RHSVar;
   
+  CCTK_COMPLEX *OldComplexVar;
+  CCTK_COMPLEX *UpdateComplexVar;
+  CCTK_COMPLEX *RHSComplexVar;
+  CCTK_COMPLEX Complex_Delta_Time = CCTK_Cmplx(CCTK_DELTA_TIME, 0);
+  
 #ifdef MOLDEBUG
   printf("Inside ICN.\nProcessor %d.\nStep %d.\nRefinement %d.\nTimestep %g.\nSpacestep %g.\nTime %g\n",
          CCTK_MyProc(cctkGH),
@@ -104,6 +113,53 @@ void MoL_ICNAdd(CCTK_ARGUMENTS)
     }
   }
   
+  for (var = 0; var < MoLNumEvolvedArrayVariables; var++)
+  {
+    OldVar = (CCTK_REAL*)CCTK_VarDataPtrI(cctkGH, 1,
+                                             EvolvedArrayVariableIndex[var]);
+    UpdateVar = (CCTK_REAL*)CCTK_VarDataPtrI(cctkGH, 0,
+                                             EvolvedArrayVariableIndex[var]);
+    RHSVar = (CCTK_REAL*)CCTK_VarDataPtrI(cctkGH, 0, 
+                                          RHSArrayVariableIndex[var]);
+    
+    groupindex = CCTK_GroupIndexFromVarI(EvolvedArrayVariableIndex[var]);
+    ierr = CCTK_GroupDynamicData(cctkGH, groupindex,
+                                 &arraydata);
+    if (ierr)
+    {
+      CCTK_VWarn(0, __LINE__, __FILE__, "MoL", 
+                 "The driver does not return group information for group '%s'.", 
+                 CCTK_GroupName(groupindex));
+    }
+    arraytotalsize = 1;
+    for (arraydim = 0; arraydim < arraydata.dim; arraydim++)
+    {
+      arraytotalsize *= arraydata.lsh[arraydim];
+    }
+
+    for (index = 0; index < arraytotalsize; index++)
+    {
+      UpdateVar[index] = OldVar[index] + CCTK_DELTA_TIME * RHSVar[index];
+    }
+  }
+  
+  for (var = 0; var < MoLNumEvolvedComplexVariables; var++)
+  {
+    OldComplexVar = (CCTK_COMPLEX*)CCTK_VarDataPtrI(cctkGH, 1,
+                                                    EvolvedComplexVariableIndex[var]);
+    UpdateComplexVar = (CCTK_COMPLEX*)CCTK_VarDataPtrI(cctkGH, 0,
+                                                       EvolvedComplexVariableIndex[var]);
+    RHSComplexVar = (CCTK_COMPLEX*)CCTK_VarDataPtrI(cctkGH, 0, 
+                                                    RHSComplexVariableIndex[var]);
+    
+    for (index = 0; index < totalsize; index++)
+    {
+      UpdateComplexVar[index] = CCTK_CmplxAdd(OldComplexVar[index],
+                                              CCTK_CmplxMul(Complex_Delta_Time,
+                                                            RHSComplexVar[index]));
+    }
+  }
+  
   return;
 
 }