2014-10-29 148 views
-3

这个问题只是为了改进目的。有一个功能如下:用不同参数调用函数C++

void Func_A(u8* arg) { 
    bool local_arg=false; 
    if(!arg) { 
     //allocate memory for arg 
     local_arg=true; 
    } 
    //process arg 

我从多个地方用NULL和非NULL参数调用此函数。 我只想问,是否有更好的方法来处理这种不local_arg或不。

+1

如果你能描述'* arg'是输入还是输出,以及'arg'为NULL时'Func_A'意味着什么,这将有所帮助。 – 2014-10-29 06:12:06

+1

您可以尝试函数的默认参数,但我不太确定这种情况对您的情况会有多大改变。 – Niall 2014-10-29 06:33:45

+2

'local_arg'在你的代码中没有做任何事情。你应该发布一个更现实的例子。 – juanchopanza 2014-10-29 06:37:37

回答

0

有时你可以使用...

u8* p = arg ? arg : new u8(...); 

...do things with *p... 

if (!arg) delete p; 

小票友,您可以编写与运行时配置的所有权的智能指针:

template <typename T> 
class Uncertain_Ownership_Ptr 
{ 
    public: 
    enum Ownership { Own_It, Dont_Own_It }; 
    Uncertain_Ownership_Ptr(T* p, Ownership own) : p_(p), own_(own) { } 
    Uncertain_Ownership_Ptr(const Uncertain_Ownership_Ptr&) = delete; 
    void operator=(const Uncertain_Ownership_Ptr&) = delete; 
    ~Uncertain_Ownership_Ptr() { if (own_ == Own_It) delete p_; } 
    T& operator*() { return *p_; } 
    const T& operator*() const { return *p_; } 
    T* operator->() { return p_; } 
    const T* operator->() const { return p_; } 
    private: 
    T* p_; 
    Ownership own_; 
}; 

...然后...

void Func_A(u8* arg) 
{ 
    Uncertain_Ownership_Ptr p(arg ? arg : new u8(...), 
     arg ? Uncertain_Ownership_Ptr::Dont_Own_It : Uncertain_Ownership_Ptr::Own_It); 

    // use *p ... 
}