用于在任何基础中流式传输整数的自定义流操纵器

时间:2011-06-25 15:37:22

标签: c++ iostream

我可以使用十六进制的std::ostream个对象输出整数,例如

std::cout << std::hex << 0xabc; //prints `abc`, not the base-10 representation

是否有适用于所有基地的操纵器?像

这样的东西
std::cout << std::base(4) << 20; //I want this to output 110

如果有,那我就没有其他问题了。 如果没有,那我可以写一个吗?是否需要我访问std::ostream的私人实施细节?

请注意,我知道我可以编写一个带数字的函数并将其转换为字符串,该字符串是任何基数中该数字的表示。或者我可以使用已经存在的。我问的是自定义流操纵器 - 它们可能吗?

提前致谢

5 个答案:

答案 0 :(得分:17)

您可以执行以下操作。我已经对代码进行了评论,以解释每个部分正在做什么,但基本上就是这个:

  • 创建一个“操纵器”结构,使用xallociword在流中存储一些数据。
  • 创建一个自定义num_put构面,用于查找操纵器并应用操作。

这是代码......

编辑:请注意,我不确定我是否正确处理了std::ios_base::internal标记 - 因为我实际上并不知道它的用途。

编辑2:我找到了std::ios_base::internal的用途,并更新了代码以便处理它。

编辑3:添加了对std::locacle::global的调用,以显示默认情况下如何使所有标准流类支持新的流操纵器,而不是必须imbue

#include <algorithm>
#include <cassert>
#include <climits>
#include <iomanip>
#include <iostream>
#include <locale>

namespace StreamManip {

// Define a base manipulator type, its what the built in stream manipulators
// do when they take parameters, only they return an opaque type.
struct BaseManip
{
    int mBase;

    BaseManip(int base) : mBase(base)
    {
        assert(base >= 2);
        assert(base <= 36);
    }

    static int getIWord()
    {
        // call xalloc once to get an index at which we can store data for this
        // manipulator.
        static int iw = std::ios_base::xalloc();
        return iw;
    }

    void apply(std::ostream& os) const
    {
        // store the base value in the manipulator.
        os.iword(getIWord()) = mBase;
    }
};

// We need this so we can apply our custom stream manipulator to the stream.
std::ostream& operator<<(std::ostream& os, const BaseManip& bm)
{
    bm.apply(os);
    return os;
}

// convience function, so we can do std::cout << base(16) << 100;
BaseManip base(int b)
{
    return BaseManip(b);
}

// A custom number output facet.  These are used by the std::locale code in
// streams.  The num_put facet handles the output of numberic values as characters
// in the stream.  Here we create one that knows about our custom manipulator.
struct BaseNumPut : std::num_put<char>
{
    // These absVal functions are needed as std::abs doesnt support 
    // unsigned types, but the templated doPutHelper works on signed and
    // unsigned types.
    unsigned long int absVal(unsigned long int a) const
    {
        return a;
    }

    unsigned long long int absVal(unsigned long long int a) const
    {
        return a;
    }

    template <class NumType>
    NumType absVal(NumType a) const
    {
        return std::abs(a);
    }

    template <class NumType>
    iter_type doPutHelper(iter_type out, std::ios_base& str, char_type fill, NumType val) const
    {
        // Read the value stored in our xalloc location.
        const int base = str.iword(BaseManip::getIWord());

        // we only want this manipulator to affect the next numeric value, so
        // reset its value.
        str.iword(BaseManip::getIWord()) = 0;

        // normal number output, use the built in putter.
        if (base == 0 || base == 10)
        {
            return std::num_put<char>::do_put(out, str, fill, val);
        }

        // We want to conver the base, so do it and output.
        // Base conversion code lifted from Nawaz's answer

        int digits[CHAR_BIT * sizeof(NumType)];
        int i = 0;
        NumType tempVal = absVal(val);

        while (tempVal != 0)
        {
            digits[i++] = tempVal % base;
            tempVal /= base;
        }

        // Get the format flags.
        const std::ios_base::fmtflags flags = str.flags();

        // Add the padding if needs by (i.e. they have used std::setw).
        // Only applies if we are right aligned, or none specified.
        if (flags & std::ios_base::right || 
            !(flags & std::ios_base::internal || flags & std::ios_base::left))
        {
            std::fill_n(out, str.width() - i, fill);
        }

        if (val < 0)
        {
            *out++ = '-';
        }

        // Handle the internal adjustment flag.
        if (flags & std::ios_base::internal)
        {
            std::fill_n(out, str.width() - i, fill);
        }

        char digitCharLc[] = "0123456789abcdefghijklmnopqrstuvwxyz";
        char digitCharUc[] = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";

        const char *digitChar = (str.flags() & std::ios_base::uppercase)
            ? digitCharUc
            : digitCharLc;

        while (i)
        {
            // out is an iterator that accepts characters
            *out++ = digitChar[digits[--i]];
        }

        // Add the padding if needs by (i.e. they have used std::setw).
        // Only applies if we are left aligned.
        if (str.flags() & std::ios_base::left)
        {
            std::fill_n(out, str.width() - i, fill);
        }

        // clear the width
        str.width(0);

        return out;
    }

    // Overrides for the virtual do_put member functions.

    iter_type do_put(iter_type out, std::ios_base& str, char_type fill, long val) const
    {
        return doPutHelper(out, str, fill, val);
    }

    iter_type do_put(iter_type out, std::ios_base& str, char_type fill, unsigned long val) const
    {
        return doPutHelper(out, str, fill, val);
    }
};

} // namespace StreamManip

int main()
{
    // Create a local the uses our custom num_put
    std::locale myLocale(std::locale(), new StreamManip::BaseNumPut());

    // Set our locacle to the global one used by default in all streams created 
    // from here on in.  Any streams created in this app will now support the
    // StreamManip::base modifier.
    std::locale::global(myLocale);

    // imbue std::cout, so it uses are custom local.
    std::cout.imbue(myLocale);
    std::cerr.imbue(myLocale);

    // Output some stuff.
    std::cout << std::setw(50) << StreamManip::base(2) << std::internal << -255 << std::endl;
    std::cout << StreamManip::base(4) << 255 << std::endl;
    std::cout << StreamManip::base(8) << 255 << std::endl;
    std::cout << StreamManip::base(10) << 255 << std::endl;
    std::cout << std::uppercase << StreamManip::base(16) << 255 << std::endl;

    return 0;
}

答案 1 :(得分:2)

自定义操纵器确实可行。请参阅示例this question。我不熟悉任何针对普遍基础的具体方法。

答案 2 :(得分:2)

你真的有两个不同的问题。我认为你问的那个是完全可以解决的。不幸的是,另一个则不那么重要了。

分配和使用流中的某些空间来保存某些流状态是一个预见到的问题。 Streams有几个成员(xallociwordpword),可以让您在流中的数组中分配一个点,并在那里读/写数据。因此,流操纵器本身是完全可能的。您基本上使用xalloc在流的数组中分配一个点以保存当前基数,以便插入运算符在转换数字时使用。

我没有看到解决方案的问题相当简单:标准库已经提供了operator<<来将int插入到流中,显然它不< / em>了解您的假设数据以保留转换基础。你不能重载它,因为它需要与现有签名完全相同的签名,所以你的重载是不明确的。

intshort等的重载是重载的成员函数。我如果你想要足够严重,你可以使用模板重载operator<<。如果我没记错的话,那就像图书馆提供的那样,甚至与非模板功能完全匹配也是首选。你仍然违反了规则,但是如果你把这样的模板放在命名空间std中,那么它至少有可能起作用。

答案 3 :(得分:2)

我尝试编写代码,并且它有一些限制。它本身不是流操纵者,因为根本不可能,正如其他人所指出的那样(特别是@Jerry)。

这是我的代码:

struct base
{
   mutable std::ostream *_out;
   int _value;

   base(int value=10) : _value(value) {}

   template<typename T>
   const base& operator << (const T & data) const
   {
        *_out << data;
        return *this;
   }
   const base& operator << (const int & data) const
   {
        switch(_value)
        {
            case 2:  
            case 4:  
            case 8:  return print(data);
            case 16: *_out << std::hex << data; break;
            default:  *_out << data; 
        }
        return *this;
   }
   const base & print(int data) const
   {
        int digits[CHAR_BIT * sizeof(int)], i = 0;
        while(data)
        {
             digits[i++] = data % _value;  
             data /= _value;
        }
        while(i) *_out << digits[--i] ;
        return *this;
   }
   friend const base& operator <<(std::ostream& out, const base& b)   
   {
       b._out = &out;
       return b;
   }
};

这是测试代码:

int main() {
   std::cout << base(2) << 255 <<", " << 54 << ", " << 20<< "\n";
   std::cout << base(4) << 255 <<", " << 54 << ", " << 20<< "\n";
   std::cout << base(8) << 255 <<", " << 54 << ", " << 20<< "\n";
   std::cout << base(16) << 255 <<", " << 54 << ", " << 20<< "\n";
}

输出:

11111111, 110110, 10100
3333, 312, 110
377, 66, 24
ff, 36, 14

在线演示:http://www.ideone.com/BWhW5

限制:

  • 基座不能更改两次。所以这将是一个错误:

    std::cout << base(4) << 879 << base(8) << 9878 ; //error
    
  • 使用base后无法使用其他操纵器:

    std::cout << base(4) << 879 << std::hex << 9878 ; //error
    std::cout << std::hex << 879 << base(8) << 9878 ; //ok
    
  • 使用std::endl后,
  • base无法使用:

    std::cout << base(4) << 879 << std::endl ; //error
    //that is why I used "\n" in the test code.
    

答案 4 :(得分:1)

我不认为语法对于任意流是可能的(使用操纵器,@ gigantt链接了一个显示一些替代非操纵器解决方案的答案)。标准操纵器仅设置在流内实现的选项。

OTOH,你当然可以使这种语法有效:

std::cout << base(4, 20);

其中base是提供流插入运算符的对象(无需返回临时string)。