WebSVN – spk – Diff – //libspk/src/File_IO.cpp

 	m_bBinary = bBinary;
 	_sFilename = sFilename.asFilename();
 	// check if there are any directories, and split the file and directory parts
 	if ( _sFilename.contains('/') ) {
-		_sDirIO.setDir(_sFilename.tokens(L"/", 1, -2).toString());
+		_sDirIO.setDir(_sFilename.tokens(L"/", 1, -2));
 		_sFile = _sFilename.token(L"/", -1);
+	}
 	else
 		_sFile = _sFilename;
 	if ( pSize ) (*pSize) = m_lSize;
 	return data;
+}
-bool CFileIO::read(unsigned char *buf, size_t iSize, bool bEndChar)
+bool CFileIO::read(wchar_t* buf, size_t iSize, bool bEndChar)
+{
-	if ( !this->isOpened() ) startRead();
+	if (!this->isOpened()) startRead();
-	if ( !this->isOpened() ) return false;
+	if (!this->isOpened()) return false;
-	if ( iSize > m_lSize ) iSize = m_lSize;
+	if (iSize > m_lSize) iSize = m_lSize;
 	try {
-		m_fId.read((char *)buf, iSize);
+		m_fId.read((char*)buf, iSize);
+	}
-	catch (std::exception &e) {
+	catch (std::exception& e) {
+		CLog::logf(CLog::Log_IO, 3, "ERROR: CFileIO::read() unable to read from file: %s (%s)", _sFilename.c_str(), e.what());
+		return false;
+	}
+	if (bEndChar) buf[iSize] = '\0';
+	return !m_fId.bad();
+}
+bool CFileIO::read(unsigned char* buf, size_t iSize, bool bEndChar)
+{
+	if (!this->isOpened()) startRead();
+	if (!this->isOpened()) return false;
+	if (iSize > m_lSize) iSize = m_lSize;
+	try {
+		m_fId.read((char*)buf, iSize);
+	}
+	catch (std::exception& e) {
 		CLog::logf(CLog::Log_IO, 3, "ERROR: CFileIO::read() unable to read from file: %s (%s)", _sFilename.c_str(), e.what());
 		return false;
+	}
-	if ( bEndChar ) buf[iSize] = '\0';
+	if (bEndChar) buf[iSize] = '\0';
 	return !m_fId.bad();
+}
 bool CFileIO::WritePartFile ( size_t *offsets, size_t numOffset )
 	File.close();
 	writeFile.close();
 	// now copy to original file
-	std::remove(_fileData());
+	CFileIO::Remove(_sFilename);
-	return std::rename(CPackages::tempDirectory() + "/temp.tmp", _fileData()) != 0;
+	return CFileIO::Rename(CPackages::tempDirectory() + L"/temp.tmp", _sFilename);
+}
 void CFileIO::setDir(const Utils::WString &dir)
+{
 	File.close();
 	this->close();
 	File.setAutoDelete(false);
 	// now copy to original file
-	if ( std::remove(_fileData()) == 0) {
+	if (CFileIO::Remove(_sFilename)) {
-		std::rename(CPackages::tempDirectory() + "/temp.tmp", _fileData());
+		if (CFileIO::Rename(CPackages::tempDirectory() + L"/temp.tmp", _sFilename))
+		{
-		size_t oldSize = m_lSize;
+			size_t oldSize = m_lSize;
-		size_t checkFileSize = m_lSize - datasize;
+			size_t checkFileSize = m_lSize - datasize;
-		this->_readFileSize();
+			this->_readFileSize();
-		if ( checkFileSize != m_lSize ) {
+			if (checkFileSize != m_lSize) {
-			CLog::log(CLog::Log_IO, 3, Utils::String("WARNING: CFileIO::TruncateFile, file size mismatch, ") + (long)checkFileSize + " => " + (long)m_lSize);
+				CLog::log(CLog::Log_IO, 3, Utils::WString(L"WARNING: CFileIO::TruncateFile, file size mismatch, ") + (long)checkFileSize + L" => " + (long)m_lSize);
+			}
+			return FILEERR_NONE;
+		}
-		return FILEERR_NONE;
+		return FILEERR_NOWRITE;
+	}
 	return FILEERR_NOWRITE;
+}
 	return WriteData ( data.c_str(), data.length() );
+}
 bool CFileIO::Exists(const Utils::WString &filename)
+{
-	std::fstream File(filename, std::ios::in | std::ios::binary);
+	std::wfstream File(filename, std::ios::in | std::ios::binary);
 	bool bRet = File.good();
 	File.close();
 	return bRet;
+}
 	infile.close();
 	return to.size();
+}
+size_t CFileIO::readLines(std::vector<Utils::WString>& to) const
+{
+	if (_sFilename.empty()) return 0;
+	std::wstring line;
+	std::wifstream infile(_sFilename.c_str(), std::ios_base::in);
+	while (getline(infile, line, L'\n')) {
+		to.push_back(Utils::WString(line).remove((char)0));
+	}
+	infile.close();
+	return to.size();
+}
 size_t CFileIO::readLines(Utils::CStringList& to) const
+{
 	if (_sFilename.empty()) return 0;
 	Utils::String line;
 bool CFileIO::write(const char *buf, size_t iSize)
+{
 	if ( !this->isOpened() ) startWrite();
 	if ( !this->isOpened() ) return false;
 	try {
-		m_fId.write((char *)buf, iSize);
+		m_fId.write((char*)buf, iSize);
+	}
 	catch (std::exception &e) {
 		CLog::logf(CLog::Log_IO, 2, "CFileIO::write() unable to write to file: %s (%s)", _sFilename.c_str(), e.what());
 		return false;
+	}
 	va_start(args, buf);
 	bool ret = this->_write((char *)buf, args);
 	va_end(args);
 	return ret;
+}
-bool CFileIO::_write(const char *buf, va_list args)
+bool CFileIO::_write(const char* buf, va_list args)
+{
-	if ( !this->isOpened() ) return false;
+	if (!this->isOpened()) return false;
 	char buffer[10000];
-	vsprintf (buffer, buf, args);
+	vsprintf(buffer, buf, args);
 	try {
 		int iLen = strlen(buffer);
-		m_fId.write((char *)buffer, iLen);
+		m_fId.write((char*)buffer, iLen);
 		m_lSize += iLen;
+	}
-	catch (std::exception &e) {
+	catch (std::exception& e) {
 		CLog::logf(CLog::Log_IO, 2, "CFileIO::write() unable to write to file: %s (%s)", _sFilename.c_str(), e.what());
 		return false;
+	}
 	return !m_fId.bad();
+}
+bool CFileIO::_write(const wchar_t* buf, va_list args)
+{
+	if (!this->isOpened()) return false;
+	wchar_t buffer[10000];
+#ifdef _WIN32
+	vswprintf_s(buffer, 10000, buf, args);
+#else
+	vswprintf(buffer, buf, args);
+#endif
+	try {
+		size_t iLen = wcslen(buffer);
+		m_fId.write((char*)buffer, iLen);
+		m_lSize += iLen;
+	}
+	catch (std::exception& e) {
+		CLog::logf(CLog::Log_IO, 2, "CFileIO::write() unable to write to file: %ls (%s)", _sFilename.c_str(), e.what());
+		return false;
+	}
+	return !m_fId.bad();
+}
 bool CFileIO::write(const char *buf, ...)
+{
 	va_list args;
 	va_start(args, buf);
+	}
 	return this->write(d, size);
+}
-Utils::String CFileIO::readEndOfLine()
+Utils::String CFileIO::readEndOfLineStr()
+{
-	if ( !this->isOpened() ) {
+	if (!this->isOpened()) {
-		if ( !startRead() )	return "";
+		if (!startRead())	return "";
+	}
+	std::string str;
+	std::getline(m_fId, str, '\n');
+	return str;
+}
+Utils::WString CFileIO::readEndOfLine()
+{
+	if (!this->isOpened()) {
+		if (!startRead())	return L"";
+	}
-	Utils::String str;
+	std::string str;
 	std::getline(m_fId, str, '\n');
 	return str;
+}
 Utils::WString CFileIO::baseName() const
+{
-	return _sFile.tokens(L".", 1, -2).toString();
+	return _sFile.tokens(L".", 1, -2);
+}
 const Utils::WString& CFileIO::fullFilename() const { return _sFilename; }
 const Utils::WString& CFileIO::filename() const { return _sFile; }
 		return Utils::WString::Null();
 	return _sFilename.tokens(L".", 1, -2) + L"." + ext;
+}
-bool CFileIO::Remove(const Utils::WString &rem)
+bool CFileIO::Remove(const Utils::WString& rem)
+{
+#ifdef _WIN32
+	return DeleteFile(rem.c_str());
+#else
 	//if ( !Exists() ) return false;
 	return (std::remove(_toFileData(rem)) == 0) ? true : false;
+#endif
+}
+bool CFileIO::Rename(const Utils::WString& from, const Utils::WString &to)
+{
+	if (std::rename(_toFileData(from).c_str(), _toFileData(to).c_str()) == 0)
+		return true;
+	return false;
+}
 bool CFileIO::remove()
+{
 	if ( this->isOpened() ) this->close();
 	if ( !this->exists() ) return false;
-	if ( std::remove(_fileData()) == 0) return true;
+	if ( CFileIO::Remove(_sFilename)) return true;
 	return false;
+}
-bool CFileIO::writeFileUTF(std::vector<Utils::String> *lines)
+bool CFileIO::writeFileUTF(std::vector<Utils::String>* lines)
+{
 	if (!lines || _sFilename.empty())
 		return false;
 	// we need to create the directory
 			return false;
+	}
 #ifdef _WIN32
 	TCHAR buf[5000];
-	wsprintf(buf, L"%hs", _sFilename.c_str());
-	FILE *id = _wfopen(buf, L"wt+,ccs=UTF-8");
+	FILE* id = _wfopen(_sFilename.c_str(), L"wt+,ccs=UTF-8");
 	if (!id)
 		return false;
 	// write the rest
-	for(auto itr = lines->begin(); itr != lines->end(); itr++)
+	for (auto itr = lines->begin(); itr != lines->end(); itr++)
+	{
 		Utils::String l = (*itr);
 		if (l.isin('\n'))
+		{
 			int max;
-			Utils::String *strs = l.tokenise("\n", &max);
+			Utils::String* strs = l.tokenise("\n", &max);
 			if (strs && max)
+			{
 				for (int i = 0; i < max; i++)
+				{
 					Utils::String line = strs[i];
 	//TODO: write utf8 file writing function
 	return false;
 #endif
+}
+bool CFileIO::writeFileUTF(std::vector<Utils::WString>* lines)
+{
+	if (!lines || _sFilename.empty())
+		return false;
+	// we need to create the directory
+	if (!_sDirIO.exists())
+	{
+		if (!_sDirIO.create())
+			return false;
+	}
+#ifdef _WIN32
+	FILE* id = _wfopen(_sFilename.c_str(), L"wt+,ccs=UTF-8");
+	if (!id)
+		return false;
+	// write the rest
+	for (auto itr = lines->begin(); itr != lines->end(); itr++)
+	{
+		Utils::WString l = (*itr);
+		if (l.contains('\n'))
+		{
+			std::vector<Utils::WString> strs;
+			if(l.tokenise(L"\n", strs) && !strs.empty())
+			{
+				for(auto itr = strs.begin(); itr != strs.end(); itr++)
+				{
+					Utils::WString line = *itr;
+					line += L"\n";
+					fwrite(l.c_str(), sizeof(TCHAR), wcslen(l.c_str()), id);
+				}
+			}
+		}
+		else
+		{
+			l += L"\n";
+			fwrite(l.c_str(), sizeof(TCHAR), wcslen(l.c_str()), id);
+		}
+	}
+	fclose(id);
+	return true;
+#else
+	//TODO: write utf8 file writing function
+	return false;
+#endif
+}
 bool CFileIO::writeFile(std::vector<Utils::String> *lines)
+{
 	if (!lines || _sFilename.empty())
 		return false;
 	out.close();
 	return true;
+}
+bool CFileIO::writeFile(const std::vector<Utils::WString> &lines) const
+{
+	if (lines.empty() || _sFilename.empty())
+		return false;
+	// we need to create the directory
+	if (!_sDirIO.exists())
+	{
+		if (!_sDirIO.create())
+			return false;
+	}
+	std::wofstream out(_sFilename.c_str());
+	if (!out)
+		return false;
+	for (auto itr = lines.begin(); itr != lines.end(); itr++)
+		out << (*itr).c_str() << std::endl;
+	out.close();
+	return true;
+}
 bool CFileIO::writeFile(Utils::WStringList* lines)
+{
 	if (!lines || _sFilename.empty())
 		return false;
+	{
 		if (!_sDirIO.create())
 			return false;
+	}
-	std::ofstream out(_sFilename.c_str());
+	std::wofstream out(_sFilename.c_str());
 	if (!out)
 		return false;
 	for (auto itr = lines->begin(); itr != lines->end(); itr++)
 		out << (*itr)->str.c_str() << std::endl;
 	return true;
+}
 bool CFileIO::Rename(const Utils::WString &toFile)
 {
-	if ( rename(_fileData(), _toFileData(toFile)) == 0)
+	return CFileIO::Rename(_sFilename, toFile);
-		return true;
-	return false;
+}
 Utils::WString CFileIO::getWindowsFilename() const
+{
 	return _sFilename.findReplace(L"/", L"\\");
+{
 	if (_sFilename.empty()) _sExt = Utils::WString::Null();
 	_sExt = _sFilename.token(L".", -1);
+}
-const char* CFileIO::_fileData() const
+std::string CFileIO::_fileData() const
+{
-	return std::string(_sFilename.begin(), _sFilename.end()).c_str();
+	return _toFileData(_sFilename);
+}
-const char* CFileIO::_toFileData(const Utils::WString &str)
+std::string CFileIO::_toFileData(const Utils::WString &str)
+{
-	return std::string(str.begin(), str.end()).c_str();
+	return std::string(str.begin(), str.end());
+}

Subversion Repositories spk

(root)//libspk/src/File_IO.cpp – Rev 196 → 197