在调用GDAL算法的时候,希望能够显示其处理进度信息,其实在GDAL的算法API中,一般最后两个参数就是进度信息的指针。下面分别实现两种进度条信息,一种是在控制台中的进度条,一种是基于QT界面的进度条(你可以参考写一个MFC的)。
对于GDAL来说,本身就实现了一个基于控制台的进度条函数,名字叫GDALTermProgress,其函数说明参考这里 ,调用这个进度函数后,会在控制台中显示一个进度信息,形状大概就是下面的样子:
0...10...20...30...40...50...60...70...80...90...100 - done.
每个2.5%就会输出一个点,然后每到10%就会输出数字,在完成结束后会输出done。
GDAL的算法中,一般最后两个参数是用来返回进度信息的,下面以栅格矢量化的接口举例说明,GDAL栅格矢量化的函数接口定义为:
CPLErr GDALPolygonize ( GDALRasterBandH hSrcBand,
GDALRasterBandH hMaskBand,
OGRLayerH hOutLayer,
int iPixValField,
char ** papszOptions,
GDALProgressFunc pfnProgress,
void * pProgressArg
)
算法说明文档参考地址:http://www.gdal.org/gdal__alg_8h.html#a3f522a9035d3512b5d414fb4752671b1。从文档中可以看到,最后两个参数说明是:
pfnProgress callback for reporting algorithm progress matching the GDALProgressFunc() semantics. May be NULL.
pProgressArg callback argument passed to pfnProgress.
上面的意思大概就是说,参数pfnProgress 是一个回调函数指针,回调函数定义参考 GDALProgressFunc(),参数pProgressArg 是回调函数pfnProgress 的第三个参数。关于GDALProgressFunc()的定义如下:
int (*GDALProgressFunc)(double dfComplete, const char *pszMessage, void *pProgressArg);
该回调函数参数说明分别是:
dfComplete 进度值,取值范围是0.0~1.0,0.0表示开始,1.0表示完成。
pszMessage 可选项,用来显示的字符串信息。通常用NULL即可。
pProgressArg 应用程序回调函数参数,是一个自定义的类(或结构体)指针。
下面以GDAL自带的进度函数GDALTermProgress为例,进行说明。GDALTermProgress的源代码在GDALSrc\gcore\gdal_misc.cpp,762行左右(GDAL1.8.1版本),代码如下:
int CPL_STDCALL GDALTermProgress( double dfComplete, const char *pszMessage,
void * pProgressArg )
{
static int nLastTick = -1;
int nThisTick = (int) (dfComplete * 40.0);
(void) pProgressArg;
nThisTick = MIN(40,MAX(0,nThisTick));
// Have we started a new progress run?
if( nThisTick < nLastTick && nLastTick >= 39 )
nLastTick = -1;
if( nThisTick <= nLastTick )
return TRUE;
while( nThisTick > nLastTick )
{
nLastTick++;
if( nLastTick % 4 == 0 )
fprintf( stdout, "%d", (nLastTick / 4) * 10 );
else
fprintf( stdout, "." );
}
if( nThisTick == 40 )
fprintf( stdout, " - done.\n" );
else
fflush( stdout );
return TRUE;
}
可以看出,上面的函数中,只用了第一个参数,后面两个参数没用,就是将第一个参数使用printf函数输出在屏幕上。基于上面的分析,可以自己实现一个自己的GDAL进度条类,首先看类声明:
/**
* @brief 进度条基类
*
* 提供进度条基类接口,来反映当前算法的进度值
*/
class IMGALG_API CProcessBase
{
public:
/**
* @brief 构造函数
*/
CProcessBase()
{
m_dPosition = 0.0;
m_iStepCount = 100;
m_iCurStep = 0;
m_bIsContinue = true;
}
/**
* @brief 析构函数
*/
virtual ~CProcessBase() {}
/**
* @brief 设置进度信息
* @param pszMsg 进度信息
*/
virtual void SetMessage(const char* pszMsg) = 0;
/**
* @brief 设置进度值
* @param dPosition 进度值
* @return 返回是否取消的状态,true为不取消,false为取消
*/
virtual bool SetPosition(double dPosition) = 0;
/**
* @brief 进度条前进一步,返回true表示继续,false表示取消
* @return 返回是否取消的状态,true为不取消,false为取消
*/
virtual bool StepIt() = 0;
/**
* @brief 设置进度个数
* @param iStepCount 进度个数
*/
virtual void SetStepCount(int iStepCount)
{
ReSetProcess();
m_iStepCount = iStepCount;
}
/**
* @brief 获取进度信息
* @return 返回当前进度信息
*/
string GetMessage()
{
return m_strMessage;
}
/**
* @brief 获取进度值
* @return 返回当前进度值
*/
double GetPosition()
{
return m_dPosition;
}
/**
* @brief 重置进度条
*/
void ReSetProcess()
{
m_dPosition = 0.0;
m_iStepCount = 100;
m_iCurStep = 0;
m_bIsContinue = true;
}
/*! 进度信息 */
string m_strMessage;
/*! 进度值 */
double m_dPosition;
/*! 进度个数 */
int m_iStepCount;
/*! 进度当前个数 */
int m_iCurStep;
/*! 是否取消,值为false时表示计算取消 */
bool m_bIsContinue;
};
该类是一个进度条基类,将一些通用的函数进行实现,比如获取进度值,获取进度信息等之类的,然后将一些需要根据使用方式不同的设置为虚函数,比如设置进度值等。下面给出一个控制台进度条类,用于显示基于控制台程序的进度值显示,参考GDALTermProgress函数写的:
/**
* @brief 控制台进度条类
*
* 提供控制台程序的进度条类接口,来反映当前算法的进度值
*/
class CConsoleProcess : public CProcessBase
{
public:
/**
* @brief 构造函数
*/
CConsoleProcess()
{
m_dPosition = 0.0;
m_iStepCount = 100;
m_iCurStep = 0;
};
/**
* @brief 析构函数
*/
~CConsoleProcess()
{
//remove(m_pszFile);
};
/**
* @brief 设置进度信息
* @param pszMsg 进度信息
*/
void SetMessage(const char* pszMsg)
{
m_strMessage = pszMsg;
printf("%s\n", pszMsg);
}
/**
* @brief 设置进度值
* @param dPosition 进度值
* @return 返回是否取消的状态,true为不取消,false为取消
*/
bool SetPosition(double dPosition)
{
m_dPosition = dPosition;
TermProgress(m_dPosition);
m_bIsContinue = true;
return true;
}
/**
* @brief 进度条前进一步
* @return 返回是否取消的状态,true为不取消,false为取消
*/
bool StepIt()
{
m_iCurStep ++;
m_dPosition = m_iCurStep*1.0 / m_iStepCount;
TermProgress(m_dPosition);
m_bIsContinue = true;
return true;
}
private:
void TermProgress(double dfComplete)
{
static int nLastTick = -1;
int nThisTick = (int) (dfComplete * 40.0);
nThisTick = MIN(40,MAX(0,nThisTick));
// Have we started a new progress run?
if( nThisTick < nLastTick && nLastTick >= 39 )
nLastTick = -1;
if( nThisTick <= nLastTick )
return ;
while( nThisTick > nLastTick )
{
nLastTick++;
if( nLastTick % 4 == 0 )
fprintf( stdout, "%d", (nLastTick / 4) * 10 );
else
fprintf( stdout, "." );
}
if( nThisTick == 40 )
fprintf( stdout, " - done.\n" );
else
fflush( stdout );
}
};
至此,一个控制台的进度条就完成了,使用方式如下(注意:之前的博客中出现的LT_ConsoleProgress就是上面这个类):
void main()
{
CConsoleProcess *pProgress = new CConsoleProcess();
//参考建立Erdas金字塔的那篇博客:http://blog.csdn.net/liminlu0314/article/details/6127755
int f = CreatePyramids("C://Work//Data//ttttt.img", pProgress);
if (f == RE_SUCCESS)
printf("计算成功/n");
else
printf("计算失败/n");
delete pProgress;
}
效果图如下:
下面开始编写一个基于QT界面的进度条类,首先是类定义,基于QT界面的,对于MFC界面的可以参考这个写一个:
/**
* @brief 进度条对话框类
*
* 提供GUI程序的进度条类接口,来反映当前算法的进度值
*/
class CProcessDlg :
public QProgressDialog,
public CProcessBase
{
Q_OBJECT
public:
/**
* @brief 构造函数
*/
CProcessDlg(QWidget *parent = 0);
/**
* @brief 析构函数
*/
~CProcessDlg();
/**
* @brief 设置进度信息
* @param pszMsg 进度信息
*/
void SetMessage(const char* pszMsg);
/**
* @brief 设置进度值
* @param dPosition 进度值
*/
bool SetPosition(double dPosition);
/**
* @brief 进度条前进一步
*/
bool StepIt();
public slots:
void updateProgress(int);
};
接下来是实现部分代码,具体和上面的控制台类一样,只不过是设置进度值都是设置在界面上的相应的控件中:
/**
* @brief 构造函数
*/
CProcessDlg::CProcessDlg(QWidget *parent)
:QProgressDialog(parent)
{
m_dPosition = 0.0;
m_iStepCount = 100;
m_iCurStep = 0;
setModal(true);
setLabelText(tr("处理中..."));
setAutoClose(false);
setAutoReset(false);
setCancelButtonText(tr("取消"));
setWindowTitle(tr("进度"));
//禁用关闭按钮
setWindowFlags(Qt::WindowTitleHint | Qt::FramelessWindowHint | Qt::WindowStaysOnTopHint);
};
/**
* @brief 析构函数
*/
CProcessDlg::~CProcessDlg()
{
};
/**
* @brief 设置进度信息
* @param pszMsg 进度信息
*/
void CProcessDlg::SetMessage(const char* pszMsg)
{
if (pszMsg != NULL)
{
m_strMessage = pszMsg;
setLabelText(QString(pszMsg));
}
}
/**
* @brief 设置进度值
* @param dPosition 进度值
*/
bool CProcessDlg::SetPosition(double dPosition)
{
m_dPosition = dPosition;
setValue( std::min( 100u, ( uint )( m_dPosition*100.0 ) ) );
QCoreApplication::instance()->processEvents();
if(this->wasCanceled())
return false;
return true;
}
/**
* @brief 进度条前进一步,返回false表示终止操作
*/
bool CProcessDlg::StepIt()
{
m_iCurStep ++;
m_dPosition = m_iCurStep*1.0 / m_iStepCount;
setValue( std::min( 100u, ( uint )( m_dPosition*100.0 ) ) );
QCoreApplication::instance()->processEvents();
if(this->wasCanceled())
return false;
return true;
}
void CProcessDlg::updateProgress(int step)
{
this->setValue(step);
QCoreApplication::instance()->processEvents();
}
至此,QT界面的进度条类已经完成,调用方式和上面控制台的基本类似,示例代码:
//省略部分代码
CProcessDlg *pPro = new CProcessDlg();
pPro->setWindowTitle(tr("正在执行栅格矢量化"));
pPro->show();
int iRev = ImagePolygonize(m_strInputName.c_str(), m_strOutputName.c_str(), pszFormat, pPro);
delete pPro;
这样,我们基本上就实现了两个进度条,一个控制台,一个QT界面的。执行的效果图如下:
那么在自己的算法中如何使用这个进度条类呢,下面给出一个简单的算法,图像反色算法:
/**
* @brief 图像反色
* @param pszSrcFile 输入文件路径
* @param pszDstFile 输出文件路径
* @param pszFormat 输出文件格式,详细参考GDAL支持数据类型
* @param pProcess 进度条指针
* @return 返回值,表示计算过程中出现的各种错误信息
*/
int ImageAnticolor(const char* pszSrcFile, const char* pszDstFile, const char* pszFormat, CProcessBase* pProcess = NULL)
{
if(pProcess != NULL)
{
pProcess->ReSetProcess();
pProcess->SetMessage("执行图像反色...");
}
if(pszSrcFile == NULL || pszDstFile == NULL)
return RE_PARAMERROR;
GDALAllRegister();
GDALDataset *poSrcDS = (GDALDataset *) GDALOpen( pszSrcFile, GA_ReadOnly );
if( poSrcDS == NULL )
{
if(pProcess != NULL)
pProcess->SetMessage("输入文件不能打开,请检查文件是否存在!");
return RE_FILENOTEXIST;
}
GDALDriver *poDriver = GetGDALDriverManager()->GetDriverByName(pszFormat);
if( poDriver == NULL )
{
if(pProcess != NULL)
pProcess->SetMessage("不能创建制定类型的文件,请检查该文件类型GDAL是否支持创建!");
GDALClose( (GDALDatasetH) poSrcDS );
return RE_FILENOTSUPPORT;
}
//获取图像宽高和波段数
int iXSize = poSrcDS->GetRasterXSize();
int iYSize = poSrcDS->GetRasterYSize();
int iBandCount = poSrcDS->GetRasterCount();
//假设输入图像也是8U的数据
GDALDataset *poDstDS = poDriver->Create(pszDstFile, iXSize, iYSize, iBandCount, GDT_Byte, NULL);
double dGeoTrans[6] = {0};
//设置仿射变换参数
poSrcDS->GetGeoTransform(dGeoTrans);
poDstDS->SetGeoTransform(dGeoTrans);
//设置图像投影信息
poDstDS->SetProjection(poSrcDS->GetProjectionRef());
DT_8U *pSrcData = new DT_8U[iXSize];
DT_8U *pDstData = new DT_8U[iXSize];
if(m_pProcess != NULL)
{
m_pProcess->SetMessage("计算图像反色...");
m_pProcess->SetStepCount(iYSize*iBandCount); //设置进度条的总步长
}
//循环波段
for(int iBand=1; iBand<=iBandCount; iBand++)
{
GDALRasterBand *pSrcBand = poSrcDS->GetRasterBand(iBand);
GDALRasterBand *pDstBand = poDstDS->GetRasterBand(iBand);
for(int i=0; i<iYSize; i++) //循环图像高
{
pSrcBand->RasterIO(GF_Read, 0, i, iXSize, 1, pSrcData, iXSize, 1, GDT_Byte, 0, 0);
for(int j=0; j<iXSize; j++) //循环图像宽
pDstData[j] = 255 - pSrcData[j];
pDstBand->RasterIO(GF_Write, 0, i, iXSize, 1, pDstData, iXSize, 1, GDT_Byte, 0, 0);
if(m_pProcess != NULL)
{
bool bIsCancel = m_pProcess->StepIt();
if(!bIsCancel)
{
RELEASE(pSrcData);
RELEASE(pDstData);
//关闭原始图像和结果图像
GDALClose( (GDALDatasetH) poDstDS );
GDALClose( (GDALDatasetH) poSrcDS );
remove(pszDstFile); //删除结果图像
if(pProcess != NULL)
pProcess->SetMessage("图像反色取消!");
return RE_CANCEL;
}
}
}
}
RELEASE(pSrcData);
RELEASE(pDstData);
//关闭原始图像和结果图像
GDALClose( (GDALDatasetH) poDstDS );
GDALClose( (GDALDatasetH) poSrcDS );
if(pProcess != NULL)
pProcess->SetMessage("图像反色完成!");
return RE_SUCCESS;
}
如果我要使用GDAL提供的算法,该怎么使用上面的进度条呢,很简单,具体是要实现一个GDALProgressFunc()函数相同的一个使用__stdcall导出的函数即可,函数定义如下:
/**
* @brief 导出符号定义
*/
#ifndef STD_API
#define STD_API __stdcall
#endif
/**
* \brief 调用GDAL进度条接口
*
* 该函数用于将GDAL算法中的进度信息导出到CProcessBase基类中,供给界面显示
*
* @param dfComplete 完成进度值,其取值为 0.0 到 1.0 之间
* @param pszMessage 进度信息
* @param pProgressArg CProcessBase的指针
*
* @return 返回TRUE表示继续计算,否则为取消
*/
int STD_API ALGTermProgress( double dfComplete, const char *pszMessage, void * pProgressArg );
下面为该函数的实现代码,请重点看第三个参数的使用方式:
/**
* \brief 调用GDAL进度条接口
*
* 该函数用于将GDAL算法中的进度信息导出到CProcessBase基类中,供给界面显示
*
* @param dfComplete 完成进度值,其取值为 0.0 到 1.0 之间
* @param pszMessage 进度信息
* @param pProgressArg CProcessBase的指针
*
* @return 返回TRUE表示继续计算,否则为取消
*/
int STD_API ALGTermProgress( double dfComplete, const char *pszMessage, void * pProgressArg )
{
if(pProgressArg != NULL)
{
CProcessBase * pProcess = (CProcessBase*) pProgressArg;
pProcess->m_bIsContinue = pProcess->SetPosition(dfComplete);
if(pProcess->m_bIsContinue)
return TRUE;
else
return FALSE;
}
else
return TRUE;
}
稍微对上面的函数体进行说明一下,关键是第三个参数,第三个参数其实就是我们上面定义的控制台类获取QT界面类的对象指针,只不过是专为一个void指针,在函数体中,我们需要将其转为基类CProcessBase的指针,然后调用该基类的设置进度值函数即可。对于该函数的使用,下面还是以栅格矢量化为例来进行说明,废话不多说,看代码:
//前面省略很多代码
GDALRasterBandH hSrcBand = (GDALRasterBandH)poSrcDS->GetRasterBand(1);
if(GDALPolygonize(hSrcBand, NULL, (OGRLayerH)poLayer, 0, NULL, ALGTermProgress, pProcess)!= CE_None)
{
GDALClose( (GDALDatasetH) poSrcDS );
if(pProcess != NULL)
{
if(!pProcess->m_bIsContinue)
{
pProcess->SetMessage("计算取消!");
return RE_USERCANCEL;
}
else
{
pProcess->SetMessage("计算失败!");
return RE_PARAMERROR;
}
}
return RE_PARAMERROR;
}
//后面省略部分代码
终于写完了。好长啊~~~
---------------------
作者:箜_Kong
来源:CSDN
原文:https://blog.csdn.net/liminlu0314/article/details/7276954?utm_source=copy
版权声明:本文为博主原创文章,转载请附上博文链接!