! $Id$ ! ! System test code FieldLSSMM !------------------------------------------------------------------------- !ESMF_MULTI_PROC_SYSTEM_TEST String used by test script to count system tests. !========================================================================= !BOP ! ! !DESCRIPTION: ! System test FieldLSSMM. ! Field SMM test. 2 components and 1 coupler, one-way coupling. ! The first component has a location stream with ! a Field whose data is set to constant value 1 ! and then transformed to the second component through ! sparse matrix multiplication operation. The destination field builds upon ! a location stream. The transformed data is then compared ! to predetermined result based on the structure of SMM operation. ! !\begin{verbatim} program ESMF_FieldLSSMMSTest #define ESMF_METHOD "program ESMF_FieldLSSMMSTest" #include "ESMF.h" ! ESMF Framework module use ESMF use ESMF_TestMod use user_model1, only : userm1_register use user_model2, only : userm2_register use user_coupler, only : usercpl_register implicit none ! Local variables integer :: pet_id, npets, rc, localrc, userrc character(len=ESMF_MAXSTR) :: cname1, cname2, cplname type(ESMF_VM):: vm type(ESMF_State) :: c1exp, c2imp type(ESMF_GridComp) :: comp1, comp2 type(ESMF_CplComp) :: cpl ! instantiate a clock, a calendar, and timesteps type(ESMF_Clock) :: clock type(ESMF_Calendar) :: gregorianCalendar type(ESMF_TimeInterval) :: timeStep type(ESMF_Time) :: startTime type(ESMF_Time) :: stopTime ! cumulative result: count failures; no failures equals "all pass" integer :: result = 0 ! individual test name character(ESMF_MAXSTR) :: testname ! individual test failure message, and final status msg character(ESMF_MAXSTR) :: failMsg, finalMsg ! set rc = ESMF_SUCCESS rc = ESMF_SUCCESS !------------------------------------------------------------------------- !------------------------------------------------------------------------- print *, "-------------------------------- " print *, "Start of System Test FieldLSSMM:" print *, "-------------------------------- " !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! Create section !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! ! Initialize framework and get back default global VM call ESMF_Initialize(vm=vm, defaultlogfilename="FieldLSSMMSTest.Log", & logkindflag=ESMF_LOGKIND_MULTI, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) call ESMF_LogSet (flush=.true.) ! Get number of PETs we are running with call ESMF_VMGet(vm, petCount=npets, localPet=pet_id, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) ! Check for correct number of PETs if ( npets < 4 ) then call ESMF_LogSetError(ESMF_RC_ARG_BAD,& msg="This system test does not run on fewer than 4 PETs.",& ESMF_CONTEXT, rcToReturn=rc) call ESMF_Finalize(rc=rc, endflag=ESMF_END_ABORT) endif ! Create the 2 model components and coupler cname1 = "user model 1" comp1 = ESMF_GridCompCreate(name=cname1, petList=(/0,1/), rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) print *, "Created component ", trim(cname1), "rc =", rc ! call ESMF_GridCompPrint(comp1, "", rc) cname2 = "user model 2" comp2 = ESMF_GridCompCreate(name=cname2, petList=(/2,3/), rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) print *, "Created component ", trim(cname2), "rc =", rc ! call ESMF_GridCompPrint(comp2, "", rc) cplname = "user one-way coupler" cpl = ESMF_CplCompCreate(name=cplname, petList=(/0,1,2,3/), rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) print *, "Created component ", trim(cplname), ", rc =", rc ! call ESMF_CplCompPrint(cpl, "", rc) print *, "Comp Creates finished" !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! Register section !------------------------------------------------------------------------- !------------------------------------------------------------------------- call ESMF_GridCompSetServices(comp1, userRoutine=userm1_register, & userRc=userrc, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) print *, "Comp SetServices finished, rc= ", rc, userrc call ESMF_GridCompSetServices(comp2, userRoutine=userm2_register, & userRc=userrc, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) print *, "Comp SetServices finished, rc= ", rc, userrc call ESMF_CplCompSetServices(cpl, userRoutine=usercpl_register, & userRc=userrc, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) print *, "Comp SetServices finished, rc= ", rc, userrc !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! Create and initialize a clock. !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! initialize calendar to be Gregorian type gregorianCalendar = ESMF_CalendarCreate(ESMF_CALKIND_GREGORIAN, & name="Gregorian", rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) ! initialize time interval to 6 hours call ESMF_TimeIntervalSet(timeStep, h=6, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) ! initialize start time to 5/01/2003 call ESMF_TimeSet(startTime, yy=2003, mm=5, dd=1, & calendar=gregorianCalendar, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) ! initialize stop time to 5/02/2003 call ESMF_TimeSet(stopTime, yy=2003, mm=5, dd=1, h=6, & calendar=gregorianCalendar, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) ! initialize the clock with the above values clock = ESMF_ClockCreate(timeStep, startTime, stopTime=stopTime, & name="Clock 1", rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! Init section !------------------------------------------------------------------------- !------------------------------------------------------------------------- c1exp = ESMF_StateCreate(name="comp1 export", & stateintent=ESMF_STATEINTENT_EXPORT, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) call ESMF_GridCompInitialize(comp1, exportState=c1exp, clock=clock, & userRc=userrc, rc=localrc) print *, "Comp 1 Initialize finished, rc =", rc if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) c2imp = ESMF_StateCreate(name="comp2 import", & stateintent=ESMF_STATEINTENT_IMPORT, rc=localrc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) call ESMF_GridCompInitialize(comp2, importState=c2imp, clock=clock, & userRc=userrc, rc=localrc) print *, "Comp 2 Initialize finished, rc =", rc if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) ! note that the coupler's import is comp1's export call ESMF_CplCompInitialize(cpl, importState=c1exp, & exportState=c2imp, clock=clock, & userRc=userrc, rc=localrc) print *, "Coupler Initialize finished, rc =", rc if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! Run section !------------------------------------------------------------------------- !------------------------------------------------------------------------- do while (.not. ESMF_ClockIsStopTime(clock, rc=rc)) call ESMF_GridCompRun(comp1, exportState=c1exp, clock=clock, & userRc=userrc, rc=localrc) print *, "Comp 1 Run returned, rc =", rc if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) call ESMF_CplCompRun(cpl, importState=c1exp, & exportState=c2imp, clock=clock, & userRc=userrc, rc=localrc) print *, "Coupler Run returned, rc =", rc if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) call ESMF_GridCompRun(comp2, importState=c2imp, clock=clock, & userRc=userrc, rc=localrc) print *, "Comp 2 Run returned, rc =", rc if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) call ESMF_ClockAdvance(clock, rc=localrc) !call ESMF_ClockPrint(clock, rc=rc) if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) enddo !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! Finalize section !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! Print result call ESMF_GridCompFinalize(comp1, exportState=c1exp, clock=clock, & userRc=userrc, rc=localrc) print *, "Comp 1 Finalize finished, rc =", rc, userrc if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) call ESMF_GridCompFinalize(comp2, importState=c2imp, clock=clock, & userRc=userrc, rc=localrc) print *, "Comp 2 Finalize finished, rc =", rc, userrc if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) call ESMF_CplCompFinalize(cpl, importState=c1exp, & exportState=c2imp, clock=clock, & userRc=userrc, rc=localrc) print *, "Coupler Finalize finished, rc =", rc, userrc if (ESMF_LogFoundError(localrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) if (ESMF_LogFoundError(userrc, ESMF_ERR_PASSTHRU, & ESMF_CONTEXT, rcToReturn=rc)) & call ESMF_Finalize(endflag=ESMF_END_ABORT) print *, "------------------------------------------------------------" print *, "------------------------------------------------------------" print *, "Test finished, pet_id = ", pet_id print *, "------------------------------------------------------------" print *, "------------------------------------------------------------" print *, "Comp Finalize returned" ! !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! Destroy section !------------------------------------------------------------------------- !------------------------------------------------------------------------- ! Clean up call ESMF_StateDestroy(c1exp, rc=rc) call ESMF_StateDestroy(c2imp, rc=rc) call ESMF_ClockDestroy(clock, rc=rc) call ESMF_CalendarDestroy(gregorianCalendar, rc=rc) call ESMF_GridCompDestroy(comp1, rc=rc) call ESMF_GridCompDestroy(comp2, rc=rc) call ESMF_CplCompDestroy(cpl, rc=rc) print *, "All Destroy routines done" !------------------------------------------------------------------------- !------------------------------------------------------------------------- 10 print *, "System Test FieldLSSMM complete." ! Normal ESMF Test output write(failMsg, *) "System Test failure" write(testname, *) "System Test FieldLSSMM: LocStream based Field SMM" if (rc .ne. ESMF_SUCCESS) then ! Separate message to console, for quick confirmation of success/failure if (rc .eq. ESMF_SUCCESS) then write(finalMsg, *) "SUCCESS: LocStream based Field SMM test finished correctly." else write(finalMsg, *) "System Test did not succeed. Error code ", rc endif write(0, *) "" write(0, *) trim(testname) write(0, *) trim(finalMsg) write(0, *) "" endif ! IMPORTANT: ESMF_STest() prints the PASS string and the # of processors in the log ! file that the scripts grep for. call ESMF_STest((rc.eq.ESMF_SUCCESS), testname, failMsg, result, ESMF_SRCLINE) call ESMF_Finalize(rc=rc) end program ESMF_FieldLSSMMSTest !\end{verbatim}