core/OOXML/Binary/Sheets/Reader/XMLReader/XLSXTableController.cpp

/*
 * (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 "XlSXTableController.h"

#include "../../../../XlsxFormat/Workbook/Sheets.h"
#include "../../../XlsxFormat/Workbook/Workbook.h"
#include "../../../../Common/OfficeFileErrorDescription.h"

#include "../../../XlsxFormat/Styles/Styles.h"
#include "../../../XlsxFormat/Styles/NumFmts.h"
#include "../../../XlsxFormat/Styles/Xfs.h"

constexpr auto SheetName = L"Sheet";

XLSXTableController::XLSXTableController(OOX::Spreadsheet::CXlsx &book)
{
    book_ = &book;
    book_->CreateWorkbook();
    book_->m_pWorkbook->m_oSheets.Init();
    tableRows_.push_back(nullptr);

    // Создадим стили
	book_->CreateStyles();

	// Добавим стили для wrap-а
	book_->m_pStyles->m_oCellXfs.Init();
	book_->m_pStyles->m_oCellXfs->m_oCount.Init();
	book_->m_pStyles->m_oCellXfs->m_oCount->SetValue(2);

	m_pStyles_ = book_->m_pStyles;

// Normall default
	OOX::Spreadsheet::CXfs* pXfs = NULL;
	pXfs = new OOX::Spreadsheet::CXfs();
	pXfs->m_oBorderId.Init();		pXfs->m_oBorderId->SetValue(0);
	pXfs->m_oFillId.Init();			pXfs->m_oFillId->SetValue(0);
	pXfs->m_oFontId.Init();			pXfs->m_oFontId->SetValue(0);
	pXfs->m_oNumFmtId.Init();		pXfs->m_oNumFmtId->SetValue(0);

	book_->m_pStyles->m_oCellXfs->m_arrItems.push_back(pXfs);

// Wrap style
	pXfs = new OOX::Spreadsheet::CXfs();
	pXfs->m_oBorderId.Init();		pXfs->m_oBorderId->SetValue(0);
	pXfs->m_oFillId.Init();			pXfs->m_oFillId->SetValue(0);
	pXfs->m_oFontId.Init();			pXfs->m_oFontId->SetValue(0);
	pXfs->m_oNumFmtId.Init();		pXfs->m_oNumFmtId->SetValue(0);

	pXfs->m_oApplyAlignment.Init();	pXfs->m_oApplyAlignment->SetValue(SimpleTypes::onoffTrue);
	pXfs->m_oAligment.Init();		pXfs->m_oAligment->m_oWrapText.Init();
	pXfs->m_oAligment->m_oWrapText->SetValue(SimpleTypes::onoffTrue);

	book_->m_pStyles->m_oCellXfs->m_arrItems.push_back(pXfs);
}


void XLSXTableController::FormBook()
{
    /// @todo сделать так чтобы блок выполнялся только один раз
    if(1)
    {
        auto pWorksheet = new OOX::Spreadsheet::CWorksheet(NULL);
        pWorksheet->m_oSheetData.Init();
        for(int i = 1; i< tableRows_.size(); i++)
        {
            addRow(tableRows_.at(i), pWorksheet, i);
        }

        addPage(pWorksheet, 1);
    }
}


void XLSXTableController::AddCell(const std::wstring &sText, INT nRow, INT nCol)
{

    size_t length = sText.length();
    // Пустую не пишем
    if ((0 == length) || (sText[0] == L'\0'))
    {
        return;
    }

    auto pCell = new OOX::Spreadsheet::CCell();
    pCell->m_oType.Init();

    pCell->m_oType->SetValue(SimpleTypes::Spreadsheet::celltypeInlineStr);
    pCell->m_oRichText.Init();
    OOX::Spreadsheet::CText *pText = new OOX::Spreadsheet::CText();
    pText->m_sText = sText;
    pCell->m_oRichText->m_arrItems.push_back(pText);

    pCell->setRowCol(nRow - 1, nCol);

    pCell->m_oStyle = 1;

    while(tableRows_.size() - 1 < nRow)
    {
        tableRows_.push_back(new OOX::Spreadsheet::CRow);
    }

    tableRows_.at(nRow)->m_arrItems.push_back(pCell);

}

void XLSXTableController::addPage(OOX::Spreadsheet::CWorksheet *page, INT pageNumber)
{
    book_->m_arWorksheets.push_back(page);

    smart_ptr<OOX::File> oWorksheetFile = page;
    const OOX::RId oRid = book_->m_pWorkbook->Add(oWorksheetFile);

    book_->m_mapWorksheets.insert(std::make_pair(oRid.ToString(), page)); // for bin

    OOX::Spreadsheet::CSheet *pSheet = new OOX::Spreadsheet::CSheet();

    std::wstring sheetName = SheetName + std::to_wstring(pageNumber);
    pSheet->m_oName = sheetName;
    pSheet->m_oSheetId.Init();
    pSheet->m_oSheetId->SetValue(pageNumber);
    pSheet->m_oRid.Init();
    pSheet->m_oRid->SetValue(oRid.ToString());

    book_->m_pWorkbook->m_oSheets.Init();
    book_->m_pWorkbook->m_oSheets->m_arrItems.push_back(pSheet);
}

_UINT32 XLSXTableController::addRow(OOX::Spreadsheet::CRow *pRow, OOX::Spreadsheet::CWorksheet *pWorkSheet,  INT nRow)
{
    if (pWorkSheet->m_oSheetData->m_arrItems.size() > 1048576)
            {
                return AVS_FILEUTILS_ERROR_CONVERT_ROWLIMITS; // ограниечние мс
            }
    pRow->m_oR.Init();
    pRow->m_oR->SetValue(nRow);
    pWorkSheet->m_oSheetData->m_arrItems.push_back(pRow);

    return 0;
}