////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 1996-2024 The Octave Project Developers
//
// See the file COPYRIGHT.md in the top-level directory of this
// distribution or <https://octave.org/copyright/>.
//
// This file is part of Octave.
//
// Octave is free software: you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Octave is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Octave; see the file COPYING.  If not, see
// <https://www.gnu.org/licenses/>.
//
////////////////////////////////////////////////////////////////////////

#if ! defined (octave_oct_stdstrm_h)
#define octave_oct_stdstrm_h 1

#include "octave-config.h"

#include <iomanip>

#include "oct-stream.h"
#include "c-file-ptr-stream.h"

OCTAVE_BEGIN_NAMESPACE(octave)

template <typename BUF_T, typename STREAM_T, typename FILE_T>
class
tstdiostream : public base_stream
{
public:

  tstdiostream (const std::string& n, FILE_T f = 0, int fid = 0,
                std::ios::openmode m = std::ios::in | std::ios::out,
                mach_info::float_format ff = mach_info::native_float_format (),
                const std::string& encoding = "utf-8",
                typename BUF_T::close_fcn cf = BUF_T::file_close)
    : base_stream (m, ff, encoding), m_name (n), m_mode (m),
      m_stream (f ? new STREAM_T (f, cf) : nullptr), m_fnum (fid)
  { }

  OCTAVE_DISABLE_CONSTRUCT_COPY_MOVE (tstdiostream)

  // Position a stream at OFFSET relative to ORIGIN.

  int seek (off_t offset, int origin)
  {
    return m_stream ? m_stream->seek (offset, origin) : -1;
  }

  // Return current stream position.

  off_t tell () { return m_stream ? m_stream->tell () : -1; }

  // Return nonzero if EOF has been reached on this stream.

  bool eof () const { return m_stream ? m_stream->eof () : true; }

  // The name of the file.

  std::string name () const { return m_name; }

  std::istream * input_stream ()
  {
    return (m_mode & std::ios::in) ? m_stream : nullptr;
  }

  std::ostream * output_stream ()
  {
    return (m_mode & std::ios::out) ? m_stream : nullptr;
  }

  // FIXME: should not have to cast away const here.
  BUF_T * rdbuf () const
  {
    return m_stream ? (const_cast<STREAM_T *> (m_stream))->rdbuf () : nullptr;
  }

  int file_number () const { return m_fnum; }

  bool bad () const { return m_stream ? m_stream->bad () : true; }

  void clear ()
  {
    if (m_stream)
      m_stream->clear ();
  }

  void do_close ()
  {
    if (m_stream)
      m_stream->stream_close ();
  }

protected:

  ~tstdiostream () { delete m_stream; }

  //--------

  std::string m_name;

  std::ios::openmode m_mode;

  STREAM_T *m_stream;

  // The file number associated with this file.
  int m_fnum;
};

class
stdiostream
  : public tstdiostream<c_file_ptr_buf, io_c_file_ptr_stream, FILE *>
{
public:

  stdiostream (const std::string& n, FILE *f = nullptr,
               std::ios::openmode m = std::ios::in | std::ios::out,
               mach_info::float_format ff = mach_info::native_float_format (),
               const std::string& encoding = "utf-8",
               c_file_ptr_buf::close_fcn cf = c_file_ptr_buf::file_close)
    : tstdiostream<c_file_ptr_buf, io_c_file_ptr_stream, FILE *>
      (n, f, f ? fileno (f) : -1, m, ff, encoding, cf) { }

  static stream
  create (const std::string& n, FILE *f = nullptr,
          std::ios::openmode m = std::ios::in | std::ios::out,
          mach_info::float_format ff = mach_info::native_float_format (),
          const std::string& encoding = "utf-8",
          c_file_ptr_buf::close_fcn cf = c_file_ptr_buf::file_close)
  {
    return stream (new stdiostream (n, f, m, ff, encoding, cf));
  }

  OCTAVE_DISABLE_CONSTRUCT_COPY_MOVE (stdiostream)

protected:

  ~stdiostream () = default;
};

#if defined (HAVE_ZLIB)

class
zstdiostream
  : public tstdiostream<c_zfile_ptr_buf, io_c_zfile_ptr_stream, gzFile>
{
public:

  zstdiostream (const std::string& n, gzFile f = nullptr, int fid = 0,
                std::ios::openmode m = std::ios::in | std::ios::out,
                mach_info::float_format ff = mach_info::native_float_format (),
                const std::string& encoding = "utf-8",
                c_zfile_ptr_buf::close_fcn cf = c_zfile_ptr_buf::file_close)
    : tstdiostream<c_zfile_ptr_buf, io_c_zfile_ptr_stream, gzFile>
      (n, f, fid, m, ff, encoding, cf) { }

  static stream
  create (const std::string& n, gzFile f = nullptr, int fid = 0,
          std::ios::openmode m = std::ios::in | std::ios::out,
          mach_info::float_format ff = mach_info::native_float_format (),
          const std::string& encoding = "utf-8",
          c_zfile_ptr_buf::close_fcn cf = c_zfile_ptr_buf::file_close)
  {
    return stream (new zstdiostream (n, f, fid, m, ff, encoding, cf));
  }

  OCTAVE_DISABLE_CONSTRUCT_COPY_MOVE (zstdiostream)

protected:

  ~zstdiostream () = default;
};

#endif

OCTAVE_END_NAMESPACE(octave)

#endif
