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(QObject *parent) :
    QObject(parent)
{
    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_strings_to_numbers_enabled = false;
    m_activesheet = 0;
    m_firstsheet = 0;
    m_table_count = 0;
}

Workbook::~Workbook()
{
}

void Workbook::save(const QString &name)
{
    //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(name);
}

/*
  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