int main( int argc, char ** argv ) { int iArg; /* -------------------------------------------------------------------- */ /* Process arguments. */ /* -------------------------------------------------------------------- */ argc = GDALGeneralCmdLineProcessor( argc, &argv, 0 ); if( argc < 1 ) exit( -argc ); for( iArg = 1; iArg < argc; iArg++ ) { if( EQUAL(argv[iArg],"-i") && iArg < argc-1 ) nIterations = atoi(argv[++iArg]); else if( EQUAL(argv[iArg],"-oi") && iArg < argc-1 ) nOpenIterations = atoi(argv[++iArg]); else if( EQUAL(argv[iArg],"-t") && iArg < argc-1 ) nThreadCount = atoi(argv[++iArg]); else if( EQUAL(argv[iArg],"-nlo") ) bLockOnOpen = FALSE; else if( pszFilename == NULL ) pszFilename = argv[iArg]; else { printf( "Unrecognised argument: %s\n", argv[iArg] ); Usage(); } } if( pszFilename == NULL ) { printf( "Need a file to operate on.\n" ); Usage(); exit( 1 ); } if( nOpenIterations > 0 ) bLockOnOpen = FALSE; /* -------------------------------------------------------------------- */ /* Get the checksum of band1. */ /* -------------------------------------------------------------------- */ GDALDatasetH hDS; GDALAllRegister(); hDS = GDALOpen( pszFilename, GA_ReadOnly ); if( hDS == NULL ) exit( 1 ); nChecksum = GDALChecksumImage( GDALGetRasterBand( hDS, 1 ), 0, 0, GDALGetRasterXSize( hDS ), GDALGetRasterYSize( hDS ) ); GDALClose( hDS ); printf( "Got checksum %d, launching %d worker threads on %s, %d iterations.\n", nChecksum, nThreadCount, pszFilename, nIterations ); /* -------------------------------------------------------------------- */ /* Fire off worker threads. */ /* -------------------------------------------------------------------- */ int iThread; pGlobalMutex = CPLCreateMutex(); CPLReleaseMutex( pGlobalMutex ); nPendingThreads = nThreadCount; for( iThread = 0; iThread < nThreadCount; iThread++ ) { if( CPLCreateThread( WorkerFunc, NULL ) == -1 ) { printf( "CPLCreateThread() failed.\n" ); exit( 1 ); } } while( nPendingThreads > 0 ) CPLSleep( 0.5 ); CPLReleaseMutex( pGlobalMutex ); printf( "All threads complete.\n" ); CSLDestroy( argv ); GDALDestroyDriverManager(); return 0; }
int main( int argc, char ** argv ) { /* -------------------------------------------------------------------- */ /* Process arguments. */ /* -------------------------------------------------------------------- */ argc = GDALGeneralCmdLineProcessor(argc, &argv, 0); if( argc < 1 ) exit(-argc); int nThreadCount = 4; bool bOpenInThreads = true; for( int iArg = 1; iArg < argc; iArg++ ) { if( iArg < argc-1 && EQUAL(argv[iArg], "-i") ) { nIterations = atoi(argv[++iArg]); } else if( iArg < argc-1 && EQUAL(argv[iArg], "-oi") ) { nOpenIterations = atoi(argv[++iArg]); } else if( iArg < argc-1 && EQUAL(argv[iArg], "-t") ) { nThreadCount = atoi(argv[++iArg]); } else if( EQUAL(argv[iArg], "-lock_on_open") ) { bLockOnOpen = true; } else if( EQUAL(argv[iArg], "-open_in_main") ) { bOpenInThreads = false; } else if( pszFilename == nullptr ) { pszFilename = argv[iArg]; } else { printf("Unrecognized argument: %s\n", argv[iArg]); Usage(); } } if( pszFilename == nullptr ) { printf("Need a file to operate on.\n"); Usage(); exit(1); } if( nOpenIterations > 0 ) bLockOnOpen = false; /* -------------------------------------------------------------------- */ /* Get the checksum of band1. */ /* -------------------------------------------------------------------- */ GDALDatasetH hDS = nullptr; GDALAllRegister(); for( int i = 0; i < 2; i++ ) { hDS = GDALOpen( pszFilename, GA_ReadOnly ); if( hDS == nullptr ) exit( 1 ); nChecksum = GDALChecksumImage(GDALGetRasterBand(hDS, 1), 0, 0, GDALGetRasterXSize(hDS), GDALGetRasterYSize(hDS)); GDALClose(hDS); } printf( "Got checksum %d, launching %d worker threads on %s, %d iterations.\n", nChecksum, nThreadCount, pszFilename, nIterations); /* -------------------------------------------------------------------- */ /* Fire off worker threads. */ /* -------------------------------------------------------------------- */ pGlobalMutex = CPLCreateMutex(); CPLReleaseMutex(pGlobalMutex); nPendingThreads = nThreadCount; std::vector<GDALDatasetH> aoDS; for( int iThread = 0; iThread < nThreadCount; iThread++ ) { hDS = nullptr; if( !bOpenInThreads ) { hDS = GDALOpen(pszFilename, GA_ReadOnly); if( !hDS ) { printf("GDALOpen() failed.\n"); exit(1); } aoDS.push_back(hDS); } if( CPLCreateThread(WorkerFunc, hDS) == -1 ) { printf("CPLCreateThread() failed.\n"); exit(1); } } while( nPendingThreads > 0 ) CPLSleep(0.5); CPLDestroyMutex(pGlobalMutex); for( size_t i = 0; i < aoDS.size(); ++i ) GDALClose(aoDS[i]); printf("All threads complete.\n"); CSLDestroy(argv); GDALDestroyDriverManager(); return 0; }