Rose/FormatRestorer.h

#ifndef ROSE_FormatRestorer_H
#define ROSE_FormatRestorer_H

#include <iostream>

class FormatRestorer {
protected:
    std::ostream &stream;
    std::ios_base::fmtflags fmt;

public:
    FormatRestorer(std::ostream &o): stream(o) {
        save(o);
    }

    ~FormatRestorer() {
        restore();
    }

    void save(std::ostream &o) {
        fmt = o.flags();
    }

    void restore() {
        stream.flags(fmt);
    }
};
    
#endif