QtXlsxWriter/src/xlsxworkbook.cpp

/****************************************************************************
** Copyright (c) 2013 Debao Zhang <hello@debao.me>
** All right reserved.
**
** Permission is hereby granted, free of charge, to any person obtaining
** a copy of this software and associated documentation files (the
** "Software"), to deal in the Software without restriction, including
** without limitation the rights to use, copy, modify, merge, publish,
** distribute, sublicense, and/or sell copies of the Software, and to
** permit persons to whom the Software is furnished to do so, subject to
** the following conditions:
**
** The above copyright notice and this permission notice shall be
** included in all copies or substantial portions of the Software.
**
** THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
** EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
** MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
** NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
** LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
** OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
** WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
**
****************************************************************************/
#include "xlsxworkbook.h"
#include "xlsxsharedstrings_p.h"
#include "xlsxworksheet.h"
#include "xlsxstyles_p.h"
#include "xlsxformat.h"
#include "xlsxpackage_p.h"
#include "xmlstreamwriter_p.h"

namespace QXlsx {

Workbook::Workbook(const QString &name, QObject *parent) :
    QObject(parent), m_fileName(name)
{
    m_sharedStrings = new SharedStrings(this);
    m_styles = new Styles(this);


    m_x_window = 240;
    m_y_window = 15;
    m_window_width = 16095;
    m_window_height = 9660;

    m_optimization_enabled = false;
    m_strings_to_numbers_enabled = false;
    m_activesheet = 0;
    m_firstsheet = 0;
    m_table_count = 0;
    m_closed = false;
}

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

void Workbook::close()
{
    if (m_closed)
        return;

    m_closed = true;
    saveWorkbook();
}

void Workbook::saveWorkbook()
{
    //Add a default worksheet if non have been added.
    if (m_worksheets.size() == 0)
        addWorksheet();

    //Ensure that at least one worksheet has been selected.
    if (m_activesheet == 0) {
        m_worksheets[0]->setHidden(false);
        m_worksheets[0]->setSelected(true);
    }

    //Set the active sheet
    foreach (Worksheet *sheet, m_worksheets) {
        if (sheet->index() == m_activesheet)
            sheet->setActived(true);
    }


    //Create the package based on current workbook
    Package package(this);
    package.createPackage();
}

QString Workbook::fileName() const
{
    return m_fileName;
}

/*!
  Reduces the amount of data stored in memory so
  that large files can be written efficiently

  Note, in this mode a row of data is written and
  then discarded when a cell in a new row is added
  via one of the worksheet write_() methods. As such
  data should be written in sequential row order once
  this mode is on.

  It is used to optimise speed and reduce memory usage.
  However, these design goals meant that it wasn’t easy
  to implement features that many users requested such as
  writing formatting and data separately.

  This feature is disabled by default.
 */
void Workbook::setOptimizationEnabled(bool enable)
{
    m_optimization_enabled = enable;
}

bool Workbook::isOptimizationEnabled() const
{
    return m_optimization_enabled;
}

/*
  Enable the worksheet.write() method to convert strings
  to numbers, where possible, using float() in order to avoid
  an Excel warning about “Numbers Stored as Text”.

  The default is false
 */
void Workbook::setStringsToNumbersEnabled(bool enable)
{
    m_strings_to_numbers_enabled = enable;
}

bool Workbook::isStringsToNumbersEnabled() const
{
    return m_strings_to_numbers_enabled;
}

void Workbook::defineName(const QString &name, const QString &formula)
{

}

Worksheet *Workbook::addWorksheet(const QString &name)
{
    QString worksheetName = name;
    int index = m_worksheets.size()+1;
    if (name.isEmpty())
        worksheetName = QString("Sheet%1").arg(index);

    Worksheet *sheet = new Worksheet(worksheetName, index, this);
    m_worksheets.append(sheet);
    return sheet;
}

Format *Workbook::addFormat()
{
    return m_styles->addFormat();
}

QList<Worksheet *> Workbook::worksheets() const
{
    return m_worksheets;
}

SharedStrings *Workbook::sharedStrings()
{
    return m_sharedStrings;
}

Styles *Workbook::styles()
{
    return m_styles;
}

void Workbook::saveToXmlFile(QIODevice *device)
{
    XmlStreamWriter writer(device);

    writer.writeStartDocument("1.0", true);
    writer.writeStartElement("workbook");
    writer.writeAttribute("xmlns", "http://schemas.openxmlformats.org/spreadsheetml/2006/main");
    writer.writeAttribute("xmlns:r", "http://schemas.openxmlformats.org/officeDocument/2006/relationships");

    writer.writeEmptyElement("fileVersion");
    writer.writeAttribute("appName", "xl");
    writer.writeAttribute("lastEdited", "4");
    writer.writeAttribute("lowestEdited", "4");
    writer.writeAttribute("rupBuild", "4505");
//    writer.writeAttribute("codeName", "{37E998C4-C9E5-D4B9-71C8-EB1FF731991C}");

    writer.writeEmptyElement("workbookPr");
    writer.writeAttribute("defaultThemeVersion", "124226");

    writer.writeStartElement("bookViews");
    writer.writeEmptyElement("workbookView");
    writer.writeAttribute("xWindow", QString::number(m_x_window));
    writer.writeAttribute("yWindow", QString::number(m_y_window));
    writer.writeAttribute("windowWidth", QString::number(m_window_width));
    writer.writeAttribute("windowHeight", QString::number(m_window_height));
    //Store the firstSheet when it isn't the default
    if (m_firstsheet > 0)
        writer.writeAttribute("firstSheet", QString::number(m_firstsheet + 1));
    //Store the activeTab when it isn't the first sheet
    if (m_activesheet > 0)
        writer.writeAttribute("activeTab", QString::number(m_activesheet));
    writer.writeEndElement();//bookviews

    writer.writeStartElement("sheets");
    foreach (Worksheet *sheet, m_worksheets) {
        writer.writeEmptyElement("sheet");
        writer.writeAttribute("name", sheet->name());
        writer.writeAttribute("sheetId", QString::number(sheet->index()));
        if (sheet->isHidden())
            writer.writeAttribute("state", "hidden");
        writer.writeAttribute("r:id", QString("rId%1").arg(sheet->index()));
    }
    writer.writeEndElement();//sheets

//    writer.writeStartElement("definedNames");
//    writer.writeEndElement();//definedNames

    writer.writeStartElement("calcPr");
    writer.writeAttribute("calcId", "124519");
    writer.writeEndElement(); //calcPr

    writer.writeEndElement();//workbook
    writer.writeEndDocument();
}

} //namespace