fileobj.hpp

Go to the documentation of this file.

// fileobj.hpp
// 
// Copyright (c) 2009 - Michele De Stefano (micdestefano@users.sourceforge.net)
// 
// Distributed under the MIT License (See accompanying file LICENSE)

#ifndef MDS_UTILS_PYTHON_FILEOBJ_HPP_INCLUDED
#define MDS_UTILS_PYTHON_FILEOBJ_HPP_INCLUDED

#include <mds_utils/python/common.hpp>
#include <mds_utils/python/obj.hpp>
#include <iostream>
#include <stdexcept>
#include <mds_utils/file_utils/cfile_stream.hpp>
#include <boost/iostreams/stream_buffer.hpp>
#include <boost/shared_ptr.hpp>

namespace mds_utils {

    namespace python {

class FileObj : public Obj {
    
    boost::shared_ptr<boost::iostreams::stream_buffer<mds_utils::file_utils::cFile_Device> >
        piosbuf;
    
    mds_utils::file_utils::cFile_Device     fdevice;
    
    boost::shared_ptr<std::iostream>        pios;
    
public:

    FileObj() {}
    
    FileObj(PyObject *po) : Obj(po) {
        if (!PyFile_Check(m_po)) throw std::runtime_error("Not a Python file object");
        
        fdevice.set_FILEp(PyFile_AsFile(m_po));
        piosbuf.reset(new boost::iostreams::stream_buffer<mds_utils::file_utils::cFile_Device>());
        piosbuf->open(fdevice);
        pios.reset(new std::iostream(piosbuf.get()));
        pios->exceptions(std::ios::badbit | std::ios::failbit);
    }
        
    operator std::iostream& () {
        return *pios;
    }
};


class iFileObj : public Obj {
    
    boost::shared_ptr<boost::iostreams::stream_buffer<mds_utils::file_utils::cFile_Source> >
        pisbuf;
    
    mds_utils::file_utils::cFile_Source     fsource;
    
    boost::shared_ptr<std::istream>         pis;
    
public:
    
    iFileObj() {}

    iFileObj(PyObject *po) : Obj(po) {
        if (!PyFile_Check(m_po)) throw std::runtime_error("Not a Python file object");
        
        fsource.set_FILEp(PyFile_AsFile(m_po));
        pisbuf.reset(new boost::iostreams::stream_buffer<mds_utils::file_utils::cFile_Source>());
        pisbuf->open(fsource);
        pis.reset(new std::istream(pisbuf.get()));
        pis->exceptions(std::ios::badbit | std::ios::failbit);
    }
    
    operator std::istream& () { return *pis; }
};


class oFileObj : public Obj {
    
    boost::shared_ptr<boost::iostreams::stream_buffer<mds_utils::file_utils::cFile_Sink> >
            posbuf;
    
    mds_utils::file_utils::cFile_Sink       fsink;
    
    boost::shared_ptr<std::ostream>         pos;
    
public:
    
    oFileObj() {}

    oFileObj(PyObject *po) : Obj(po) {
        if (!PyFile_Check(m_po)) throw std::runtime_error("Not a Python file object");
    
        fsink.set_FILEp(PyFile_AsFile(m_po));
        posbuf.reset(new boost::iostreams::stream_buffer<mds_utils::file_utils::cFile_Sink>());
        posbuf->open(fsink);
        pos.reset(new std::ostream(posbuf.get()));
        pos->exceptions(std::ios::badbit | std::ios::failbit);
    }
    
    operator std::ostream& () { return *pos; }
};
    }
}
#endif