tahoma2d/toonz/sources/image/zcc/tfile.cpp

#include "tfile.h"

/*
namespace {
	bool IsWindowsNT() {
		return (LONG)GetVersion()>=0;
	}
};
*/
using namespace TFileConsts;

TFile::TFile(const TFilePath &fname, uint32 flags)
	: m_hFile(NULL), m_ec(0)
{
	open_internal(fname, /*NULL,*/ flags);
}

//------------------------------------------------------------------------------

TFile::~TFile()
{
	close();
}

//------------------------------------------------------------------------------

bool TFile::open(const TFilePath &fname, uint32 flags)
{
	return open_internal(fname, flags);
}

//------------------------------------------------------------------------------

bool TFile::open_internal(const TFilePath &fname, /*const wchar_t *pwszFilename,*/ uint32 flags)
{
	close();

	// At least one of the read/write flags must be set.
	assert(flags & (kRead | kWrite));

	DWORD dwDesiredAccess = 0;

	if (flags & kRead)
		dwDesiredAccess = GENERIC_READ;
	if (flags & kWrite)
		dwDesiredAccess |= GENERIC_WRITE;

	// Win32 docs are screwed here -- FILE_SHARE_xxx is the inverse of a deny flag.

	DWORD dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
	if (flags & kDenyRead)
		dwShareMode = FILE_SHARE_WRITE;
	if (flags & kDenyWrite)
		dwShareMode &= ~FILE_SHARE_WRITE;

	// One of the creation flags must be set.
	assert(flags & kCreationMask);

	DWORD dwCreationDisposition;

	uint32 creationType = flags & kCreationMask;

	switch (creationType) {
	case kOpenExisting:
		dwCreationDisposition = OPEN_EXISTING;
		break;
	case kOpenAlways:
		dwCreationDisposition = OPEN_ALWAYS;
		break;
	case kCreateAlways:
		dwCreationDisposition = CREATE_ALWAYS;
		break;
	case kCreateNew:
		dwCreationDisposition = CREATE_NEW;
		break;
	case kTruncateExisting:
		dwCreationDisposition = TRUNCATE_EXISTING;
		break;
	default:
		assert(0);
		return false;
	}

	assert((flags & (kSequential | kRandomAccess)) != (kSequential | kRandomAccess));

	DWORD dwAttributes = FILE_ATTRIBUTE_NORMAL;

	if (flags & kSequential)
		dwAttributes |= FILE_FLAG_SEQUENTIAL_SCAN;
	if (flags & kRandomAccess)
		dwAttributes |= FILE_FLAG_RANDOM_ACCESS;
	if (flags & kWriteThrough)
		dwAttributes |= FILE_FLAG_WRITE_THROUGH;
	if (flags & kUnbuffered)
		dwAttributes |= FILE_FLAG_NO_BUFFERING;
	/*
if (pwszFilename && !IsWindowsNT())
	pszFilename = VDFastTextWToA(pwszFilename);
*/
	//if (pszFilename)
	m_hFile = CreateFileW(fname.getWideString().c_str(), dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwAttributes, NULL);
	/*
else
	m_hFile = CreateFileW(pwszFilename, dwDesiredAccess, dwShareMode, NULL, dwCreationDisposition, dwAttributes, NULL);
*/
	//VDFastTextFree();

	// INVALID_HANDLE_VALUE isn't NULL.  *sigh*

	if (m_hFile == INVALID_HANDLE_VALUE) {
		m_hFile = NULL;

		m_ec = GetLastError();

		if (m_ec == ERROR_FILE_NOT_FOUND)
			return false;

		if (m_ec == ERROR_PATH_NOT_FOUND && creationType == kOpenExisting)
			return false;

		/*
  VDStringA fname(TFileSplitPathRight(pszFilename?VDString(pszFilename):VDTextWToA(VDStringW(pwszFilename))));

	throw MyWin32Error("Cannot open file \"%s\":\n%%s", err, fname.c_str());
  */
	}

	m_FilePosition = 0;

	return true;
}

//------------------------------------------------------------------------------

bool TFile::close()
{
	if (m_hFile) {
		HANDLE h = m_hFile;
		m_hFile = NULL;
		if (!CloseHandle(h)) {
			m_ec = GetLastError();
			return false;
		}
	}
	return true;
}

//------------------------------------------------------------------------------

bool TFile::truncate()
{
	DWORD rc = SetEndOfFile(m_hFile);
	if (rc)
		m_ec = GetLastError();
	return 0 != rc;
}

//------------------------------------------------------------------------------

TINT32 TFile::read(void *buffer, TINT32 length)
{
	DWORD dwActual;

	if (!ReadFile(m_hFile, buffer, (DWORD)length, &dwActual, NULL)) {
		m_ec = GetLastError();
		return 0;
	}

	m_FilePosition += dwActual;

	return dwActual;
}

//------------------------------------------------------------------------------

TINT32 TFile::write(const void *buffer, TINT32 length)
{
	DWORD dwActual;

	if (!WriteFile(m_hFile, buffer, (DWORD)length, &dwActual, NULL) || dwActual != (DWORD)length) {
		m_ec = GetLastError();
		return 0;
	}

	m_FilePosition += dwActual;

	return dwActual;
}

//------------------------------------------------------------------------------

bool TFile::seek(TINT64 newPos, SeekMode mode)
{
	DWORD dwMode;

	switch (mode) {
	case seekStart:
		dwMode = FILE_BEGIN;
		break;
	case seekCur:
		dwMode = FILE_CURRENT;
		break;
	case seekEnd:
		dwMode = FILE_END;
		break;
	default:
		assert(0);
		return false;
	}

	union {
		TINT64 pos;
		LONG l[2];
	} u = {newPos};

	u.l[0] = SetFilePointer(m_hFile, u.l[0], &u.l[1], dwMode);

	if (u.l[0] == -1 && GetLastError() != NO_ERROR) {
		m_ec = GetLastError();
		return false;
	}

	m_FilePosition = u.pos;
	return true;
}

//------------------------------------------------------------------------------

bool TFile::skip(TINT64 delta)
{
	if (!delta)
		return true;

	char buf[1024];

	if (delta <= sizeof buf)
		return (TINT32)delta == read(buf, (TINT32)delta);
	else
		return seek(delta, seekCur);
}

//------------------------------------------------------------------------------

TINT64 TFile::size()
{
	union {
		UINT64 siz;
		DWORD l[2];
	} u;

	u.l[0] = GetFileSize(m_hFile, &u.l[1]);

	DWORD err;

	if (u.l[0] == (DWORD)-1L && (err = GetLastError()) != NO_ERROR)
	//throw MyWin32Error("Error retrieving file size: %%s", err);
	{
		m_ec = GetLastError();
		return (TINT64)(-1);
	}
	return (TINT64)u.siz;
}

//------------------------------------------------------------------------------

TINT64 TFile::tell()
{
	return m_FilePosition;
}

//------------------------------------------------------------------------------

bool TFile::isOpen()
{
	return m_hFile != 0;
}

//------------------------------------------------------------------------------

string TFile::getLastError()
{
	LPVOID lpMsgBuf;
	FormatMessage(
		FORMAT_MESSAGE_ALLOCATE_BUFFER |
			FORMAT_MESSAGE_FROM_SYSTEM |
			FORMAT_MESSAGE_IGNORE_INSERTS,
		NULL,
		m_ec,
		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
		(LPTSTR)&lpMsgBuf,
		0,
		NULL);
	string s = string(reinterpret_cast<char *>(lpMsgBuf));
	LocalFree(lpMsgBuf);
	return s;
}