CMCore/inc/hgl/io/MemoryOutputStream.h

#ifndef HGL_IO_MEMORY_OUTPUT_STREAM_INCLUDE
#define HGL_IO_MEMORY_OUTPUT_STREAM_INCLUDE

#include<hgl/io/OutputStream.h>
namespace hgl
{
    namespace io
    {
        /**
         * 内存数据输出流，将数据输出到一个内存块中。
         */
        class MemoryOutputStream:public OutputStream                                                ///内存数据输出流
        {
        protected:

            uint8 *buf;

            int64 max_size;
            int64 buf_size;
            int64 cur_pos;

            bool one;

        public:

            MemoryOutputStream()
            {
                buf=0;
                max_size=0;
                buf_size=0;
                cur_pos=0;
                one=false;
            }

            virtual ~MemoryOutputStream()
            {
                Close();
            }

            void *GetData()const{return buf;}

            /**
             * 创建一个当前流数据的内存拷贝，此函数需用户自行delete[]
             * @param len 存放数据长度的指针
             * @return 创建好的内存拷贝
             */
            void *CreateCopyData(int64 *len)const
            {
                if(buf_size<=0)
                    return(nullptr);

                uint8 *data=new uint8[buf_size+1];
                memcpy(data,buf,buf_size);
                data[buf_size]=0;

                if(len)
                    *len=buf_size;

                return data;
            }

            /**
             * 关联一个数据区到当前输出流
             * @param ptr 数据指针
             * @param size 数据长度字节数
             * @param one_instance 是否仅此一份实例(如果是，将由MemoryOutputStream类负责释放)
             * @return 是否成功
             */
            bool Link(void *ptr,int64 size,bool one_instance=false)
            {
                if(!ptr||!size)
                    return(false);

                buf=(uint8 *)ptr;
                buf_size=size;
                max_size=size;
                cur_pos=0;

                one=one_instance;

                return(true);
            }

            /**
             * 更新关联的数据区长度，不复位cur_pos
             * @param ptr 数据指针
             * @param size 数据长度字节数
             * @return 是否成功
             */
            bool Update(void *ptr,int64 size)
            {
                if(!ptr||!size)
                    return(false);

                buf=(uint8 *)ptr;
                buf_size=size;
                max_size=0;

                return(true);
            }

            void Unlink()
            {
                buf=0;
                buf_size=0;
                max_size=0;
            }

            bool Create(int64 size)
            {
                Close();

                if(size<=0)
                    return(false);

                buf=(uint8 *)hgl_malloc(size);

                if(!buf)
                    return(false);

                one=true;
                cur_pos=0;
                buf_size=size;
                max_size=size;

                return(true);
            }

            void Close() override
            {
                if(buf)
                {
                    if(one)
                        hgl_free(buf);

                    buf=0;
                }

                buf_size=0;
                max_size=0;
            }

            void Clear()
            {
                cur_pos=0;
                buf_size=0;
            }

            int64 Write(const void *ptr,int64 size) override
            {
                if(!ptr||size<0)
                    return(-1);

                if(!buf)
                {
                    if(!Create(size))
                        return(-1);
                }

                if(cur_pos+size>buf_size)
                {
                    if(one)
                    {
                        buf_size=cur_pos+size;

                        if(buf_size>max_size)
                        {
                            max_size=power_to_2(buf_size);

                            buf=(uint8 *)hgl_realloc(buf,max_size);
                        }
                    }
                    else
                        size=buf_size-cur_pos;
                }

                if(size<=0)
                    return(0);

                memcpy(buf+cur_pos,ptr,size);

                cur_pos+=size;

                return size;
            }

            bool    CanRestart()const override{return true;}
            bool    CanSeek()const override{return true;}
            bool    CanSize()const override{return true;}

            bool    Restart() override
            {
                cur_pos=0;
                return(true);
            }

            int64   Seek(int64 off,SeekOrigin so=SeekOrigin::Begin) override
            {
                if(!CanSeek())return(-1);

                if(so==SeekOrigin::Current)
                {
                    off+=cur_pos;
                }
                else
                if(so==SeekOrigin::End)
                {
                    off+=buf_size;
                }

                if(off<0||off>=buf_size)
                    return(-1);

                cur_pos=off;
                return cur_pos;
            }

            int64   Tell()const override
            {
                return cur_pos;
            }

            int64   GetSize()const override
            {
                return buf_size;
            }

            int64   Available()const override
            {
                return buf_size-cur_pos;
            }
        };//class MemoryOutputStream
    }//namespace io
}//namespace hgl
#endif//HGL_IO_MEMORY_OUTPUT_STREAM_INCLUDE