TnbLzFile.h

[詳解]

#pragma once
#include "TnbReader.h"
 
 
 
#include <lzexpand.h>
#pragma comment(lib,"lz32.lib")
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CLzFileReader : public IReader
{
    INT     m_iHandle;      
    CStr    m_strName;      
 
    void m_Close(void)
    {
        m_strName.Empty();
        if ( m_iHandle >= 0 )
        {
            ::LZClose(m_iHandle);
            m_iHandle = -1;
        }
    }
 
public:
 
    CLzFileReader(void) : m_iHandle(-1)
    {
    }
 
    virtual ~CLzFileReader(void)
    {
        m_Close();
    }
 
    bool Open(LPCTSTR lpszLzFile, bool boIsShare = true)
    {
        CStr s = lpszLzFile;
        Close();
        CWorkMemT<OFSTRUCT> work;
        INT r = ::LZOpenFile(
                    s.GetBuffer(), work, static_cast<WORD>(OF_READ | (boIsShare ? OF_SHARE_DENY_NONE : OF_SHARE_EXCLUSIVE)));
        if ( r >= 0 )
        {
            m_strName = lpszLzFile;
            m_iHandle = r;
        }
        return r >= 0;
    }
 
    bool Copy(LPCTSTR lpszDstFile)
    {
        if ( ! CanRead() ){ return false; }
        CWorkMemT<OFSTRUCT> work;
        CStr s = lpszDstFile;
        INT r = ::LZOpenFile(s.GetBuffer(), work, OF_CREATE);
        if ( r >= 0 )
        {
            if ( ::LZCopy(m_iHandle, r) >= 0 )
            {
                return true;
            }
        }       
        return false;
    }
 
    virtual LPCTSTR GetOpenName(void) const
    {
        return m_strName;
    }
 
    virtual void Close(void)
    {
        m_Close();
    }
 
    virtual bool CanRead(void) const
    {
        return m_iHandle >= 0;
    }
 
    virtual LONGLONG GetSize(void) const
    {
        LONGLONG c = Seek(0, CURRENT);
        LONGLONG r = Seek(0, END);
        Seek(c, TOP);
        return r;
    }
 
    virtual LONGLONG Seek(LONGLONG llOffset, ESeekMode eSeek = TOP) const
    {
         return ::LZSeek(m_iHandle, ToInt(llOffset), eSeek);
    }
 
    virtual size_t Read(size_t size, LPVOID _P) const
    {
        INT r = ::LZRead(m_iHandle, static_cast<char *>(_P), ToInt(size));
        if ( r < 0 )
        {
            throw CReadFailureException(ERROR_ACCESS_DENIED);
        }
        return r;
    }
 
    static CStr GetExpandedName(LPCTSTR lpszLzFile)
    {
        CStr strSrc = lpszLzFile;
        CStr strDst;
        if ( ::GetExpandedName(strSrc.GetBuffer(), strDst.GetBuffer(MAX_PATH)) >= 0 )
        {
            strDst.ReleaseBuffer();
            return strDst;
        }
        return CStr();
    }
};
 
 
 
}; // TNB