#ifndef VTR_OSTREAM_GUARD_H
#define VTR_OSTREAM_GUARD_H

namespace vtr {

//A RAII guard class to ensure restoration of output stream format
class OsFormatGuard {
    public:
        explicit OsFormatGuard(std::ostream& os)
            : os_(os)
            , flags_(os_.flags()) //Save formatting flag state
            , width_(os_.width())
            , precision_(os.precision())
            , fill_(os.fill())
        {}

        ~OsFormatGuard() {
            os_.flags(flags_); //Restore
            os_.width(width_);
            os_.precision(precision_);
            os_.fill(fill_);
        }

        OsFormatGuard(const OsFormatGuard&) = delete;
        OsFormatGuard& operator=(const OsFormatGuard&) = delete;
        OsFormatGuard(const OsFormatGuard&&) = delete;
        OsFormatGuard& operator=(const OsFormatGuard&&) = delete;
    private:
        std::ostream& os_;
        std::ios::fmtflags flags_;
        std::streamsize width_;
        std::streamsize precision_;
        char fill_;

};

} //namespace

#endif