C模板实例化取决于if子句

前端之家收集整理的这篇文章主要介绍了C模板实例化取决于if子句前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
目前我正在做:
if(dimension == 2)
{
    typedef itk::Image<short,2>      ImageType;
    typedef itk::Image<unsigned int,2>   IntegralImageType;
    m_pApp->train<2,ImageType,IntegralImageType>();
}
else
{
    typedef itk::Image<short,3>      ImageType;
    typedef itk::Image<unsigned int,3>   IntegralImageType;
    m_pApp->train<3,IntegralImageType>();
}

但我想这样做:

if (dimension == 2)
    DIMENSION = 2;
    else
    DIMENSION = 3;

    typedef itk::Image<short,DIMENSION>      ImageType;
    typedef itk::Image<unsigned int,DIMENSION>   IntegralImageType;
    m_pApp->train<DIMENSION,IntegralImageType>();

我无法做到这一点,因为c需要const变量用于模板实例化.
有没有这样的方法呢?

解决方法

您可以使用模板参数定义函数
template<unsigned N>
void train(){
    typedef itk::Image<short,N>      ImageType;
    typedef itk::Image<unsigned int,N>   IntegralImageType;
    m_pApp->train<N,IntegralImageType>();
}

然后:

if (dimension == 2)
    train<2>();
else
    train<3>();

请注意,此代码将实例化两个模板(将为它们生成代码),因为在编译时无法知道将使用哪个模板.

猜你在找的C&C++相关文章