StarMirror/xlnt
1
0
Fork 0
mirror of https://github.com/tfussell/xlnt.git synced 2024-03-22 13:11:17 +08:00
Code Issues Projects Releases Wiki Activity

reformat xlsx_consumer.cpp

Browse Source
This commit is contained in:
Thomas Fussell 2016-11-19 17:19:38 -05:00
parent c89b36f749
commit daab3cf8ba
1 changed files with 823 additions and 807 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

280
source/detail/xlsx_consumer.cpp
View File

@ -24,14 +24,14 @@
#include <cctype>
#include <numeric> // for std::accumulate
#include <detail/xlsx_consumer.hpp>
#include <detail/constants.hpp>
#include <detail/custom_value_traits.hpp>
#include <detail/workbook_impl.hpp>
#include <detail/xlsx_consumer.hpp>
#include <detail/zip.hpp>
#include <xlnt/cell/cell.hpp>
#include <xlnt/utils/path.hpp>
#include <xlnt/packaging/manifest.hpp>
#include <xlnt/utils/path.hpp>
#include <xlnt/workbook/const_worksheet_iterator.hpp>
#include <xlnt/workbook/workbook.hpp>
#include <xlnt/worksheet/worksheet.hpp>
@ -41,7 +41,7 @@ namespace std {
template <>
struct hash<xml::qname>
{
std::size_t operator()(const xml::qname& k) const
std::size_t operator()(const xml::qname &k) const
{
static std::hash<std::string> hasher;
return hasher(k.string());
@ -134,9 +134,9 @@ std::vector<xlnt::relationship> read_relationships(const xlnt::path &part, xlnt:
parser.next_expect(xml::parser::event_type::start_element, xmlns, "Relationship");
relationships.emplace_back(parser.attribute("Id"),
parser.attribute<xlnt::relationship::type>("Type"), source,
xlnt::uri(parser.attribute("Target")), xlnt::target_mode::internal);
parser.next_expect(xml::parser::event_type::end_element,
xlnt::constants::get_namespace("relationships"), "Relationship");
xlnt::uri(parser.attribute("Target")),
xlnt::target_mode::internal);
parser.next_expect(xml::parser::event_type::end_element, xmlns, "Relationship");
}
parser.next_expect(xml::parser::event_type::end_element, xmlns, "Relationships");
@ -146,8 +146,8 @@ std::vector<xlnt::relationship> read_relationships(const xlnt::path &part, xlnt:
void check_document_type(const std::string &document_content_type)
{
if (document_content_type != "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet.main+xml"
&& document_content_type != "application/vnd.openxmlformats-officedocument.spreadsheetml.template.main+xml ")
if (document_content_type != "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet.main+xml" &&
document_content_type != "application/vnd.openxmlformats-officedocument.spreadsheetml.template.main+xml ")
{
throw xlnt::invalid_file(document_content_type);
}
@ -177,6 +177,7 @@ xml::parser &xlsx_consumer::parser()
void xlsx_consumer::read_part(const std::vector<relationship> &rel_chain)
{
// ignore namespace declarations except in parts of these types
const auto using_namespaces = std::vector<relationship::type>
{
relationship::type::office_document,
@ -216,7 +217,6 @@ void xlsx_consumer::read_part(const std::vector<relationship> &rel_chain)
break;
case relationship::type::office_document:
check_document_type(manifest.get_content_type(part_path));
read_workbook();
break;
@ -272,24 +272,60 @@ void xlsx_consumer::read_part(const std::vector<relationship> &rel_chain)
read_worksheet(rel_chain.back().get_id());
break;
case relationship::type::thumbnail: break;
case relationship::type::calculation_chain: break;
case relationship::type::hyperlink: break;
case relationship::type::comments: break;
case relationship::type::vml_drawing: break;
case relationship::type::unknown: break;
case relationship::type::printer_settings: break;
case relationship::type::custom_property: break;
case relationship::type::drawings: break;
case relationship::type::pivot_table_cache_definition: break;
case relationship::type::pivot_table_cache_records: break;
case relationship::type::query_table: break;
case relationship::type::shared_workbook: break;
case relationship::type::revision_log: break;
case relationship::type::shared_workbook_user_data: break;
case relationship::type::single_cell_table_definitions: break;
case relationship::type::table_definition: break;
case relationship::type::image: break;
case relationship::type::thumbnail:
break;
case relationship::type::calculation_chain:
read_calculation_chain();
break;
case relationship::type::hyperlink:
break;
case relationship::type::comments:
break;
case relationship::type::vml_drawing:
break;
case relationship::type::unknown:
break;
case relationship::type::printer_settings:
break;
case relationship::type::custom_property:
break;
case relationship::type::drawings:
break;
case relationship::type::pivot_table_cache_definition:
break;
case relationship::type::pivot_table_cache_records:
break;
case relationship::type::query_table:
break;
case relationship::type::shared_workbook:
break;
case relationship::type::revision_log:
break;
case relationship::type::shared_workbook_user_data:
break;
case relationship::type::single_cell_table_definitions:
break;
case relationship::type::table_definition:
break;
case relationship::type::image:
break;
}
parser_ = nullptr;
@ -323,12 +359,11 @@ void xlsx_consumer::populate_workbook()
{
if (manifest.has_relationship(root_path, rel_type))
{
read_part({ manifest.get_relationship(root_path, rel_type) });
read_part({manifest.get_relationship(root_path, rel_type)});
}
}
const auto workbook_rel = manifest.get_relationship(root_path,
relationship::type::office_document);
const auto workbook_rel = manifest.get_relationship(root_path, relationship::type::office_document);
const auto workbook_parts_first = std::vector<relationship::type>
{
@ -341,7 +376,7 @@ void xlsx_consumer::populate_workbook()
{
if (manifest.has_relationship(workbook_rel.get_target().get_path(), rel_type))
{
read_part({ workbook_rel, manifest.get_relationship(workbook_rel.get_target().get_path(), rel_type) });
read_part({workbook_rel, manifest.get_relationship(workbook_rel.get_target().get_path(), rel_type)});
}
}
@ -358,7 +393,7 @@ void xlsx_consumer::populate_workbook()
{
for (auto rel : manifest.get_relationships(workbook_rel.get_target().get_path(), rel_type))
{
read_part({ workbook_rel, rel });
read_part({workbook_rel, rel});
}
}
}
@ -392,14 +427,18 @@ void xlsx_consumer::read_manifest()
if (parser.name() == "Default")
{
manifest.register_default_type(parser.attribute("Extension"),
parser.attribute("ContentType"));
auto extension = parser.attribute("Extension");
auto content_type = parser.attribute("ContentType");
manifest.register_default_type(extension, content_type);
parser.next_expect(xml::parser::event_type::end_element, xmlns, "Default");
}
else if (parser.name() == "Override")
{
manifest.register_override_type(path(parser.attribute("PartName")),
parser.attribute("ContentType"));
auto part_name = parser.attribute("PartName");
auto content_type = parser.attribute("ContentType");
manifest.register_override_type(path(part_name), content_type);
parser.next_expect(xml::parser::event_type::end_element, xmlns, "Override");
}
}
@ -408,6 +447,12 @@ void xlsx_consumer::read_manifest()
for (const auto &package_rel : package_rels)
{
if (package_rel.get_type() == relationship::type::office_document)
{
auto content_type = manifest.get_content_type(manifest.canonicalize({ package_rel }));
check_document_type(content_type);
}
manifest.register_relationship(uri("/"),
package_rel.get_type(),
package_rel.get_target(),
@ -419,8 +464,7 @@ void xlsx_consumer::read_manifest()
{
auto relationship_source = path(relationship_source_string);
if (relationship_source == path("_rels/.rels")
|| relationship_source.extension() != "rels") continue;
if (relationship_source == path("_rels/.rels") || relationship_source.extension() != "rels") continue;
path part(relationship_source.parent().parent());
part = part.append(relationship_source.split_extension().first);
@ -433,8 +477,11 @@ void xlsx_consumer::read_manifest()
for (const auto &part_rel : part_rels)
{
path target_path(source_directory.append(part_rel.get_target().get_path()));
manifest.register_relationship(source, part_rel.get_type(),
part_rel.get_target(), part_rel.get_target_mode(), part_rel.get_id());
manifest.register_relationship(source,
part_rel.get_type(),
part_rel.get_target(),
part_rel.get_target_mode(),
part_rel.get_id());
}
}
}
@ -451,73 +498,43 @@ void xlsx_consumer::read_extended_properties()
{
{
xml::qname(xmlns, "Application"),
[](xlsx_consumer &c)
{
c.target_.set_application(c.read_text());
}
[](xlsx_consumer &c) { c.target_.set_application(c.read_text()); }
},
{
xml::qname(xmlns, "DocSecurity"),
[](xlsx_consumer &c)
{
c.target_.set_doc_security(std::stoi(c.read_text()));
}
[](xlsx_consumer &c) { c.target_.set_doc_security(std::stoi(c.read_text())); }
},
{
xml::qname(xmlns, "ScaleCrop"),
[](xlsx_consumer &c)
{
c.target_.set_scale_crop(is_true(c.read_text()));
}
[](xlsx_consumer &c) { c.target_.set_scale_crop(is_true(c.read_text())); }
},
{
xml::qname(xmlns, "Company"),
[](xlsx_consumer &c)
{
c.target_.set_company(c.read_text());
}
[](xlsx_consumer &c) { c.target_.set_company(c.read_text()); }
},
{
xml::qname(xmlns, "SharedDoc"),
[](xlsx_consumer &c)
{
c.target_.set_shared_doc(is_true(c.read_text()));
}
[](xlsx_consumer &c) { c.target_.set_shared_doc(is_true(c.read_text())); }
},
{
xml::qname(xmlns, "HyperlinksChanged"),
[](xlsx_consumer &c)
{
c.target_.set_hyperlinks_changed(is_true(c.read_text()));
}
[](xlsx_consumer &c) { c.target_.set_hyperlinks_changed(is_true(c.read_text())); }
},
{
xml::qname(xmlns, "AppVersion"),
[](xlsx_consumer &c)
{
c.target_.set_app_version(c.read_text());
}
[](xlsx_consumer &c) { c.target_.set_app_version(c.read_text()); }
},
{
xml::qname(xmlns, "LinksUpToDate"),
[](xlsx_consumer &c)
{
c.target_.set_links_up_to_date(is_true(c.read_text()));
}
[](xlsx_consumer &c) { c.target_.set_links_up_to_date(is_true(c.read_text())); }
},
{
xml::qname(xmlns, "Template"),
[](xlsx_consumer &c)
{
c.read_text();
}
[](xlsx_consumer &c) { c.read_text(); }
},
{
xml::qname(xmlns, "TotalTime"),
[](xlsx_consumer &c)
{
c.read_text();
}
[](xlsx_consumer &c) { c.read_text(); }
},
{
xml::qname(xmlns, "HeadingPairs"),
@ -567,7 +584,7 @@ void xlsx_consumer::read_extended_properties()
c.parser().next_expect(xml::parser::event_type::end_element, xmlns_vt, "vector");
}
},
}
});
}
@ -655,8 +672,9 @@ void xlsx_consumer::read_workbook()
while (true)
{
if (parser().peek() == xml::parser::event_type::end_element
|| parser().peek() == xml::parser::event_type::end_namespace_decl) break;
if (parser().peek() == xml::parser::event_type::end_element ||
parser().peek() == xml::parser::event_type::end_namespace_decl)
break;
parser().next_expect(xml::parser::event_type::start_element);
parser().content(xml::parser::content_type::complex);
@ -667,6 +685,7 @@ void xlsx_consumer::read_workbook()
{
target_.d_->has_file_version_ = true;
target_.d_->file_version_.app_name = parser().attribute("appName");
if (parser().attribute_present("lastEdited"))
{
target_.d_->file_version_.last_edited = string_to_size_t(parser().attribute("lastEdited"));
@ -897,8 +916,8 @@ void xlsx_consumer::read_workbook()
}
else if (qname == xml::qname(xmlns, "workbookProtection"))
{
while (parser().peek() != xml::parser::event_type::end_element
|| parser().qname() != xml::qname(xmlns, "workbookProtection"))
while (parser().peek() != xml::parser::event_type::end_element ||
parser().qname() != xml::qname(xmlns, "workbookProtection"))
{
parser().next();
}
@ -916,7 +935,7 @@ void xlsx_consumer::read_calculation_chain()
{
}
void xlsx_consumer::read_chartsheet(const std::string &/*title*/)
void xlsx_consumer::read_chartsheet(const std::string & /*title*/)
{
}
@ -932,7 +951,7 @@ void xlsx_consumer::read_custom_xml_mappings()
{
}
void xlsx_consumer::read_dialogsheet(const std::string &/*title*/)
void xlsx_consumer::read_dialogsheet(const std::string & /*title*/)
{
}
@ -1087,13 +1106,13 @@ void xlsx_consumer::read_stylesheet()
struct formatting_record
{
std::pair<class alignment, bool> alignment = { {}, 0 };
std::pair<std::size_t, bool> border_id = { 0, false };
std::pair<std::size_t, bool> fill_id = { 0, false };
std::pair<std::size_t, bool> font_id = { 0, false };
std::pair<std::size_t, bool> number_format_id = { 0, false };
std::pair<class protection, bool> protection = { {}, false };
std::pair<std::size_t, bool> style_id = { 0, false };
std::pair<class alignment, bool> alignment = {{}, 0};
std::pair<std::size_t, bool> border_id = {0, false};
std::pair<std::size_t, bool> fill_id = {0, false};
std::pair<std::size_t, bool> font_id = {0, false};
std::pair<std::size_t, bool> number_format_id = {0, false};
std::pair<class protection, bool> protection = {{}, false};
std::pair<std::size_t, bool> style_id = {0, false};
};
struct style_data
@ -1419,8 +1438,8 @@ void xlsx_consumer::read_stylesheet()
}
else if (parser().qname() == xml::qname(xmlns, "colors"))
{
while (parser().peek() != xml::parser::event_type::end_element
|| parser().qname() != xml::qname(xmlns, "colors"))
while (parser().peek() != xml::parser::event_type::end_element ||
parser().qname() != xml::qname(xmlns, "colors"))
{
if (parser().next() == xml::parser::event_type::start_element)
{
@ -1471,8 +1490,8 @@ void xlsx_consumer::read_stylesheet()
throw xlnt::exception("counts don't match");
}
}
else if (parser().qname() == xml::qname(xmlns, "cellStyleXfs")
|| parser().qname() == xml::qname(xmlns, "cellXfs"))
else if (parser().qname() == xml::qname(xmlns, "cellStyleXfs") ||
parser().qname() == xml::qname(xmlns, "cellXfs"))
{
auto in_style_records = parser().name() == "cellStyleXfs";
auto count = parser().attribute<std::size_t>("count");
@ -1489,33 +1508,32 @@ void xlsx_consumer::read_stylesheet()
: style_records.emplace(style_records.end()));
auto apply_alignment_present = parser().attribute_present("applyAlignment");
auto alignment_applied = apply_alignment_present
&& is_true(parser().attribute("applyAlignment"));
auto alignment_applied = apply_alignment_present && is_true(parser().attribute("applyAlignment"));
record.alignment.second = alignment_applied;
auto border_applied = parser().attribute_present("applyBorder")
&& is_true(parser().attribute("applyBorder"));
auto border_index = parser().attribute_present("borderId")
? string_to_size_t(parser().attribute("borderId")) : 0;
record.border_id = { border_index, border_applied };
record.border_id = {border_index, border_applied};
auto fill_applied = parser().attribute_present("applyFill")
&& is_true(parser().attribute("applyFill"));
auto fill_index = parser().attribute_present("fillId")
? string_to_size_t(parser().attribute("fillId")) : 0;
record.fill_id = { fill_index, fill_applied };
record.fill_id = {fill_index, fill_applied};
auto font_applied = parser().attribute_present("applyFont")
&& is_true(parser().attribute("applyFont"));
auto font_index = parser().attribute_present("fontId")
? string_to_size_t(parser().attribute("fontId")) : 0;
record.font_id = { font_index, font_applied };
record.font_id = {font_index, font_applied};
auto number_format_applied = parser().attribute_present("applyNumberFormat")
&& is_true(parser().attribute("applyNumberFormat"));
auto number_format_id = parser().attribute_present("numFmtId")
? string_to_size_t(parser().attribute("numFmtId")) : 0;
record.number_format_id = { number_format_id, number_format_applied };
record.number_format_id = {number_format_id, number_format_applied};
auto apply_protection_present = parser().attribute_present("applyProtection");
auto protection_applied = apply_protection_present
@ -1524,7 +1542,7 @@ void xlsx_consumer::read_stylesheet()
if (parser().attribute_present("xfId") && parser().name() == "cellXfs")
{
record.style_id = { parser().attribute<std::size_t>("xfId"), true };
record.style_id = {parser().attribute<std::size_t>("xfId"), true};
}
while (true)
@ -1537,34 +1555,38 @@ void xlsx_consumer::read_stylesheet()
{
if (parser().attribute_present("wrapText"))
{
record.alignment.first.wrap(is_true(parser().attribute("wrapText")));
auto value = is_true(parser().attribute("wrapText"));
record.alignment.first.wrap(value);
}
if (parser().attribute_present("shrinkToFit"))
{
record.alignment.first.shrink(is_true(parser().attribute("shrinkToFit")));
auto value = is_true(parser().attribute("shrinkToFit"));
record.alignment.first.shrink(value);
}
if (parser().attribute_present("indent"))
{
record.alignment.first.indent(parser().attribute<int>("indent"));
auto value = parser().attribute<int>("indent");
record.alignment.first.indent(value);
}
if (parser().attribute_present("textRotation"))
{
record.alignment.first.rotation(parser().attribute<int>("textRotation"));
auto value = parser().attribute<int>("textRotation");
record.alignment.first.rotation(value);
}
if (parser().attribute_present("vertical"))
{
record.alignment.first.vertical(
parser().attribute<xlnt::vertical_alignment>("vertical"));
auto value = parser().attribute<xlnt::vertical_alignment>("vertical");
record.alignment.first.vertical(value);
}
if (parser().attribute_present("horizontal"))
{
record.alignment.first.horizontal(
parser().attribute<xlnt::horizontal_alignment>("horizontal"));
auto value = parser().attribute<xlnt::horizontal_alignment>("horizontal");
record.alignment.first.horizontal(value);
}
record.alignment.second = !apply_alignment_present || alignment_applied;
@ -1580,11 +1602,10 @@ void xlsx_consumer::read_stylesheet()
}
parser().next_expect(xml::parser::event_type::end_element, xmlns, "xf");
}
if ((in_style_records && count != style_records.size())
|| (!in_style_records && count != format_records.size()))
if ((in_style_records && count != style_records.size()) ||
(!in_style_records && count != format_records.size()))
{
throw xlnt::exception("counts don't match");
}
@ -1766,10 +1787,7 @@ void xlsx_consumer::read_worksheet(const std::string &rel_id)
auto title = std::find_if(target_.d_->sheet_title_rel_id_map_.begin(),
target_.d_->sheet_title_rel_id_map_.end(),
[&](const std::pair<std::string, std::string> &p)
{
return p.second == rel_id;
})->first;
[&](const std::pair<std::string, std::string> &p) { return p.second == rel_id; })->first;
auto id = sheet_title_id_map_[title];
auto index = sheet_title_index_map_[title];
@ -1982,8 +2000,7 @@ void xlsx_consumer::read_worksheet(const std::string &rel_id)
else if (parser().qname() == xml::qname(xmlns, "f"))
{
has_formula = true;
has_shared_formula = parser().attribute_present("t")
&& parser().attribute("t") == "shared";
has_shared_formula = parser().attribute_present("t") && parser().attribute("t") == "shared";
parser().next_expect(xml::parser::event_type::characters);
formula_value_string = parser().value();
}
@ -2003,7 +2020,7 @@ void xlsx_consumer::read_worksheet(const std::string &rel_id)
cell.set_formula(formula_value_string);
}
if (has_type && (type == "inlineStr" || type =="str"))
if (has_type && (type == "inlineStr" || type == "str"))
{
cell.set_value(value_string);
}
@ -2134,8 +2151,8 @@ void xlsx_consumer::read_worksheet(const std::string &rel_id)
{
parser().attribute_map();
while (parser().peek() != xml::parser::event_type::end_element
|| parser().qname() != xml::qname(xmlns, "sheetPr"))
while (parser().peek() != xml::parser::event_type::end_element ||
parser().qname() != xml::qname(xmlns, "sheetPr"))
{
if (parser().next() == xml::parser::event_type::start_element)
{
@ -2149,8 +2166,8 @@ void xlsx_consumer::read_worksheet(const std::string &rel_id)
{
parser().attribute_map();
while (parser().peek() != xml::parser::event_type::end_element
|| parser().qname() != xml::qname(xmlns, "headerFooter"))
while (parser().peek() != xml::parser::event_type::end_element ||
parser().qname() != xml::qname(xmlns, "headerFooter"))
{
if (parser().next() == xml::parser::event_type::start_element)
{
@ -2169,8 +2186,8 @@ void xlsx_consumer::read_worksheet(const std::string &rel_id)
{
parser().attribute_map();
while (parser().peek() != xml::parser::event_type::end_element
|| parser().qname() != xml::qname(xmlns, "protectedRanges"))
while (parser().peek() != xml::parser::event_type::end_element ||
parser().qname() != xml::qname(xmlns, "protectedRanges"))
{
if (parser().next() == xml::parser::event_type::start_element)
{
@ -2191,8 +2208,8 @@ void xlsx_consumer::read_worksheet(const std::string &rel_id)
if (manifest.has_relationship(sheet_path, xlnt::relationship::type::comments))
{
auto comments_part = manifest.canonicalize({workbook_rel, sheet_rel,
manifest.get_relationship(sheet_path, xlnt::relationship::type::comments)});
auto comments_part = manifest.canonicalize(
{workbook_rel, sheet_rel, manifest.get_relationship(sheet_path, xlnt::relationship::type::comments)});
auto receive = xml::parser::receive_default;
xml::parser parser(archive_->open(comments_part.string()), comments_part.string(), receive);
@ -2218,7 +2235,6 @@ void xlsx_consumer::read_worksheet(const std::string &rel_id)
void xlsx_consumer::read_vml_drawings(worksheet ws)
{
}
void xlsx_consumer::read_comments(worksheet ws)