TnbArchiveFile.h

[詳解]

#pragma once
#include "TnbArchiver.h"
#include "TnbTemporaryDirectory.h"
#include "TnbFile.h"
#include "TnbArchiveFileFinder.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
class CArchiveFileReader : public IReader
{
    DEFSUPER(IReader);
public:
 
    CArchiveFileReader(void)
    {
    }
 
    virtual ~CArchiveFileReader(void)
    {
        m_Close();
    }
 
    bool Open(CArchiverDllHandler& arcdll, LPCTSTR lpszArcFile, LPCTSTR lpszName, LPCTSTR lpszOpt = NULL)
    {
        Close();
        if ( ! arcdll.IsReady() || arcdll.IsRunning() )
        {
            return false;
        }
        CArchiver arc(&arcdll);
        arc.SetTargetArchiveFile(lpszArcFile);
        CStr strPath = m_tempDir.Create();
        arc.SetWorkDirectory(strPath);
        bool r = (arc.ExtractWithDirectory(lpszName, lpszOpt) == 0);
        if ( r )
        {
            CStr ss = strPath + _T("\\") + lpszName;
            ss.Replace(_T("\\\\"), _T("\\"));
            r = m_reader.Open(ss);
        }
        if ( ! r )
        {
            m_tempDir.Free();
        }
        return r;
    }
 
    bool Open(const CArchiveFileFinder& ff, LPCTSTR lpszOpt = NULL)
    {
        const TIndividualInfo* P = ff.RefPtr();
        if ( ff.IsFinding() && P != NULL )
        {
            #ifdef _UNICODE
                return Open(*ff.GetDllHandler(), CStr(ff.GetTargetArchiveFileName()), CStr(P->szFileName), lpszOpt);
            #else
                return Open(*ff.GetDllHandler(), ff.GetTargetArchiveFileName(), P->szFileName, lpszOpt);
            #endif
        }
        return false;
    }
 
    virtual LPCTSTR GetOpenName(void) const
    {
        return m_reader.GetOpenName();
    }
 
    virtual void Close(void)
    {
        m_Close();
    }
 
    virtual bool CanRead(void) const
    {
        return m_reader.CanRead();
    }
 
    virtual LONGLONG GetSize(void) const
    {
        return m_reader.GetSize();
    }
 
    virtual LONGLONG Seek(LONGLONG llOffset, ESeekMode eSeek = TOP) const
    {
        return m_reader.Seek(llOffset, eSeek);
    }
 
    virtual size_t Read(size_t size, LPVOID _P) const
    {
        return m_reader.Read(size, _P);
    }
 
private:
    void m_Close(void)
    {
        m_reader.Close();
        m_tempDir.Free();
    }
    CFileReader                 m_reader;   
    CTemporaryDirectoryT<'CRA'> m_tempDir;  
 
    friend class CArchiveFileReaderTest;
};
 
 
 
}; // TNB