1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "Test PetscFormatConvertGetSize().\n"; 3c4762a1bSJed Brown 4c4762a1bSJed Brown #include <petscsys.h> 5c4762a1bSJed Brown #include <petscviewer.h> 6c4762a1bSJed Brown 7c4762a1bSJed Brown PetscErrorCode TestPetscVSNPrintf(char*,size_t,size_t*,const char*,...); 8c4762a1bSJed Brown 9c4762a1bSJed Brown int main(int argc,char **argv) 10c4762a1bSJed Brown { 11c4762a1bSJed Brown PetscErrorCode ierr; 12c4762a1bSJed Brown size_t sz,fullLength; 13c4762a1bSJed Brown char *newformatstr,buffer[128],longstr[256],superlongstr[10000]; 14c4762a1bSJed Brown const char *formatstr = "Greetings %D %3.2f %g\n"; 15c4762a1bSJed Brown PetscInt i,twentytwo = 22; 16c4762a1bSJed Brown 17c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 18c4762a1bSJed Brown 19c4762a1bSJed Brown /* test that PetscFormatConvertGetSize() correctly counts needed amount of space */ 20c4762a1bSJed Brown ierr = PetscFormatConvertGetSize(formatstr,&sz);CHKERRQ(ierr); 21c4762a1bSJed Brown #if !defined(PETSC_USE_64BIT_INDICES) 22c4762a1bSJed Brown if (sz != 27) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Format size %d should be 27\n",(int)sz); 23c4762a1bSJed Brown #else 24c4762a1bSJed Brown if (sz != 29) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Format size %d should be 29\n",(int)sz); 25c4762a1bSJed Brown #endif 26c4762a1bSJed Brown ierr = PetscMalloc1(sz,&newformatstr);CHKERRQ(ierr); 27c4762a1bSJed Brown ierr = PetscFormatConvert(formatstr,newformatstr);CHKERRQ(ierr); 28c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,newformatstr,twentytwo,3.47,3.0);CHKERRQ(ierr); 29c4762a1bSJed Brown ierr = PetscFree(newformatstr);CHKERRQ(ierr); 30c4762a1bSJed Brown 31c4762a1bSJed Brown /* Test correct count is returned with %g format */ 32c4762a1bSJed Brown ierr = PetscSNPrintfCount(buffer,sizeof(buffer),"Test %g %g\n",&sz,3.33,2.7);CHKERRQ(ierr); 33c4762a1bSJed Brown ierr = PetscStrlen(buffer,&fullLength);CHKERRQ(ierr); 34c4762a1bSJed Brown if (sz != fullLength+1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"PetscSNPrintfCount() count should be %d it is %d\n",(int)fullLength+1,(int)sz); 35c4762a1bSJed Brown 36c4762a1bSJed Brown /* test that TestPetscVSNPrintf() fullLength argument returns required space for the string when buffer is long enough */ 37c4762a1bSJed Brown ierr = TestPetscVSNPrintf(buffer,sizeof(buffer),&fullLength,"Greetings %s","This is my string");CHKERRQ(ierr); 38c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,"buffer :%s: fullLength %d\n",buffer,(int)fullLength);CHKERRQ(ierr); 39c4762a1bSJed Brown 40c4762a1bSJed Brown /* test that TestPetscVSNPrintf() fullLength argument returns required space for the string when buffer is not long enough */ 41c4762a1bSJed Brown for (i=0; i<255; i++) {longstr[i] = 's';} longstr[255] = 0; 42c4762a1bSJed Brown ierr = TestPetscVSNPrintf(buffer,sizeof(buffer),&fullLength,"Greetings %s",longstr);CHKERRQ(ierr); 43c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,"longstr fullLength %d\n",(int)fullLength);CHKERRQ(ierr); 44c4762a1bSJed Brown 45c4762a1bSJed Brown /* test that PetscPrintf() works for strings longer than the default buffer size */ 46c4762a1bSJed Brown for (i=0; i<9998; i++) {superlongstr[i] = 's';} superlongstr[9998] = 't'; superlongstr[9999] = 0; 47c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,"Greetings %s",superlongstr);CHKERRQ(ierr); 48c4762a1bSJed Brown 49c4762a1bSJed Brown /* test that PetscSynchronizedPrintf() works for strings longer than the default buffer size */ 50c4762a1bSJed Brown ierr = PetscSynchronizedPrintf(PETSC_COMM_WORLD,"Greetings %s",superlongstr);CHKERRQ(ierr); 51c4762a1bSJed Brown ierr = PetscSynchronizedFlush(PETSC_COMM_WORLD,stdout);CHKERRQ(ierr); 52c4762a1bSJed Brown 53c4762a1bSJed Brown /* test that PetscSynchronizedFPrintf() works for strings longer than the default buffer size */ 54c4762a1bSJed Brown ierr = PetscSynchronizedFPrintf(PETSC_COMM_WORLD,stdout,"Greetings %s",superlongstr);CHKERRQ(ierr); 55c4762a1bSJed Brown ierr = PetscSynchronizedFlush(PETSC_COMM_WORLD,stdout);CHKERRQ(ierr); 56c4762a1bSJed Brown 57c4762a1bSJed Brown /* test that PetscSynchronizedFPrintf() works for strings longer than the default buffer size */ 58c4762a1bSJed Brown ierr = PetscViewerASCIIPushSynchronized(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 59c4762a1bSJed Brown ierr = PetscViewerASCIISynchronizedPrintf(PETSC_VIEWER_STDOUT_WORLD,"Greetings %s",superlongstr);CHKERRQ(ierr); 60c4762a1bSJed Brown ierr = PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 61c4762a1bSJed Brown ierr = PetscViewerASCIIPopSynchronized(PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 62c4762a1bSJed Brown 63c4762a1bSJed Brown /* add new line to end of file so that diff does not warn about it being missing */ 64c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_WORLD,"\n");CHKERRQ(ierr); 65c4762a1bSJed Brown ierr = PetscFinalize(); 66c4762a1bSJed Brown return ierr; 67c4762a1bSJed Brown } 68c4762a1bSJed Brown 69c4762a1bSJed Brown PetscErrorCode TestPetscVSNPrintf(char *str,size_t l_str,size_t *fullLength,const char* format,...) 70c4762a1bSJed Brown { 71c4762a1bSJed Brown va_list Argp; 72c4762a1bSJed Brown PetscErrorCode ierr; 73c4762a1bSJed Brown 74c4762a1bSJed Brown PetscFunctionBegin; 75c4762a1bSJed Brown va_start(Argp,format); 76c4762a1bSJed Brown ierr = PetscVSNPrintf(str,l_str,format,fullLength,Argp);CHKERRQ(ierr); 77c4762a1bSJed Brown PetscFunctionReturn(0); 78c4762a1bSJed Brown } 79c4762a1bSJed Brown /*TEST 80c4762a1bSJed Brown 81c4762a1bSJed Brown test: 82c4762a1bSJed Brown nsize: 2 83*dfd57a17SPierre Jolivet requires: defined(PETSC_HAVE_VA_COPY) 84c4762a1bSJed Brown 85c4762a1bSJed Brown TEST*/ 86