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 */ 20*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFormatConvertGetSize(formatstr,&sz)); 2155ed0643SJacob Faibussowitsch if (PetscDefined(USE_64BIT_INDICES)) { 222c71b3e2SJacob Faibussowitsch PetscCheckFalse(sz != 29,PETSC_COMM_WORLD,PETSC_ERR_PLIB,"Format size %zu should be 29",sz); 2355ed0643SJacob Faibussowitsch } else { 242c71b3e2SJacob Faibussowitsch PetscCheckFalse(sz != 27,PETSC_COMM_WORLD,PETSC_ERR_PLIB,"Format size %zu should be 27",sz); 2555ed0643SJacob Faibussowitsch } 26*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscMalloc1(sz,&newformatstr)); 27*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFormatConvert(formatstr,newformatstr)); 28*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,newformatstr,twentytwo,3.47,3.0)); 29*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscFree(newformatstr)); 30c4762a1bSJed Brown 31c4762a1bSJed Brown /* Test correct count is returned with %g format */ 32*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSNPrintfCount(buffer,sizeof(buffer),"Test %g %g\n",&sz,3.33,2.7)); 33*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscStrlen(buffer,&fullLength)); 342c71b3e2SJacob Faibussowitsch PetscCheckFalse(sz != fullLength+1,PETSC_COMM_SELF,PETSC_ERR_PLIB,"PetscSNPrintfCount() count should be %d it is %d",(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 */ 37*5f80ce2aSJacob Faibussowitsch CHKERRQ(TestPetscVSNPrintf(buffer,sizeof(buffer),&fullLength,"Greetings %s","This is my string")); 38*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"buffer :%s: fullLength %d\n",buffer,(int)fullLength)); 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; 42*5f80ce2aSJacob Faibussowitsch CHKERRQ(TestPetscVSNPrintf(buffer,sizeof(buffer),&fullLength,"Greetings %s",longstr)); 43*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"longstr fullLength %d\n",(int)fullLength)); 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; 47*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"Greetings %s",superlongstr)); 48c4762a1bSJed Brown 49c4762a1bSJed Brown /* test that PetscSynchronizedPrintf() works for strings longer than the default buffer size */ 50*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSynchronizedPrintf(PETSC_COMM_WORLD,"Greetings %s",superlongstr)); 51*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSynchronizedFlush(PETSC_COMM_WORLD,stdout)); 52c4762a1bSJed Brown 53c4762a1bSJed Brown /* test that PetscSynchronizedFPrintf() works for strings longer than the default buffer size */ 54*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSynchronizedFPrintf(PETSC_COMM_WORLD,stdout,"Greetings %s",superlongstr)); 55*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSynchronizedFlush(PETSC_COMM_WORLD,stdout)); 56c4762a1bSJed Brown 57c4762a1bSJed Brown /* test that PetscSynchronizedFPrintf() works for strings longer than the default buffer size */ 58*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPushSynchronized(PETSC_VIEWER_STDOUT_WORLD)); 59*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIISynchronizedPrintf(PETSC_VIEWER_STDOUT_WORLD,"Greetings %s",superlongstr)); 60*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerFlush(PETSC_VIEWER_STDOUT_WORLD)); 61*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscViewerASCIIPopSynchronized(PETSC_VIEWER_STDOUT_WORLD)); 62c4762a1bSJed Brown 63c4762a1bSJed Brown /* add new line to end of file so that diff does not warn about it being missing */ 64*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscPrintf(PETSC_COMM_WORLD,"\n")); 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 73c4762a1bSJed Brown PetscFunctionBegin; 74c4762a1bSJed Brown va_start(Argp,format); 75*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscVSNPrintf(str,l_str,format,fullLength,Argp)); 76c4762a1bSJed Brown PetscFunctionReturn(0); 77c4762a1bSJed Brown } 78c4762a1bSJed Brown /*TEST 79c4762a1bSJed Brown 80c4762a1bSJed Brown test: 81c4762a1bSJed Brown nsize: 2 82dfd57a17SPierre Jolivet requires: defined(PETSC_HAVE_VA_COPY) 83c4762a1bSJed Brown 84c4762a1bSJed Brown TEST*/ 85