#include "pch.h"
|
#include "BasicExcel.hpp"
|
|
namespace YCompoundFiles
|
{
|
/********************************** Start of Class Block *************************************/
|
// PURPOSE: Manage a file by treating it as blocks of data of a certain size.
|
Block::Block() :
|
blockSize_(512), fileSize_(0), indexEnd_(0),
|
filename_(0) {}
|
|
bool Block::Create(const wchar_t* filename)
|
// PURPOSE: Create a new block file and open it.
|
// PURPOSE: If file is present, truncate it and then open it.
|
// PROMISE: Return true if file is successfully created and opened, false if otherwise.
|
{
|
// Create new file
|
size_t filenameLength = wcslen(filename) * 2;
|
char* name = new char[filenameLength+1];
|
wcstombs(name, filename, filenameLength);
|
name[filenameLength] = 0;
|
|
file_.open(name, ios_base::out | ios_base::trunc);
|
file_.close();
|
file_.clear();
|
|
// Open the file
|
bool ret = this->Open(filename);
|
delete[] name;
|
return ret;
|
}
|
|
bool Block::Open(const wchar_t* filename, ios_base::openmode mode)
|
// PURPOSE: Open an existing block file.
|
// PROMISE: Return true if file is successfully opened, false if otherwise.
|
{
|
// Open existing file for reading or writing or both
|
size_t filenameLength = wcslen(filename) * 2;
|
filename_.resize(filenameLength+1, 0);
|
wcstombs(&*(filename_.begin()), filename, filenameLength);
|
|
file_.open(&*(filename_.begin()), mode | ios_base::binary);
|
if (!file_.is_open()) return false;
|
|
mode_ = mode;
|
|
// Calculate filesize
|
if (mode & ios_base::in)
|
{
|
file_.seekg(0, ios_base::end);
|
fileSize_ = file_.tellg();
|
}
|
else if (mode & ios_base::out)
|
{
|
file_.seekp(0, ios_base::end);
|
fileSize_ = file_.tellp();
|
}
|
else
|
{
|
this->Close();
|
return false;
|
}
|
|
// Calculate last index + 1
|
indexEnd_ = fileSize_/blockSize_ + (fileSize_ % blockSize_ ? 1 : 0);
|
return true;
|
}
|
|
bool Block::Close()
|
// PURPOSE: Close the opened block file.
|
// PROMISE: Return true if file is successfully closed, false if otherwise.
|
{
|
file_.close();
|
file_.clear();
|
filename_.clear();
|
fileSize_ = 0;
|
indexEnd_ = 0;
|
blockSize_ = 512;
|
return !file_.is_open();
|
}
|
|
bool Block::IsOpen()
|
// PURPOSE: Check if the block file is still opened.
|
// PROMISE: Return true if file is still opened, false if otherwise.
|
{
|
return file_.is_open();
|
}
|
|
bool Block::Read(size_t index, char* block)
|
// PURPOSE: Read a block of data from the opened file at the index position.
|
// EXPLAIN: index is from [0..].
|
// PROMISE: Return true if data are successfully read, false if otherwise.
|
{
|
if (!(mode_ & ios_base::in)) return false;
|
if (index < indexEnd_)
|
{
|
file_.seekg(index * blockSize_);
|
file_.read(block, blockSize_);
|
return !file_.fail();
|
}
|
else return false;
|
}
|
|
bool Block::Write(size_t index, const char* block)
|
// PURPOSE: Write a block of data to the opened file at the index position.
|
// EXPLAIN: index is from [0..].
|
// PROMISE: Return true if data are successfully written, false if otherwise.
|
{
|
if (!(mode_ & ios_base::out)) return false;
|
file_.seekp(index * blockSize_);
|
file_.write(block, blockSize_);
|
if (indexEnd_ <= index)
|
{
|
indexEnd_ = index + 1;
|
fileSize_ += blockSize_;
|
}
|
file_.close();
|
file_.clear();
|
file_.open(&*(filename_.begin()), mode_ | ios_base::binary);
|
return file_.is_open();
|
}
|
|
bool Block::Swap(size_t index1, size_t index2)
|
// PURPOSE: Swap two blocks of data in the opened file at the index positions.
|
// EXPLAIN: index1 and index2 are from [0..].
|
// PROMISE: Return true if data are successfully swapped, false if otherwise.
|
{
|
if (!(mode_ & ios_base::out)) return false;
|
if (index1 < indexEnd_ && index2 < indexEnd_)
|
{
|
if (index1 == index2) return true;
|
|
char* block1 = new char[blockSize_];
|
if (!this->Read(index1, block1)) return false;
|
|
char* block2 = new char[blockSize_];
|
if (!this->Read(index2, block2)) return false;
|
|
if (!this->Write(index1, block2)) return false;
|
if (!this->Write(index2, block1)) return false;
|
|
delete[] block1;
|
delete[] block2;
|
return true;
|
}
|
else return false;
|
}
|
|
bool Block::Move(size_t from, size_t to)
|
// PURPOSE: Move a block of data in the opened file from an index position to another index position.
|
// EXPLAIN: from and to are from [0..].
|
// PROMISE: Return true if data are successfully moved, false if otherwise.
|
{
|
if (!(mode_ & ios_base::out)) return false;
|
if (from < indexEnd_ && to < indexEnd_)
|
{
|
if (to > from)
|
{
|
for (size_t i=from; i!=to; ++i)
|
{
|
if (!this->Swap(i, i+1)) return false;
|
}
|
}
|
else
|
{
|
for (size_t i=from; i!=to; --i)
|
{
|
if (!this->Swap(i, i-1)) return false;
|
}
|
}
|
return true;
|
}
|
else return false;
|
}
|
|
bool Block::Insert(size_t index, const char* block)
|
// PURPOSE: Insert a new block of data in the opened file at the index position.
|
// EXPLAIN: index is from [0..].
|
// PROMISE: Return true if data are successfully inserted, false if otherwise.
|
{
|
if (!(mode_ & ios_base::out)) return false;
|
if (index <= indexEnd_)
|
{
|
// Write block to end of file
|
if (!this->Write(indexEnd_, block)) return false;
|
|
// Move block to index if necessary
|
if (index < indexEnd_-1) return this->Move(indexEnd_-1, index);
|
else return true;
|
}
|
else
|
{
|
// Write block to index after end of file
|
return this->Write(index, block);
|
}
|
}
|
|
bool Block::Erase(size_t index)
|
// PURPOSE: Erase a block of data in the opened file at the index position.
|
// EXPLAIN: index is from [0..].
|
// PROMISE: Return true if data are successfully erased, false if otherwise.
|
{
|
if (!(mode_ & ios_base::out)) return false;
|
if (index < indexEnd_)
|
{
|
fileSize_ -= blockSize_;
|
indexEnd_ -= 1;
|
|
// Read entire file except the block to be deleted into memory.
|
char* buffer = new char[fileSize_];
|
for (size_t i=0, j=0; i!=indexEnd_+1; ++i)
|
{
|
file_.seekg(i*blockSize_);
|
if (i != index)
|
{
|
file_.read(buffer+j*blockSize_, blockSize_);
|
++j;
|
}
|
}
|
file_.close();
|
file_.open(&*(filename_.begin()), ios_base::out | ios_base::trunc | ios_base::binary);
|
file_.write(buffer, fileSize_); // Write the new file.
|
file_.close();
|
file_.open(&*(filename_.begin()), mode_ | ios_base::binary);
|
delete[] buffer;
|
return true;
|
}
|
else return false;
|
}
|
|
bool Block::Erase(vector<size_t>& indices)
|
// PURPOSE: Erase blocks of data in the opened file at the index positions.
|
// EXPLAIN: Each index in indices is from [0..].
|
// PROMISE: Return true if data are successfully erased, false if otherwise.
|
{
|
if (!(mode_ & ios_base::out)) return false;
|
|
// Read entire file except the blocks to be deleted into memory.
|
size_t maxIndices = indices.size();
|
fileSize_ -= maxIndices*blockSize_;
|
char* buffer = new char[fileSize_];
|
for (size_t i=0, k=0; i!=indexEnd_; ++i)
|
{
|
file_.seekg(i*blockSize_);
|
bool toDelete = false;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (i == indices[j])
|
{
|
toDelete = true;
|
break;
|
}
|
}
|
if (!toDelete)
|
{
|
file_.read(buffer+k*blockSize_, blockSize_);
|
++k;
|
}
|
}
|
indexEnd_ -= maxIndices;
|
|
file_.close();
|
file_.open(&*(filename_.begin()), ios_base::out | ios_base::trunc | ios_base::binary);
|
file_.write(buffer, fileSize_); // Write the new file.
|
file_.close();
|
file_.open(&*(filename_.begin()), mode_ | ios_base::binary);
|
delete[] buffer;
|
return true;
|
}
|
/********************************** End of Class Block ***************************************/
|
|
/********************************** Start of Class Header ************************************/
|
// PURPOSE: Read and write data to a compound file header.
|
CompoundFile::Header::Header() :
|
fileType_(0xE11AB1A1E011CFD0LL),
|
uk1_(0), uk2_(0), uk3_(0), uk4_(0), uk5_(0x003B), uk6_(0x0003), uk7_(-2),
|
log2BigBlockSize_(9), log2SmallBlockSize_(6),
|
uk8_(0), uk9_(0), uk10_(0), uk11_(0x00001000),
|
SBATStart_(-2), SBATCount_(0),
|
XBATStart_(-2), XBATCount_(0),
|
BATCount_(1), propertiesStart_(1)
|
{
|
BATArray_[0] = 0; // Initial BAT indices at block 0 (=block 1 in Block)
|
fill (BATArray_+1, BATArray_+109, -1); // Rest of the BATArray is empty
|
Initialize();
|
}
|
|
void CompoundFile::Header::Write(char* block)
|
// PURPOSE: Write header information into a block of data.
|
// REQUIRE: Block of data must be at least 512 bytes in size.
|
{
|
LittleEndian::Write(block, fileType_, 0x0000, 8);
|
LittleEndian::Write(block, uk1_, 0x0008, 4);
|
LittleEndian::Write(block, uk2_, 0x000C, 4);
|
LittleEndian::Write(block, uk3_, 0x0010, 4);
|
LittleEndian::Write(block, uk4_, 0x0014, 4);
|
LittleEndian::Write(block, uk5_, 0x0018, 2);
|
LittleEndian::Write(block, uk6_, 0x001A, 2);
|
LittleEndian::Write(block, uk7_, 0x001C, 2);
|
LittleEndian::Write(block, log2BigBlockSize_, 0x001E, 2);
|
LittleEndian::Write(block, log2SmallBlockSize_, 0x0020, 4);
|
LittleEndian::Write(block, uk8_, 0x0024, 4);
|
LittleEndian::Write(block, uk9_, 0x0028, 4);
|
LittleEndian::Write(block, BATCount_, 0x002C, 4);
|
LittleEndian::Write(block, propertiesStart_, 0x0030, 4);
|
LittleEndian::Write(block, uk10_, 0x0034, 4);
|
LittleEndian::Write(block, uk11_, 0x0038, 4);
|
LittleEndian::Write(block, SBATStart_, 0x003C, 4);
|
LittleEndian::Write(block, SBATCount_, 0x0040, 4);
|
LittleEndian::Write(block, XBATStart_, 0x0044, 4);
|
LittleEndian::Write(block, XBATCount_, 0x0048, 4);
|
for (size_t i=0; i<109; ++i) LittleEndian::Write(block, BATArray_[i], 0x004C+i*4, 4);
|
}
|
|
void CompoundFile::Header::Read(char* block)
|
// PURPOSE: Read header information from a block of data.
|
// REQUIRE: Block of data must be at least 512 bytes in size.
|
{
|
LittleEndian::Read(block, fileType_, 0x0000, 8);
|
LittleEndian::Read(block, uk1_, 0x0008, 4);
|
LittleEndian::Read(block, uk2_, 0x000C, 4);
|
LittleEndian::Read(block, uk3_, 0x0010, 4);
|
LittleEndian::Read(block, uk4_, 0x0014, 4);
|
LittleEndian::Read(block, uk5_, 0x0018, 2);
|
LittleEndian::Read(block, uk6_, 0x001A, 2);
|
LittleEndian::Read(block, uk7_, 0x001C, 2);
|
LittleEndian::Read(block, log2BigBlockSize_, 0x001E, 2);
|
LittleEndian::Read(block, log2SmallBlockSize_, 0x0020, 4);
|
LittleEndian::Read(block, uk8_, 0x0024, 4);
|
LittleEndian::Read(block, uk9_, 0x0028, 4);
|
LittleEndian::Read(block, BATCount_, 0x002C, 4);
|
LittleEndian::Read(block, propertiesStart_, 0x0030, 4);
|
LittleEndian::Read(block, uk10_, 0x0034, 4);
|
LittleEndian::Read(block, uk11_, 0x0038, 4);
|
LittleEndian::Read(block, SBATStart_, 0x003C, 4);
|
LittleEndian::Read(block, SBATCount_, 0x0040, 4);
|
LittleEndian::Read(block, XBATStart_, 0x0044, 4);
|
LittleEndian::Read(block, XBATCount_, 0x0048, 4);
|
for (size_t i=0; i<109; ++i) LittleEndian::Read(block, BATArray_[i], 0x004C+i*4, 4);
|
Initialize();
|
}
|
|
void CompoundFile::Header::Initialize()
|
{
|
bigBlockSize_ = (size_t)pow(2.0, log2BigBlockSize_); // Calculate each big block size.
|
smallBlockSize_ = (size_t)pow(2.0, log2SmallBlockSize_); // Calculate each small block size.
|
}
|
/********************************** End of Class Header **************************************/
|
|
/********************************** Start of Class Property **********************************/
|
// PURPOSE: Read and write data to a compound file property.
|
CompoundFile::Property::Property() :
|
nameSize_(0),
|
propertyType_(1), nodeColor_(1),
|
previousProp_(-1), nextProp_(-1), childProp_(-1),
|
uk1_(0), uk2_(0), uk3_(0), uk4_(0), uk5_(0),
|
seconds1_(0), days1_(0), seconds2_(0), days2_(0),
|
startBlock_(-2), size_(0)
|
{
|
fill (name_, name_+32, 0);
|
}
|
|
void CompoundFile::Property::Write(char* block)
|
// PURPOSE: Write property information from a block of data.
|
// REQUIRE: Block of data must be at least 128 bytes in size.
|
{
|
LittleEndian::WriteString(block, name_, 0x00, 32);
|
LittleEndian::Write(block, nameSize_, 0x40, 2);
|
LittleEndian::Write(block, propertyType_, 0x42, 1);
|
LittleEndian::Write(block, nodeColor_, 0x43, 1);
|
LittleEndian::Write(block, previousProp_, 0x44, 4);
|
LittleEndian::Write(block, nextProp_, 0x48, 4);
|
LittleEndian::Write(block, childProp_, 0x4C, 4);
|
LittleEndian::Write(block, uk1_, 0x50, 4);
|
LittleEndian::Write(block, uk2_, 0x54, 4);
|
LittleEndian::Write(block, uk3_, 0x58, 4);
|
LittleEndian::Write(block, uk4_, 0x5C, 4);
|
LittleEndian::Write(block, uk5_, 0x60, 4);
|
LittleEndian::Write(block, seconds1_, 0x64, 4);
|
LittleEndian::Write(block, days1_, 0x68, 4);
|
LittleEndian::Write(block, seconds2_, 0x6C, 4);
|
LittleEndian::Write(block, days2_, 0x70, 4);
|
LittleEndian::Write(block, startBlock_, 0x74, 4);
|
LittleEndian::Write(block, size_, 0x78, 4);
|
}
|
|
void CompoundFile::Property::Read(char* block)
|
// PURPOSE: Read property information from a block of data.
|
// REQUIRE: Block of data must be at least 128 bytes in size.
|
{
|
LittleEndian::ReadString(block, name_, 0x00, 32);
|
LittleEndian::Read(block, nameSize_, 0x40, 2);
|
LittleEndian::Read(block, propertyType_, 0x42, 1);
|
LittleEndian::Read(block, nodeColor_, 0x43, 1);
|
LittleEndian::Read(block, previousProp_, 0x44, 4);
|
LittleEndian::Read(block, nextProp_, 0x48, 4);
|
LittleEndian::Read(block, childProp_, 0x4C, 4);
|
LittleEndian::Read(block, uk1_, 0x50, 4);
|
LittleEndian::Read(block, uk2_, 0x54, 4);
|
LittleEndian::Read(block, uk3_, 0x58, 4);
|
LittleEndian::Read(block, uk4_, 0x5C, 4);
|
LittleEndian::Read(block, uk5_, 0x60, 4);
|
LittleEndian::Read(block, seconds1_, 0x64, 4);
|
LittleEndian::Read(block, days1_, 0x68, 4);
|
LittleEndian::Read(block, seconds2_, 0x6C, 4);
|
LittleEndian::Read(block, days2_, 0x70, 4);
|
LittleEndian::Read(block, startBlock_, 0x74, 4);
|
LittleEndian::Read(block, size_, 0x78, 4);
|
}
|
/********************************** End of Class Property ************************************/
|
|
/********************************** Start of Class PropertyTree **********************************/
|
CompoundFile::PropertyTree::PropertyTree() {};
|
|
CompoundFile::PropertyTree::~PropertyTree()
|
{
|
size_t maxChildren = children_.size();
|
for (size_t i=0; i<maxChildren; ++i) delete children_[i];
|
}
|
/********************************** End of Class PropertyTree ************************************/
|
|
/********************************** Start of Class CompoundFile ******************************/
|
// PURPOSE: Manage a compound file.
|
CompoundFile::CompoundFile() :
|
block_(512), properties_(0), propertyTrees_(0),
|
blocksIndices_(0), sblocksIndices_(0) {};
|
|
CompoundFile::~CompoundFile() {this->Close();}
|
|
/************************* Compound File Functions ***************************/
|
bool CompoundFile::Create(const wchar_t* filename)
|
// PURPOSE: Create a new compound file and open it.
|
// PURPOSE: If file is present, truncate it and then open it.
|
// PROMISE: Return true if file is successfully created and opened, false if otherwise.
|
{
|
Close();
|
file_.Create(filename);
|
|
// Write compound file header
|
header_ = Header();
|
SaveHeader();
|
|
// Save BAT
|
blocksIndices_.clear();
|
blocksIndices_.resize(128, -1);
|
blocksIndices_[0] = -3;
|
blocksIndices_[1] = -2;
|
SaveBAT();
|
|
// Save properties
|
Property* root = new Property;
|
wcscpy(root->name_, L"Root Entry");
|
root->propertyType_ = 5;
|
properties_.push_back(root);
|
SaveProperties();
|
|
// Set property tree
|
propertyTrees_ = new PropertyTree;
|
propertyTrees_->parent_ = 0;
|
propertyTrees_->self_ = properties_[0];
|
propertyTrees_->index_ = 0;
|
currentDirectory_ = propertyTrees_;
|
|
return true;
|
}
|
|
bool CompoundFile::Open(const wchar_t* filename, ios_base::openmode mode)
|
// PURPOSE: Open an existing compound file.
|
// PROMISE: Return true if file is successfully opened, false if otherwise.
|
{
|
Close();
|
if (!file_.Open(filename, mode)) return false;
|
|
// Load header
|
if (!LoadHeader()) return false;
|
|
// Load BAT information
|
LoadBAT();
|
|
// Load properties
|
propertyTrees_ = new PropertyTree;
|
LoadProperties();
|
currentDirectory_ = propertyTrees_;
|
|
return true;
|
}
|
|
bool CompoundFile::Close()
|
// PURPOSE: Close the opened compound file.
|
// PURPOSE: Reset BAT indices, SBAT indices, properties and properties tree information.
|
// PROMISE: Return true if file is successfully closed, false if otherwise.
|
{
|
blocksIndices_.clear();
|
sblocksIndices_.clear();
|
|
size_t maxProperties = properties_.size();
|
for (size_t i=0; i<maxProperties; ++i)
|
{
|
if (properties_[i]) delete properties_[i];
|
}
|
properties_.clear();
|
|
if (propertyTrees_)
|
{
|
delete propertyTrees_;
|
propertyTrees_ = 0;
|
}
|
|
previousDirectories_.clear();
|
currentDirectory_ = 0;
|
|
return file_.Close();
|
}
|
|
bool CompoundFile::IsOpen()
|
// PURPOSE: Check if the compound file is still opened.
|
// PROMISE: Return true if file is still opened, false if otherwise.
|
{
|
return file_.IsOpen();
|
}
|
|
/************************* Directory Functions ***************************/
|
int CompoundFile::ChangeDirectory(const wchar_t* path)
|
// PURPOSE: Change to a different directory in the compound file.
|
// PROMISE: Current directory will not be changed if directory is not present.
|
{
|
previousDirectories_.push_back(currentDirectory_);
|
|
// Handle special cases
|
if (wcscmp(path, L".") == 0)
|
{
|
// Current directory
|
previousDirectories_.pop_back();
|
return SUCCESS;
|
}
|
if (wcscmp(path, L"..") == 0)
|
{
|
// Go up 1 directory
|
if (currentDirectory_->parent_ != 0)
|
{
|
currentDirectory_ = currentDirectory_->parent_;
|
}
|
previousDirectories_.pop_back();
|
return SUCCESS;
|
}
|
if (wcscmp(path, L"\\") == 0)
|
{
|
// Go to root directory
|
currentDirectory_ = propertyTrees_;
|
previousDirectories_.pop_back();
|
return SUCCESS;
|
}
|
|
// Handle normal cases
|
size_t ipos = 0;
|
size_t npos = 0;
|
size_t pathLength = wcslen(path);
|
if (pathLength > 0 && path[0] == L'\\')
|
{
|
// Start from root directory
|
currentDirectory_ = propertyTrees_;
|
++ipos;
|
++npos;
|
}
|
do
|
{
|
for (; npos<pathLength; ++npos)
|
{
|
if (path[npos] == L'\\') break;
|
}
|
|
wchar_t* directory = new wchar_t[npos-ipos+1];
|
copy (path+ipos, path+npos, directory);
|
directory[npos-ipos] = 0;
|
currentDirectory_ = FindProperty(currentDirectory_, directory);
|
delete[] directory;
|
ipos = npos + 1;
|
npos = ipos;
|
if (currentDirectory_ == 0)
|
{
|
// Directory not found
|
currentDirectory_ = previousDirectories_.back();
|
previousDirectories_.pop_back();
|
return DIRECTORY_NOT_FOUND;
|
}
|
} while (npos < pathLength);
|
previousDirectories_.pop_back();
|
return SUCCESS;
|
}
|
|
int CompoundFile::MakeDirectory(const wchar_t* path)
|
// PURPOSE: Create a new directory in the compound file.
|
// PROMISE: Directory will not be created if it is already present or
|
// PROMISE: a file with the same name is present.
|
{
|
previousDirectories_.push_back(currentDirectory_);
|
Property* property = new Property;
|
property->propertyType_ = 1;
|
int ret = MakeProperty(path, property);
|
currentDirectory_ = previousDirectories_.back();
|
previousDirectories_.pop_back();
|
SaveHeader();
|
SaveBAT();
|
SaveProperties();
|
return ret;
|
}
|
|
int CompoundFile::PresentWorkingDirectory(wchar_t* path)
|
// PURPOSE: Get the full path of the current directory in the compound file.
|
// REQUIRE: path must be large enough to receive the full path information.
|
{
|
previousDirectories_.push_back(currentDirectory_);
|
vector<wchar_t> fullpath;
|
do
|
{
|
size_t directoryLength = wcslen(currentDirectory_->self_->name_);
|
vector<wchar_t> directory(directoryLength+1);
|
directory[0] = L'\\';
|
copy (currentDirectory_->self_->name_,
|
currentDirectory_->self_->name_+directoryLength,
|
directory.begin()+1);
|
fullpath.insert(fullpath.begin(), directory.begin(), directory.end());
|
} while (currentDirectory_ = currentDirectory_->parent_);
|
|
fullpath.erase(fullpath.begin(), fullpath.begin()+11);
|
if (fullpath.empty()) fullpath.push_back(L'\\');
|
copy (fullpath.begin(), fullpath.end(), path);
|
path[fullpath.size()] = 0;
|
currentDirectory_ = previousDirectories_.back();
|
previousDirectories_.pop_back();
|
return SUCCESS;
|
}
|
|
int CompoundFile::PresentWorkingDirectory(vector<wchar_t>& path)
|
// PURPOSE: Get the full path of the current directory in the compound file.
|
{
|
previousDirectories_.push_back(currentDirectory_);
|
path.clear();
|
do
|
{
|
size_t directoryLength = wcslen(currentDirectory_->self_->name_);
|
vector<wchar_t> directory(directoryLength+1);
|
directory[0] = L'\\';
|
copy (currentDirectory_->self_->name_,
|
currentDirectory_->self_->name_+directoryLength,
|
directory.begin()+1);
|
path.insert(path.begin(), directory.begin(), directory.end());
|
} while (currentDirectory_ = currentDirectory_->parent_);
|
|
path.erase(path.begin(), path.begin()+11);
|
if (path.empty()) path.push_back(L'\\');
|
currentDirectory_ = previousDirectories_.back();
|
previousDirectories_.pop_back();
|
return SUCCESS;
|
}
|
|
int CompoundFile::RemoveDirectory(const wchar_t* path)
|
// PURPOSE: Remove a directory in the compound file.
|
// PROMISE: Directory will not be removed if it has subdirectories or files under it.
|
{
|
PropertyTree* directory = FindProperty(path);
|
if (directory == 0) return DIRECTORY_NOT_FOUND;
|
if (directory->self_->childProp_ != -1) return DIRECTORY_NOT_EMPTY;
|
DeletePropertyTree(directory);
|
SaveHeader();
|
SaveBAT();
|
SaveProperties();
|
return SUCCESS;
|
}
|
|
int CompoundFile::DelTree(const wchar_t* path)
|
// PURPOSE: Remove everything in the path in the compound file, including
|
// PURPOSE: any files and subdirectories.
|
{
|
previousDirectories_.push_back(currentDirectory_);
|
PropertyTree* directory = FindProperty(path);
|
if (directory == 0) return DIRECTORY_NOT_FOUND;
|
if (directory->self_->childProp_ != -1)
|
{
|
size_t maxChildren = directory->children_.size();
|
wchar_t* curpath = new wchar_t[65535];
|
for (size_t i=0; i<maxChildren; ++i)
|
{
|
currentDirectory_ = directory->children_[i];
|
PresentWorkingDirectory(curpath);
|
if (directory->children_[i]->self_->propertyType_ == 1)
|
{
|
// Directory
|
DelTree(curpath);
|
}
|
else if (directory->children_[i]->self_->propertyType_ == 2)
|
{
|
// File
|
RemoveFile(curpath);
|
}
|
}
|
directory->self_->childProp_ = -1;
|
delete[] curpath;
|
}
|
|
if (directory->self_->propertyType_ == 1)
|
{
|
// Directory
|
RemoveDirectory(path);
|
}
|
else if (directory->self_->propertyType_ == 2)
|
{
|
// File
|
RemoveFile(path);
|
}
|
|
currentDirectory_ = previousDirectories_.back();
|
previousDirectories_.pop_back();
|
return SUCCESS;
|
}
|
|
int CompoundFile::DirectoryList(vector<vector<wchar_t> >& list, const wchar_t* path)
|
{
|
previousDirectories_.push_back(currentDirectory_);
|
if (path != 0)
|
{
|
int ret = ChangeDirectory(path);
|
if (ret != SUCCESS) return ret;
|
}
|
list.clear();
|
size_t maxChildren = currentDirectory_->children_.size();
|
vector<wchar_t> name(32);
|
for (size_t i=0; i<maxChildren; ++i)
|
{
|
wcscpy(&*(name.begin()), currentDirectory_->children_[i]->self_->name_);
|
list.push_back(name);
|
}
|
currentDirectory_ = previousDirectories_.back();
|
previousDirectories_.pop_back();
|
return SUCCESS;
|
}
|
|
|
/************************* File Functions ***************************/
|
int CompoundFile::MakeFile(const wchar_t* path)
|
// PURPOSE: Create a new file in the compound file.
|
// PROMISE: File will not be created if it is already present or
|
// PROMISE: a directory with the same name is present.
|
{
|
previousDirectories_.push_back(currentDirectory_);
|
Property* property = new Property;
|
property->propertyType_ = 2;
|
int ret = MakeProperty(path, property);
|
currentDirectory_ = previousDirectories_.back();
|
previousDirectories_.pop_back();
|
SaveHeader();
|
SaveBAT();
|
SaveProperties();
|
return ret;
|
}
|
|
int CompoundFile::RemoveFile(const wchar_t* path)
|
// PURPOSE: Remove a file in the compound file.
|
{
|
int ret = WriteFile(path, 0, 0);
|
if (ret == SUCCESS)
|
{
|
DeletePropertyTree(FindProperty(path));
|
SaveHeader();
|
SaveBAT();
|
SaveProperties();
|
return SUCCESS;
|
}
|
else return ret;
|
}
|
|
int CompoundFile::FileSize(const wchar_t* path, size_t& size)
|
// PURPOSE: Get the size of a file in the compound file.
|
// PROMISE: Return the data size stored in the Root Entry if path = "\".
|
// PROMISE: size will not be set if file is not present in the compound file.
|
{
|
// Special case of reading root entry
|
if (wcscmp(path, L"\\") == 0)
|
{
|
size = propertyTrees_->self_->size_;
|
return SUCCESS;
|
}
|
|
// Check to see if file is present in the specified directory.
|
PropertyTree* property = FindProperty(path);
|
if (property == 0) return FILE_NOT_FOUND;
|
else
|
{
|
size = property->self_->size_;
|
return SUCCESS;
|
}
|
}
|
|
int CompoundFile::ReadFile(const wchar_t* path, char* data)
|
// PURPOSE: Read a file's data in the compound file.
|
// REQUIRE: data must be large enough to receive the file's data.
|
// REQUIRE: The required data size can be obtained by using FileSize().
|
// PROMISE: Returns the small blocks of data stored by the Root Entry if path = "\".
|
// PROMISE: data will not be set if file is not present in the compound file.
|
{
|
// Special case of reading root entry
|
char* buffer;
|
if (wcscmp(path, L"\\") == 0)
|
{
|
buffer = new char[DataSize(propertyTrees_->self_->startBlock_, true)];
|
ReadData(propertyTrees_->self_->startBlock_, buffer, true);
|
copy (buffer, buffer+propertyTrees_->self_->size_, data);
|
delete[] buffer;
|
return SUCCESS;
|
}
|
|
// Check to see if file is present in the specified directory.
|
PropertyTree* property = FindProperty(path);
|
if (property == 0) return FILE_NOT_FOUND;
|
|
if (property->self_->size_ >= 4096)
|
{
|
// Data stored in normal big blocks
|
buffer = new char[DataSize(property->self_->startBlock_, true)];
|
ReadData(property->self_->startBlock_, buffer, true);
|
}
|
else
|
{
|
// Data stored in small blocks
|
buffer = new char[DataSize(property->self_->startBlock_, false)];
|
ReadData(property->self_->startBlock_, buffer, false);
|
}
|
// Truncated the retrieved data to the actual file size.
|
copy (buffer, buffer+property->self_->size_, data);
|
delete[] buffer;
|
return SUCCESS;
|
}
|
|
int CompoundFile::ReadFile(const wchar_t* path, vector<char>& data)
|
// PURPOSE: Read a file's data in the compound file.
|
// PROMISE: Returns the small blocks of data stored by the Root Entry if path = "\".
|
// PROMISE: data will not be set if file is not present in the compound file.
|
{
|
data.clear();
|
size_t dataSize;
|
int ret = FileSize(path, dataSize);
|
if (ret != SUCCESS) return ret;
|
|
data.resize(dataSize);
|
return ReadFile(path, &*(data.begin()));
|
}
|
|
int CompoundFile::WriteFile(const wchar_t* path, const char* data, size_t size)
|
// PURPOSE: Write data to a file in the compound file.
|
// PROMISE: The file's original data will be replaced by the new data.
|
{
|
PropertyTree* property = FindProperty(path);
|
if (property == 0) return FILE_NOT_FOUND;
|
|
if (property->self_->size_ >= 4096)
|
{
|
if (size >= 4096) property->self_->startBlock_ = WriteData(data, size, property->self_->startBlock_, true);
|
else
|
{
|
property->self_->startBlock_ = WriteData(0, 0, property->self_->startBlock_, true);
|
property->self_->startBlock_ = WriteData(data, size, property->self_->startBlock_, false);
|
}
|
}
|
else
|
{
|
if (size < 4096) property->self_->startBlock_ = WriteData(data, size, property->self_->startBlock_, false);
|
else
|
{
|
property->self_->startBlock_ = WriteData(0, 0, property->self_->startBlock_, false);
|
property->self_->startBlock_ = WriteData(data, size, property->self_->startBlock_, true);
|
}
|
}
|
property->self_->size_ = size;
|
SaveHeader();
|
SaveBAT();
|
SaveProperties();
|
return SUCCESS;
|
}
|
|
int CompoundFile::WriteFile(const wchar_t* path, const vector<char>& data, size_t size)
|
// PURPOSE: Write data to a file in the compound file.
|
// PROMISE: The file's original data will be replaced by the new data.
|
{
|
return WriteFile(path, &*(data.begin()), size);
|
}
|
|
/*************ANSI char compound file, directory and file functions******************/
|
bool CompoundFile::Create(const char* filename)
|
{
|
size_t filenameLength = strlen(filename);
|
wchar_t* wname = new wchar_t[filenameLength+1];
|
mbstowcs(wname, filename, filenameLength);
|
wname[filenameLength] = 0;
|
bool ret = Create(wname);
|
delete[] wname;
|
return ret;
|
}
|
|
bool CompoundFile::Open(const char* filename, ios_base::openmode mode)
|
{
|
size_t filenameLength = strlen(filename);
|
wchar_t* wname = new wchar_t[filenameLength+1];
|
mbstowcs(wname, filename, filenameLength);
|
wname[filenameLength] = 0;
|
bool ret = Open(wname, mode);
|
delete[] wname;
|
return ret;
|
}
|
|
int CompoundFile::ChangeDirectory(const char* path)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = ChangeDirectory(wpath);
|
delete[] wpath;
|
return ret;
|
}
|
|
int CompoundFile::MakeDirectory(const char* path)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = MakeDirectory(wpath);
|
delete[] wpath;
|
return ret;
|
}
|
|
int CompoundFile::PresentWorkingDirectory(char* path)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
int ret = PresentWorkingDirectory(wpath);
|
if (ret == SUCCESS)
|
{
|
pathLength = wcslen(wpath) * 2;
|
wcstombs(path, wpath, pathLength);
|
path[pathLength] = 0;
|
}
|
delete[] wpath;
|
return ret;
|
}
|
|
int CompoundFile::PresentWorkingDirectory(vector<char>& path)
|
{
|
vector<wchar_t> wpath;
|
int ret = PresentWorkingDirectory(wpath);
|
if (ret == SUCCESS)
|
{
|
size_t pathLength = wpath.size() * 2;
|
path.resize(pathLength);
|
wcstombs(&*(path.begin()), &*(wpath.begin()), pathLength);
|
path[pathLength] = 0;
|
}
|
return ret;
|
}
|
|
int CompoundFile::RemoveDirectory(const char* path)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = RemoveDirectory(wpath);
|
delete[] wpath;
|
return ret;
|
}
|
|
int CompoundFile::DelTree(const char* path)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = DelTree(wpath);
|
delete[] wpath;
|
return ret;
|
}
|
|
int CompoundFile::MakeFile(const char* path)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = MakeFile(wpath);
|
delete[] wpath;
|
return ret;
|
}
|
|
int CompoundFile::RemoveFile(const char* path)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = RemoveFile(wpath);
|
delete[] wpath;
|
return ret;
|
}
|
|
int CompoundFile::FileSize(const char* path, size_t& size)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = FileSize(wpath, size);
|
delete[] wpath;
|
return ret;
|
}
|
int CompoundFile::ReadFile(const char* path, char* data)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = ReadFile(wpath, data);
|
delete[] wpath;
|
return ret;
|
}
|
int CompoundFile::ReadFile(const char* path, vector<char>& data)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = ReadFile(wpath, data);
|
delete[] wpath;
|
return ret;
|
}
|
int CompoundFile::WriteFile(const char* path, char* data, size_t size)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = WriteFile(wpath, data, size);
|
delete[] wpath;
|
return ret;
|
}
|
int CompoundFile::WriteFile(const char* path, vector<char>& data, size_t size)
|
{
|
size_t pathLength = strlen(path);
|
wchar_t* wpath = new wchar_t[pathLength+1];
|
mbstowcs(wpath, path, pathLength);
|
wpath[pathLength] = 0;
|
int ret = WriteFile(wpath, data, size);
|
delete[] wpath;
|
return ret;
|
}
|
|
/*********************** Inaccessible General Functions ***************************/
|
void CompoundFile::IncreaseLocationReferences(vector<size_t> indices)
|
// PURPOSE: Increase block location references in header, BAT indices and properties,
|
// PURPOSE: which will be affected by the insertion of new indices contained in indices.
|
// PROMISE: Block location references which are smaller than all the new indices
|
// PROMISE: will not be affected.
|
// PROMISE: SBAT location references will not be affected.
|
// PROMISE: Changes will not be written to compound file.
|
{
|
size_t maxIndices = indices.size();
|
|
// Change BAT Array references
|
{for (size_t i=0; i<109 && header_.BATArray_[i]!=-1; ++i)
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (header_.BATArray_[i] >= indices[j] &&
|
header_.BATArray_[i] != -1) ++count;
|
}
|
header_.BATArray_[i] += count;
|
}}
|
|
// Change XBAT start block if any
|
if (header_.XBATCount_)
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (header_.XBATStart_ >= indices[j] &&
|
header_.XBATStart_ != -2) ++count;
|
}
|
header_.XBATStart_ += count;
|
}
|
|
// Change SBAT start block if any
|
if (header_.SBATCount_)
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (header_.SBATStart_ >= indices[j] &&
|
header_.SBATStart_ != -2) ++count;
|
}
|
header_.SBATStart_ += count;
|
}
|
|
// Change BAT block indices
|
size_t maxBATindices = blocksIndices_.size();
|
{for (size_t i=0; i<maxBATindices && blocksIndices_[i]!=-1; ++i)
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (blocksIndices_[i] > indices[j] &&
|
blocksIndices_[i] != -2 &&
|
blocksIndices_[i] != -3) ++count;
|
}
|
blocksIndices_[i] += count;
|
}}
|
|
// Change properties start block
|
size_t count = 0;
|
{for (size_t i=0; i<maxIndices; ++i)
|
{
|
if (header_.propertiesStart_ >= indices[i] &&
|
header_.propertiesStart_ != -2) ++count;
|
}}
|
header_.propertiesStart_ += count;
|
|
// Change individual properties start block if their size is more than 4096
|
size_t maxProperties = properties_.size();
|
if (!properties_.empty())
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (properties_[0]->startBlock_ >= indices[j] &&
|
properties_[0]->startBlock_ != -2) ++count;
|
}
|
properties_[0]->startBlock_ += count;
|
}
|
{for (size_t i=1; i<maxProperties; ++i)
|
{
|
if (properties_[i]->size_ >= 4096)
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (properties_[i]->startBlock_ >= indices[j] &&
|
properties_[i]->startBlock_ != -2) ++count;
|
}
|
properties_[i]->startBlock_ += count;
|
}
|
}}
|
}
|
|
void CompoundFile::DecreaseLocationReferences(vector<size_t> indices)
|
// PURPOSE: Decrease block location references in header, BAT indices and properties,
|
// PURPOSE: which will be affected by the deletion of indices contained in indices.
|
// PROMISE: BAT indices pointing to a deleted index will be redirected to point to
|
// PROMISE: the location where the deleted index original points to.
|
// PROMISE: Block location references which are smaller than all the new indices
|
// PROMISE: will not be affected.
|
// PROMISE: SBAT location references will not be affected.
|
// PROMISE: Changes will not be written to compound file.
|
{
|
size_t maxIndices = indices.size();
|
|
// Change BAT Array references
|
{for (size_t i=0; i<109 && header_.BATArray_[i]!=-1; ++i)
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (header_.BATArray_[i] > indices[j] &&
|
header_.BATArray_[i] != -1) ++count;
|
}
|
header_.BATArray_[i] -= count;
|
}}
|
|
// Change XBAT start block if any
|
if (header_.XBATCount_)
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (header_.XBATStart_ > indices[j] &&
|
header_.XBATStart_ != -2) ++count;
|
}
|
header_.XBATStart_ -= count;
|
}
|
|
// Change SBAT start block if any
|
if (header_.SBATCount_)
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (header_.SBATStart_ > indices[j] &&
|
header_.SBATStart_ != -2) ++count;
|
}
|
header_.SBATStart_ -= count;
|
}
|
|
// Change BAT block indices
|
// Redirect BAT indices pointing to a deleted index to point to
|
// the location where the deleted index original points to.
|
size_t maxBATindices = blocksIndices_.size();
|
{for (size_t i=0; i<maxBATindices && blocksIndices_[i]!=-1; ++i)
|
{
|
bool end;
|
do
|
{
|
end = true;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (blocksIndices_[i] == indices[j])
|
{
|
blocksIndices_[i] = blocksIndices_[indices[j]];
|
end = false;
|
break;
|
}
|
}
|
} while (!end);
|
}}
|
// Erase indices to be deleted from the block indices
|
sort (indices.begin(), indices.end(), greater<size_t>());
|
{for (size_t i=0; i<maxIndices; ++i)
|
{
|
blocksIndices_.erase(blocksIndices_.begin()+indices[i]);
|
blocksIndices_.push_back(-1);
|
}}
|
|
// Decrease block location references for affected block indices.
|
{for (size_t i=0; i<maxBATindices && blocksIndices_[i]!=-1; ++i)
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (blocksIndices_[i] > indices[j] &&
|
blocksIndices_[i] != -2 &&
|
blocksIndices_[i] != -3) ++count;
|
}
|
blocksIndices_[i] -= count;
|
}}
|
|
// Change properties start block
|
size_t count = 0;
|
{for (size_t i=0; i<maxIndices; ++i)
|
{
|
if (header_.propertiesStart_ > indices[i] &&
|
header_.propertiesStart_ != -2) ++count;
|
}}
|
header_.propertiesStart_ -= count;
|
|
size_t maxProperties = properties_.size();
|
// Change Root Entry start block
|
if (!properties_.empty())
|
{
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (properties_[0]->startBlock_ > indices[j] &&
|
properties_[0]->startBlock_ != -2) ++count;
|
}
|
properties_[0]->startBlock_ -= count;
|
}
|
{for (size_t i=1; i<maxProperties; ++i)
|
{
|
if (properties_[i]->size_ >= 4096)
|
{
|
// Change individual properties start block if their size is more than 4096
|
size_t count = 0;
|
for (size_t j=0; j<maxIndices; ++j)
|
{
|
if (properties_[i]->startBlock_ > indices[j] &&
|
properties_[i]->startBlock_ != -2) ++count;
|
}
|
properties_[i]->startBlock_ -= count;
|
}
|
}}
|
}
|
|
void CompoundFile::SplitPath(const wchar_t* path,
|
wchar_t*& parentpath,
|
wchar_t*& propertyname)
|
// PURPOSE: Get a path's parent path and its name.
|
// EXPLAIN: E.g. path = "\\Abc\\def\\ghi => parentpath = "\\Abc\\def", propertyname = "ghi".
|
// REQUIRE: Calling function is responsible for deleting the memory created for
|
// REQUIRE: parentpath and propertyname.
|
{
|
size_t pathLength = wcslen(path);
|
|
int npos;
|
for (npos=pathLength-1; npos>0; --npos)
|
{
|
if (path[npos] == L'\\') break;
|
}
|
|
if (npos != 0)
|
{
|
// Get parent path if available
|
parentpath = new wchar_t[npos+1];
|
copy (path, path+npos, parentpath);
|
parentpath[npos] = 0;
|
++npos;
|
}
|
|
// Get property name (ignore initial "\" if present)
|
if (npos==0 && pathLength > 0 && path[0] == L'\\') ++npos;
|
propertyname = new wchar_t[pathLength-npos+1];
|
copy (path+npos, path+pathLength, propertyname);
|
propertyname[pathLength-npos] = 0;
|
}
|
|
/*********************** Inaccessible Header Functions ***************************/
|
bool CompoundFile::LoadHeader()
|
// PURPOSE: Load header information for compound file.
|
// PROMISE: Return true if file header contain magic number, false if otherwise.
|
{
|
file_.Read(0, &*(block_.begin()));
|
header_.Read(&*(block_.begin()));
|
|
// Check magic number to see if it is a compound file
|
if (header_.fileType_ != 0xE11AB1A1E011CFD0LL) return false;
|
|
block_.resize(header_.bigBlockSize_); // Resize buffer block
|
file_.SetBlockSize(header_.bigBlockSize_); // Resize block array block size
|
return true;
|
}
|
|
void CompoundFile::SaveHeader()
|
// PURPOSE: Save header information for compound file.
|
{
|
header_.Write(&*(block_.begin()));
|
file_.Write(0, &*(block_.begin()));
|
}
|
|
/*********************** Inaccessible BAT Functions ***************************/
|
void CompoundFile::LoadBAT()
|
// PURPOSE: Load all block allocation table information for compound file.
|
{
|
// Read BAT indices
|
{for (size_t i=0; i<header_.BATCount_; ++i)
|
{
|
// Load blocksIndices_
|
blocksIndices_.resize(blocksIndices_.size()+128, -1);
|
file_.Read(header_.BATArray_[i]+1, &*(block_.begin()));
|
for (size_t j=0; j<128; ++j)
|
{
|
LittleEndian::Read(&*(block_.begin()), blocksIndices_[j+i*128], j*4, 4);
|
}
|
}}
|
|
// Read XBAT indices
|
{for (size_t i=0; i<header_.XBATCount_; ++i)
|
{
|
blocksIndices_.resize(blocksIndices_.size()+128, -1);
|
file_.Read(header_.XBATStart_+i+1, &*(block_.begin()));
|
for (size_t j=0; j<128; ++j)
|
{
|
LittleEndian::Read(&*(block_.begin()), blocksIndices_[j+((i+109)*128)], j*4, 4);
|
}
|
}}
|
|
// Read SBAT indices
|
{for (size_t i=0; i<header_.SBATCount_; ++i)
|
{
|
sblocksIndices_.resize(sblocksIndices_.size()+128, -1);
|
file_.Read(header_.SBATStart_+i+1, &*(block_.begin()));
|
for (size_t j=0; j<128; ++j)
|
{
|
LittleEndian::Read(&*(block_.begin()), sblocksIndices_[j+i*128], j*4, 4);
|
}
|
}}
|
}
|
|
void CompoundFile::SaveBAT()
|
// PURPOSE: Save all block allocation table information for compound file.
|
{
|
// Write BAT indices
|
{for (size_t i=0; i<header_.BATCount_; ++i)
|
{
|
for (size_t j=0; j<128; ++j)
|
{
|
LittleEndian::Write(&*(block_.begin()), blocksIndices_[j+i*128], j*4, 4);
|
}
|
file_.Write(header_.BATArray_[i]+1, &*(block_.begin()));
|
}}
|
|
// Write XBAT indices
|
{for (size_t i=0; i<header_.XBATCount_; ++i)
|
{
|
for (size_t j=0; j<128; ++j)
|
{
|
LittleEndian::Write(&*(block_.begin()), blocksIndices_[j+((i+109)*128)], j*4, 4);
|
}
|
file_.Write(header_.XBATStart_+i+1, &*(block_.begin()));
|
}}
|
|
// Write SBAT indices
|
{for (size_t i=0; i<header_.SBATCount_; ++i)
|
{
|
for (size_t j=0; j<128; ++j)
|
{
|
LittleEndian::Write(&*(block_.begin()), sblocksIndices_[j+i*128], j*4, 4);
|
}
|
file_.Write(header_.SBATStart_+i+1, &*(block_.begin()));
|
}}
|
}
|
|
size_t CompoundFile::DataSize(size_t startIndex, bool isBig)
|
// PURPOSE: Gets the total size occupied by a property, starting from startIndex.
|
// EXPLAIN: isBig is true if property uses big blocks, false if it uses small blocks.
|
// PROMISE: Returns the total size occupied by the property which is the total
|
// PROMISE: number of blocks occupied multiply by the block size.
|
{
|
vector<size_t> indices;
|
if (isBig)
|
{
|
GetBlockIndices(startIndex, indices, true);
|
return indices.size()*header_.bigBlockSize_;
|
}
|
else
|
{
|
GetBlockIndices(startIndex, indices, false);
|
return indices.size()*header_.smallBlockSize_;
|
}
|
}
|
|
size_t CompoundFile::ReadData(size_t startIndex, char* data, bool isBig)
|
// PURPOSE: Read a property's data, starting from startIndex.
|
// REQUIRE: data must be large enough to receive the property's data
|
// REQUIRE: The required data size can be obtained by using DataSize().
|
// EXPLAIN: isBig is true if property uses big blocks, false if it uses small blocks.
|
// PROMISE: Returns the total size occupied by the property which is the total
|
// PROMISE: number of blocks occupied multiply by the block size.
|
{
|
vector<size_t> indices;
|
if (isBig)
|
{
|
GetBlockIndices(startIndex, indices, true);
|
size_t maxIndices = indices.size();
|
for (size_t i=0; i<maxIndices; ++i)
|
{
|
file_.Read(indices[i]+1, data+i*header_.bigBlockSize_);
|
}
|
return maxIndices*header_.bigBlockSize_;
|
}
|
else
|
{
|
GetBlockIndices(startIndex, indices, false);
|
size_t minIndex = *min_element(indices.begin(), indices.end());
|
size_t maxIndex = *max_element(indices.begin(), indices.end());
|
size_t smallBlocksPerBigBlock = header_.bigBlockSize_ / header_.smallBlockSize_;
|
size_t minBlock = minIndex / smallBlocksPerBigBlock;
|
size_t maxBlock = maxIndex / smallBlocksPerBigBlock +
|
(maxIndex % smallBlocksPerBigBlock ? 1 : 0);
|
size_t totalBlocks = maxBlock - minBlock;
|
char* buffer = new char[DataSize(properties_[0]->startBlock_, true)];
|
ReadData(properties_[0]->startBlock_, buffer, true);
|
|
size_t maxIndices = indices.size();
|
for (size_t i=0; i<maxIndices; ++i)
|
{
|
size_t start = (indices[i] - minBlock*smallBlocksPerBigBlock)*header_.smallBlockSize_;
|
copy (buffer+start,
|
buffer+start+header_.smallBlockSize_,
|
data+i*header_.smallBlockSize_);
|
}
|
delete[] buffer;
|
return maxIndices*header_.smallBlockSize_;
|
}
|
}
|
|
size_t CompoundFile::WriteData(const char* data, size_t size, int startIndex, bool isBig)
|
// PURPOSE: Write data to a property, starting from startIndex.
|
// EXPLAIN: startIndex can be -2 if property initially has no data.
|
// EXPLAIN: isBig is true if property uses big blocks, false if it uses small blocks.
|
// PROMISE: The file's original data will be replaced by the new data.
|
// PROMISE: Returns the startIndex of new data for the property.
|
{
|
if (isBig)
|
{
|
if (size==0 && startIndex==-2) return startIndex;
|
|
// Get present indices
|
vector<size_t> indices;
|
GetBlockIndices(startIndex, indices, true);
|
size_t maxPresentBlocks = indices.size();
|
|
// Calculate how many blocks does the data need
|
size_t extraSize = size % header_.bigBlockSize_;
|
size_t maxNewBlocks = size / header_.bigBlockSize_ + (extraSize ? 1 : 0);
|
|
// Readjust indices and remove blocks if new data size is smaller than original
|
int extraBlocks = maxPresentBlocks - maxNewBlocks;
|
if (extraBlocks > 0)
|
{
|
// Place new end marker
|
if (maxNewBlocks != 0) blocksIndices_[indices[maxNewBlocks]-1] = -2;
|
else startIndex = -2;
|
|
// Get indices of blocks to delete
|
vector<size_t> indicesToRemove(extraBlocks);
|
copy (indices.begin()+maxNewBlocks, indices.end(), indicesToRemove.begin());
|
indices.erase(indices.begin()+maxNewBlocks, indices.end());
|
|
// Remove extra blocks and readjust indices
|
FreeBlocks(indicesToRemove, true);
|
}
|
|
// Write blocks into available space
|
size_t remainingFullBlocks = size / header_.bigBlockSize_;
|
size_t curIndex=0;
|
if (maxPresentBlocks != 0)
|
{
|
for (; remainingFullBlocks && curIndex<maxPresentBlocks;
|
--remainingFullBlocks, ++curIndex)
|
{
|
file_.Write(indices[curIndex]+1, data+curIndex*header_.bigBlockSize_);
|
}
|
}
|
|
// Check if all blocks have been written
|
size_t index;
|
if (indices.empty()) index = 0;
|
else if (curIndex == 0) index = indices[0];
|
else index = (startIndex != -2) ? indices[curIndex-1] : 0;
|
if (remainingFullBlocks != 0)
|
{
|
// Require extra blocks to write data (i.e. new data is larger than original data
|
do
|
{
|
size_t newIndex = GetFreeBlockIndex(true); // Get new free block to write data
|
if (startIndex == -2) startIndex = newIndex; // Get start index
|
else LinkBlocks(index, newIndex, true); // Link last index to new index
|
file_.Write(newIndex+1, data+curIndex*header_.bigBlockSize_);
|
++curIndex;
|
index = newIndex;
|
} while (--remainingFullBlocks);
|
}
|
|
if (extraSize != 0)
|
{
|
size_t newIndex;
|
if (curIndex >= maxPresentBlocks)
|
{
|
// No more free blocks to write extra block data
|
newIndex = GetFreeBlockIndex(true); // Get new free block to write data
|
if (startIndex == -2) startIndex = newIndex;
|
else LinkBlocks(index, newIndex,true);
|
}
|
else newIndex = indices[curIndex];
|
|
// Write extra block after increasing its size to the minimum block size
|
vector<char> tempdata(header_.bigBlockSize_, 0);
|
copy (data+curIndex*header_.bigBlockSize_, data+curIndex*header_.bigBlockSize_+extraSize, tempdata.begin());
|
file_.Write(newIndex+1, &*(tempdata.begin()));
|
}
|
return startIndex;
|
}
|
else
|
{
|
if (size==0 && startIndex==-2) return startIndex;
|
|
if (size != 0 && properties_[0]->startBlock_ == -2)
|
{
|
size_t newIndex = GetFreeBlockIndex(true);
|
fill (block_.begin(), block_.end(), 0);
|
file_.Insert(newIndex, &*(block_.begin()));
|
IncreaseLocationReferences(vector<size_t>(1, newIndex));
|
properties_[0]->startBlock_ = newIndex;
|
properties_[0]->size_ = header_.bigBlockSize_;
|
}
|
|
// Get present indices
|
vector<size_t> indices;
|
GetBlockIndices(startIndex, indices, false);
|
size_t maxPresentBlocks = indices.size();
|
|
// Calculate how many blocks does the data need
|
size_t extraSize = size % header_.smallBlockSize_;
|
size_t maxNewBlocks = size / header_.smallBlockSize_ + (extraSize ? 1 : 0);
|
|
vector<char> smallBlocksData;
|
int extraBlocks = maxPresentBlocks - maxNewBlocks;
|
if (extraBlocks > 0)
|
{
|
// Readjust indices and remove blocks
|
// Place new end marker
|
if (maxNewBlocks != 0) sblocksIndices_[indices[maxNewBlocks]-1] = -2;
|
else startIndex = -2;
|
|
// Get indices of blocks to delete
|
vector<size_t> indicesToRemove(extraBlocks);
|
copy (indices.begin()+maxNewBlocks, indices.end(), indicesToRemove.begin());
|
indices.erase(indices.begin()+maxNewBlocks, indices.end());
|
|
// Remove extra blocks and readjust indices
|
FreeBlocks(indicesToRemove, false);
|
}
|
else if (extraBlocks < 0)
|
{
|
size_t maxBlocks = properties_[0]->size_ / header_.bigBlockSize_ +
|
(properties_[0]->size_ % header_.bigBlockSize_ ? 1 : 0);
|
size_t actualSize = maxBlocks * header_.bigBlockSize_;
|
smallBlocksData.resize(actualSize);
|
ReadData(properties_[0]->startBlock_, &*(smallBlocksData.begin()), true);
|
smallBlocksData.resize(properties_[0]->size_);
|
|
// Readjust indices and add blocks
|
size_t newBlocksNeeded = -extraBlocks;
|
size_t index = maxPresentBlocks - 1;
|
for (size_t i=0; i<newBlocksNeeded; ++i)
|
{
|
size_t newIndex = GetFreeBlockIndex(false); // Get new free block to write data
|
if (startIndex == -2) startIndex = newIndex; // Get start index
|
else LinkBlocks(index, newIndex, false); // Link last index to new index
|
smallBlocksData.insert(smallBlocksData.begin()+newIndex,
|
header_.smallBlockSize_, 0);
|
index = newIndex;
|
}
|
properties_[0]->size_ = newBlocksNeeded * header_.smallBlockSize_;
|
}
|
if (smallBlocksData.empty())
|
{
|
size_t maxBlocks = properties_[0]->size_ / header_.bigBlockSize_ +
|
(properties_[0]->size_ % header_.bigBlockSize_ ? 1 : 0);
|
size_t actualSize = maxBlocks * header_.bigBlockSize_;
|
smallBlocksData.resize(actualSize);
|
ReadData(properties_[0]->startBlock_, &*(smallBlocksData.begin()), true);
|
smallBlocksData.resize(properties_[0]->size_);
|
}
|
|
// Write blocks
|
GetBlockIndices(startIndex, indices, false);
|
size_t fullBlocks = size / header_.smallBlockSize_;
|
for (size_t i=0; i<fullBlocks; ++i)
|
{
|
copy (data+i*header_.smallBlockSize_,
|
data+i*header_.smallBlockSize_+header_.smallBlockSize_,
|
smallBlocksData.begin()+indices[i]*header_.smallBlockSize_);
|
}
|
if (extraSize != 0)
|
{
|
copy (data+fullBlocks*header_.smallBlockSize_,
|
data+fullBlocks*header_.smallBlockSize_+extraSize,
|
smallBlocksData.begin()+indices[fullBlocks]*header_.smallBlockSize_);
|
}
|
WriteData(&*(smallBlocksData.begin()), properties_[0]->size_,
|
properties_[0]->startBlock_, true);
|
return startIndex;
|
}
|
}
|
|
void CompoundFile::GetBlockIndices(size_t startIndex, vector<size_t>& indices, bool isBig)
|
// PURPOSE: Get the indices of blocks where data are stored, starting from startIndex.
|
// EXPLAIN: isBig is true if property uses big blocks, false if it uses small blocks.
|
{
|
indices.clear();
|
if (isBig)
|
{
|
for (size_t i=startIndex; i!=-2; i=blocksIndices_[i]) indices.push_back(i);
|
}
|
else
|
{
|
for (size_t i=startIndex; i!=-2; i=sblocksIndices_[i]) indices.push_back(i);
|
}
|
}
|
|
size_t CompoundFile::GetFreeBlockIndex(bool isBig)
|
// PURPOSE: Get the index of a new block where data can be stored.
|
// EXPLAIN: isBig is true if property uses big blocks, false if it uses small blocks.
|
// PROMISE: It does not physically create a new block in the compound file.
|
// PROMISE: It only adjust BAT arrays and indices or SBAT arrays and indices so that
|
// PROMISE: it gives the index of a new block where data can be inserted.
|
{
|
size_t index;
|
if (isBig)
|
{
|
// Find first free location
|
index = distance(blocksIndices_.begin(),
|
find(blocksIndices_.begin(),
|
blocksIndices_.end(), -1));
|
if (index == blocksIndices_.size())
|
{
|
ExpandBATArray(true);
|
index = distance(blocksIndices_.begin(),
|
find(blocksIndices_.begin(),
|
blocksIndices_.end(), -1));
|
}
|
blocksIndices_[index] = -2;
|
}
|
else
|
{
|
// Find first free location
|
index = distance(sblocksIndices_.begin(),
|
find(sblocksIndices_.begin(),
|
sblocksIndices_.end(), -1));
|
if (index == sblocksIndices_.size())
|
{
|
ExpandBATArray(false);
|
index = distance(sblocksIndices_.begin(),
|
find(sblocksIndices_.begin(),
|
sblocksIndices_.end(), -1));
|
}
|
sblocksIndices_[index] = -2;
|
}
|
return index;
|
}
|
|
void CompoundFile::ExpandBATArray(bool isBig)
|
// PURPOSE: Create a new block of BAT or SBAT indices.
|
// EXPLAIN: isBig is true if property uses big blocks, false if it uses small blocks.
|
{
|
size_t newIndex;
|
fill (block_.begin(), block_.end(), -1);
|
|
if (isBig)
|
{
|
size_t BATindex = distance(&header_.BATArray_[0],
|
find(header_.BATArray_,
|
header_.BATArray_+109, -1));
|
if (BATindex < 109)
|
{
|
// Set new BAT index location
|
newIndex = blocksIndices_.size(); // New index location
|
file_.Insert(newIndex+1, &*(block_.begin()));
|
IncreaseLocationReferences(vector<size_t>(1, newIndex));
|
|
// Update BAT array
|
header_.BATArray_[BATindex] = newIndex;
|
++header_.BATCount_;
|
}
|
else
|
{
|
// No free BAT indices. Increment using XBAT
|
// Set new XBAT index location
|
if (header_.XBATCount_ != 0)
|
{
|
newIndex = header_.XBATStart_ + header_.XBATCount_;
|
file_.Insert(newIndex, &*(block_.begin()));
|
IncreaseLocationReferences(vector<size_t>(1, newIndex));
|
}
|
else
|
{
|
newIndex = blocksIndices_.size();
|
file_.Insert(newIndex, &*(block_.begin()));
|
IncreaseLocationReferences(vector<size_t>(1, newIndex));
|
header_.XBATStart_ = newIndex;
|
}
|
++header_.XBATCount_;
|
}
|
blocksIndices_.insert(blocksIndices_.begin()+newIndex, -3);
|
blocksIndices_.resize(blocksIndices_.size()+127, -1);
|
}
|
else
|
{
|
// Set new SBAT index location
|
if (header_.SBATCount_ != 0)
|
{
|
newIndex = header_.SBATStart_ + header_.SBATCount_;
|
file_.Insert(newIndex, &*(block_.begin()));
|
IncreaseLocationReferences(vector<size_t>(1, newIndex));
|
}
|
else
|
{
|
newIndex = GetFreeBlockIndex(true);
|
file_.Insert(newIndex, &*(block_.begin()));
|
IncreaseLocationReferences(vector<size_t>(1, newIndex));
|
header_.SBATStart_ = newIndex;
|
}
|
++header_.SBATCount_;
|
sblocksIndices_.resize(sblocksIndices_.size()+128, -1);
|
}
|
}
|
|
void CompoundFile::LinkBlocks(size_t from, size_t to, bool isBig)
|
// PURPOSE: Link one BAT index to another.
|
// EXPLAIN: isBig is true if property uses big blocks, false if it uses small blocks.
|
{
|
if (isBig) blocksIndices_[from] = to;
|
else sblocksIndices_[from] = to;
|
}
|
|
void CompoundFile::FreeBlocks(vector<size_t>& indices, bool isBig)
|
// PURPOSE: Delete blocks of data from compound file.
|
// EXPLAIN: indices contains indices to blocks of data to be deleted.
|
// EXPLAIN: isBig is true if property uses big blocks, false if it uses small blocks.
|
{
|
if (isBig)
|
{
|
// Decrease all location references before deleting blocks from file.
|
DecreaseLocationReferences(indices);
|
size_t maxIndices = indices.size();
|
{for (size_t i=0; i<maxIndices; ++i) ++indices[i];} // Increase by 1 because Block index 1 corresponds to index 0 here
|
file_.Erase(indices);
|
|
// Shrink BAT indices if necessary
|
vector<size_t> indicesToRemove;
|
while (distance(find(blocksIndices_.begin(),
|
blocksIndices_.end(),-1),
|
blocksIndices_.end()) >= 128)
|
{
|
blocksIndices_.resize(blocksIndices_.size()-128);
|
if (header_.XBATCount_ != 0)
|
{
|
// Shrink XBAT first
|
--header_.XBATCount_;
|
indicesToRemove.push_back(header_.XBATStart_+header_.XBATCount_+1); // Add 1 because Block index 1 corresponds to index 0 here
|
if (header_.XBATCount_ == 0) header_.XBATStart_ = -2;
|
}
|
else
|
{
|
// No XBAT, delete last occupied BAT array element
|
size_t BATindex = distance(&header_.BATArray_[0],
|
find(header_.BATArray_,
|
header_.BATArray_+109, -1));
|
if (BATindex != 109)
|
{
|
--header_.BATCount_;
|
indicesToRemove.push_back(header_.BATArray_[BATindex-1]+1); // Add 1 because Block index 1 corresponds to index 0 here
|
header_.BATArray_[BATindex-1] = -1;
|
}
|
}
|
}
|
// Erase extra BAT indices if present
|
if (!indicesToRemove.empty()) file_.Erase(indicesToRemove);
|
}
|
else
|
{
|
// Erase block
|
size_t maxIndices = indices.size();
|
size_t maxBlocks = properties_[0]->size_ / header_.bigBlockSize_ +
|
(properties_[0]->size_ % header_.bigBlockSize_ ? 1 : 0);
|
size_t size = maxBlocks * header_.bigBlockSize_;
|
char* data = new char[size];
|
ReadData(properties_[0]->startBlock_, data, true);
|
size_t maxSmallBlocks = properties_[0]->size_ / header_.smallBlockSize_;
|
char* newdata = new char[properties_[0]->size_-maxIndices*header_.smallBlockSize_];
|
{for (size_t i=0, j=0; i<maxSmallBlocks; ++i)
|
{
|
if (find(indices.begin(), indices.end(), i) == indices.end())
|
{
|
copy (data+i*header_.smallBlockSize_,
|
data+i*header_.smallBlockSize_+header_.smallBlockSize_,
|
newdata+j*header_.smallBlockSize_);
|
++j;
|
}
|
}}
|
properties_[0]->startBlock_ = WriteData(newdata, properties_[0]->size_-maxIndices*header_.smallBlockSize_,
|
properties_[0]->startBlock_, true);
|
properties_[0]->size_ -= maxIndices*header_.smallBlockSize_;
|
delete[] data;
|
delete[] newdata;
|
|
// Change SBAT indices
|
size_t maxSBATindices = sblocksIndices_.size();
|
{for (size_t i=0; i<maxIndices; ++i)
|
{
|
for (size_t j=0; j<maxSBATindices; ++j)
|
{
|
if (j == indices[i]) continue;
|
if (sblocksIndices_[j] == indices[i]) sblocksIndices_[j] = sblocksIndices_[indices[i]];
|
if (sblocksIndices_[j] > indices[i] &&
|
sblocksIndices_[j] != -1 &&
|
sblocksIndices_[j] != -2) --sblocksIndices_[j];
|
}
|
}}
|
sort (indices.begin(), indices.end(), greater<size_t>());
|
{for (size_t i=0; i<maxIndices; ++i)
|
{
|
sblocksIndices_.erase(sblocksIndices_.begin()+indices[i]);
|
sblocksIndices_.push_back(-1);
|
}}
|
vector<size_t> indicesToRemove;
|
while (distance(find(sblocksIndices_.begin(),
|
sblocksIndices_.end(),-1),
|
sblocksIndices_.end()) >= 128)
|
{
|
// Shrink SBAT indices if necessary
|
sblocksIndices_.resize(sblocksIndices_.size()-128);
|
--header_.SBATCount_;
|
indicesToRemove.push_back(header_.SBATStart_+header_.SBATCount_);
|
if (header_.SBATCount_ == 0) header_.SBATStart_ = -2;
|
}
|
FreeBlocks(indicesToRemove, true);
|
}
|
}
|
|
/*********************** Inaccessible Properties Functions ***************************/
|
void CompoundFile::LoadProperties()
|
// PURPOSE: Load properties information for compound file.
|
{
|
// Read properties' data from compound file.
|
size_t propertiesSize = DataSize(header_.propertiesStart_, true);
|
char* buffer = new char[propertiesSize];
|
ReadData(header_.propertiesStart_, buffer, true);
|
|
// Split properties' data into individual property.
|
size_t maxPropertiesBlock = propertiesSize / header_.bigBlockSize_;
|
size_t propertiesPerBlock = header_.bigBlockSize_ / 128;
|
size_t maxProperties = maxPropertiesBlock * propertiesPerBlock;
|
size_t maxBlocks = maxProperties / propertiesPerBlock +
|
(maxProperties % propertiesPerBlock ? 1 : 0);
|
|
for (size_t i=0; i<maxBlocks; ++i)
|
{
|
for (size_t j=0; j<4; ++j)
|
{
|
// Read individual property
|
Property* property = new Property;
|
property->Read(buffer+i*512+j*128);
|
if (wcslen(property->name_) == 0)
|
{
|
delete property;
|
break;
|
}
|
properties_.push_back(property);
|
}
|
}
|
delete[] buffer;
|
|
// Generate property trees
|
propertyTrees_->parent_ = 0;
|
propertyTrees_->self_ = properties_[0];
|
propertyTrees_->index_ = 0;
|
|
InsertPropertyTree(propertyTrees_,
|
properties_[properties_[0]->childProp_],
|
properties_[0]->childProp_);
|
}
|
|
void CompoundFile::SaveProperties()
|
// PURPOSE: Save properties information for compound file.
|
{
|
// Calculate total size required by properties
|
size_t maxProperties = properties_.size();
|
size_t propertiesPerBlock = header_.bigBlockSize_ / 128;
|
size_t maxBlocks = maxProperties / propertiesPerBlock +
|
(maxProperties % propertiesPerBlock ? 1 : 0);
|
size_t propertiesSize = maxBlocks*header_.bigBlockSize_;
|
char* buffer = new char[propertiesSize];
|
{for (size_t i=0; i<propertiesSize; ++i) buffer[i] = 0;}
|
{for (size_t i=0; i<maxProperties; ++i)
|
{
|
// Save individual property
|
properties_[i]->Write(buffer+i*128);
|
}}
|
|
// Write properties' data to compound file.
|
WriteData(buffer, propertiesSize, header_.propertiesStart_, true);
|
delete[] buffer;
|
}
|
|
int CompoundFile::MakeProperty(const wchar_t* path, CompoundFile::Property* property)
|
// PURPOSE: Create a new property in the compound file.
|
// EXPLAIN: path is the full path name for the property.
|
// EXPLAIN: property contains information on the type of property to be created.
|
{
|
wchar_t* parentpath = 0;
|
wchar_t* propertyname = 0;
|
|
// Change to the specified directory. If specified directory is not present,
|
// create it.
|
if (wcslen(path) != 0)
|
{
|
if (path[0] == L'\\') currentDirectory_ = propertyTrees_;
|
}
|
SplitPath(path, parentpath, propertyname);
|
|
if (propertyname != 0)
|
{
|
if (parentpath != 0)
|
{
|
if (ChangeDirectory(parentpath) != SUCCESS)
|
{
|
int ret = MakeDirectory(parentpath);
|
if (ret != SUCCESS)
|
{
|
delete[] parentpath;
|
delete[] propertyname;
|
return ret;
|
}
|
else ChangeDirectory(parentpath);
|
}
|
delete[] parentpath;
|
}
|
|
// Insert property into specified directory
|
size_t propertynameLength = wcslen(propertyname);
|
if (propertynameLength >= 32)
|
{
|
delete[] propertyname;
|
return NAME_TOO_LONG;
|
}
|
wcscpy(property->name_, propertyname);
|
delete[] propertyname;
|
property->nameSize_ = propertynameLength*2+2;
|
if (FindProperty(currentDirectory_, property->name_) == 0)
|
{
|
// Find location to insert property
|
size_t maxProperties = properties_.size();
|
size_t index;
|
for (index=1; index<maxProperties; ++index)
|
{
|
if (*(properties_[index]) > *property) break;
|
}
|
if (index != maxProperties)
|
{
|
// Change references for all properties affected by the new property
|
IncreasePropertyReferences(propertyTrees_, index);
|
}
|
properties_.insert(properties_.begin()+index, property);
|
InsertPropertyTree(currentDirectory_, property, index);
|
return SUCCESS;
|
}
|
else return DUPLICATE_PROPERTY;
|
}
|
else
|
{
|
if (parentpath != 0) delete[] parentpath;
|
return INVALID_PATH;
|
}
|
}
|
|
CompoundFile::PropertyTree* CompoundFile::FindProperty(size_t index)
|
// PURPOSE: Find property in the compound file, given the index of the property.
|
// PROMISE: Returns a pointer to the property tree of the property if property
|
// PROMISE: is present, 0 if otherwise.
|
{
|
if (previousDirectories_.empty()) previousDirectories_.push_back(propertyTrees_);
|
PropertyTree* currentTree = previousDirectories_.back();
|
if (currentTree->index_ != index)
|
{
|
size_t maxChildren = currentTree->children_.size();
|
for (size_t i=0; i<maxChildren; ++i)
|
{
|
previousDirectories_.push_back(currentTree->children_[i]);
|
PropertyTree* child = FindProperty(index);
|
if (child != 0)
|
{
|
previousDirectories_.pop_back();
|
return child;
|
}
|
}
|
}
|
else
|
{
|
previousDirectories_.pop_back();
|
return currentTree;
|
}
|
previousDirectories_.pop_back();
|
return 0;
|
}
|
|
CompoundFile::PropertyTree* CompoundFile::FindProperty(const wchar_t* path)
|
// PURPOSE: Find property in the compound file, given the path of the property.
|
// PROMISE: Returns a pointer to the property tree of the property if property
|
// PROMISE: is present, 0 if otherwise.
|
{
|
previousDirectories_.push_back(currentDirectory_);
|
|
// Change to specified directory
|
wchar_t* parentpath = 0;
|
wchar_t* filename = 0;
|
|
if (wcslen(path) != 0)
|
{
|
if (path[0] == L'\\') currentDirectory_ = propertyTrees_;
|
}
|
|
SplitPath(path, parentpath, filename);
|
if (parentpath != 0)
|
{
|
int ret = ChangeDirectory(parentpath);
|
delete[] parentpath;
|
if (ret != SUCCESS)
|
{
|
// Cannot change to specified directory
|
if (filename != 0) delete[] filename;
|
currentDirectory_ = previousDirectories_.back();
|
previousDirectories_.pop_back();
|
PropertyTree* property = 0;
|
return property;
|
}
|
}
|
|
// Check to see if file is present in the specified directory.
|
PropertyTree* property = 0;
|
if (filename != 0)
|
{
|
property = FindProperty(currentDirectory_, filename);
|
delete[] filename;
|
}
|
currentDirectory_ = previousDirectories_.back();
|
previousDirectories_.pop_back();
|
return property;
|
}
|
|
CompoundFile::PropertyTree*
|
CompoundFile::FindProperty(CompoundFile::PropertyTree* parentTree,
|
wchar_t* name)
|
// PURPOSE: Find property in the compound file, given the parent property tree and its name.
|
// PROMISE: Returns a pointer to the property tree of the property if property
|
// PROMISE: is present, 0 if otherwise.
|
{
|
if (parentTree->self_->childProp_ != -1)
|
{
|
size_t maxChildren = parentTree->children_.size();
|
for (size_t i=0; i<maxChildren; ++i)
|
{
|
if (wcscmp(parentTree->children_[i]->self_->name_, name) == 0)
|
{
|
return parentTree->children_[i];
|
}
|
}
|
}
|
return 0;
|
}
|
|
void CompoundFile::InsertPropertyTree(CompoundFile::PropertyTree* parentTree,
|
CompoundFile::Property* property,
|
size_t index)
|
// PURPOSE: Insert a property and all its siblings and children into the property tree.
|
// REQUIRE: If the property is a new property and its index is already occupied by
|
// REQUIRE: another property, the calling function has to call IncreasePropertyReferences()
|
// REQUIRE: first before calling this function.
|
// EXPLAIN: This function is used by LoadProperty() to initialize the property trees
|
// EXPLAIN: and MakeProperty() thus resulting in the above requirements.
|
// EXPLAIN: parentTree is the parent of the new property.
|
// EXPLAIN: property is the property to be added.
|
// EXPLAIN: index is the index of the new property.
|
// PROMISE: The property will be added as the parent tree's child and the parent's
|
// PROMISE: child property and all the its children previous property and next property
|
// PROMISE: will be readjusted to accomodate the next property.
|
{
|
PropertyTree* tree = new PropertyTree;
|
tree->parent_ = parentTree;
|
tree->self_ = property;
|
tree->index_ = index;
|
|
if (property->previousProp_ != -1)
|
{
|
InsertPropertyTree(parentTree,
|
properties_[property->previousProp_],
|
property->previousProp_);
|
}
|
|
if (property->nextProp_ != -1)
|
{
|
InsertPropertyTree(parentTree,
|
properties_[property->nextProp_],
|
property->nextProp_);
|
}
|
|
if (property->childProp_ != -1)
|
{
|
InsertPropertyTree(tree,
|
properties_[property->childProp_],
|
property->childProp_);
|
}
|
|
// Sort children
|
size_t maxChildren = parentTree->children_.size();
|
size_t i;
|
for (i=0; i<maxChildren; ++i)
|
{
|
if (index < parentTree->children_[i]->index_) break;
|
}
|
parentTree->children_.insert(parentTree->children_.begin()+i, tree);
|
|
// Update children indices
|
UpdateChildrenIndices(parentTree);
|
}
|
|
void CompoundFile::DeletePropertyTree(CompoundFile::PropertyTree* tree)
|
// PURPOSE: Delete a property from properties.
|
// EXPLAIN: tree is the property tree to be deleted.
|
// PROMISE: The tree's parent's child property and all the its children previous property
|
// PROMISE: and next property will be readjusted to accomodate the deleted property.
|
{
|
// Decrease all property references
|
DecreasePropertyReferences(propertyTrees_, tree->index_);
|
|
// Remove property
|
if (properties_[tree->index_]) delete properties_[tree->index_];
|
properties_.erase(properties_.begin()+tree->index_);
|
|
// Remove property from property trees
|
size_t maxChildren = tree->parent_->children_.size();
|
size_t i;
|
for (i=0; i<maxChildren; ++i)
|
{
|
if (tree->parent_->children_[i]->index_ == tree->index_) break;
|
}
|
tree->parent_->children_.erase(tree->parent_->children_.begin()+i);
|
|
// Update children indices
|
UpdateChildrenIndices(tree->parent_);
|
}
|
|
void CompoundFile::UpdateChildrenIndices(CompoundFile::PropertyTree* parentTree)
|
{
|
// Update indices for 1st to middle child
|
size_t maxChildren = parentTree->children_.size();
|
if (maxChildren != 0)
|
{
|
vector<PropertyTree*>& children = parentTree->children_;
|
size_t prevChild = 0;
|
children[0]->self_->previousProp_ = -1;
|
children[0]->self_->nextProp_ = -1;
|
size_t curChild;
|
for (curChild=1; curChild<=maxChildren/2; ++curChild)
|
{
|
children[curChild]->self_->previousProp_ = children[prevChild]->index_;
|
children[curChild]->self_->nextProp_ = -1;
|
prevChild = curChild;
|
}
|
|
// Update middle child
|
--curChild;
|
children[curChild]->parent_->self_->childProp_ = children[curChild]->index_;
|
|
// Update from middle to last child
|
size_t nextChild = curChild + 1;
|
if (nextChild < maxChildren)
|
{
|
children[curChild]->self_->nextProp_ = children[nextChild]->index_;
|
for (++curChild, ++nextChild;
|
nextChild<maxChildren;
|
++curChild, ++nextChild)
|
{
|
children[curChild]->self_->previousProp_ = -1;
|
children[curChild]->self_->nextProp_ = children[nextChild]->index_;
|
|
}
|
children[curChild]->self_->previousProp_ = -1;
|
children[curChild]->self_->nextProp_ = -1;
|
}
|
}
|
else
|
{
|
parentTree->self_->childProp_ = -1;
|
}
|
}
|
|
void CompoundFile::IncreasePropertyReferences(CompoundFile::PropertyTree* parentTree,
|
size_t index)
|
// PURPOSE: Increase all property references (previous property, next property
|
// PURPOSE: and child property) which will be affected by the insertion of the new index.
|
// EXPLAIN: The recursive method of going through each property tree is used instead of
|
// EXPLAIN: using the iterative method of going through each property in properties_ is
|
// EXPLAIN: because the index in property tree needs to be updated also.
|
{
|
if (parentTree->index_ >= index) ++parentTree->index_;
|
if (parentTree->self_->previousProp_ != -1)
|
{
|
if (parentTree->self_->previousProp_ >= index)
|
{
|
++parentTree->self_->previousProp_;
|
}
|
}
|
if (parentTree->self_->nextProp_!= -1)
|
{
|
if (parentTree->self_->nextProp_ >= index)
|
{
|
++parentTree->self_->nextProp_;
|
}
|
}
|
if (parentTree->self_->childProp_ != -1)
|
{
|
if (parentTree->self_->childProp_ >= index)
|
{
|
++parentTree->self_->childProp_;
|
}
|
}
|
|
size_t maxChildren = parentTree->children_.size();
|
for (size_t i=0; i<maxChildren; ++i)
|
{
|
IncreasePropertyReferences(parentTree->children_[i], index);
|
}
|
}
|
|
void CompoundFile::DecreasePropertyReferences(CompoundFile::PropertyTree* parentTree, size_t index)
|
// PURPOSE: Decrease all property references (previous property, next property
|
// PURPOSE: and child property) which will be affected by the deletion of the index.
|
// EXPLAIN: The recursive method of going through each property tree is used instead of
|
// EXPLAIN: using the iterative method of going through each property in properties_ is
|
// EXPLAIN: because the index in property tree needs to be updated also.
|
{
|
if (parentTree->index_ > index) --parentTree->index_;
|
if (parentTree->self_->previousProp_ != -1)
|
{
|
if (parentTree->self_->previousProp_ > index)
|
{
|
--parentTree->self_->previousProp_;
|
}
|
}
|
if (parentTree->self_->nextProp_!= -1)
|
{
|
if (parentTree->self_->nextProp_ > index)
|
{
|
--parentTree->self_->nextProp_;
|
}
|
}
|
if (parentTree->self_->childProp_ != -1)
|
{
|
if (parentTree->self_->childProp_ > index)
|
{
|
--parentTree->self_->childProp_;
|
}
|
}
|
|
size_t maxChildren = parentTree->children_.size();
|
for (size_t i=0; i<maxChildren; ++i)
|
{
|
DecreasePropertyReferences(parentTree->children_[i], index);
|
}
|
}
|
} // YCompoundFiles namespace end
|
|
namespace YExcel
|
{
|
using namespace YCompoundFiles;
|
/************************************************************************************************************/
|
Record::Record() : dataSize_(0), recordSize_(4) {};
|
Record::~Record() {};
|
size_t Record::Read(const char* data)
|
{
|
LittleEndian::Read(data, code_, 0, 2); // Read operation code.
|
LittleEndian::Read(data, dataSize_, 2, 2); // Read size of record.
|
data_.assign(data+4, data+4+dataSize_);
|
|
recordSize_ = 4 + dataSize_;
|
|
// Check if next record is a continue record
|
continueIndices_.clear();
|
short code;
|
LittleEndian::Read(data, code, dataSize_+4, 2);
|
while (code == CODE::CONTINUE)
|
{
|
continueIndices_.push_back(dataSize_);
|
|
size_t size;
|
LittleEndian::Read(data, size, recordSize_+2, 2);
|
data_.insert(data_.end(), data+recordSize_+4, data+recordSize_+4+size);
|
dataSize_ += size;
|
recordSize_ += 4 + size;
|
|
LittleEndian::Read(data, code, recordSize_, 2);
|
};
|
return recordSize_;
|
}
|
size_t Record::Write(char* data)
|
{
|
LittleEndian::Write(data, code_, 0, 2); // Write operation code.
|
size_t npos = 2;
|
|
if (continueIndices_.empty())
|
{
|
size_t size = dataSize_;
|
size_t i=0;
|
while (size > 8224)
|
{
|
LittleEndian::Write(data, 8224, npos, 2); // Write size of record.
|
npos += 2;
|
size -= 8224;
|
copy (data_.begin()+i*8224, data_.begin()+(i+1)*8224, data+npos);
|
npos += 8224;
|
|
if (size != 0)
|
{
|
++i;
|
LittleEndian::Write(data, 0x3C, npos, 2); // Write CONTINUE code.
|
npos += 2;
|
}
|
}
|
|
LittleEndian::Write(data, size, npos, 2); // Write size of record.
|
npos += 2;
|
copy (data_.begin()+i*8224, data_.begin()+i*8224+size, data+npos);
|
npos += size;
|
}
|
else
|
{
|
size_t maxContinue = continueIndices_.size();
|
size_t size = continueIndices_[0];
|
LittleEndian::Write(data, size, npos, 2); // Write size of record
|
npos += 2;
|
copy (data_.begin(), data_.begin()+size, data+npos);
|
npos += size;
|
size_t c=0;
|
for (c=1; c<maxContinue; ++c)
|
{
|
LittleEndian::Write(data, 0x3C, npos, 2); // Write CONTINUE code.
|
npos += 2;
|
size = continueIndices_[c] - continueIndices_[c-1];
|
LittleEndian::Write(data, size, npos, 2);
|
npos += 2;
|
copy (data_.begin()+continueIndices_[c-1],
|
data_.begin()+continueIndices_[c],
|
data+npos);
|
npos += size;
|
}
|
LittleEndian::Write(data, 0x3C, npos, 2); // Write CONTINUE code.
|
npos += 2;
|
size = data_.size() - continueIndices_[c-1];
|
LittleEndian::Write(data, size, npos, 2);
|
npos += 2;
|
copy (data_.begin()+continueIndices_[c-1],
|
data_.end(),
|
data+npos);
|
npos += size;
|
}
|
return npos;
|
}
|
size_t Record::DataSize() {return dataSize_;}
|
size_t Record::RecordSize() {return recordSize_;}
|
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
BOF::BOF() : Record() {code_ = CODE::BOF; dataSize_ = 16; recordSize_ = 20;}
|
size_t BOF::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, version_, 0, 2);
|
LittleEndian::Read(data_, type_, 2, 2);
|
LittleEndian::Read(data_, buildIdentifier_, 4, 2);
|
LittleEndian::Read(data_, buildYear_, 6, 2);
|
LittleEndian::Read(data_, fileHistoryFlags_, 8, 4);
|
LittleEndian::Read(data_, lowestExcelVersion_, 12, 4);
|
return RecordSize();
|
}
|
size_t BOF::Write(char* data)
|
{
|
data_.resize(dataSize_);
|
LittleEndian::Write(data_, version_, 0, 2);
|
LittleEndian::Write(data_, type_, 2, 2);
|
LittleEndian::Write(data_, buildIdentifier_, 4, 2);
|
LittleEndian::Write(data_, buildYear_, 6, 2);
|
LittleEndian::Write(data_, fileHistoryFlags_, 8, 4);
|
LittleEndian::Write(data_, lowestExcelVersion_, 12, 4);
|
return Record::Write(data);
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
YEOF::YEOF() : Record() {code_ = CODE::YEOF; dataSize_ = 0; recordSize_ = 4;}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
SmallString::SmallString() : name_(0), wname_(0) {};
|
SmallString::~SmallString() {Reset();}
|
SmallString::SmallString(const SmallString& s) :
|
name_(0), wname_(0), unicode_(s.unicode_)
|
{
|
if (s.name_)
|
{
|
size_t len = strlen(s.name_);
|
name_ = new char[len+1];
|
strcpy(name_, s.name_);
|
}
|
if (s.wname_)
|
{
|
size_t len = wcslen(s.wname_);
|
wname_ = new wchar_t[len+1];
|
wcscpy(wname_, s.wname_);
|
}
|
}
|
SmallString& SmallString::operator=(const SmallString& s)
|
{
|
Reset();
|
unicode_ = s.unicode_;
|
if (s.name_)
|
{
|
size_t len = strlen(s.name_);
|
name_ = new char[len+1];
|
strcpy(name_, s.name_);
|
}
|
if (s.wname_)
|
{
|
size_t len = wcslen(s.wname_);
|
wname_ = new wchar_t[len+1];
|
wcscpy(wname_, s.wname_);
|
}
|
return *this;
|
}
|
const SmallString& SmallString::operator=(const char* str)
|
{
|
unicode_ = 0;
|
Reset();
|
size_t len = strlen(str);
|
name_ = new char[len+1];
|
strcpy(name_, str);
|
return *this;
|
}
|
const SmallString& SmallString::operator=(const wchar_t* str)
|
{
|
unicode_ = 1;
|
Reset();
|
size_t len = wcslen(str);
|
wname_ = new wchar_t[len+1];
|
wcscpy(wname_, str);
|
return *this;
|
}
|
void SmallString::Reset()
|
{
|
if (name_) {delete[] name_; name_ = 0;}
|
if (wname_) {delete[] wname_; wname_ = 0;}
|
}
|
size_t SmallString::Read(const char* data)
|
{
|
Reset();
|
char stringSize;
|
LittleEndian::Read(data, stringSize, 0, 1);
|
LittleEndian::Read(data, unicode_, 1, 1);
|
size_t bytesRead = 2;
|
if (unicode_ == 0)
|
{
|
// ANSI string
|
name_ = new char[stringSize+1];
|
LittleEndian::ReadString(data, name_, 2, stringSize);
|
name_[stringSize] = 0;
|
bytesRead += stringSize;
|
}
|
else
|
{
|
// UNICODE
|
wname_ = new wchar_t[stringSize+1];
|
LittleEndian::ReadString(data, wname_, 2, stringSize);
|
wname_[stringSize] = 0;
|
bytesRead += stringSize*2;
|
}
|
return bytesRead;
|
}
|
size_t SmallString::Write(char* data)
|
{
|
size_t stringSize = 0;
|
size_t bytesWrite = 0;
|
if (unicode_ == 0)
|
{
|
// ANSI string
|
if (name_)
|
{
|
stringSize = strlen(name_);
|
LittleEndian::Write(data, stringSize, 0, 1);
|
LittleEndian::Write(data, unicode_, 1, 1);
|
LittleEndian::WriteString(data, name_, 2, stringSize);
|
bytesWrite = 2 + stringSize;
|
}
|
else
|
{
|
LittleEndian::Write(data, stringSize, 0, 1);
|
LittleEndian::Write(data, unicode_, 1, 1);
|
bytesWrite = 2;
|
}
|
}
|
else
|
{
|
// UNICODE
|
if (wname_)
|
{
|
stringSize = wcslen(wname_);
|
LittleEndian::Write(data, stringSize, 0, 1);
|
LittleEndian::Write(data, unicode_, 1, 1);
|
LittleEndian::WriteString(data, wname_, 2, stringSize);
|
bytesWrite = 2 + stringSize*2;
|
}
|
else
|
{
|
LittleEndian::Write(data, stringSize, 0, 1);
|
LittleEndian::Write(data, unicode_, 1, 1);
|
bytesWrite = 2;
|
}
|
}
|
return bytesWrite;
|
}
|
size_t SmallString::DataSize() {return (unicode_ == 0) ? StringSize()+2 : StringSize()*2+2;}
|
size_t SmallString::RecordSize() {return DataSize();}
|
size_t SmallString::StringSize()
|
{
|
if (unicode_ == 0)
|
{
|
if (name_) return strlen(name_);
|
}
|
else
|
{
|
if (wname_) return wcslen(wname_);
|
}
|
return 0;
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
LargeString::LargeString() : unicode_(-1), richtext_(0), phonetic_(0) {};
|
LargeString::~LargeString() {};
|
LargeString::LargeString(const LargeString& s) :
|
name_(s.name_), wname_(s.wname_),
|
unicode_(s.unicode_), richtext_(s.richtext_), phonetic_(s.phonetic_) {};
|
LargeString& LargeString::operator=(const LargeString& s)
|
{
|
unicode_ = s.unicode_;
|
richtext_ = s.richtext_;
|
phonetic_ = s.phonetic_;
|
name_ = s.name_;
|
wname_ = s.wname_;
|
return *this;
|
}
|
const LargeString& LargeString::operator=(const char* str)
|
{
|
unicode_ = 0;
|
richtext_ = 0;
|
phonetic_ = 0;
|
wname_.clear();
|
size_t len = strlen(str);
|
name_.resize(len+1);
|
strcpy(&*(name_.begin()), str);
|
return *this;
|
}
|
const LargeString& LargeString::operator=(const wchar_t* str)
|
{
|
unicode_ = 1;
|
richtext_ = 0;
|
phonetic_ = 0;
|
name_.clear();
|
size_t len = wcslen(str);
|
wname_.resize(len+1);
|
wcscpy(&*(wname_.begin()), str);
|
return *this;
|
}
|
size_t LargeString::Read(const char* data)
|
{
|
size_t stringSize;
|
LittleEndian::Read(data, stringSize, 0, 2);
|
LittleEndian::Read(data, unicode_, 2, 1);
|
size_t npos = 3;
|
if (unicode_ & 8)
|
{
|
LittleEndian::Read(data, richtext_, npos, 2);
|
npos += 2;
|
}
|
if (unicode_ & 4) LittleEndian::Read(data, phonetic_, npos, 4);
|
name_.clear();
|
wname_.clear();
|
size_t bytesRead = 2;
|
if (stringSize>0) bytesRead += ContinueRead(data+2, stringSize);
|
else bytesRead = 3;
|
return bytesRead;
|
}
|
size_t LargeString::ContinueRead(const char* data, size_t size)
|
{
|
if (size == 0) return 0;
|
|
char unicode;
|
LittleEndian::Read(data, unicode, 0, 1);
|
if (unicode_ == -1) unicode_ = unicode;
|
if (unicode_ & 1)
|
{
|
// Present stored string is uncompressed (16 bit)
|
size_t npos = 1;
|
if (richtext_) npos += 2;
|
if (phonetic_) npos += 4;
|
|
size_t strpos = wname_.size();
|
wname_.resize(strpos+size, 0);
|
if (unicode & 1)
|
{
|
LittleEndian::ReadString(data, &*(wname_.begin())+strpos, npos, size);
|
npos += size * SIZEOFWCHAR_T;
|
}
|
else
|
{
|
// String to be read is in ANSI
|
vector<char> name(size);
|
LittleEndian::ReadString(data, &*(name.begin()), npos, size);
|
mbstowcs(&*(wname_.begin())+strpos, &*(name.begin()), size);
|
npos += size;
|
}
|
if (richtext_) npos += 4*richtext_;
|
if (phonetic_) npos += phonetic_;
|
return npos;
|
}
|
else
|
{
|
// Present stored string has character compression (8 bit)
|
size_t npos = 1;
|
if (richtext_) npos += 2;
|
if (phonetic_) npos += 4;
|
|
size_t strpos = name_.size();
|
name_.resize(strpos+size*2, 0);
|
if (unicode & 1)
|
{
|
// String to be read is in unicode
|
vector<wchar_t> name(size);
|
LittleEndian::ReadString(data, &*(name.begin()), npos, size);
|
wcstombs(&*(name_.begin())+strpos, &*(name.begin()), size*2);
|
npos += size * SIZEOFWCHAR_T;
|
}
|
else
|
{
|
LittleEndian::ReadString(data, &*(name_.begin())+strpos, npos, size);
|
npos += size;
|
}
|
if (richtext_) npos += 4*richtext_;
|
if (phonetic_) npos += phonetic_;
|
return npos;
|
}
|
}
|
size_t LargeString::Write(char* data)
|
{
|
size_t stringSize = 0;
|
size_t bytesWrite = 0;
|
if (unicode_ & 1)
|
{
|
// UNICODE
|
unicode_ = 1; // Don't handle richtext or phonetic for now.
|
if (!wname_.empty())
|
{
|
stringSize = wname_.size();
|
LittleEndian::Write(data, stringSize, 0, 2);
|
LittleEndian::Write(data, unicode_, 2, 1);
|
LittleEndian::WriteString(data, &*(wname_.begin()), 3, stringSize);
|
bytesWrite = 3 + stringSize * SIZEOFWCHAR_T;
|
}
|
else
|
{
|
LittleEndian::Write(data, stringSize, 0, 2);
|
LittleEndian::Write(data, unicode_, 2, 1);
|
bytesWrite = 3;
|
}
|
}
|
else
|
{
|
// ANSI string
|
unicode_ = 0; // Don't handle richtext or phonetic for now.
|
if (!name_.empty())
|
{
|
stringSize = name_.size();
|
LittleEndian::Write(data, stringSize, 0, 2);
|
LittleEndian::Write(data, unicode_, 2, 1);
|
LittleEndian::WriteString(data, &*(name_.begin()), 3, stringSize);
|
bytesWrite = 3 + stringSize;
|
}
|
else
|
{
|
LittleEndian::Write(data, stringSize, 0, 2);
|
LittleEndian::Write(data, unicode_, 2, 1);
|
bytesWrite = 3;
|
}
|
}
|
return bytesWrite;
|
}
|
size_t LargeString::DataSize()
|
{
|
size_t dataSize = StringSize() + 3;
|
if (richtext_) dataSize += 2 + 4*richtext_;
|
if (phonetic_) dataSize += 4 + phonetic_;
|
return dataSize;
|
}
|
size_t LargeString::RecordSize() {return DataSize();}
|
size_t LargeString::StringSize()
|
{
|
if (unicode_ & 1) return wname_.size() * SIZEOFWCHAR_T;
|
else return name_.size();
|
}
|
/************************************************************************************************************/
|
|
|
/************************************************************************************************************/
|
Workbook::Workbook()
|
{
|
bof_.version_ = 1536;
|
bof_.type_ = 5;
|
bof_.buildIdentifier_ = 6560;
|
bof_.buildYear_ = 1997;
|
bof_.fileHistoryFlags_ = 49353;
|
bof_.lowestExcelVersion_ = 774;
|
}
|
|
size_t Workbook::Read(const char* data)
|
{
|
size_t bytesRead = 0;
|
short code;
|
LittleEndian::Read(data, code, 0, 2);
|
while (code != CODE::YEOF)
|
{
|
switch (code)
|
{
|
case CODE::BOF:
|
bytesRead += bof_.Read(data+bytesRead);
|
break;
|
|
case CODE::WINDOW1:
|
bytesRead += window1_.Read(data+bytesRead);
|
break;
|
|
case CODE::FONT:
|
fonts_.push_back(Font());
|
bytesRead += fonts_.back().Read(data+bytesRead);
|
break;
|
|
case CODE::XF:
|
XFs_.push_back(XF());
|
bytesRead += XFs_.back().Read(data+bytesRead);
|
break;
|
|
case CODE::STYLE:
|
styles_.push_back(Style());
|
bytesRead += styles_.back().Read(data+bytesRead);
|
break;
|
|
case CODE::BOUNDSHEET:
|
boundSheets_.push_back(BoundSheet());
|
bytesRead += boundSheets_.back().Read(data+bytesRead);
|
break;
|
|
case CODE::SST:
|
bytesRead += sst_.Read(data+bytesRead);
|
break;
|
|
// case CODE::EXTSST:
|
// bytesRead += extSST_.Read(data+bytesRead);
|
// break;
|
|
default:
|
Record rec;
|
bytesRead += rec.Read(data+bytesRead);
|
}
|
LittleEndian::Read(data, code, bytesRead, 2);
|
}
|
bytesRead += eof_.RecordSize();
|
return bytesRead;
|
}
|
size_t Workbook::Write(char* data)
|
{
|
size_t bytesWritten = 0;
|
|
bytesWritten += bof_.Write(data+bytesWritten);
|
|
bytesWritten += window1_.Write(data+bytesWritten);
|
|
size_t maxFonts = fonts_.size();
|
{for (size_t i=0; i<maxFonts; ++i) {bytesWritten += fonts_[i].Write(data+bytesWritten);}}
|
|
size_t maxXFs = XFs_.size();
|
{for (size_t i=0; i<maxXFs; ++i) {bytesWritten += XFs_[i].Write(data+bytesWritten);}}
|
|
size_t maxStyles = styles_.size();
|
{for (size_t i=0; i<maxStyles; ++i) {bytesWritten += styles_[i].Write(data+bytesWritten);}}
|
|
size_t maxBoundSheets = boundSheets_.size();
|
{for (size_t i=0; i<maxBoundSheets; ++i) {bytesWritten += boundSheets_[i].Write(data+bytesWritten);}}
|
|
bytesWritten += sst_.Write(data+bytesWritten);
|
// bytesWritten += extSST_.Write(data+bytesWritten);
|
|
bytesWritten += eof_.Write(data+bytesWritten);
|
|
return bytesWritten;
|
}
|
size_t Workbook::DataSize()
|
{
|
size_t size = 0;
|
size += bof_.RecordSize();
|
size += window1_.RecordSize();
|
|
size_t maxFonts = fonts_.size();
|
{for (size_t i=0; i<maxFonts; ++i) {size += fonts_[i].RecordSize();}}
|
|
size_t maxXFs = XFs_.size();
|
{for (size_t i=0; i<maxXFs; ++i) {size += XFs_[i].RecordSize();}}
|
|
size_t maxStyles = styles_.size();
|
{for (size_t i=0; i<maxStyles; ++i) {size += styles_[i].RecordSize();}}
|
|
size_t maxBoundSheets = boundSheets_.size();
|
{for (size_t i=0; i<maxBoundSheets; ++i) {size += boundSheets_[i].RecordSize();}}
|
|
size += sst_.RecordSize();
|
// size += extSST_.RecordSize();
|
size += eof_.RecordSize();
|
return size;
|
}
|
size_t Workbook::RecordSize() {return DataSize();}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Workbook::Window1::Window1() : Record(),
|
horizontalPos_(0x78), verticalPos_(0x78), width_(0x3B1F), height_(0x2454),
|
options_(0x38), activeWorksheetIndex_(0), firstVisibleTabIndex_(0), selectedWorksheetNo_(1),
|
worksheetTabBarWidth_(0x258) {code_ = CODE::WINDOW1; dataSize_ = 18; recordSize_ = 22;}
|
size_t Workbook::Window1::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, horizontalPos_, 0, 2);
|
LittleEndian::Read(data_, verticalPos_, 2, 2);
|
LittleEndian::Read(data_, width_, 4, 2);
|
LittleEndian::Read(data_, height_, 6, 2);
|
LittleEndian::Read(data_, options_, 8, 2);
|
LittleEndian::Read(data_, activeWorksheetIndex_, 10, 2);
|
LittleEndian::Read(data_, firstVisibleTabIndex_, 12, 2);
|
LittleEndian::Read(data_, selectedWorksheetNo_, 14, 2);
|
LittleEndian::Read(data_, worksheetTabBarWidth_, 16, 2);
|
return RecordSize();
|
}
|
size_t Workbook::Window1::Write(char* data)
|
{
|
data_.resize(dataSize_);
|
LittleEndian::Write(data_, horizontalPos_, 0, 2);
|
LittleEndian::Write(data_, verticalPos_, 2, 2);
|
LittleEndian::Write(data_, width_, 4, 2);
|
LittleEndian::Write(data_, height_, 6, 2);
|
LittleEndian::Write(data_, options_, 8, 2);
|
LittleEndian::Write(data_, activeWorksheetIndex_, 10, 2);
|
LittleEndian::Write(data_, firstVisibleTabIndex_, 12, 2);
|
LittleEndian::Write(data_, selectedWorksheetNo_, 14, 2);
|
LittleEndian::Write(data_, worksheetTabBarWidth_, 16, 2);
|
return Record::Write(data);
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Workbook::Font::Font() : Record(),
|
height_(200), options_(0), colourIndex_(0x7FFF), weight_(400), escapementType_(0),
|
underlineType_(0), family_(0), characterSet_(0), unused_(0)
|
{
|
code_ = CODE::FONT;
|
dataSize_ = 14;
|
recordSize_ = 18;
|
name_ = L"Arial";
|
name_.unicode_ = 1;
|
}
|
size_t Workbook::Font::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, height_, 0, 2);
|
LittleEndian::Read(data_, options_, 2, 2);
|
LittleEndian::Read(data_, colourIndex_, 4, 2);
|
LittleEndian::Read(data_, weight_, 6, 2);
|
LittleEndian::Read(data_, escapementType_, 8, 2);
|
LittleEndian::Read(data_, underlineType_, 10, 1);
|
LittleEndian::Read(data_, family_, 11, 1);
|
LittleEndian::Read(data_, characterSet_, 12, 1);
|
LittleEndian::Read(data_, unused_, 13, 1);
|
name_.Read(&*(data_.begin())+14);
|
return RecordSize();
|
}
|
size_t Workbook::Font::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, height_, 0, 2);
|
LittleEndian::Write(data_, options_, 2, 2);
|
LittleEndian::Write(data_, colourIndex_, 4, 2);
|
LittleEndian::Write(data_, weight_, 6, 2);
|
LittleEndian::Write(data_, escapementType_, 8, 2);
|
LittleEndian::Write(data_, underlineType_, 10, 1);
|
LittleEndian::Write(data_, family_, 11, 1);
|
LittleEndian::Write(data_, characterSet_, 12, 1);
|
LittleEndian::Write(data_, unused_, 13, 1);
|
name_.Write(&*(data_.begin())+14);
|
return Record::Write(data);
|
}
|
size_t Workbook::Font::DataSize() {return (dataSize_ = 14 + name_.RecordSize());}
|
size_t Workbook::Font::RecordSize() {return (recordSize_ = DataSize()+4);}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Workbook::XF::XF() : Record(),
|
fontRecordIndex_(0), formatRecordIndex_(0), protectionType_(0xFFF5), alignment_(0x20), rotation_(0x00),
|
textProperties_(0x00), usedAttributes_(0x00), borderLines_(0x0000), colour1_(0x0000), colour2_(0x20C0)
|
{code_ = CODE::XF; dataSize_ = 20; recordSize_ = 24;}
|
size_t Workbook::XF::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, fontRecordIndex_, 0, 2);
|
LittleEndian::Read(data_, formatRecordIndex_, 2, 2);
|
LittleEndian::Read(data_, protectionType_, 4, 2);
|
LittleEndian::Read(data_, alignment_, 6, 1);
|
LittleEndian::Read(data_, rotation_, 7, 1);
|
LittleEndian::Read(data_, textProperties_, 8, 1);
|
LittleEndian::Read(data_, usedAttributes_, 9, 1);
|
LittleEndian::Read(data_, borderLines_, 10, 4);
|
LittleEndian::Read(data_, colour1_, 14, 4);
|
LittleEndian::Read(data_, colour2_, 18, 2);
|
return RecordSize();
|
}
|
size_t Workbook::XF::Write(char* data)
|
{
|
data_.resize(dataSize_);
|
LittleEndian::Write(data_, fontRecordIndex_, 0, 2);
|
LittleEndian::Write(data_, formatRecordIndex_, 2, 2);
|
LittleEndian::Write(data_, protectionType_, 4, 2);
|
LittleEndian::Write(data_, alignment_, 6, 1);
|
LittleEndian::Write(data_, rotation_, 7, 1);
|
LittleEndian::Write(data_, textProperties_, 8, 1);
|
LittleEndian::Write(data_, usedAttributes_, 9, 1);
|
LittleEndian::Write(data_, borderLines_, 10, 4);
|
LittleEndian::Write(data_, colour1_, 14, 4);
|
LittleEndian::Write(data_, colour2_, 18, 2);
|
return Record::Write(data);
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Workbook::Style::Style() : Record(),
|
XFRecordIndex_(0x8000), identifier_(0), level_(0xFF)
|
{code_ = CODE::STYLE; dataSize_ = 2; recordSize_ = 6;}
|
size_t Workbook::Style::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, XFRecordIndex_, 0, 2);
|
if (XFRecordIndex_ & 0x8000)
|
{
|
// Built-in styles
|
LittleEndian::Read(data_, identifier_, 2, 1);
|
LittleEndian::Read(data_, level_, 3, 1);
|
}
|
else
|
{
|
// User-defined styles
|
name_.Read(&*(data_.begin())+2);
|
}
|
return RecordSize();
|
}
|
size_t Workbook::Style::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, XFRecordIndex_, 0, 2);
|
if (XFRecordIndex_ & 0x8000)
|
{
|
// Built-in styles
|
LittleEndian::Write(data_, identifier_, 2, 1);
|
LittleEndian::Write(data_, level_, 3, 1);
|
}
|
else
|
{
|
// User-defined styles
|
name_.Write(&*(data_.begin())+2);
|
}
|
return Record::Write(data);
|
}
|
size_t Workbook::Style::DataSize() {return (dataSize_ = (XFRecordIndex_ & 0x8000) ? 4 : 2+name_.RecordSize());}
|
size_t Workbook::Style::RecordSize() {return (recordSize_ = DataSize()+4);}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Workbook::BoundSheet::BoundSheet() : Record(),
|
BOFpos_(0x0000), visibility_(0), type_(0)
|
{
|
code_ = CODE::BOUNDSHEET;
|
dataSize_ = 6;
|
dataSize_ = 10;
|
name_ = "Sheet1";
|
name_.unicode_ = false;
|
}
|
size_t Workbook::BoundSheet::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, BOFpos_, 0, 4);
|
LittleEndian::Read(data_, visibility_, 4, 1);
|
LittleEndian::Read(data_, type_, 5, 1);
|
name_.Read(&*(data_.begin())+6);
|
return RecordSize();
|
}
|
size_t Workbook::BoundSheet::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, BOFpos_, 0, 4);
|
LittleEndian::Write(data_, visibility_, 4, 1);
|
LittleEndian::Write(data_, type_, 5, 1);
|
name_.Write(&*(data_.begin())+6);
|
return Record::Write(data);
|
}
|
size_t Workbook::BoundSheet::DataSize() {return (dataSize_ = 6+name_.RecordSize());}
|
size_t Workbook::BoundSheet::RecordSize() {return (recordSize_ = DataSize()+4);}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Workbook::SharedStringTable::SharedStringTable() : Record(),
|
stringsTotal_(0), uniqueStringsTotal_(0) {code_ = CODE::SST; dataSize_ = 8; recordSize_ = 12;}
|
size_t Workbook::SharedStringTable::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, stringsTotal_, 0, 4);
|
LittleEndian::Read(data_, uniqueStringsTotal_, 4, 4);
|
strings_.clear();
|
strings_.resize(uniqueStringsTotal_);
|
|
size_t npos = 8;
|
if (continueIndices_.empty())
|
{
|
for (size_t i=0; i<uniqueStringsTotal_; ++i)
|
{
|
npos += strings_[i].Read(&*(data_.begin())+npos);
|
}
|
}
|
else
|
{
|
// Require special handling since CONTINUE records are present
|
size_t maxContinue = continueIndices_.size();
|
|
for (size_t i=0, c=0; i<uniqueStringsTotal_; ++i)
|
{
|
char unicode;
|
size_t stringSize;
|
LittleEndian::Read(data_, stringSize, npos, 2);
|
LittleEndian::Read(data_, unicode, npos+2, 1);
|
size_t multiplier = unicode & 1 ? 2 : 1;
|
if (c >= maxContinue || npos+stringSize*multiplier+3 <= continueIndices_[c])
|
{
|
// String to be read is not split into two records
|
npos += strings_[i].Read(&*(data_.begin())+npos);
|
}
|
else
|
{
|
// String to be read is split into two or more records
|
int bytesRead = 2;// Start from unicode field
|
|
int size = continueIndices_[c] - npos - 1 - bytesRead;
|
++c;
|
if (size > 0)
|
{
|
size /= multiplier; // Number of characters available for string in current record.
|
bytesRead += strings_[i].ContinueRead(&*(data_.begin())+npos+bytesRead, size);
|
stringSize -= size;
|
size = 0;
|
}
|
while (c<maxContinue && npos+stringSize+1>continueIndices_[c])
|
{
|
size_t dataSize = (continueIndices_[c] - continueIndices_[c-1] - 1) / multiplier;
|
bytesRead += strings_[i].ContinueRead(&*(data_.begin())+npos+bytesRead, dataSize);
|
stringSize -= dataSize + 1;
|
++c;
|
};
|
if (stringSize>0)
|
{
|
bytesRead += strings_[i].ContinueRead(&*(data_.begin())+npos+bytesRead, stringSize);
|
}
|
npos += bytesRead;
|
}
|
}
|
}
|
return npos + 4*(npos/8224 + 1);
|
}
|
size_t Workbook::SharedStringTable::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, stringsTotal_, 0, 4);
|
LittleEndian::Write(data_, uniqueStringsTotal_, 4, 4);
|
|
size_t maxContinue = continueIndices_.size();
|
for (size_t i=0, c=0, npos=8; i<uniqueStringsTotal_; ++i)
|
{
|
npos += strings_[i].Write(&*(data_.begin())+npos);
|
if (c<maxContinue && npos==continueIndices_[c]) ++c;
|
else if (c<maxContinue && npos > continueIndices_[c])
|
{
|
// Insert unicode flag where appropriate for CONTINUE records.
|
data_.insert(data_.begin()+continueIndices_[c], strings_[i].unicode_);
|
data_.pop_back();
|
++c;
|
++npos;
|
}
|
}
|
return Record::Write(data);
|
}
|
size_t Workbook::SharedStringTable::DataSize()
|
{
|
dataSize_ = 8;
|
continueIndices_.clear();
|
size_t curMax = 8224;
|
for (size_t i=0; i<uniqueStringsTotal_; ++i)
|
{
|
size_t stringSize = strings_[i].StringSize();
|
if (dataSize_+stringSize+3 <= curMax)
|
{
|
dataSize_ += stringSize + 3;
|
}
|
else
|
{
|
// If have >= 12 bytes (2 for size, 1 for unicode and >=9 for data, can split string
|
// otherwise, end record and start continue record.
|
bool unicode = strings_[i].unicode_ & 1;
|
if (curMax - dataSize_ >= 12)
|
{
|
if (unicode && !((curMax-dataSize_)%2)) --curMax; // Make sure space reserved for unicode strings is even.
|
continueIndices_.push_back(curMax);
|
stringSize -= (curMax - dataSize_ - 3);
|
dataSize_ = curMax;
|
curMax += 8224;
|
|
size_t additionalContinueRecords = unicode ? stringSize/8222 : stringSize/8223; // 8222 or 8223 because the first byte is for unicode identifier
|
for (size_t j=0; j<additionalContinueRecords; ++j)
|
{
|
if (unicode)
|
{
|
--curMax;
|
continueIndices_.push_back(curMax);
|
curMax += 8223;
|
dataSize_ += 8223;
|
stringSize -= 8222;
|
}
|
else
|
{
|
continueIndices_.push_back(curMax);
|
curMax += 8224;
|
dataSize_ += 8224;
|
stringSize -= 8223;
|
}
|
}
|
dataSize_ += stringSize + 1;
|
}
|
else
|
{
|
continueIndices_.push_back(dataSize_);
|
curMax = dataSize_ + 8224;
|
if (dataSize_+stringSize+3 < curMax)
|
{
|
dataSize_ += stringSize + 3;
|
}
|
else
|
{
|
// If have >= 12 bytes (2 for size, 1 for unicode and >=9 for data, can split string
|
// otherwise, end record and start continue record.
|
if (curMax - dataSize_ >= 12)
|
{
|
if (unicode && !((curMax-dataSize_)%2)) --curMax; // Make sure space reserved for unicode strings is even.
|
continueIndices_.push_back(curMax);
|
stringSize -= (curMax - dataSize_ - 3);
|
dataSize_ = curMax;
|
curMax += 8224;
|
|
size_t additionalContinueRecords = unicode ? stringSize/8222 : stringSize/8223; // 8222 or 8223 because the first byte is for unicode identifier
|
for (size_t j=0; j<additionalContinueRecords; ++j)
|
{
|
if (unicode)
|
{
|
--curMax;
|
continueIndices_.push_back(curMax);
|
curMax += 8223;
|
dataSize_ += 8223;
|
stringSize -= 8222;
|
}
|
else
|
{
|
continueIndices_.push_back(curMax);
|
curMax += 8224;
|
dataSize_ += 8224;
|
stringSize -= 8223;
|
}
|
}
|
dataSize_ += stringSize + 1;
|
}
|
}
|
}
|
}
|
}
|
return dataSize_;
|
}
|
size_t Workbook::SharedStringTable::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
return (recordSize_ = dataSize + 4*(dataSize/8224 + 1));
|
}
|
/************************************************************************************************************/
|
Workbook::ExtSST::ExtSST() : Record(),
|
stringsTotal_(0), streamPos_(0), firstStringPos_(0), unused_(0)
|
{
|
code_ = CODE::EXTSST;
|
dataSize_ = 2;
|
recordSize_ = 6;
|
}
|
|
size_t Workbook::ExtSST::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, stringsTotal_, 0, 2);
|
|
size_t maxPortions = (dataSize_-2) / 8;
|
streamPos_.clear();
|
streamPos_.resize(maxPortions);
|
firstStringPos_.clear();
|
firstStringPos_.resize(maxPortions);
|
unused_.clear();
|
unused_.resize(maxPortions);
|
|
for (size_t i=0, npos=2; i<maxPortions; ++i)
|
{
|
LittleEndian::Read(data_, streamPos_[i], npos, 4);
|
LittleEndian::Read(data_, firstStringPos_[i], npos+4, 2);
|
LittleEndian::Read(data_, unused_[i], npos+6, 2);
|
npos += 8;
|
}
|
return RecordSize();
|
}
|
|
size_t Workbook::ExtSST::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, stringsTotal_, 0, 2);
|
|
size_t maxPortions = streamPos_.size();
|
for (size_t i=0, npos=2; i<maxPortions; ++i)
|
{
|
LittleEndian::Write(data_, streamPos_[i], npos, 4);
|
LittleEndian::Write(data_, firstStringPos_[i], npos+4, 2);
|
LittleEndian::Write(data_, unused_[i], npos+6, 2);
|
npos += 8;
|
}
|
return Record::Write(data);
|
}
|
|
size_t Workbook::ExtSST::DataSize()
|
{
|
dataSize_ = 2 + streamPos_.size()*8;
|
dataSize_ += (int)(dataSize_/8224)*4;
|
return dataSize_;
|
}
|
|
size_t Workbook::ExtSST::RecordSize() {return (recordSize_ = DataSize()+(int)((2+streamPos_.size()*8)/8224)*4)+4;}
|
/************************************************************************************************************/
|
|
|
|
|
/************************************************************************************************************/
|
Worksheet::Worksheet()
|
{
|
bof_.version_ = 1536;
|
bof_.type_ = 16;
|
bof_.buildIdentifier_ = 6560;
|
bof_.buildYear_ = 1997;
|
bof_.fileHistoryFlags_ = 49353;
|
bof_.lowestExcelVersion_ = 774;
|
}
|
|
size_t Worksheet::Read(const char* data)
|
{
|
size_t bytesRead = 0;
|
short code;
|
LittleEndian::Read(data, code, 0, 2);
|
while (code != CODE::YEOF)
|
{
|
switch (code)
|
{
|
case CODE::BOF:
|
bytesRead += bof_.Read(data+bytesRead);
|
break;
|
|
case CODE::INDEX:
|
bytesRead += index_.Read(data+bytesRead);
|
break;
|
|
case CODE::DIMENSIONS:
|
bytesRead += dimensions_.Read(data+bytesRead);
|
break;
|
|
case CODE::ROW:
|
bytesRead += cellTable_.Read(data+bytesRead);
|
break;
|
|
case CODE::WINDOW2:
|
bytesRead += window2_.Read(data+bytesRead);
|
break;
|
|
default:
|
Record rec;
|
bytesRead += rec.Read(data+bytesRead);
|
}
|
LittleEndian::Read(data, code, bytesRead, 2);
|
}
|
bytesRead += eof_.RecordSize();
|
return bytesRead;
|
}
|
size_t Worksheet::Write(char* data)
|
{
|
size_t bytesWritten = 0;
|
bytesWritten += bof_.Write(data+bytesWritten);
|
|
bytesWritten += index_.Write(data+bytesWritten);
|
|
bytesWritten += dimensions_.Write(data+bytesWritten);
|
|
bytesWritten += cellTable_.Write(data+bytesWritten);
|
|
bytesWritten += window2_.Write(data+bytesWritten);
|
|
bytesWritten += eof_.Write(data+bytesWritten);
|
|
return bytesWritten;
|
}
|
size_t Worksheet::DataSize()
|
{
|
size_t dataSize = 0;
|
dataSize += bof_.RecordSize();
|
dataSize += index_.RecordSize();
|
dataSize += dimensions_.RecordSize();
|
dataSize += cellTable_.RecordSize();
|
dataSize += window2_.RecordSize();
|
dataSize += eof_.RecordSize();
|
return dataSize;
|
}
|
size_t Worksheet::RecordSize() {return DataSize();}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Worksheet::Index::Index() : Record(),
|
unused1_(0), firstUsedRowIndex_(0), firstUnusedRowIndex_(0), unused2_(0)
|
{code_ = CODE::INDEX; dataSize_ = 16; recordSize_ = 20; DBCellPos_.resize(1);}
|
size_t Worksheet::Index::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, unused1_, 0, 4);
|
LittleEndian::Read(data_, firstUsedRowIndex_, 4, 4);
|
LittleEndian::Read(data_, firstUnusedRowIndex_, 8, 4);
|
LittleEndian::Read(data_, unused2_, 12, 4);
|
size_t nm = int(firstUnusedRowIndex_ - firstUsedRowIndex_ - 1) / 32 + 1;
|
DBCellPos_.clear();
|
DBCellPos_.resize(nm);
|
if (dataSize_>16)
|
{
|
for (size_t i=0; i<nm; ++i)
|
{
|
LittleEndian::Read(data_, DBCellPos_[i], 16+i*4, 4);
|
}
|
}
|
return RecordSize();
|
}
|
size_t Worksheet::Index::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, unused1_, 0, 4);
|
LittleEndian::Write(data_, firstUsedRowIndex_, 4, 4);
|
LittleEndian::Write(data_, firstUnusedRowIndex_, 8, 4);
|
LittleEndian::Write(data_, unused2_, 12, 4);
|
size_t nm = DBCellPos_.size();
|
for (size_t i=0; i<nm; ++i)
|
{
|
LittleEndian::Write(data_, DBCellPos_[i], 16+i*4, 4);
|
}
|
return Record::Write(data);
|
}
|
size_t Worksheet::Index::DataSize() {return (dataSize_ = 16 + DBCellPos_.size()*4);}
|
size_t Worksheet::Index::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
return (recordSize_ = dataSize + 4*(dataSize/8224 + 1));
|
}
|
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Worksheet::Dimensions::Dimensions() : Record(),
|
firstUsedRowIndex_(0), lastUsedRowIndexPlusOne_(0),
|
firstUsedColIndex_(0), lastUsedColIndexPlusOne_(0),
|
unused_(0) {code_ = CODE::DIMENSIONS; dataSize_ = 14; recordSize_ = 18;}
|
size_t Worksheet::Dimensions::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, firstUsedRowIndex_, 0, 4);
|
LittleEndian::Read(data_, lastUsedRowIndexPlusOne_, 4, 4);
|
LittleEndian::Read(data_, firstUsedColIndex_, 8, 2);
|
LittleEndian::Read(data_, lastUsedColIndexPlusOne_, 10, 2);
|
LittleEndian::Read(data_, unused_, 12, 2);
|
return RecordSize();
|
}
|
size_t Worksheet::Dimensions::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, firstUsedRowIndex_, 0, 4);
|
LittleEndian::Write(data_, lastUsedRowIndexPlusOne_, 4, 4);
|
LittleEndian::Write(data_, firstUsedColIndex_, 8, 2);
|
LittleEndian::Write(data_, lastUsedColIndexPlusOne_, 10, 2);
|
LittleEndian::Write(data_, unused_, 12, 2);
|
return Record::Write(data);
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Worksheet::CellTable::RowBlock::CellBlock::Blank::Blank() : Record(),
|
rowIndex_(0), colIndex_(0), XFRecordIndex_(0) {code_ = CODE::BLANK; dataSize_ = 6; recordSize_ = 10;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Blank::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, rowIndex_, 0, 2);
|
LittleEndian::Read(data_, colIndex_, 2, 2);
|
LittleEndian::Read(data_, XFRecordIndex_, 4, 2);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Blank::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, rowIndex_, 0, 2);
|
LittleEndian::Write(data_, colIndex_, 2, 2);
|
LittleEndian::Write(data_, XFRecordIndex_, 4, 2);
|
return Record::Write(data);
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::BoolErr::BoolErr() : Record(),
|
rowIndex_(0), colIndex_(0), XFRecordIndex_(0), value_(0), error_(0)
|
{code_ = CODE::BOOLERR; dataSize_ = 8; recordSize_ = 12;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::BoolErr::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, rowIndex_, 0, 2);
|
LittleEndian::Read(data_, colIndex_, 2, 2);
|
LittleEndian::Read(data_, XFRecordIndex_, 4, 2);
|
LittleEndian::Read(data_, value_, 6, 1);
|
LittleEndian::Read(data_, error_, 7, 1);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::BoolErr::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, rowIndex_, 0, 2);
|
LittleEndian::Write(data_, colIndex_, 2, 2);
|
LittleEndian::Write(data_, XFRecordIndex_, 4, 2);
|
LittleEndian::Write(data_, value_, 6, 1);
|
LittleEndian::Write(data_, error_, 7, 1);
|
return Record::Write(data);
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::LabelSST::LabelSST() : Record(),
|
rowIndex_(0), colIndex_(0), XFRecordIndex_(0), SSTRecordIndex_(0)
|
{code_ = CODE::LABELSST; dataSize_ = 10; recordSize_ = 14;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::LabelSST::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, rowIndex_, 0, 2);
|
LittleEndian::Read(data_, colIndex_, 2, 2);
|
LittleEndian::Read(data_, XFRecordIndex_, 4, 2);
|
LittleEndian::Read(data_, SSTRecordIndex_, 6, 4);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::LabelSST::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, rowIndex_, 0, 2);
|
LittleEndian::Write(data_, colIndex_, 2, 2);
|
LittleEndian::Write(data_, XFRecordIndex_, 4, 2);
|
LittleEndian::Write(data_, SSTRecordIndex_, 6, 4);
|
return Record::Write(data);
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::MulBlank::MulBlank() : Record(),
|
rowIndex_(0), firstColIndex_(0), lastColIndex_(0)
|
{code_ = CODE::MULBLANK; dataSize_ = 10; recordSize_ = 14;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::MulBlank::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, rowIndex_, 0, 2);
|
LittleEndian::Read(data_, firstColIndex_, 2, 2);
|
LittleEndian::Read(data_, lastColIndex_, dataSize_-2, 2);
|
size_t nc = lastColIndex_ - firstColIndex_ + 1;
|
XFRecordIndices_.clear();
|
XFRecordIndices_.resize(nc);
|
for (size_t i=0; i<nc; ++i)
|
{
|
LittleEndian::Read(data_, XFRecordIndices_[i], 4+i*2, 2);
|
}
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::MulBlank::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, rowIndex_, 0, 2);
|
LittleEndian::Write(data_, firstColIndex_, 2, 2);
|
LittleEndian::Write(data_, lastColIndex_, dataSize_-2, 2);
|
size_t nc = XFRecordIndices_.size();
|
for (size_t i=0; i<nc; ++i)
|
{
|
LittleEndian::Write(data_, XFRecordIndices_[i], 4+i*2, 2);
|
}
|
return Record::Write(data);
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::MulBlank::DataSize() {return (dataSize_ = 6 + XFRecordIndices_.size()*2);}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::MulBlank::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
return (recordSize_ = dataSize + 4*(dataSize/8224 + 1));
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::MulRK::XFRK::XFRK() :
|
XFRecordIndex_(0), RKValue_(0) {};
|
void Worksheet::CellTable::RowBlock::CellBlock::MulRK::XFRK::Read(const char* data)
|
{
|
LittleEndian::Read(data, XFRecordIndex_, 0, 2);
|
LittleEndian::Read(data, RKValue_, 2, 4);
|
}
|
void Worksheet::CellTable::RowBlock::CellBlock::MulRK::XFRK::Write(char* data)
|
{
|
LittleEndian::Write(data, XFRecordIndex_, 0, 2);
|
LittleEndian::Write(data, RKValue_, 2, 4);
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::MulRK::MulRK() : Record(),
|
rowIndex_(0), firstColIndex_(0), lastColIndex_(0) {code_ = CODE::MULRK; dataSize_ = 10; recordSize_ = 14;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::MulRK::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, rowIndex_, 0, 2);
|
LittleEndian::Read(data_, firstColIndex_, 2, 2);
|
LittleEndian::Read(data_, lastColIndex_, dataSize_-2, 2);
|
size_t nc = lastColIndex_ - firstColIndex_ + 1;
|
XFRK_.clear();
|
XFRK_.resize(nc);
|
for (size_t i=0; i<nc; ++i)
|
{
|
XFRK_[i].Read(&*(data_.begin())+4+i*6);
|
}
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::MulRK::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, rowIndex_, 0, 2);
|
LittleEndian::Write(data_, firstColIndex_, 2, 2);
|
LittleEndian::Write(data_, lastColIndex_, dataSize_-2, 2);
|
size_t nc = XFRK_.size();
|
for (size_t i=0; i<nc; ++i)
|
{
|
XFRK_[i].Write(&*(data_.begin())+4+i*6);
|
}
|
return Record::Write(data);
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::MulRK::DataSize() {return (dataSize_ = 6 + XFRK_.size()*6);}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::MulRK::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
return (recordSize_ = dataSize + 4*(dataSize/8224 + 1));
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::Number::Number() : Record(),
|
rowIndex_(0), colIndex_(0), XFRecordIndex_(0), value_(0) {code_ = CODE::NUMBER; dataSize_ = 14; recordSize_ = 18;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Number::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, rowIndex_, 0, 2);
|
LittleEndian::Read(data_, colIndex_, 2, 2);
|
LittleEndian::Read(data_, XFRecordIndex_, 4, 2);
|
long long value;
|
LittleEndian::Read(data_, value, 6, 8);
|
intdouble_.intvalue_ = value;
|
value_ = intdouble_.doublevalue_;
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Number::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, rowIndex_, 0, 2);
|
LittleEndian::Write(data_, colIndex_, 2, 2);
|
LittleEndian::Write(data_, XFRecordIndex_, 4, 2);
|
intdouble_.doublevalue_ = value_;
|
long long value = intdouble_.intvalue_;
|
LittleEndian::Write(data_, value, 6, 8);
|
return Record::Write(data);
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::RK::RK() : Record(),
|
rowIndex_(0), colIndex_(0), XFRecordIndex_(0), value_(0) {code_ = CODE::RK; dataSize_ = 10; recordSize_ = 14;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::RK::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, rowIndex_, 0, 2);
|
LittleEndian::Read(data_, colIndex_, 2, 2);
|
LittleEndian::Read(data_, XFRecordIndex_, 4, 2);
|
LittleEndian::Read(data_, value_, 6, 4);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::RK::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, rowIndex_, 0, 2);
|
LittleEndian::Write(data_, colIndex_, 2, 2);
|
LittleEndian::Write(data_, XFRecordIndex_, 4, 2);
|
LittleEndian::Write(data_, value_, 6, 4);
|
return Record::Write(data);
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::Formula::Formula() : Record(),
|
rowIndex_(0), colIndex_(0), XFRecordIndex_(0), options_(0), unused_(0), type_(-1)
|
{code_ = CODE::FORMULA; dataSize_ = 18; recordSize_ = 22;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, rowIndex_, 0, 2);
|
LittleEndian::Read(data_, colIndex_, 2, 2);
|
LittleEndian::Read(data_, XFRecordIndex_, 4, 2);
|
LittleEndian::ReadString(data_, result_, 6, 8);
|
LittleEndian::Read(data_, options_, 14, 2);
|
LittleEndian::Read(data_, unused_, 16, 2);
|
RPNtoken_.clear();
|
RPNtoken_.resize(dataSize_-18);
|
LittleEndian::ReadString(data_, &*(RPNtoken_.begin()), 18, dataSize_-18);
|
|
size_t offset = dataSize_ + 4;
|
short code;
|
LittleEndian::Read(data, code, offset, 2);
|
switch (code)
|
{
|
case CODE::ARRAY:
|
type_ = code;
|
array_.Read(data+offset);
|
offset += array_.RecordSize();
|
break;
|
|
case CODE::SHRFMLA:
|
type_ = code;
|
shrfmla_.Read(data+offset);
|
offset += shrfmla_.RecordSize();
|
break;
|
|
case CODE::SHRFMLA1:
|
type_ = code;
|
shrfmla1_.Read(data+offset);
|
offset += shrfmla1_.RecordSize();
|
break;
|
|
case CODE::TABLE:
|
type_ = code;
|
table_.Read(data+offset);
|
offset += table_.RecordSize();
|
break;
|
}
|
LittleEndian::Read(data, code, offset, 2);
|
if (code == CODE::STRING) string_.Read(data+offset);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, rowIndex_, 0, 2);
|
LittleEndian::Write(data_, colIndex_, 2, 2);
|
LittleEndian::Write(data_, XFRecordIndex_, 4, 2);
|
LittleEndian::WriteString(data_, result_, 6, 8);
|
LittleEndian::Write(data_, options_, 14, 2);
|
LittleEndian::Write(data_, unused_, 16, 2);
|
LittleEndian::WriteString(data_, &*(RPNtoken_.begin()), 18, RPNtoken_.size());
|
Record::Write(data);
|
|
size_t offset = dataSize_ + 4;
|
switch (type_)
|
{
|
case CODE::ARRAY:
|
array_.Write(data+offset);
|
offset += array_.RecordSize();
|
break;
|
|
case CODE::SHRFMLA:
|
shrfmla_.Write(data+offset);
|
offset += shrfmla_.RecordSize();
|
break;
|
|
case CODE::SHRFMLA1:
|
shrfmla1_.Write(data+offset);
|
offset += shrfmla1_.RecordSize();
|
break;
|
|
case CODE::TABLE:
|
table_.Write(data+offset);
|
offset += table_.RecordSize();
|
break;
|
}
|
if (string_.DataSize() != 0) string_.Write(data+offset);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::DataSize() {return (dataSize_ = 18 + RPNtoken_.size());}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
recordSize_ = dataSize + 4*(dataSize/8224 + 1);
|
|
switch (type_)
|
{
|
case CODE::ARRAY:
|
recordSize_ += array_.RecordSize();
|
break;
|
|
case CODE::SHRFMLA:
|
recordSize_ += shrfmla_.RecordSize();
|
break;
|
|
case CODE::SHRFMLA1:
|
recordSize_ += shrfmla1_.RecordSize();
|
break;
|
|
case CODE::TABLE:
|
recordSize_ += table_.RecordSize();
|
break;
|
}
|
if (string_.DataSize() != 0) recordSize_ += string_.RecordSize();
|
return (recordSize_);
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::Formula::Array::Array() : Record(),
|
firstRowIndex_(0), lastRowIndex_(0), firstColIndex_(0), lastColIndex_(0),
|
options_(0), unused_(0)
|
{code_ = CODE::ARRAY; dataSize_ = 12; recordSize_ = 16;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::Array::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, firstRowIndex_, 0, 2);
|
LittleEndian::Read(data_, lastRowIndex_, 2, 2);
|
LittleEndian::Read(data_, firstColIndex_, 4, 1);
|
LittleEndian::Read(data_, lastColIndex_, 5, 1);
|
LittleEndian::Read(data_, options_, 6, 2);
|
LittleEndian::Read(data_, unused_, 8, 4);
|
formula_.clear();
|
formula_.resize(dataSize_-12);
|
LittleEndian::ReadString(data_, &*(formula_.begin()), 12, dataSize_-12);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::Array::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, firstRowIndex_, 0, 2);
|
LittleEndian::Write(data_, lastRowIndex_, 2, 2);
|
LittleEndian::Write(data_, firstColIndex_, 4, 1);
|
LittleEndian::Write(data_, lastColIndex_, 5, 1);
|
LittleEndian::Write(data_, options_, 6, 2);
|
LittleEndian::Write(data_, unused_, 8, 4);
|
LittleEndian::WriteString(data_, &*(formula_.begin()), 12, formula_.size());
|
return Record::Write(data);
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::Array::DataSize() {return (dataSize_ = 12 + formula_.size());}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::Array::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
return (recordSize_ = dataSize + 4*(dataSize/8224 + 1));
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla::ShrFmla() : Record(),
|
firstRowIndex_(0), lastRowIndex_(0), firstColIndex_(0), lastColIndex_(0),
|
unused_(0)
|
{code_ = CODE::SHRFMLA; dataSize_ = 8; recordSize_ = 12;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, firstRowIndex_, 0, 2);
|
LittleEndian::Read(data_, lastRowIndex_, 2, 2);
|
LittleEndian::Read(data_, firstColIndex_, 4, 1);
|
LittleEndian::Read(data_, lastColIndex_, 5, 1);
|
LittleEndian::Read(data_, unused_, 6, 2);
|
formula_.clear();
|
formula_.resize(dataSize_-8);
|
LittleEndian::ReadString(data_, &*(formula_.begin()), 8, dataSize_-8);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, firstRowIndex_, 0, 2);
|
LittleEndian::Write(data_, lastRowIndex_, 2, 2);
|
LittleEndian::Write(data_, firstColIndex_, 4, 1);
|
LittleEndian::Write(data_, lastColIndex_, 5, 1);
|
LittleEndian::Write(data_, unused_, 6, 2);
|
LittleEndian::WriteString(data_, &*(formula_.begin()), 8, formula_.size());
|
return Record::Write(data);
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla::DataSize() {return (dataSize_ = 8 + formula_.size());}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
return (recordSize_ = dataSize + 4*(dataSize/8224 + 1));
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla1::ShrFmla1() : Record(),
|
firstRowIndex_(0), lastRowIndex_(0), firstColIndex_(0), lastColIndex_(0),
|
unused_(0)
|
{code_ = CODE::SHRFMLA1; dataSize_ = 8; recordSize_ = 12;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla1::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, firstRowIndex_, 0, 2);
|
LittleEndian::Read(data_, lastRowIndex_, 2, 2);
|
LittleEndian::Read(data_, firstColIndex_, 4, 1);
|
LittleEndian::Read(data_, lastColIndex_, 5, 1);
|
LittleEndian::Read(data_, unused_, 6, 2);
|
formula_.clear();
|
formula_.resize(dataSize_-8);
|
LittleEndian::ReadString(data_, &*(formula_.begin()), 8, dataSize_-8);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla1::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, firstRowIndex_, 0, 2);
|
LittleEndian::Write(data_, lastRowIndex_, 2, 2);
|
LittleEndian::Write(data_, firstColIndex_, 4, 1);
|
LittleEndian::Write(data_, lastColIndex_, 5, 1);
|
LittleEndian::Write(data_, unused_, 6, 2);
|
LittleEndian::WriteString(data_, &*(formula_.begin()), 8, formula_.size());
|
return Record::Write(data);
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla1::DataSize() {return (dataSize_ = 8 + formula_.size());}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::ShrFmla1::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
return (recordSize_ = dataSize + 4*(dataSize/8224 + 1));
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::Formula::Table::Table() : Record(),
|
firstRowIndex_(0), lastRowIndex_(0), firstColIndex_(0), lastColIndex_(0), options_(0),
|
inputCellRowIndex_(0), inputCellColIndex_(0),
|
inputCellColumnInputRowIndex_(0), inputCellColumnInputColIndex_(0)
|
{code_ = CODE::TABLE; dataSize_ = 16; recordSize_ = 20;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::Table::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, firstRowIndex_, 0, 2);
|
LittleEndian::Read(data_, lastRowIndex_, 2, 2);
|
LittleEndian::Read(data_, firstColIndex_, 4, 1);
|
LittleEndian::Read(data_, lastColIndex_, 5, 1);
|
LittleEndian::Read(data_, options_, 6, 2);
|
LittleEndian::Read(data_, inputCellRowIndex_, 8, 2);
|
LittleEndian::Read(data_, inputCellColIndex_, 10, 2);
|
LittleEndian::Read(data_, inputCellColumnInputRowIndex_, 12, 2);
|
LittleEndian::Read(data_, inputCellColumnInputColIndex_, 14, 2);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::Table::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, firstRowIndex_, 0, 2);
|
LittleEndian::Write(data_, lastRowIndex_, 2, 2);
|
LittleEndian::Write(data_, firstColIndex_, 4, 1);
|
LittleEndian::Write(data_, lastColIndex_, 5, 1);
|
LittleEndian::Write(data_, options_, 6, 2);
|
LittleEndian::Write(data_, inputCellRowIndex_, 8, 2);
|
LittleEndian::Write(data_, inputCellColIndex_, 10, 2);
|
LittleEndian::Write(data_, inputCellColumnInputRowIndex_, 12, 2);
|
LittleEndian::Write(data_, inputCellColumnInputColIndex_, 14, 2);
|
return Record::Write(data);
|
}
|
|
Worksheet::CellTable::RowBlock::CellBlock::Formula::String::String() : Record()
|
{code_ = CODE::STRING; dataSize_ = 0; recordSize_ = 4;}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::String::Read(const char* data)
|
{
|
Record::Read(data);
|
string_.clear();
|
string_.resize(dataSize_);
|
LittleEndian::ReadString(data_, &*(string_.begin()), 0, dataSize_);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::String::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::WriteString(data_, &*(string_.begin()), 0, string_.size());
|
return Record::Write(data);
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::String::DataSize() {return (dataSize_ = string_.size());}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Formula::String::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
return (recordSize_ = dataSize + 4*(dataSize/8224 + 1));
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Worksheet::CellTable::RowBlock::Row::Row() : Record(),
|
rowIndex_(0), firstCellColIndex_(0), lastCellColIndexPlusOne_(0), height_(255),
|
unused1_(0), unused2_(0), options_(0) {code_ = CODE::ROW; dataSize_ = 16; recordSize_ = 20;}
|
size_t Worksheet::CellTable::RowBlock::Row::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, rowIndex_, 0, 2);
|
LittleEndian::Read(data_, firstCellColIndex_, 2, 2);
|
LittleEndian::Read(data_, lastCellColIndexPlusOne_, 4, 2);
|
LittleEndian::Read(data_, height_, 6, 2);
|
LittleEndian::Read(data_, unused1_, 8, 2);
|
LittleEndian::Read(data_, unused2_, 10, 2);
|
LittleEndian::Read(data_, options_, 12, 4);
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::Row::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, rowIndex_, 0, 2);
|
LittleEndian::Write(data_, firstCellColIndex_, 2, 2);
|
LittleEndian::Write(data_, lastCellColIndexPlusOne_, 4, 2);
|
LittleEndian::Write(data_, height_, 6, 2);
|
LittleEndian::Write(data_, unused1_, 8, 2);
|
LittleEndian::Write(data_, unused2_, 10, 2);
|
LittleEndian::Write(data_, options_, 12, 4);
|
return Record::Write(data);
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Worksheet::CellTable::RowBlock::CellBlock::CellBlock() :
|
type_(-1), normalType_(true) {};
|
Worksheet::CellTable::RowBlock::CellBlock::~CellBlock() {};
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Read(const char* data)
|
{
|
size_t bytesRead = 0;
|
LittleEndian::Read(data, type_, 0, 2);
|
switch (type_)
|
{
|
case CODE::BLANK:
|
bytesRead += blank_.Read(data);
|
break;
|
|
case CODE::BOOLERR:
|
bytesRead += boolerr_.Read(data);
|
break;
|
|
case CODE::LABELSST:
|
bytesRead += labelsst_.Read(data);
|
break;
|
|
case CODE::MULBLANK:
|
bytesRead += mulblank_.Read(data);
|
break;
|
|
case CODE::MULRK:
|
bytesRead += mulrk_.Read(data);
|
break;
|
|
case CODE::NUMBER:
|
bytesRead += number_.Read(data);
|
break;
|
|
case CODE::RK:
|
bytesRead += rk_.Read(data);
|
break;
|
|
case CODE::FORMULA:
|
bytesRead += formula_.Read(data);
|
break;
|
}
|
return bytesRead;
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::Write(char* data)
|
{
|
size_t bytesWritten = 0;
|
switch (type_)
|
{
|
case CODE::BLANK:
|
bytesWritten += blank_.Write(data);
|
break;
|
|
case CODE::BOOLERR:
|
bytesWritten += boolerr_.Write(data);
|
break;
|
|
case CODE::LABELSST:
|
bytesWritten += labelsst_.Write(data);
|
break;
|
|
case CODE::MULBLANK:
|
bytesWritten += mulblank_.Write(data);
|
break;
|
|
case CODE::MULRK:
|
bytesWritten += mulrk_.Write(data);
|
break;
|
|
case CODE::NUMBER:
|
bytesWritten += number_.Write(data);
|
break;
|
|
case CODE::RK:
|
bytesWritten += rk_.Write(data);
|
break;
|
|
case CODE::FORMULA:
|
bytesWritten += formula_.Write(data);
|
break;
|
}
|
return bytesWritten;
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::DataSize()
|
{
|
switch (type_)
|
{
|
case CODE::BLANK:
|
return blank_.DataSize();
|
|
case CODE::BOOLERR:
|
return boolerr_.DataSize();
|
|
case CODE::LABELSST:
|
return labelsst_.DataSize();
|
|
case CODE::MULBLANK:
|
return mulblank_.DataSize();
|
|
case CODE::MULRK:
|
return mulrk_.DataSize();
|
|
case CODE::NUMBER:
|
return number_.DataSize();
|
|
case CODE::RK:
|
return rk_.DataSize();
|
|
case CODE::FORMULA:
|
return formula_.DataSize();
|
}
|
abort();
|
}
|
size_t Worksheet::CellTable::RowBlock::CellBlock::RecordSize()
|
{
|
switch (type_)
|
{
|
case CODE::BLANK:
|
return blank_.RecordSize();
|
|
case CODE::BOOLERR:
|
return boolerr_.RecordSize();
|
|
case CODE::LABELSST:
|
return labelsst_.RecordSize();
|
|
case CODE::MULBLANK:
|
return mulblank_.RecordSize();
|
|
case CODE::MULRK:
|
return mulrk_.RecordSize();
|
|
case CODE::NUMBER:
|
return number_.RecordSize();
|
|
case CODE::RK:
|
return rk_.RecordSize();
|
|
case CODE::FORMULA:
|
return formula_.RecordSize();
|
}
|
abort();
|
}
|
short Worksheet::CellTable::RowBlock::CellBlock::RowIndex()
|
{
|
switch (type_)
|
{
|
case CODE::BLANK:
|
return blank_.rowIndex_;
|
|
case CODE::BOOLERR:
|
return boolerr_.rowIndex_;
|
|
case CODE::LABELSST:
|
return labelsst_.rowIndex_;
|
|
case CODE::MULBLANK:
|
return mulblank_.rowIndex_;
|
|
case CODE::MULRK:
|
return mulrk_.rowIndex_;
|
|
case CODE::NUMBER:
|
return number_.rowIndex_;
|
|
case CODE::RK:
|
return rk_.rowIndex_;
|
|
case CODE::FORMULA:
|
return formula_.rowIndex_;
|
}
|
abort();
|
}
|
short Worksheet::CellTable::RowBlock::CellBlock::ColIndex()
|
{
|
switch (type_)
|
{
|
case CODE::BLANK:
|
return blank_.colIndex_;
|
|
case CODE::BOOLERR:
|
return boolerr_.colIndex_;
|
|
case CODE::LABELSST:
|
return labelsst_.colIndex_;
|
|
case CODE::MULBLANK:
|
return mulblank_.firstColIndex_;
|
|
case CODE::MULRK:
|
return mulrk_.firstColIndex_;
|
|
case CODE::NUMBER:
|
return number_.colIndex_;
|
|
case CODE::RK:
|
return rk_.colIndex_;
|
|
case CODE::FORMULA:
|
return formula_.colIndex_;
|
}
|
abort();
|
}
|
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Worksheet::CellTable::RowBlock::DBCell::DBCell() : Record(),
|
firstRowOffset_(0) {code_ = CODE::DBCELL; dataSize_ = 4; recordSize_ = 8;}
|
size_t Worksheet::CellTable::RowBlock::DBCell::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, firstRowOffset_, 0, 4);
|
size_t nm = (dataSize_-4) / 2;
|
offsets_.clear();
|
offsets_.resize(nm);
|
for (size_t i=0; i<nm; ++i)
|
{
|
LittleEndian::Read(data_, offsets_[i], 4+i*2, 2);
|
}
|
return RecordSize();
|
}
|
size_t Worksheet::CellTable::RowBlock::DBCell::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, firstRowOffset_, 0, 4);
|
size_t nm = offsets_.size();
|
for (size_t i=0; i<nm; ++i)
|
{
|
LittleEndian::Write(data_, offsets_[i], 4+i*2, 2);
|
}
|
return Record::Write(data);
|
}
|
size_t Worksheet::CellTable::RowBlock::DBCell::DataSize() {return (dataSize_ = 4+offsets_.size()*2);}
|
size_t Worksheet::CellTable::RowBlock::DBCell::RecordSize()
|
{
|
size_t dataSize = DataSize();
|
return (recordSize_ = dataSize + 4*(dataSize/8224 + 1));
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
size_t Worksheet::CellTable::RowBlock::Read(const char* data)
|
{
|
size_t bytesRead = 0;
|
short code;
|
LittleEndian::Read(data, code, 0, 2);
|
Row row;
|
CellBlock cellBlock;
|
cellBlocks_.reserve(1000);
|
while (code != CODE::DBCELL)
|
{
|
switch (code)
|
{
|
case CODE::ROW:
|
rows_.push_back(row);
|
bytesRead += rows_.back().Read(data+bytesRead);
|
break;
|
|
case CODE::BLANK:
|
case CODE::BOOLERR:
|
case CODE::LABELSST:
|
case CODE::MULBLANK:
|
case CODE::MULRK:
|
case CODE::NUMBER:
|
case CODE::RK:
|
case CODE::FORMULA:
|
cellBlocks_.push_back(cellBlock);
|
if (cellBlocks_.size()%1000==0) cellBlocks_.reserve(cellBlocks_.size()+1000);
|
bytesRead += cellBlocks_[cellBlocks_.size()-1].Read(data+bytesRead);
|
break;
|
|
default:
|
Record rec;
|
bytesRead += rec.Read(data+bytesRead);
|
}
|
LittleEndian::Read(data, code, bytesRead, 2);
|
}
|
bytesRead += dbcell_.Read(data+bytesRead);
|
return bytesRead;
|
}
|
size_t Worksheet::CellTable::RowBlock::Write(char* data)
|
{
|
size_t bytesWritten = 0;
|
size_t maxRows = rows_.size();
|
{for (size_t i=0; i<maxRows; ++i)
|
{
|
bytesWritten += rows_[i].Write(data+bytesWritten);
|
}}
|
|
size_t maxCellBlocks = cellBlocks_.size();
|
{for (size_t i=0; i<maxCellBlocks; ++i)
|
{
|
bytesWritten += cellBlocks_[i].Write(data+bytesWritten);
|
}}
|
|
bytesWritten += dbcell_.Write(data+bytesWritten);
|
return bytesWritten;
|
}
|
size_t Worksheet::CellTable::RowBlock::DataSize()
|
{
|
size_t dataSize = 0;
|
size_t maxRows = rows_.size();
|
{for (size_t i=0; i<maxRows; ++i) dataSize += rows_[i].RecordSize();}
|
|
size_t maxCellBlocks = cellBlocks_.size();
|
{for (size_t i=0; i<maxCellBlocks; ++i) dataSize += cellBlocks_[i].RecordSize();}
|
|
dataSize += dbcell_.RecordSize();
|
return dataSize;
|
}
|
size_t Worksheet::CellTable::RowBlock::RecordSize() {return DataSize();}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
size_t Worksheet::CellTable::Read(const char* data)
|
{
|
size_t bytesRead = 0;
|
|
short code;
|
LittleEndian::Read(data, code, 0, 2);
|
RowBlock rowBlock;
|
rowBlocks_.reserve(1000);
|
while (code == CODE::ROW)
|
{
|
rowBlocks_.push_back(rowBlock);
|
bytesRead += rowBlocks_.back().Read(data+bytesRead);
|
LittleEndian::Read(data, code, bytesRead, 2);
|
}
|
return bytesRead;
|
}
|
size_t Worksheet::CellTable::Write(char* data)
|
{
|
size_t bytesWritten = 0;
|
size_t maxRowBlocks_ = rowBlocks_.size();
|
for (size_t i=0; i<maxRowBlocks_; ++i)
|
{
|
bytesWritten += rowBlocks_[i].Write(data+bytesWritten);
|
}
|
return bytesWritten;
|
}
|
size_t Worksheet::CellTable::DataSize()
|
{
|
size_t dataSize = 0;
|
size_t maxRowBlocks_ = rowBlocks_.size();
|
for (size_t i=0; i<maxRowBlocks_; ++i) dataSize += rowBlocks_[i].RecordSize();
|
return dataSize;
|
}
|
size_t Worksheet::CellTable::RecordSize() {return DataSize();}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
Worksheet::Window2::Window2() : Record(),
|
options_(1718), firstVisibleRowIndex_(0), firstVisibleColIndex_(0), gridLineColourIndex_(64),
|
unused1_(0), magnificationFactorPageBreakPreview_(0), magnificationFactorNormalView_(0), unused2_(0)
|
{code_ = CODE::WINDOW2; dataSize_ = 18; recordSize_ = 22;}
|
|
size_t Worksheet::Window2::Read(const char* data)
|
{
|
Record::Read(data);
|
LittleEndian::Read(data_, options_, 0, 2);
|
LittleEndian::Read(data_, firstVisibleRowIndex_, 2, 2);
|
LittleEndian::Read(data_, firstVisibleColIndex_, 4, 2);
|
LittleEndian::Read(data_, gridLineColourIndex_, 6, 2);
|
LittleEndian::Read(data_, unused1_, 8, 2);
|
LittleEndian::Read(data_, magnificationFactorPageBreakPreview_, 10, 2);
|
LittleEndian::Read(data_, magnificationFactorNormalView_, 12, 2);
|
LittleEndian::Read(data_, unused2_, 14, 4);
|
return RecordSize();
|
}
|
size_t Worksheet::Window2::Write(char* data)
|
{
|
data_.resize(DataSize());
|
LittleEndian::Write(data_, options_, 0, 2);
|
LittleEndian::Write(data_, firstVisibleRowIndex_, 2, 2);
|
LittleEndian::Write(data_, firstVisibleColIndex_, 4, 2);
|
LittleEndian::Write(data_, gridLineColourIndex_, 6, 2);
|
LittleEndian::Write(data_, unused1_, 8, 2);
|
LittleEndian::Write(data_, magnificationFactorPageBreakPreview_, 10, 2);
|
LittleEndian::Write(data_, magnificationFactorNormalView_, 12, 2);
|
LittleEndian::Write(data_, unused2_, 14, 4);
|
return Record::Write(data);
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
// Returns true if the supplied rk value contains an integer.
|
bool IsRKValueAnInteger(int rkValue)
|
{
|
return (rkValue & 2);
|
}
|
|
// Returns true if the supplied rk value contains a double.
|
bool IsRKValueADouble(int rkValue)
|
{
|
return !(rkValue & 2);
|
}
|
|
// Convert a rk value to a double.
|
double GetDoubleFromRKValue(int rkValue)
|
{
|
union
|
{
|
long long intvalue_;
|
double doublevalue_;
|
} intdouble;
|
|
bool isMultiplied = rkValue & 1;
|
rkValue >>= 2;
|
intdouble.intvalue_ = rkValue;
|
intdouble.intvalue_ <<= 34;
|
if (isMultiplied) intdouble.doublevalue_ *= 0.01;
|
return intdouble.doublevalue_;
|
}
|
|
// Convert a rk value to an integer.
|
int GetIntegerFromRKValue(int rkValue)
|
{
|
bool isMultiplied = rkValue & 1;
|
rkValue >>= 2;
|
if (isMultiplied) rkValue *= 0.01;
|
return rkValue;
|
}
|
|
// Convert a double to a rk value.
|
int GetRKValueFromDouble(double value)
|
{
|
union
|
{
|
long long intvalue_;
|
double doublevalue_;
|
} intdouble;
|
|
bool isMultiplied = false;
|
int testVal1 = value;
|
testVal1 *= 100;
|
int testVal2 = value * 100;
|
if (testVal1!=testVal2)
|
{
|
isMultiplied = true;
|
value *= 100;
|
}
|
|
intdouble.doublevalue_ = value;
|
intdouble.intvalue_ >>= 34;
|
|
int rkValue = intdouble.intvalue_;
|
rkValue <<= 2;
|
rkValue |= isMultiplied;
|
return rkValue;
|
}
|
|
// Convert an integer to a rk value.
|
int GetRKValueFromInteger(int value)
|
{
|
value <<= 2;
|
value |= 2;
|
return value;
|
}
|
|
// Returns true if the supplied double can be stored as a rk value.
|
bool CanStoreAsRKValue(double value)
|
{
|
int testVal1 = value * 100;
|
testVal1 *= 100;
|
int testVal2 = value * 10000;
|
if (testVal1!=testVal2) return false;
|
else return true;
|
}
|
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
BasicExcel::BasicExcel() {};
|
BasicExcel::BasicExcel(const char* filename)
|
{
|
Load(filename);
|
}
|
|
BasicExcel::~BasicExcel()
|
{
|
if (file_.IsOpen()) file_.Close();
|
}
|
|
// Create a new Excel workbook with a given number of spreadsheets (Minimum 1)
|
void BasicExcel::New(int sheets)
|
{
|
workbook_ = Workbook();
|
worksheets_.clear();
|
|
workbook_.fonts_.resize(4);
|
workbook_.XFs_.resize(21);
|
workbook_.styles_.resize(6);
|
workbook_.boundSheets_.resize(1);
|
worksheets_.resize(1);
|
UpdateYExcelWorksheet();
|
|
for (int i=0; i<sheets-1; ++i) AddWorksheet();
|
}
|
|
// Load an Excel workbook from a file.
|
bool BasicExcel::Load(const char* filename)
|
{
|
if (file_.IsOpen()) file_.Close();
|
if (file_.Open(filename))
|
{
|
workbook_ = Workbook();
|
worksheets_.clear();
|
|
vector<char> data;
|
file_.ReadFile("Workbook", data);
|
Read(&*(data.begin()), data.size());
|
UpdateYExcelWorksheet();
|
return true;
|
}
|
else return false;
|
}
|
|
bool BasicExcel::Load(const wchar_t* filename)
|
{
|
if (file_.IsOpen()) file_.Close();
|
if (file_.Open(filename))
|
{
|
workbook_ = Workbook();
|
worksheets_.clear();
|
|
vector<char> data;
|
file_.ReadFile("Workbook", data);
|
Read(&*(data.begin()), data.size());
|
UpdateYExcelWorksheet();
|
return true;
|
}
|
else return false;
|
}
|
|
// Save current Excel workbook to opened file.
|
bool BasicExcel::Save()
|
{
|
if (file_.IsOpen())
|
{
|
// Prepare Raw Worksheets for saving.
|
UpdateWorksheets();
|
|
AdjustStreamPositions();
|
|
// Calculate bytes needed for a workbook.
|
size_t minBytes = workbook_.RecordSize();
|
size_t maxWorkSheets = worksheets_.size();
|
for (size_t i=0; i<maxWorkSheets; ++i)
|
{
|
minBytes += worksheets_[i].RecordSize();
|
}
|
|
// Create new workbook.
|
vector<char> data(minBytes,0);
|
Write(&*(data).begin());
|
|
if (file_.WriteFile("Workbook", data, data.size())!=CompoundFile::SUCCESS) return false;
|
return true;
|
}
|
else return false;
|
}
|
|
// Save current Excel workbook to a file.
|
bool BasicExcel::SaveAs(const char* filename)
|
{
|
if (file_.IsOpen()) file_.Close();
|
|
if (!file_.Create(filename)) return false;
|
if (file_.MakeFile("Workbook")!=CompoundFile::SUCCESS) return false;
|
|
return Save();
|
}
|
|
// Total number of Excel worksheets in current Excel workbook.
|
size_t BasicExcel::GetTotalWorkSheets()
|
{
|
return worksheets_.size();
|
}
|
|
// Get a pointer to an Excel worksheet at the given index.
|
// Index starts from 0.
|
// Returns 0 if index is invalid.
|
BasicExcelWorksheet* BasicExcel::GetWorksheet(size_t sheetIndex)
|
{
|
return &(yesheets_[sheetIndex]);
|
}
|
|
// Get a pointer to an Excel worksheet that has given ANSI name.
|
// Returns 0 if there is no Excel worksheet with the given name.
|
BasicExcelWorksheet* BasicExcel::GetWorksheet(const char* name)
|
{
|
size_t maxWorksheets = yesheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (workbook_.boundSheets_[i].name_.unicode_ & 1)
|
{
|
size_t len = strlen(name);
|
wstring wsName(len + 1, 0);
|
mbstowcs(&wsName[0], name, len);
|
if (wcsncmp(&wsName[0], workbook_.boundSheets_[i].name_.wname_, wcslen(&wsName[0])) == 0) return &(yesheets_[i]);
|
} else {
|
if (strcmp(name, workbook_.boundSheets_[i].name_.name_) == 0) return &(yesheets_[i]);
|
}
|
}
|
|
|
return 0;
|
}
|
|
// Get a pointer to an Excel worksheet that has given Unicode name.
|
// Returns 0 if there is no Excel worksheet with the given name.
|
BasicExcelWorksheet* BasicExcel::GetWorksheet(const wchar_t* name)
|
{
|
size_t maxWorksheets = yesheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (!(workbook_.boundSheets_[i].name_.unicode_ & 1))
|
{
|
size_t len = wcslen(name) * 2;
|
string sName(len + 1, 0);
|
wcstombs(&sName[0], name, len);
|
if (strncmp(&sName[0], workbook_.boundSheets_[i].name_.name_, strlen(&sName[0])) == 0) return &(yesheets_[i]);
|
} else {
|
if (wcscmp(name, workbook_.boundSheets_[i].name_.wname_) == 0) return &(yesheets_[i]);
|
}
|
}
|
return 0;
|
}
|
|
// Add a new Excel worksheet to the given index.
|
// Name given to worksheet is SheetX, where X is a number which starts from 1.
|
// Index starts from 0.
|
// Worksheet is added to the last position if sheetIndex == -1.
|
// Returns a pointer to the worksheet if successful, 0 if otherwise.
|
BasicExcelWorksheet* BasicExcel::AddWorksheet(int sheetIndex)
|
{
|
size_t sheetNo = yesheets_.size() + 1;
|
BasicExcelWorksheet* yesheet = 0;
|
do
|
{
|
char sname[50];
|
sprintf(sname, "Sheet%d", sheetNo++);
|
yesheet = AddWorksheet(sname, sheetIndex);
|
} while (!yesheet);
|
return yesheet;
|
}
|
|
// Add a new Excel worksheet with given ANSI name to the given index.
|
// Index starts from 0.
|
// Worksheet is added to the last position if sheetIndex == -1.
|
// Returns a pointer to the worksheet if successful, 0 if otherwise.
|
BasicExcelWorksheet* BasicExcel::AddWorksheet(const char* name, int sheetIndex)
|
{
|
size_t maxWorksheets = yesheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (workbook_.boundSheets_[i].name_.unicode_ & 1) continue;
|
if (strcmp(name, workbook_.boundSheets_[i].name_.name_) == 0) return 0;
|
}
|
|
Workbook::BoundSheet* boundSheet;
|
Worksheet* worksheet;
|
BasicExcelWorksheet* yesheet;
|
if (sheetIndex == -1)
|
{
|
workbook_.boundSheets_.push_back(Workbook::BoundSheet());
|
worksheets_.push_back(Worksheet());
|
yesheets_.push_back(BasicExcelWorksheet(this, worksheets_.size()-1));
|
boundSheet = &(workbook_.boundSheets_.back());
|
worksheet = &(worksheets_.back());
|
yesheet = &(yesheets_.back());
|
}
|
else
|
{
|
boundSheet = &*(workbook_.boundSheets_.insert(workbook_.boundSheets_.begin()+sheetIndex, Workbook::BoundSheet()));
|
worksheet = &*(worksheets_.insert(worksheets_.begin()+sheetIndex, Worksheet()));
|
yesheet = &*(yesheets_.insert(yesheets_.begin()+sheetIndex, BasicExcelWorksheet(this, sheetIndex)));
|
size_t maxSheets = worksheets_.size();
|
for (size_t i=sheetIndex+1; i<maxSheets; ++i)
|
{
|
yesheets_[i].sheetIndex_ = i;
|
}
|
}
|
boundSheet->name_ = name;
|
worksheet->window2_.options_ &= ~0x200;
|
return yesheet;
|
}
|
|
// Add a new Excel worksheet with given Unicode name to the given index.
|
// Index starts from 0.
|
// Worksheet is added to the last position if sheetIndex == -1.
|
// Returns a pointer to the worksheet if successful, 0 if otherwise.
|
BasicExcelWorksheet* BasicExcel::AddWorksheet(const wchar_t* name, int sheetIndex)
|
{
|
size_t maxWorksheets = yesheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (!(workbook_.boundSheets_[i].name_.unicode_ & 1)) continue;
|
if (wcscmp(name, workbook_.boundSheets_[i].name_.wname_) == 0) return 0;
|
}
|
|
Workbook::BoundSheet* boundSheet;
|
Worksheet* worksheet;
|
BasicExcelWorksheet* yesheet;
|
if (sheetIndex == -1)
|
{
|
workbook_.boundSheets_.push_back(Workbook::BoundSheet());
|
worksheets_.push_back(Worksheet());
|
yesheets_.push_back(BasicExcelWorksheet(this, worksheets_.size()-1));
|
boundSheet = &(workbook_.boundSheets_.back());
|
worksheet = &(worksheets_.back());
|
yesheet = &(yesheets_.back());
|
}
|
else
|
{
|
boundSheet = &*(workbook_.boundSheets_.insert(workbook_.boundSheets_.begin()+sheetIndex, Workbook::BoundSheet()));
|
worksheet = &*(worksheets_.insert(worksheets_.begin()+sheetIndex, Worksheet()));
|
yesheet = &*(yesheets_.insert(yesheets_.begin()+sheetIndex, BasicExcelWorksheet(this, sheetIndex)));
|
size_t maxSheets = worksheets_.size();
|
for (size_t i=sheetIndex+1; i<maxSheets; ++i)
|
{
|
yesheets_[i].sheetIndex_ = i;
|
}
|
}
|
boundSheet->name_ = name;
|
worksheet->window2_.options_ &= ~0x200;
|
return yesheet;
|
}
|
|
// Delete an Excel worksheet at the given index.
|
// Index starts from 0.
|
// Returns true if successful, false if otherwise.
|
bool BasicExcel::DeleteWorksheet(size_t sheetIndex)
|
{
|
if (sheetIndex<workbook_.boundSheets_.size())
|
{
|
workbook_.boundSheets_.erase(workbook_.boundSheets_.begin()+sheetIndex);
|
worksheets_.erase(worksheets_.begin()+sheetIndex);
|
yesheets_.erase(yesheets_.begin()+sheetIndex);
|
return true;
|
}
|
else return false;
|
}
|
|
// Delete an Excel worksheet that has given ANSI name.
|
// Returns true if successful, false if otherwise.
|
bool BasicExcel::DeleteWorksheet(const char* name)
|
{
|
size_t maxWorksheets = yesheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (workbook_.boundSheets_[i].name_.unicode_ & 1) continue;
|
if (strcmp(name, workbook_.boundSheets_[i].name_.name_) == 0) return DeleteWorksheet(i);
|
}
|
return false;
|
}
|
|
// Delete an Excel worksheet that has given Unicode name.
|
// Returns true if successful, false if otherwise.
|
bool BasicExcel::DeleteWorksheet(const wchar_t* name)
|
{
|
size_t maxWorksheets = worksheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (!(workbook_.boundSheets_[i].name_.unicode_ & 1)) continue;
|
if (wcscmp(name, workbook_.boundSheets_[i].name_.wname_) == 0) return DeleteWorksheet(i);
|
}
|
return false;
|
}
|
|
// Get the worksheet name at the given index.
|
// Index starts from 0.
|
// Returns 0 if name is in Unicode format.
|
char* BasicExcel::GetAnsiSheetName(size_t sheetIndex)
|
{
|
if (!(workbook_.boundSheets_[sheetIndex].name_.unicode_ & 1))
|
{
|
return workbook_.boundSheets_[sheetIndex].name_.name_;
|
}
|
else return 0;
|
}
|
|
// Get the worksheet name at the given index.
|
// Index starts from 0.
|
// Returns 0 if name is in Ansi format.
|
wchar_t* BasicExcel::GetUnicodeSheetName(size_t sheetIndex)
|
{
|
if (workbook_.boundSheets_[sheetIndex].name_.unicode_ & 1)
|
{
|
return workbook_.boundSheets_[sheetIndex].name_.wname_;
|
}
|
else return 0;
|
}
|
|
// Get the worksheet name at the given index.
|
// Index starts from 0.
|
// Returns false if name is in Unicode format.
|
bool BasicExcel::GetSheetName(size_t sheetIndex, char* name)
|
{
|
if (!(workbook_.boundSheets_[sheetIndex].name_.unicode_ & 1))
|
{
|
strcpy(name, workbook_.boundSheets_[sheetIndex].name_.name_);
|
return true;
|
}
|
else return false;
|
}
|
|
// Get the worksheet name at the given index.
|
// Index starts from 0.
|
// Returns false if name is in Ansi format.
|
bool BasicExcel::GetSheetName(size_t sheetIndex, wchar_t* name)
|
{
|
if (workbook_.boundSheets_[sheetIndex].name_.unicode_ & 1)
|
{
|
wcscpy(name, workbook_.boundSheets_[sheetIndex].name_.wname_);
|
return true;
|
}
|
else return false;
|
}
|
|
// Rename an Excel worksheet at the given index to the given ANSI name.
|
// Index starts from 0.
|
// Returns true if successful, false if otherwise.
|
bool BasicExcel::RenameWorksheet(size_t sheetIndex, const char* to)
|
{
|
size_t maxWorksheets = yesheets_.size();
|
if (sheetIndex < maxWorksheets)
|
{
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (workbook_.boundSheets_[i].name_.unicode_ & 1) continue;
|
if (strcmp(to, workbook_.boundSheets_[i].name_.name_) == 0) return false;
|
}
|
workbook_.boundSheets_[sheetIndex].name_ = to;
|
return true;
|
}
|
else return false;
|
}
|
|
// Rename an Excel worksheet at the given index to the given Unicode name.
|
// Index starts from 0.
|
// Returns true if successful, false if otherwise.
|
bool BasicExcel::RenameWorksheet(size_t sheetIndex, const wchar_t* to)
|
{
|
size_t maxWorksheets = yesheets_.size();
|
if (sheetIndex < maxWorksheets)
|
{
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (!(workbook_.boundSheets_[i].name_.unicode_ & 1)) continue;
|
if (wcscmp(to, workbook_.boundSheets_[i].name_.wname_) == 0) return false;
|
}
|
workbook_.boundSheets_[sheetIndex].name_ = to;
|
return true;
|
}
|
else return false;
|
}
|
|
// Rename an Excel worksheet that has given ANSI name to another ANSI name.
|
// Returns true if successful, false if otherwise.
|
bool BasicExcel::RenameWorksheet(const char* from, const char* to)
|
{
|
size_t maxWorksheets = yesheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (workbook_.boundSheets_[i].name_.unicode_ & 1) continue;
|
if (strcmp(from, workbook_.boundSheets_[i].name_.name_) == 0)
|
{
|
for (size_t j=0; j<maxWorksheets; ++j)
|
{
|
if (workbook_.boundSheets_[j].name_.unicode_ & 1) continue;
|
if (strcmp(to, workbook_.boundSheets_[j].name_.name_) == 0) return false;
|
}
|
workbook_.boundSheets_[i].name_ = to;
|
return true;
|
}
|
}
|
return false;
|
}
|
|
// Rename an Excel worksheet that has given Unicode name to another Unicode name.
|
// Returns true if successful, false if otherwise.
|
bool BasicExcel::RenameWorksheet(const wchar_t* from, const wchar_t* to)
|
{
|
size_t maxWorksheets = worksheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (!(workbook_.boundSheets_[i].name_.unicode_ & 1)) continue;
|
if (wcscmp(from, workbook_.boundSheets_[i].name_.wname_) == 0)
|
{
|
for (size_t j=0; j<maxWorksheets; ++j)
|
{
|
if (!(workbook_.boundSheets_[j].name_.unicode_ & 1)) continue;
|
if (wcscmp(to, workbook_.boundSheets_[j].name_.wname_) == 0) return false;
|
}
|
workbook_.boundSheets_[i].name_ = to;
|
return true;
|
}
|
}
|
return false;
|
}
|
|
size_t BasicExcel::Read(const char* data, size_t dataSize)
|
{
|
size_t bytesRead = 0;
|
BOF bof;
|
short code;
|
LittleEndian::Read(data, code, 0, 2);
|
Record rec;
|
while (code == CODE::BOF)
|
{
|
bof.Read(data+bytesRead);
|
switch (bof.type_)
|
{
|
case WORKBOOK_GLOBALS:
|
bytesRead += workbook_.Read(data+bytesRead);
|
break;
|
|
case VISUAL_BASIC_MODULE:
|
bytesRead += rec.Read(data+bytesRead);
|
break;
|
|
case WORKSHEET:
|
worksheets_.push_back(Worksheet());
|
bytesRead += worksheets_.back().Read(data+bytesRead);
|
break;
|
|
case CHART:
|
bytesRead += rec.Read(data+bytesRead);
|
break;
|
|
default:
|
bytesRead += rec.Read(data+bytesRead);
|
break;
|
}
|
if (bytesRead < dataSize) LittleEndian::Read(data, code, bytesRead, 2);
|
else break;
|
}
|
return bytesRead;
|
}
|
|
size_t BasicExcel::Write(char* data)
|
{
|
size_t bytesWritten = 0;
|
bytesWritten += workbook_.Write(data+bytesWritten);
|
|
size_t maxWorkSheets = worksheets_.size();
|
for (size_t i=0; i<maxWorkSheets; ++i)
|
{
|
bytesWritten += worksheets_[i].Write(data+bytesWritten);
|
}
|
return bytesWritten;
|
}
|
|
void BasicExcel::AdjustStreamPositions()
|
{
|
// AdjustExtSSTPositions();
|
AdjustBoundSheetBOFPositions();
|
AdjustDBCellPositions();
|
}
|
|
void BasicExcel::AdjustBoundSheetBOFPositions()
|
{
|
size_t offset = workbook_.RecordSize();
|
size_t maxBoundSheets = workbook_.boundSheets_.size();
|
for (size_t i=0; i<maxBoundSheets; ++i)
|
{
|
workbook_.boundSheets_[i].BOFpos_ = offset;
|
offset += worksheets_[i].RecordSize();
|
}
|
}
|
|
void BasicExcel::AdjustDBCellPositions()
|
{
|
size_t offset = workbook_.RecordSize();
|
size_t maxSheets = worksheets_.size();
|
for (size_t i=0; i<maxSheets; ++i)
|
{
|
offset += worksheets_[i].bof_.RecordSize();
|
offset += worksheets_[i].index_.RecordSize();
|
offset += worksheets_[i].dimensions_.RecordSize();
|
|
size_t maxRowBlocks_ = worksheets_[i].cellTable_.rowBlocks_.size();
|
for (size_t j=0; j<maxRowBlocks_; ++j)
|
{
|
size_t firstRowOffset = 0;
|
|
size_t maxRows = worksheets_[i].cellTable_.rowBlocks_[j].rows_.size();
|
{for (size_t k=0; k<maxRows; ++k)
|
{
|
offset += worksheets_[i].cellTable_.rowBlocks_[j].rows_[k].RecordSize();
|
firstRowOffset += worksheets_[i].cellTable_.rowBlocks_[j].rows_[k].RecordSize();
|
}}
|
size_t cellOffset = firstRowOffset - 20; // a ROW record is 20 bytes long
|
|
size_t maxCellBlocks = worksheets_[i].cellTable_.rowBlocks_[j].cellBlocks_.size();
|
{for (size_t k=0; k<maxCellBlocks; ++k)
|
{
|
offset += worksheets_[i].cellTable_.rowBlocks_[j].cellBlocks_[k].RecordSize();
|
firstRowOffset += worksheets_[i].cellTable_.rowBlocks_[j].cellBlocks_[k].RecordSize();
|
}}
|
|
// Adjust Index DBCellPos_ absolute offset
|
worksheets_[i].index_.DBCellPos_[j] = offset;
|
|
offset += worksheets_[i].cellTable_.rowBlocks_[j].dbcell_.RecordSize();
|
|
// Adjust DBCell first row offsets
|
worksheets_[i].cellTable_.rowBlocks_[j].dbcell_.firstRowOffset_ = firstRowOffset;
|
|
// Adjust DBCell offsets
|
size_t l=0;
|
{for (size_t k=0; k<maxRows; ++k)
|
{
|
for (; l<maxCellBlocks; ++l)
|
{
|
if (worksheets_[i].cellTable_.rowBlocks_[j].rows_[k].rowIndex_ <=
|
worksheets_[i].cellTable_.rowBlocks_[j].cellBlocks_[l].RowIndex())
|
{
|
worksheets_[i].cellTable_.rowBlocks_[j].dbcell_.offsets_[k] = cellOffset;
|
break;
|
}
|
cellOffset += worksheets_[i].cellTable_.rowBlocks_[j].cellBlocks_[l].RecordSize();
|
}
|
cellOffset = 0;
|
}}
|
}
|
|
offset += worksheets_[i].cellTable_.RecordSize();
|
offset += worksheets_[i].window2_.RecordSize();
|
offset += worksheets_[i].eof_.RecordSize();
|
}
|
}
|
|
void BasicExcel::AdjustExtSSTPositions()
|
{
|
size_t offset = workbook_.bof_.RecordSize();
|
offset += workbook_.bof_.RecordSize();
|
offset += workbook_.window1_.RecordSize();
|
|
size_t maxFonts = workbook_.fonts_.size();
|
{for (size_t i=0; i<maxFonts; ++i) {offset += workbook_.fonts_[i].RecordSize();}}
|
|
size_t maxXFs = workbook_.XFs_.size();
|
{for (size_t i=0; i<maxXFs; ++i) {offset += workbook_.XFs_[i].RecordSize();}}
|
|
size_t maxStyles = workbook_.styles_.size();
|
{for (size_t i=0; i<maxStyles; ++i) {offset += workbook_.styles_[i].RecordSize();}}
|
|
size_t maxBoundSheets = workbook_.boundSheets_.size();
|
{for (size_t i=0; i<maxBoundSheets; ++i) {offset += workbook_.boundSheets_[i].RecordSize();}}
|
|
workbook_.extSST_.stringsTotal_ = 10;
|
size_t maxPortions = workbook_.sst_.uniqueStringsTotal_ / workbook_.extSST_.stringsTotal_ +
|
(workbook_.sst_.uniqueStringsTotal_%workbook_.extSST_.stringsTotal_ ? 1 : 0);
|
workbook_.extSST_.streamPos_.resize(maxPortions);
|
workbook_.extSST_.firstStringPos_.resize(maxPortions);
|
workbook_.extSST_.unused_.resize(maxPortions);
|
|
size_t relativeOffset = 8;
|
for (size_t i=0; i<maxPortions; ++i)
|
{
|
workbook_.extSST_.streamPos_[i] = offset + 4 + relativeOffset;
|
workbook_.extSST_.firstStringPos_[i] = 4 + relativeOffset;
|
workbook_.extSST_.unused_[i] = 0;
|
|
for (size_t j=0; j<workbook_.extSST_.stringsTotal_; ++j)
|
{
|
if (i*workbook_.extSST_.stringsTotal_+j >= workbook_.sst_.strings_.size()) break;
|
size_t stringSize = workbook_.sst_.strings_[i*workbook_.extSST_.stringsTotal_+j].StringSize();
|
if (relativeOffset+stringSize+3 < 8224)
|
{
|
relativeOffset += stringSize + 3;
|
}
|
else
|
{
|
// If have >= 12 bytes (2 for size, 1 for unicode and >=9 for data, can split string
|
// otherwise, end record and start continue record.
|
if (8224 - relativeOffset >= 12)
|
{
|
stringSize -= (8224 - relativeOffset - 3);
|
offset += 12 + relativeOffset;
|
relativeOffset = 0;
|
|
size_t additionalContinueRecords = stringSize / 8223; // 8223 because the first byte is for unicode
|
for (size_t k=0; k<additionalContinueRecords; ++k)
|
{
|
stringSize -= 8223;
|
}
|
relativeOffset += stringSize + 1;
|
}
|
else
|
{
|
if (relativeOffset+stringSize+3 < 8224)
|
{
|
relativeOffset += stringSize + 3;
|
}
|
else
|
{
|
// If have >= 12 bytes (2 for size, 1 for unicode and >=9 for data, can split string
|
// otherwise, end record and start continue record.
|
if (8224 - relativeOffset >= 12)
|
{
|
stringSize -= (8224 - relativeOffset - 3);
|
offset += 12 + relativeOffset;
|
relativeOffset = 0;
|
|
size_t additionalContinueRecords = stringSize / 8223; // 8223 because the first byte is for unicode
|
for (size_t k=0; k<additionalContinueRecords; ++k)
|
{
|
stringSize -= 8223;
|
}
|
relativeOffset += stringSize + 1;
|
}
|
}
|
}
|
}
|
}
|
}
|
}
|
|
// Update yesheets_ using information from worksheets_.
|
void BasicExcel::UpdateYExcelWorksheet()
|
{
|
size_t maxWorksheets = worksheets_.size();
|
yesheets_.clear();
|
yesheets_.reserve(maxWorksheets);
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
yesheets_.push_back(BasicExcelWorksheet(this,i));
|
}
|
}
|
|
// Update worksheets_ using information from yesheets_.
|
void BasicExcel::UpdateWorksheets()
|
{
|
// Constants.
|
const size_t maxWorksheets = yesheets_.size();
|
Worksheet::CellTable::RowBlock rowBlock;
|
Worksheet::CellTable::RowBlock::CellBlock cellBlock;
|
Worksheet::CellTable::RowBlock::Row row;
|
Worksheet::CellTable::RowBlock::CellBlock::MulRK::XFRK xfrk;
|
LargeString largeString;
|
|
map<vector<char>, size_t> stringMap;
|
map<vector<char>, size_t>::iterator stringMapIt;
|
map<vector<wchar_t>, size_t> wstringMap;
|
map<vector<wchar_t>, size_t>::iterator wstringMapIt;
|
|
// Reset worksheets and string table.
|
worksheets_.clear();
|
worksheets_.resize(maxWorksheets);
|
|
workbook_.sst_.stringsTotal_ = 0;
|
workbook_.sst_.uniqueStringsTotal_ = 0;
|
workbook_.sst_.strings_.clear();
|
|
for (size_t s=0; s<maxWorksheets; ++s)
|
{
|
size_t maxRows = yesheets_[s].GetTotalRows();
|
size_t maxCols = yesheets_[s].GetTotalCols();
|
|
// Modify Index
|
worksheets_[s].index_.firstUsedRowIndex_ = 100000; // Use 100000 to indicate that firstUsedRowIndex is not set yet since maximum allowed rows in Excel is 65535.
|
worksheets_[s].index_.firstUnusedRowIndex_ = maxRows;
|
|
// Modify Dimensions
|
worksheets_[s].dimensions_.firstUsedRowIndex_ = 100000; // Use 100000 to indicate that firstUsedRowIndex is not set yet since maximum allowed rows in Excel is 65535.
|
worksheets_[s].dimensions_.firstUsedColIndex_ = 1000; // Use 1000 to indicate that firstUsedColIndex is not set yet since maximum allowed columns in Excel is 255.
|
worksheets_[s].dimensions_.lastUsedRowIndexPlusOne_ = maxRows;
|
worksheets_[s].dimensions_.lastUsedColIndexPlusOne_ = maxCols;
|
|
// Make first sheet selected and other sheets unselected
|
if (s > 0) worksheets_[s].window2_.options_ &= ~0x200;
|
|
// References and pointers to shorten code
|
vector<Worksheet::CellTable::RowBlock>& rRowBlocks = worksheets_[s].cellTable_.rowBlocks_;
|
vector<Worksheet::CellTable::RowBlock::CellBlock>* pCellBlocks = NULL;
|
Worksheet::CellTable::RowBlock::CellBlock* pCell;
|
rRowBlocks.resize(maxRows/32 + (maxRows%32 ? 1 : 0));
|
for (size_t r=0, curRowBlock=0; r<maxRows; ++r)
|
{
|
if (r%32==0)
|
{
|
// New row block for every 32 rows.
|
pCellBlocks = &(rRowBlocks[curRowBlock++].cellBlocks_);
|
}
|
bool newRow = true; // Keep track whether current row contains data.
|
pCellBlocks->reserve(1000);
|
for (size_t c=0; c<maxCols; ++c)
|
{
|
BasicExcelCell* cell = yesheets_[s].Cell(r,c);
|
int cellType = cell->Type();
|
if (cellType != BasicExcelCell::UNDEFINED) // Current cell contains some data
|
{
|
if (worksheets_[s].index_.firstUsedRowIndex_ == 100000)
|
{
|
// Set firstUsedRowIndex.
|
worksheets_[s].index_.firstUsedRowIndex_ = r;
|
worksheets_[s].dimensions_.firstUsedRowIndex_ = r;
|
|
// Resize DBCellPos.
|
size_t nm = int(worksheets_[s].index_.firstUnusedRowIndex_ - worksheets_[s].index_.firstUsedRowIndex_ - 1) / 32 + 1;
|
worksheets_[s].index_.DBCellPos_.resize(nm);
|
}
|
if (worksheets_[s].dimensions_.firstUsedColIndex_ == 1000)
|
{
|
// Set firstUsedColIndex.
|
worksheets_[s].dimensions_.firstUsedColIndex_ = c;
|
}
|
|
if (newRow)
|
{
|
// Prepare Row and DBCell for new row with data.
|
Worksheet::CellTable::RowBlock& rRowBlock = rRowBlocks[curRowBlock-1];
|
rRowBlock.rows_.push_back(row);
|
rRowBlock.rows_.back().rowIndex_ = r;
|
rRowBlock.rows_.back().lastCellColIndexPlusOne_ = maxCols;
|
rRowBlock.dbcell_.offsets_.push_back(0);
|
newRow = false;
|
}
|
|
// Create new cellblock to store cell.
|
pCellBlocks->push_back(cellBlock);
|
if (pCellBlocks->size()%1000==0) pCellBlocks->reserve(pCellBlocks->size()+1000);
|
pCell = &(pCellBlocks->back());
|
|
// Store cell.
|
switch(cellType)
|
{
|
case BasicExcelCell::INT:
|
{
|
// Check whether it is a single cell or range of cells.
|
size_t cl = c + 1;
|
for (; cl<maxCols; ++cl)
|
{
|
BasicExcelCell* cellNext = yesheets_[s].Cell(r,cl);
|
if (cellNext->Type()==BasicExcelCell::UNDEFINED ||
|
cellNext->Type()!=cell->Type()) break;
|
}
|
|
if (cl > c+1)
|
{
|
// MULRK cells
|
pCell->type_ = CODE::MULRK;
|
pCell->normalType_ = true;
|
pCell->mulrk_.rowIndex_ = r;
|
pCell->mulrk_.firstColIndex_ = c;
|
pCell->mulrk_.lastColIndex_ = cl - 1;
|
pCell->mulrk_.XFRK_.resize(cl-c);
|
for (size_t i=0; c<cl; ++c, ++i)
|
{
|
cell = yesheets_[s].Cell(r,c);
|
pCell->mulrk_.XFRK_[i].RKValue_ = GetRKValueFromInteger(cell->GetInteger());
|
}
|
--c;
|
}
|
else
|
{
|
// Single cell
|
pCell->normalType_ = true;
|
pCell->type_ = CODE::RK;
|
pCell->rk_.rowIndex_ = r;
|
pCell->rk_.colIndex_ = c;
|
pCell->rk_.value_ = GetRKValueFromInteger(cell->GetInteger());
|
}
|
break;
|
}
|
|
case BasicExcelCell::DOUBLE:
|
{
|
// Check whether it is a single cell or range of cells.
|
// Double values which cannot be stored as RK values will be stored as single cells.
|
bool canStoreAsRKValue = CanStoreAsRKValue(cell->GetDouble());
|
size_t cl = c + 1;
|
for (; cl<maxCols; ++cl)
|
{
|
BasicExcelCell* cellNext = yesheets_[s].Cell(r,cl);
|
if (cellNext->Type()==BasicExcelCell::UNDEFINED ||
|
cellNext->Type()!=cell->Type() ||
|
canStoreAsRKValue!=CanStoreAsRKValue(cellNext->GetDouble())) break;
|
}
|
|
if (cl > c+1 && canStoreAsRKValue)
|
{
|
// MULRK cells
|
pCell->type_ = CODE::MULRK;
|
pCell->normalType_ = true;
|
pCell->mulrk_.rowIndex_ = r;
|
pCell->mulrk_.firstColIndex_ = c;
|
pCell->mulrk_.lastColIndex_ = cl - 1;
|
pCell->mulrk_.XFRK_.resize(cl-c);
|
for (size_t i=0; c<cl; ++c, ++i)
|
{
|
cell = yesheets_[s].Cell(r,c);
|
pCell->mulrk_.XFRK_[i].RKValue_ = GetRKValueFromDouble(cell->GetDouble());
|
}
|
--c;
|
}
|
else
|
{
|
// Single cell
|
pCell->normalType_ = true;
|
if (canStoreAsRKValue)
|
{
|
pCell->type_ = CODE::RK;
|
pCell->rk_.rowIndex_ = r;
|
pCell->rk_.colIndex_ = c;
|
pCell->rk_.value_ = GetRKValueFromDouble(cell->GetDouble());
|
}
|
else
|
{
|
pCell->type_ = CODE::NUMBER;
|
pCell->number_.rowIndex_ = r;
|
pCell->number_.colIndex_ = c;
|
pCell->number_.value_ = cell->GetDouble();
|
}
|
}
|
break;
|
}
|
|
case BasicExcelCell::STRING:
|
{
|
// Fill cell information
|
pCell->type_ = CODE::LABELSST;
|
pCell->normalType_ = true;
|
pCell->labelsst_.rowIndex_ = r;
|
pCell->labelsst_.colIndex_ = c;
|
|
// Get cell string
|
vector<char> str(cell->GetStringLength()+1);
|
cell->Get(&*(str.begin()));
|
str.pop_back(); // Remove null character because LargeString does not store null character.
|
|
// Check if string is present in Shared string table.
|
++workbook_.sst_.stringsTotal_;
|
size_t maxUniqueStrings = workbook_.sst_.uniqueStringsTotal_;
|
size_t strIndex = 0;
|
stringMapIt = stringMap.find(str);
|
if (stringMapIt != stringMap.end()) strIndex = stringMapIt->second;
|
else strIndex = maxUniqueStrings;
|
|
if (strIndex < maxUniqueStrings)
|
{
|
// String is present in Shared string table.
|
pCell->labelsst_.SSTRecordIndex_ = strIndex;
|
}
|
else
|
{
|
// New unique string.
|
stringMap[str] = maxUniqueStrings;
|
workbook_.sst_.strings_.push_back(largeString);
|
workbook_.sst_.strings_[maxUniqueStrings].name_ = str;
|
workbook_.sst_.strings_[maxUniqueStrings].unicode_ = 0;
|
pCell->labelsst_.SSTRecordIndex_ = maxUniqueStrings;
|
++workbook_.sst_.uniqueStringsTotal_;
|
}
|
break;
|
}
|
|
case BasicExcelCell::WSTRING:
|
{
|
// Fill cell information
|
pCell->type_ = CODE::LABELSST;
|
pCell->normalType_ = true;
|
pCell->labelsst_.rowIndex_ = r;
|
pCell->labelsst_.colIndex_ = c;
|
|
// Get cell string
|
vector<wchar_t> str(cell->GetStringLength()+1);
|
cell->Get(&*(str.begin()));
|
str.pop_back(); // Remove null character because LargeString does not store null character.
|
|
// Check if string is present in Shared string table.
|
++workbook_.sst_.stringsTotal_;
|
size_t maxUniqueStrings = workbook_.sst_.strings_.size();
|
size_t strIndex = 0;
|
wstringMapIt = wstringMap.find(str);
|
if (wstringMapIt != wstringMap.end()) strIndex = wstringMapIt->second;
|
else strIndex = maxUniqueStrings;
|
|
if (strIndex < maxUniqueStrings)
|
{
|
// String is present in Shared string table.
|
pCell->labelsst_.SSTRecordIndex_ = strIndex;
|
}
|
else
|
{
|
// New unique string
|
wstringMap[str] = maxUniqueStrings;
|
workbook_.sst_.strings_.push_back(largeString);
|
workbook_.sst_.strings_[maxUniqueStrings].wname_ = str;
|
workbook_.sst_.strings_[maxUniqueStrings].unicode_ = 1;
|
pCell->labelsst_.SSTRecordIndex_ = maxUniqueStrings;
|
++workbook_.sst_.uniqueStringsTotal_;
|
}
|
break;
|
}
|
}
|
}
|
}
|
}
|
|
// If worksheet has no data
|
if (worksheets_[s].index_.firstUsedRowIndex_ == 100000)
|
{
|
// Set firstUsedRowIndex.
|
worksheets_[s].index_.firstUsedRowIndex_ = 0;
|
worksheets_[s].dimensions_.firstUsedRowIndex_ = 0;
|
|
// Resize DBCellPos.
|
size_t nm = int(worksheets_[s].index_.firstUnusedRowIndex_ - worksheets_[s].index_.firstUsedRowIndex_ - 1) / 32 + 1;
|
worksheets_[s].index_.DBCellPos_.resize(nm);
|
}
|
if (worksheets_[s].dimensions_.firstUsedColIndex_ == 1000)
|
{
|
// Set firstUsedColIndex.
|
worksheets_[s].dimensions_.firstUsedColIndex_ = 0;
|
}
|
}
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
BasicExcelWorksheet::BasicExcelWorksheet(BasicExcel* excel, size_t sheetIndex) :
|
excel_(excel), sheetIndex_(sheetIndex)
|
{
|
UpdateCells();
|
}
|
|
// Get the current worksheet name.
|
// Returns 0 if name is in Unicode format.
|
char* BasicExcelWorksheet::GetAnsiSheetName()
|
{
|
if (!(excel_->workbook_.boundSheets_[sheetIndex_].name_.unicode_ & 1))
|
{
|
return excel_->workbook_.boundSheets_[sheetIndex_].name_.name_;
|
}
|
else return 0;
|
}
|
|
// Get the current worksheet name.
|
// Returns 0 if name is in Ansi format.
|
wchar_t* BasicExcelWorksheet::GetUnicodeSheetName()
|
{
|
if (excel_->workbook_.boundSheets_[sheetIndex_].name_.unicode_ & 1)
|
{
|
return excel_->workbook_.boundSheets_[sheetIndex_].name_.wname_;
|
}
|
else return 0;
|
}
|
|
// Get the current worksheet name.
|
// Returns false if name is in Unicode format.
|
bool BasicExcelWorksheet::GetSheetName(char* name)
|
{
|
if (!(excel_->workbook_.boundSheets_[sheetIndex_].name_.unicode_ & 1))
|
{
|
strcpy(name, excel_->workbook_.boundSheets_[sheetIndex_].name_.name_);
|
return true;
|
}
|
else return false;
|
}
|
|
// Get the current worksheet name.
|
// Returns false if name is in Ansi format.
|
bool BasicExcelWorksheet::GetSheetName(wchar_t* name)
|
{
|
if (excel_->workbook_.boundSheets_[sheetIndex_].name_.unicode_ & 1)
|
{
|
wcscpy(name, excel_->workbook_.boundSheets_[sheetIndex_].name_.wname_);
|
return true;
|
}
|
else return false;
|
}
|
|
// Rename current Excel worksheet to another ANSI name.
|
// Returns true if successful, false if otherwise.
|
bool BasicExcelWorksheet::Rename(const char* to)
|
{
|
size_t maxWorksheets = excel_->workbook_.boundSheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (excel_->workbook_.boundSheets_[i].name_.unicode_ & 1) continue;
|
if (strcmp(to, excel_->workbook_.boundSheets_[i].name_.name_) == 0) return false;
|
}
|
|
excel_->workbook_.boundSheets_[sheetIndex_].name_ = to;
|
return true;
|
}
|
|
// Rename current Excel worksheet to another Unicode name.
|
// Returns true if successful, false if otherwise.
|
bool BasicExcelWorksheet::Rename(const wchar_t* to)
|
{
|
size_t maxWorksheets = excel_->workbook_.boundSheets_.size();
|
for (size_t i=0; i<maxWorksheets; ++i)
|
{
|
if (!(excel_->workbook_.boundSheets_[i].name_.unicode_ & 1)) continue;
|
if (wcscmp(to, excel_->workbook_.boundSheets_[i].name_.wname_) == 0) return false;
|
}
|
|
excel_->workbook_.boundSheets_[sheetIndex_].name_ = to;
|
return true;
|
}
|
|
///< Print entire worksheet to an output stream, separating each column with the defined delimiter and enclosing text using the defined textQualifier.
|
///< Leave out the textQualifier argument if do not wish to have any text qualifiers.
|
void BasicExcelWorksheet::Print(ostream& os, char delimiter, char textQualifier)
|
{
|
for (size_t r=0; r<maxRows_; ++r)
|
{
|
for (size_t c=0; c<maxCols_; ++c)
|
{
|
BasicExcelCell* cell = Cell(r,c);
|
switch (cell->Type())
|
{
|
case BasicExcelCell::UNDEFINED:
|
break;
|
|
case BasicExcelCell::INT:
|
os << cell->GetInteger();
|
break;
|
|
case BasicExcelCell::DOUBLE:
|
os << setprecision(15) << cell->GetDouble();
|
break;
|
|
case BasicExcelCell::STRING:
|
{
|
if (textQualifier != '\0')
|
{
|
// Get string.
|
size_t maxLength = cell->GetStringLength();
|
vector<char> cellString(maxLength+1);
|
cell->Get(&*(cellString.begin()));
|
|
// Duplicate textQualifier if found in string.
|
vector<char>::iterator it;
|
size_t npos = 0;
|
while ((it=find(cellString.begin()+npos, cellString.end(), textQualifier)) != cellString.end())
|
{
|
npos = distance(cellString.begin(), cellString.insert(it, textQualifier)) + 2;
|
}
|
|
// Print out string enclosed with textQualifier.
|
os << textQualifier << &*(cellString.begin()) << textQualifier;
|
}
|
else os << cell->GetString();
|
break;
|
}
|
|
case BasicExcelCell::WSTRING:
|
{
|
// Print out string enclosed with textQualifier (does not work).
|
//os << textQualifier << cell->GetWString() << textQualifier;
|
break;
|
}
|
}
|
if (c < maxCols_-1) os << delimiter;
|
}
|
os << endl;
|
}
|
}
|
|
// Total number of rows in current Excel worksheet.
|
size_t BasicExcelWorksheet::GetTotalRows()
|
{
|
return maxRows_;
|
}
|
|
// Total number of columns in current Excel worksheet.
|
size_t BasicExcelWorksheet::GetTotalCols()
|
{
|
return maxCols_;
|
}
|
|
// Return a pointer to an Excel cell.
|
// row and col starts from 0.
|
// Returns 0 if row exceeds 65535 or col exceeds 255.
|
BasicExcelCell* BasicExcelWorksheet::Cell(size_t row, size_t col)
|
{
|
// Check to ensure row and col does not exceed maximum allowable range for an Excel worksheet.
|
if (row>65535 || col>255) return 0;
|
|
// Increase size of cells matrix if necessary
|
if (col>=maxCols_)
|
{
|
// Increase number of columns.
|
maxCols_ = col + 1;
|
for (size_t i=0; i<maxRows_; ++i) cells_[i].resize(maxCols_);
|
}
|
if (row>=maxRows_)
|
{
|
// Increase number of rows.
|
maxRows_ = row + 1;
|
cells_.resize(maxRows_, vector<BasicExcelCell>(maxCols_));
|
}
|
|
return &(cells_[row][col]);
|
}
|
|
// Erase content of a cell. row and col starts from 0.
|
// Returns true if successful, false if row or col exceeds range.
|
bool BasicExcelWorksheet::EraseCell(size_t row, size_t col)
|
{
|
if (row<maxRows_ && col<maxCols_)
|
{
|
cells_[row][col].EraseContents();
|
return true;
|
}
|
else return false;
|
}
|
|
// Update cells using information from BasicExcel.worksheets_.
|
void BasicExcelWorksheet::UpdateCells()
|
{
|
// Define some reference
|
Worksheet::Dimensions& dimension = excel_->worksheets_[sheetIndex_].dimensions_;
|
vector<Worksheet::CellTable::RowBlock>& rRowBlocks = excel_->worksheets_[sheetIndex_].cellTable_.rowBlocks_;
|
|
vector<wchar_t> wstr;
|
vector<char> str;
|
|
maxRows_ = dimension.lastUsedRowIndexPlusOne_;
|
maxCols_ = dimension.lastUsedColIndexPlusOne_;
|
|
// Resize the cells to the size of the worksheet
|
vector<BasicExcelCell> cellCol(maxCols_);
|
cells_.resize(maxRows_, cellCol);
|
|
size_t maxRowBlocks = rRowBlocks.size();
|
for (size_t i=0; i<maxRowBlocks; ++i)
|
{
|
vector<Worksheet::CellTable::RowBlock::CellBlock>& rCellBlocks = rRowBlocks[i].cellBlocks_;
|
size_t maxCells = rCellBlocks.size();
|
for (size_t j=0; j<maxCells; ++j)
|
{
|
size_t row = rCellBlocks[j].RowIndex();
|
size_t col = rCellBlocks[j].ColIndex();
|
switch (rCellBlocks[j].type_)
|
{
|
case CODE::BLANK:
|
break;
|
|
case CODE::BOOLERR:
|
if (rCellBlocks[j].boolerr_.error_ == 0)
|
{
|
cells_[row][col].Set(rCellBlocks[j].boolerr_.value_);
|
}
|
break;
|
|
case CODE::LABELSST:
|
{
|
vector<LargeString>& ss = excel_->workbook_.sst_.strings_;
|
if (ss[rCellBlocks[j].labelsst_.SSTRecordIndex_].unicode_ & 1)
|
{
|
wstr = ss[rCellBlocks[j].labelsst_.SSTRecordIndex_].wname_;
|
wstr.resize(wstr.size()+1);
|
wstr.back() = L'\0';
|
cells_[row][col].Set(&*(wstr.begin()));
|
}
|
else
|
{
|
str = ss[rCellBlocks[j].labelsst_.SSTRecordIndex_].name_;
|
str.resize(str.size()+1);
|
str.back() = '\0';
|
cells_[row][col].Set(&*(str.begin()));
|
}
|
break;
|
}
|
|
case CODE::MULBLANK:
|
break;
|
|
case CODE::MULRK:
|
{
|
size_t maxCols = rCellBlocks[j].mulrk_.lastColIndex_ - rCellBlocks[j].mulrk_.firstColIndex_ + 1;
|
for (size_t k=0; k<maxCols; ++k, ++col)
|
{
|
// Get values of the whole range
|
int rkValue = rCellBlocks[j].mulrk_.XFRK_[k].RKValue_;
|
if (IsRKValueAnInteger(rkValue))
|
{
|
cells_[row][col].Set(GetIntegerFromRKValue(rkValue));
|
}
|
else
|
{
|
cells_[row][col].Set(GetDoubleFromRKValue(rkValue));
|
}
|
}
|
break;
|
}
|
|
case CODE::NUMBER:
|
cells_[row][col].Set(rCellBlocks[j].number_.value_);
|
break;
|
|
case CODE::RK:
|
{
|
int rkValue = rCellBlocks[j].rk_.value_;
|
if (IsRKValueAnInteger(rkValue))
|
{
|
cells_[row][col].Set(GetIntegerFromRKValue(rkValue));
|
}
|
else
|
{
|
cells_[row][col].Set(GetDoubleFromRKValue(rkValue));
|
}
|
break;
|
}
|
|
case CODE::FORMULA:
|
break;
|
}
|
}
|
}
|
}
|
/************************************************************************************************************/
|
|
/************************************************************************************************************/
|
BasicExcelCell::BasicExcelCell() : type_(UNDEFINED) {};
|
|
// Get type of value stored in current Excel cell.
|
// Returns one of the enums.
|
int BasicExcelCell::Type() const {return type_;}
|
|
// Get an integer value.
|
// Returns false if cell does not contain an integer or a double.
|
bool BasicExcelCell::Get(int& val) const
|
{
|
if (type_ == INT)
|
{
|
val = ival_;
|
return true;
|
}
|
else if (type_ == DOUBLE)
|
{
|
val = (int)dval_;
|
return true;
|
}
|
else return false;
|
}
|
|
// Get a double value.
|
// Returns false if cell does not contain a double or an integer.
|
bool BasicExcelCell::Get(double& val) const
|
{
|
if (type_ == DOUBLE)
|
{
|
val = dval_;
|
return true;
|
}
|
else if (type_ == INT)
|
{
|
val = (double)ival_;
|
return true;
|
}
|
else return false;
|
}
|
|
// Get an ANSI string.
|
// Returns false if cell does not contain an ANSI string.
|
bool BasicExcelCell::Get(char* str) const
|
{
|
if (type_ == STRING)
|
{
|
if (str_.empty()) *str = '\0';
|
else strcpy(str, &*(str_.begin()));
|
return true;
|
}
|
else return false;
|
}
|
|
// Get an Unicode string.
|
// Returns false if cell does not contain an Unicode string.
|
bool BasicExcelCell::Get(wchar_t* str) const
|
{
|
if (type_ == WSTRING)
|
{
|
if (wstr_.empty()) *str = L'\0';
|
else wcscpy(str, &*(wstr_.begin()));
|
return true;
|
}
|
else return false;
|
}
|
|
// Return length of ANSI or Unicode string (excluding null character).
|
size_t BasicExcelCell::GetStringLength() const
|
{
|
if (type_ == STRING) return str_.size() - 1;
|
else return wstr_.size() - 1;
|
}
|
|
// Get an integer value.
|
// Returns 0 if cell does not contain an integer.
|
int BasicExcelCell::GetInteger() const
|
{
|
int val;
|
if (Get(val)) return val;
|
else return 0;
|
}
|
|
// Get a double value.
|
// Returns 0.0 if cell does not contain a double.
|
double BasicExcelCell::GetDouble() const
|
{
|
double val;
|
if (Get(val)) return val;
|
else return 0.0;
|
}
|
|
// Get an ANSI string.
|
// Returns 0 if cell does not contain an ANSI string.
|
const char* BasicExcelCell::GetString() const
|
{
|
vector<char> str(str_.size());
|
if (!str.empty() && Get(&*(str.begin()))) return &*(str_.begin());
|
else return 0;
|
}
|
|
// Get an Unicode string.
|
// Returns 0 if cell does not contain an Unicode string.
|
const wchar_t* BasicExcelCell::GetWString() const
|
{
|
vector<wchar_t> wstr(wstr_.size());
|
if (!wstr.empty() && Get(&*(wstr.begin()))) return &*(wstr_.begin());
|
else return 0;
|
}
|
|
// Set content of current Excel cell to an integer.
|
void BasicExcelCell::Set(int val)
|
{
|
SetInteger(val);
|
}
|
|
// Set content of current Excel cell to a double.
|
void BasicExcelCell::Set(double val)
|
{
|
SetDouble(val);
|
}
|
|
// Set content of current Excel cell to an ANSI string.
|
void BasicExcelCell::Set(const char* str)
|
{
|
SetString(str);
|
}
|
|
// Set content of current Excel cell to an Unicode string.
|
void BasicExcelCell::Set(const wchar_t* str)
|
{
|
SetWString(str);
|
}
|
|
// Set content of current Excel cell to an integer.
|
void BasicExcelCell::SetInteger(int val)
|
{
|
type_ = INT;
|
ival_ = val;
|
}
|
|
// Set content of current Excel cell to a double.
|
void BasicExcelCell::SetDouble(double val)
|
{
|
type_ = DOUBLE;
|
dval_ = val;
|
}
|
|
// Set content of current Excel cell to an ANSI string.
|
void BasicExcelCell::SetString(const char* str)
|
{
|
size_t length = strlen(str);
|
if (length > 0)
|
{
|
type_ = WSTRING;
|
wstr_ = vector<wchar_t>(length + 1);
|
mbstowcs(&wstr_[0], str, length);
|
str_.clear();
|
|
//type_ = STRING;
|
//str_ = vector<char>(length+1);
|
//strcpy(&*(str_.begin()), str);
|
//wstr_.clear();
|
}
|
else EraseContents();
|
}
|
|
// Set content of current Excel cell to an Unicode string.
|
void BasicExcelCell::SetWString(const wchar_t* str)
|
{
|
size_t length = wcslen(str);
|
if (length > 0)
|
{
|
type_ = WSTRING;
|
wstr_ = vector<wchar_t>(length+1);
|
wcscpy(&*(wstr_.begin()), str);
|
str_.clear();
|
}
|
else EraseContents();
|
}
|
|
// Erase the content of current Excel cell.
|
// Set type to UNDEFINED.
|
void BasicExcelCell::EraseContents()
|
{
|
type_ = UNDEFINED;
|
str_.clear();
|
wstr_.clear();
|
}
|
|
///< Print cell to output stream.
|
///< Print a null character if cell is undefined.
|
ostream& operator<<(ostream& os, const BasicExcelCell& cell)
|
{
|
switch (cell.Type())
|
{
|
case BasicExcelCell::UNDEFINED:
|
os << '\0';
|
break;
|
|
case BasicExcelCell::INT:
|
os << cell.GetInteger();
|
break;
|
|
case BasicExcelCell::DOUBLE:
|
os << cell.GetDouble();
|
break;
|
|
case BasicExcelCell::STRING:
|
os << cell.GetString();
|
break;
|
|
case BasicExcelCell::WSTRING:
|
os << cell.GetWString();
|
break;
|
}
|
return os;
|
}
|
|
} // YExcel namespace end
|