281 lines
8.0 KiB
C++
281 lines
8.0 KiB
C++
/*
|
|
* (c) Copyright Ascensio System SIA 2010-2023
|
|
*
|
|
* This program is a free software product. You can redistribute it and/or
|
|
* modify it under the terms of the GNU Affero General Public License (AGPL)
|
|
* version 3 as published by the Free Software Foundation. In accordance with
|
|
* Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
|
|
* that Ascensio System SIA expressly excludes the warranty of non-infringement
|
|
* of any third-party rights.
|
|
*
|
|
* This program is distributed WITHOUT ANY WARRANTY; without even the implied
|
|
* warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. For
|
|
* details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
|
|
*
|
|
* You can contact Ascensio System SIA at 20A-6 Ernesta Birznieka-Upish
|
|
* street, Riga, Latvia, EU, LV-1050.
|
|
*
|
|
* The interactive user interfaces in modified source and object code versions
|
|
* of the Program must display Appropriate Legal Notices, as required under
|
|
* Section 5 of the GNU AGPL version 3.
|
|
*
|
|
* Pursuant to Section 7(b) of the License you must retain the original Product
|
|
* logo when distributing the program. Pursuant to Section 7(e) we decline to
|
|
* grant you any rights under trademark law for use of our trademarks.
|
|
*
|
|
* All the Product's GUI elements, including illustrations and icon sets, as
|
|
* well as technical writing content are licensed under the terms of the
|
|
* Creative Commons Attribution-ShareAlike 4.0 International. See the License
|
|
* terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
|
|
*
|
|
*/
|
|
|
|
#include "SharedStrings.h"
|
|
|
|
#include "../../Common/SimpleTypes_Shared.h"
|
|
|
|
#include "../../XlsbFormat/Biff12_records/BeginSst.h"
|
|
#include "../../XlsbFormat/Biff12_unions/SHAREDSTRINGS.h"
|
|
#include "../../XlsbFormat/Biff12_records/SSTItem.h"
|
|
|
|
#include "../../Binary/XlsbFormat/FileTypes_SpreadsheetBin.h"
|
|
|
|
namespace OOX
|
|
{
|
|
namespace Spreadsheet
|
|
{
|
|
CSharedStrings::CSharedStrings(OOX::Document* pMain) : OOX::File(pMain), OOX::IFileContainer(pMain)
|
|
{
|
|
m_nCount = 0;
|
|
m_bSpreadsheets = true;
|
|
|
|
CXlsx* xlsx = dynamic_cast<CXlsx*>(File::m_pMainDocument);
|
|
if ((xlsx) && (!xlsx->m_pSharedStrings))
|
|
xlsx->m_pSharedStrings = this;
|
|
}
|
|
CSharedStrings::CSharedStrings(OOX::Document* pMain, const CPath& oRootPath, const CPath& oPath) : OOX::File(pMain), OOX::IFileContainer(pMain)
|
|
{
|
|
m_nCount = 0;
|
|
m_bSpreadsheets = true;
|
|
|
|
CXlsx* xlsx = dynamic_cast<CXlsx*>(File::m_pMainDocument);
|
|
if ((xlsx) && (!xlsx->m_pSharedStrings))
|
|
xlsx->m_pSharedStrings = this;
|
|
|
|
read( oRootPath, oPath );
|
|
}
|
|
CSharedStrings::~CSharedStrings()
|
|
{
|
|
ClearItems();
|
|
}
|
|
void CSharedStrings::readBin(const CPath& oPath)
|
|
{
|
|
CXlsb* xlsb = dynamic_cast<CXlsb*>(File::m_pMainDocument);
|
|
if (xlsb)
|
|
{
|
|
XLSB::SharedStringsStreamPtr sharedStringsStream(new XLSB::SharedStringsStream);
|
|
|
|
xlsb->ReadBin(oPath, sharedStringsStream.get());
|
|
|
|
if (sharedStringsStream != nullptr)
|
|
{
|
|
auto ptr = static_cast<XLSB::SHAREDSTRINGS*>(sharedStringsStream->m_SHAREDSTRINGS.get());
|
|
if (ptr != nullptr)
|
|
{
|
|
ReadAttributes(ptr->m_BrtBeginSst);
|
|
|
|
for(auto &sstItem : ptr->m_arBrtSSTItem)
|
|
{
|
|
CSi* pItem = new CSi();
|
|
auto ptr = static_cast<XLSB::SSTItem*>(sstItem.get());
|
|
if(ptr != nullptr)
|
|
{
|
|
pItem->fromBin(ptr->richStr);
|
|
}
|
|
m_arrItems.push_back(pItem);
|
|
m_nCount++;
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
//sharedStringsStream.reset();
|
|
}
|
|
|
|
}
|
|
XLS::BaseObjectPtr CSharedStrings::WriteBin() const
|
|
{
|
|
XLSB::SharedStringsStreamPtr sharedStringsStream(new XLSB::SharedStringsStream);
|
|
auto ptr(new XLSB::SHAREDSTRINGS);
|
|
XLS::BaseObjectPtr objectPtr(ptr);
|
|
|
|
auto atribPtr(new XLSB::BeginSst);
|
|
ptr->m_BrtBeginSst = XLS::BaseObjectPtr{atribPtr};
|
|
if(m_oCount.IsInit())
|
|
atribPtr->cstTotal = m_oCount->GetValue();
|
|
if(m_oUniqueCount.IsInit())
|
|
atribPtr->cstUnique = m_oUniqueCount->GetValue();
|
|
|
|
for(auto i:m_arrItems)
|
|
{
|
|
ptr->m_arBrtSSTItem.push_back(i->toBin());
|
|
}
|
|
return objectPtr;
|
|
}
|
|
void CSharedStrings::read(const CPath& oPath)
|
|
{
|
|
//don't use this. use read(const CPath& oRootPath, const CPath& oFilePath)
|
|
CPath oRootPath;
|
|
read(oRootPath, oPath);
|
|
}
|
|
void CSharedStrings::read(const CPath& oRootPath, const CPath& oPath)
|
|
{
|
|
m_oReadPath = oPath;
|
|
IFileContainer::Read( oRootPath, oPath );
|
|
|
|
if( m_oReadPath.GetExtention() == _T(".bin"))
|
|
{
|
|
readBin(m_oReadPath);
|
|
return;
|
|
}
|
|
|
|
XmlUtils::CXmlLiteReader oReader;
|
|
|
|
if ( !oReader.FromFile( m_oReadPath.GetPath() ) )
|
|
{
|
|
//test un-upper(lower)case open - CALACATA GREECE.xlsx
|
|
if (!m_oReadPath.FileInDirectoryCorrect())
|
|
return;
|
|
if ( !oReader.FromFile( m_oReadPath.GetPath() ) )
|
|
return;
|
|
}
|
|
|
|
if ( !oReader.ReadNextNode() )
|
|
return;
|
|
|
|
std::wstring sName = XmlUtils::GetNameNoNS(oReader.GetName());
|
|
if ( _T("sst") == sName )
|
|
{
|
|
ReadAttributes( oReader );
|
|
|
|
m_nCount = 0;
|
|
|
|
if ( !oReader.IsEmptyNode() )
|
|
{
|
|
int nSharedStringsDepth = oReader.GetDepth();
|
|
while ( oReader.ReadNextSiblingNode( nSharedStringsDepth ) )
|
|
{
|
|
sName = XmlUtils::GetNameNoNS(oReader.GetName());
|
|
|
|
if ( _T("si") == sName )
|
|
{
|
|
CSi* pItem = new CSi();
|
|
*pItem = oReader;
|
|
|
|
m_arrItems.push_back(pItem );
|
|
m_nCount++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
void CSharedStrings::write(const CPath& oPath, const CPath& oDirectory, CContentTypes& oContent) const
|
|
{
|
|
CXlsb* xlsb = dynamic_cast<CXlsb*>(File::m_pMainDocument);
|
|
if ((xlsb) && (xlsb->m_bWriteToXlsb))
|
|
{
|
|
XLS::BaseObjectPtr object = WriteBin();
|
|
xlsb->WriteBin(oPath, object.get());
|
|
}
|
|
else
|
|
{
|
|
NSStringUtils::CStringBuilder writer;
|
|
writer.WriteString(_T("<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?><sst xmlns=\"http://schemas.openxmlformats.org/spreadsheetml/2006/main\""));
|
|
WritingStringNullableAttrInt(L"count", m_oCount, m_oCount->GetValue());
|
|
WritingStringNullableAttrInt(L"uniqueCount", m_oUniqueCount, m_oUniqueCount->GetValue());
|
|
writer.WriteString(_T(">"));
|
|
|
|
for(size_t i = 0; i < m_arrItems.size(); i++)
|
|
{
|
|
m_arrItems[i]->toXML(writer);
|
|
}
|
|
|
|
writer.WriteString(_T("</sst>"));
|
|
std::wstring sPath = oPath.GetPath();
|
|
NSFile::CFileBinary::SaveToFile(sPath.c_str(), writer.GetData());
|
|
}
|
|
oContent.Registration( type().OverrideType(), oDirectory, oPath.GetFilename() );
|
|
}
|
|
const OOX::FileType CSharedStrings::type() const
|
|
{
|
|
CXlsb* xlsb = dynamic_cast<CXlsb*>(File::m_pMainDocument);
|
|
if ((xlsb) && (xlsb->m_bWriteToXlsb))
|
|
{
|
|
return OOX::SpreadsheetBin::FileTypes::SharedStringsBin;
|
|
}
|
|
return OOX::Spreadsheet::FileTypes::SharedStrings;
|
|
}
|
|
const CPath CSharedStrings::DefaultDirectory() const
|
|
{
|
|
return type().DefaultDirectory();
|
|
}
|
|
const CPath CSharedStrings::DefaultFileName() const
|
|
{
|
|
CXlsb* xlsb = dynamic_cast<CXlsb*>(File::m_pMainDocument);
|
|
if ((xlsb) && (xlsb->m_bWriteToXlsb))
|
|
{
|
|
CPath name = type().DefaultFileName();
|
|
|
|
name.SetExtention(L"bin");
|
|
return name;
|
|
}
|
|
else
|
|
{
|
|
return type().DefaultFileName();
|
|
}
|
|
}
|
|
const CPath& CSharedStrings::GetReadPath()
|
|
{
|
|
return m_oReadPath;
|
|
}
|
|
const int CSharedStrings::AddSi(CSi* pSi)
|
|
{
|
|
int nIndex = m_nCount++;
|
|
m_arrItems.push_back(pSi);
|
|
return nIndex;
|
|
}
|
|
void CSharedStrings::ClearItems()
|
|
{
|
|
m_nCount = 0;
|
|
for(size_t i = 0; i < m_arrItems.size(); i++)
|
|
{
|
|
if ( m_arrItems[i] )delete m_arrItems[i];
|
|
|
|
m_arrItems[i] = NULL;
|
|
}
|
|
m_arrItems.clear();
|
|
}
|
|
void CSharedStrings::ReadAttributes(XmlUtils::CXmlLiteReader& oReader)
|
|
{
|
|
WritingElement_ReadAttributes_Start( oReader )
|
|
WritingElement_ReadAttributes_Read_if ( oReader, _T("count"), m_oCount )
|
|
WritingElement_ReadAttributes_Read_if ( oReader, _T("uniqueCount"), m_oUniqueCount )
|
|
WritingElement_ReadAttributes_End( oReader )
|
|
}
|
|
void CSharedStrings::ReadAttributes(XLS::BaseObjectPtr& obj)
|
|
{
|
|
auto ptr = static_cast<XLSB::BeginSst*>(obj.get());
|
|
|
|
if(ptr != nullptr)
|
|
{
|
|
m_oCount = ptr->cstTotal;
|
|
m_oUniqueCount = ptr->cstUnique;
|
|
}
|
|
|
|
}
|
|
|
|
} //Spreadsheet
|
|
} // namespace OOX
|
|
|