定义类模板的友元函数模板

时间:2016-06-17 12:51:41

标签: c++ templates friend-function function-templates class-template

我想定义一个类模板的函数模板。代码看起来像这样。

template<int M>
struct test{
private:
    int value;

    template<int N = 2 * M>
    friend auto foo(test const t){
        test<N> r;
        r.value = t.value;
        return r;
    }
};

int main(){
    test<1> t;
    foo(t);// expected to return test<2>
    foo<1>(t);// expected to return test<1>
    foo<3>(t);// expected to return test<3>
}

但它不会编译。与以前的问题相比,下面列出了不同之处。

  1. 函数模板的结果涉及类模板的另一个实例。似乎必须在外部定义函数模板。但我不确定。
  2. 函数模板使用默认模板参数。因此,如果函数模板在类外部定义,则需要辅助函数模板。
  3. 使用g++ -std=c++1z编译错误:

    a.cpp: In instantiation of 'auto foo(test<M>) [with int N = 2; int M = 1]':
    a.cpp:16:10:   required from here
    a.cpp:4:9: error: 'int test<2>::value' is private
         int value;
             ^
    a.cpp:9:17: error: within this context
             r.value = t.value;
                     ^
    a.cpp: In instantiation of 'auto foo(test<M>) [with int N = 3; int M = 1]':
    a.cpp:18:13:   required from here
    a.cpp:4:9: error: 'int test<3>::value' is private
         int value;
             ^
    a.cpp:9:17: error: within this context
             r.value = t.value;
                     ^
    

    可能的解决方法,但也不正确。

    template<int M>
    struct test{
    private:
        int value;
    
        template<int NA, int NR>
        friend test<NR> foo_impl(test<NA> const&);
    };
    
    template<int NA, int NR>
    test<NR> foo_impl(test<NA> const& t){
        test<NR> r;
        r.value = t.value;
        return r;
    }
    
    template<int NR, int NA>
    auto foo(test<NA> const& t){
        return foo_impl<NA, NR>;
    }
    
    template<int NA>
    auto foo(test<NA> const& t){
        return foo_impl<NA, NA * 2>(t);
    }
    
    int main(){
        test<1> t;
        foo(t);
        foo<3>(t);
        foo<1>(t);
    }
    

    错误:

    t.cpp: In function 'int main()':
    t.cpp:31:13: error: call of overloaded 'foo(test<1>&)' is ambiguous
         foo<1>(t);
                 ^
    t.cpp:18:6: note: candidate: auto foo(const test<NR>&) [with int NR = 1; int NA = 1]
     auto foo(test<NA> const& t){
          ^
    t.cpp:23:6: note: candidate: auto foo(const test<NA>&) [with int NA = 1]
     auto foo(test<NA> const& t){
          ^
    

1 个答案:

答案 0 :(得分:6)

编辑后,一种可能的方法(可能不比您的解决方法更好 - 但适用于所有调用)将使用参数N的默认值:

template<int M>
struct test{
private:
    int value;

    template<int K, int U, int P>
    friend test<P> foo(test<U> const t);

};

template <int N = 0, int M, int P = ((N == 0) ? M * 2 : N)>
test<P> foo (test<M> const t) {
  test<P> r;
  r.value = t.value;
  return r;
}

int main(){
    test<1> t;
    test<2> p1 = foo(t);
    test<3> p2 = foo<3>(t);
    test<1> p3 = foo<1>(t);
}

这可能不比你的版本漂亮......

这里的问题是,您希望foo<N>成为test<M>的朋友,而您希望它成为任何test<...>的朋友。您应该执行以下操作:

template<int M>
struct test{
private:
    int value;

    template<int U, int K>
    friend test<K> foo(test<U> const t);
};

template <int M, int N = 2 * M>
test<N> foo (test<M> const t) {
    test<N> r;
    r.value = t.value;
    return r;
}

int main(){
    test<1> t;
    foo(t);
}

在这里你说:

  

任何函数test<K> foo<U, K> (test<U> const)都是test<...>的朋友。