123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710 |
- // Copyright 2016 - 2021 The excelize Authors. All rights reserved. Use of
- // this source code is governed by a BSD-style license that can be found in
- // the LICENSE file.
- //
- // Package excelize providing a set of functions that allow you to write to
- // and read from XLSX / XLSM / XLTM files. Supports reading and writing
- // spreadsheet documents generated by Microsoft Excel™ 2007 and later. Supports
- // complex components by high compatibility, and provided streaming API for
- // generating or reading data from a worksheet with huge amounts of data. This
- // library needs Go version 1.15 or later.
- package excelize
- import (
- "encoding/xml"
- "fmt"
- "strconv"
- "strings"
- )
- // PivotTableOption directly maps the format settings of the pivot table.
- type PivotTableOption struct {
- pivotTableSheetName string
- DataRange string
- PivotTableRange string
- Rows []PivotTableField
- Columns []PivotTableField
- Data []PivotTableField
- Filter []PivotTableField
- RowGrandTotals bool
- ColGrandTotals bool
- ShowDrill bool
- UseAutoFormatting bool
- PageOverThenDown bool
- MergeItem bool
- CompactData bool
- ShowError bool
- ShowRowHeaders bool
- ShowColHeaders bool
- ShowRowStripes bool
- ShowColStripes bool
- ShowLastColumn bool
- PivotTableStyleName string
- }
- // PivotTableField directly maps the field settings of the pivot table.
- // Subtotal specifies the aggregation function that applies to this data
- // field. The default value is sum. The possible values for this attribute
- // are:
- //
- // Average
- // Count
- // CountNums
- // Max
- // Min
- // Product
- // StdDev
- // StdDevp
- // Sum
- // Var
- // Varp
- //
- // Name specifies the name of the data field. Maximum 255 characters
- // are allowed in data field name, excess characters will be truncated.
- type PivotTableField struct {
- Data string
- Name string
- Subtotal string
- DefaultSubtotal bool
- }
- // AddPivotTable provides the method to add pivot table by given pivot table
- // options. Note that the same fields can not in Columns, Rows and Filter
- // fields at the same time.
- //
- // For example, create a pivot table on the Sheet1!$G$2:$M$34 area with the
- // region Sheet1!$A$1:$E$31 as the data source, summarize by sum for sales:
- //
- // package main
- //
- // import (
- // "fmt"
- // "math/rand"
- //
- // "github.com/xuri/excelize/v2"
- // )
- //
- // func main() {
- // f := excelize.NewFile()
- // // Create some data in a sheet
- // month := []string{"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"}
- // year := []int{2017, 2018, 2019}
- // types := []string{"Meat", "Dairy", "Beverages", "Produce"}
- // region := []string{"East", "West", "North", "South"}
- // f.SetSheetRow("Sheet1", "A1", &[]string{"Month", "Year", "Type", "Sales", "Region"})
- // for i := 0; i < 30; i++ {
- // f.SetCellValue("Sheet1", fmt.Sprintf("A%d", i+2), month[rand.Intn(12)])
- // f.SetCellValue("Sheet1", fmt.Sprintf("B%d", i+2), year[rand.Intn(3)])
- // f.SetCellValue("Sheet1", fmt.Sprintf("C%d", i+2), types[rand.Intn(4)])
- // f.SetCellValue("Sheet1", fmt.Sprintf("D%d", i+2), rand.Intn(5000))
- // f.SetCellValue("Sheet1", fmt.Sprintf("E%d", i+2), region[rand.Intn(4)])
- // }
- // if err := f.AddPivotTable(&excelize.PivotTableOption{
- // DataRange: "Sheet1!$A$1:$E$31",
- // PivotTableRange: "Sheet1!$G$2:$M$34",
- // Rows: []excelize.PivotTableField{{Data: "Month", DefaultSubtotal: true}, {Data: "Year"}},
- // Filter: []excelize.PivotTableField{{Data: "Region"}},
- // Columns: []excelize.PivotTableField{{Data: "Type", DefaultSubtotal: true}},
- // Data: []excelize.PivotTableField{{Data: "Sales", Name: "Summarize", Subtotal: "Sum"}},
- // RowGrandTotals: true,
- // ColGrandTotals: true,
- // ShowDrill: true,
- // ShowRowHeaders: true,
- // ShowColHeaders: true,
- // ShowLastColumn: true,
- // }); err != nil {
- // fmt.Println(err)
- // }
- // if err := f.SaveAs("Book1.xlsx"); err != nil {
- // fmt.Println(err)
- // }
- // }
- //
- func (f *File) AddPivotTable(opt *PivotTableOption) error {
- // parameter validation
- dataSheet, pivotTableSheetPath, err := f.parseFormatPivotTableSet(opt)
- if err != nil {
- return err
- }
- pivotTableID := f.countPivotTables() + 1
- pivotCacheID := f.countPivotCache() + 1
- sheetRelationshipsPivotTableXML := "../pivotTables/pivotTable" + strconv.Itoa(pivotTableID) + ".xml"
- pivotTableXML := strings.Replace(sheetRelationshipsPivotTableXML, "..", "xl", -1)
- pivotCacheXML := "xl/pivotCache/pivotCacheDefinition" + strconv.Itoa(pivotCacheID) + ".xml"
- err = f.addPivotCache(pivotCacheID, pivotCacheXML, opt, dataSheet)
- if err != nil {
- return err
- }
- // workbook pivot cache
- workBookPivotCacheRID := f.addRels(f.getWorkbookRelsPath(), SourceRelationshipPivotCache, fmt.Sprintf("/xl/pivotCache/pivotCacheDefinition%d.xml", pivotCacheID), "")
- cacheID := f.addWorkbookPivotCache(workBookPivotCacheRID)
- pivotCacheRels := "xl/pivotTables/_rels/pivotTable" + strconv.Itoa(pivotTableID) + ".xml.rels"
- // rId not used
- _ = f.addRels(pivotCacheRels, SourceRelationshipPivotCache, fmt.Sprintf("../pivotCache/pivotCacheDefinition%d.xml", pivotCacheID), "")
- err = f.addPivotTable(cacheID, pivotTableID, pivotTableXML, opt)
- if err != nil {
- return err
- }
- pivotTableSheetRels := "xl/worksheets/_rels/" + strings.TrimPrefix(pivotTableSheetPath, "xl/worksheets/") + ".rels"
- f.addRels(pivotTableSheetRels, SourceRelationshipPivotTable, sheetRelationshipsPivotTableXML, "")
- f.addContentTypePart(pivotTableID, "pivotTable")
- f.addContentTypePart(pivotCacheID, "pivotCache")
- return nil
- }
- // parseFormatPivotTableSet provides a function to validate pivot table
- // properties.
- func (f *File) parseFormatPivotTableSet(opt *PivotTableOption) (*xlsxWorksheet, string, error) {
- if opt == nil {
- return nil, "", ErrParameterRequired
- }
- pivotTableSheetName, _, err := f.adjustRange(opt.PivotTableRange)
- if err != nil {
- return nil, "", fmt.Errorf("parameter 'PivotTableRange' parsing error: %s", err.Error())
- }
- opt.pivotTableSheetName = pivotTableSheetName
- dataRange := f.getDefinedNameRefTo(opt.DataRange, pivotTableSheetName)
- if dataRange == "" {
- dataRange = opt.DataRange
- }
- dataSheetName, _, err := f.adjustRange(dataRange)
- if err != nil {
- return nil, "", fmt.Errorf("parameter 'DataRange' parsing error: %s", err.Error())
- }
- dataSheet, err := f.workSheetReader(dataSheetName)
- if err != nil {
- return dataSheet, "", err
- }
- pivotTableSheetPath, ok := f.sheetMap[trimSheetName(pivotTableSheetName)]
- if !ok {
- return dataSheet, pivotTableSheetPath, fmt.Errorf("sheet %s is not exist", pivotTableSheetName)
- }
- return dataSheet, pivotTableSheetPath, err
- }
- // adjustRange adjust range, for example: adjust Sheet1!$E$31:$A$1 to Sheet1!$A$1:$E$31
- func (f *File) adjustRange(rangeStr string) (string, []int, error) {
- if len(rangeStr) < 1 {
- return "", []int{}, ErrParameterRequired
- }
- rng := strings.Split(rangeStr, "!")
- if len(rng) != 2 {
- return "", []int{}, ErrParameterInvalid
- }
- trimRng := strings.Replace(rng[1], "$", "", -1)
- coordinates, err := f.areaRefToCoordinates(trimRng)
- if err != nil {
- return rng[0], []int{}, err
- }
- x1, y1, x2, y2 := coordinates[0], coordinates[1], coordinates[2], coordinates[3]
- if x1 == x2 && y1 == y2 {
- return rng[0], []int{}, ErrParameterInvalid
- }
- // Correct the coordinate area, such correct C1:B3 to B1:C3.
- if x2 < x1 {
- x1, x2 = x2, x1
- }
- if y2 < y1 {
- y1, y2 = y2, y1
- }
- return rng[0], []int{x1, y1, x2, y2}, nil
- }
- // getPivotFieldsOrder provides a function to get order list of pivot table
- // fields.
- func (f *File) getPivotFieldsOrder(opt *PivotTableOption) ([]string, error) {
- order := []string{}
- dataRange := f.getDefinedNameRefTo(opt.DataRange, opt.pivotTableSheetName)
- if dataRange == "" {
- dataRange = opt.DataRange
- }
- dataSheet, coordinates, err := f.adjustRange(dataRange)
- if err != nil {
- return order, fmt.Errorf("parameter 'DataRange' parsing error: %s", err.Error())
- }
- for col := coordinates[0]; col <= coordinates[2]; col++ {
- coordinate, _ := CoordinatesToCellName(col, coordinates[1])
- name, err := f.GetCellValue(dataSheet, coordinate)
- if err != nil {
- return order, err
- }
- order = append(order, name)
- }
- return order, nil
- }
- // addPivotCache provides a function to create a pivot cache by given properties.
- func (f *File) addPivotCache(pivotCacheID int, pivotCacheXML string, opt *PivotTableOption, ws *xlsxWorksheet) error {
- // validate data range
- definedNameRef := true
- dataRange := f.getDefinedNameRefTo(opt.DataRange, opt.pivotTableSheetName)
- if dataRange == "" {
- definedNameRef = false
- dataRange = opt.DataRange
- }
- dataSheet, coordinates, err := f.adjustRange(dataRange)
- if err != nil {
- return fmt.Errorf("parameter 'DataRange' parsing error: %s", err.Error())
- }
- // data range has been checked
- order, _ := f.getPivotFieldsOrder(opt)
- hcell, _ := CoordinatesToCellName(coordinates[0], coordinates[1])
- vcell, _ := CoordinatesToCellName(coordinates[2], coordinates[3])
- pc := xlsxPivotCacheDefinition{
- SaveData: false,
- RefreshOnLoad: true,
- CreatedVersion: pivotTableVersion,
- RefreshedVersion: pivotTableVersion,
- MinRefreshableVersion: pivotTableVersion,
- CacheSource: &xlsxCacheSource{
- Type: "worksheet",
- WorksheetSource: &xlsxWorksheetSource{
- Ref: hcell + ":" + vcell,
- Sheet: dataSheet,
- },
- },
- CacheFields: &xlsxCacheFields{},
- }
- if definedNameRef {
- pc.CacheSource.WorksheetSource = &xlsxWorksheetSource{Name: opt.DataRange}
- }
- for _, name := range order {
- defaultRowsSubtotal, rowOk := f.getPivotTableFieldNameDefaultSubtotal(name, opt.Rows)
- defaultColumnsSubtotal, colOk := f.getPivotTableFieldNameDefaultSubtotal(name, opt.Columns)
- sharedItems := xlsxSharedItems{
- Count: 0,
- }
- s := xlsxString{}
- if (rowOk && !defaultRowsSubtotal) || (colOk && !defaultColumnsSubtotal) {
- s = xlsxString{
- V: "",
- }
- sharedItems.Count++
- sharedItems.S = &s
- }
- pc.CacheFields.CacheField = append(pc.CacheFields.CacheField, &xlsxCacheField{
- Name: name,
- SharedItems: &sharedItems,
- })
- }
- pc.CacheFields.Count = len(pc.CacheFields.CacheField)
- pivotCache, err := xml.Marshal(pc)
- f.saveFileList(pivotCacheXML, pivotCache)
- return err
- }
- // addPivotTable provides a function to create a pivot table by given pivot
- // table ID and properties.
- func (f *File) addPivotTable(cacheID, pivotTableID int, pivotTableXML string, opt *PivotTableOption) error {
- // validate pivot table range
- _, coordinates, err := f.adjustRange(opt.PivotTableRange)
- if err != nil {
- return fmt.Errorf("parameter 'PivotTableRange' parsing error: %s", err.Error())
- }
- hcell, _ := CoordinatesToCellName(coordinates[0], coordinates[1])
- vcell, _ := CoordinatesToCellName(coordinates[2], coordinates[3])
- pivotTableStyle := func() string {
- if opt.PivotTableStyleName == "" {
- return "PivotStyleLight16"
- }
- return opt.PivotTableStyleName
- }
- pt := xlsxPivotTableDefinition{
- Name: fmt.Sprintf("Pivot Table%d", pivotTableID),
- CacheID: cacheID,
- RowGrandTotals: &opt.RowGrandTotals,
- ColGrandTotals: &opt.ColGrandTotals,
- UpdatedVersion: pivotTableVersion,
- MinRefreshableVersion: pivotTableVersion,
- ShowDrill: &opt.ShowDrill,
- UseAutoFormatting: &opt.UseAutoFormatting,
- PageOverThenDown: &opt.PageOverThenDown,
- MergeItem: &opt.MergeItem,
- CreatedVersion: pivotTableVersion,
- CompactData: &opt.CompactData,
- ShowError: &opt.ShowError,
- DataCaption: "Values",
- Location: &xlsxLocation{
- Ref: hcell + ":" + vcell,
- FirstDataCol: 1,
- FirstDataRow: 1,
- FirstHeaderRow: 1,
- },
- PivotFields: &xlsxPivotFields{},
- RowItems: &xlsxRowItems{
- Count: 1,
- I: []*xlsxI{
- {
- []*xlsxX{{}, {}},
- },
- },
- },
- ColItems: &xlsxColItems{
- Count: 1,
- I: []*xlsxI{{}},
- },
- PivotTableStyleInfo: &xlsxPivotTableStyleInfo{
- Name: pivotTableStyle(),
- ShowRowHeaders: opt.ShowRowHeaders,
- ShowColHeaders: opt.ShowColHeaders,
- ShowRowStripes: opt.ShowRowStripes,
- ShowColStripes: opt.ShowColStripes,
- ShowLastColumn: opt.ShowLastColumn,
- },
- }
- // pivot fields
- _ = f.addPivotFields(&pt, opt)
- // count pivot fields
- pt.PivotFields.Count = len(pt.PivotFields.PivotField)
- // data range has been checked
- _ = f.addPivotRowFields(&pt, opt)
- _ = f.addPivotColFields(&pt, opt)
- _ = f.addPivotPageFields(&pt, opt)
- _ = f.addPivotDataFields(&pt, opt)
- pivotTable, err := xml.Marshal(pt)
- f.saveFileList(pivotTableXML, pivotTable)
- return err
- }
- // addPivotRowFields provides a method to add row fields for pivot table by
- // given pivot table options.
- func (f *File) addPivotRowFields(pt *xlsxPivotTableDefinition, opt *PivotTableOption) error {
- // row fields
- rowFieldsIndex, err := f.getPivotFieldsIndex(opt.Rows, opt)
- if err != nil {
- return err
- }
- for _, fieldIdx := range rowFieldsIndex {
- if pt.RowFields == nil {
- pt.RowFields = &xlsxRowFields{}
- }
- pt.RowFields.Field = append(pt.RowFields.Field, &xlsxField{
- X: fieldIdx,
- })
- }
- // count row fields
- if pt.RowFields != nil {
- pt.RowFields.Count = len(pt.RowFields.Field)
- }
- return err
- }
- // addPivotPageFields provides a method to add page fields for pivot table by
- // given pivot table options.
- func (f *File) addPivotPageFields(pt *xlsxPivotTableDefinition, opt *PivotTableOption) error {
- // page fields
- pageFieldsIndex, err := f.getPivotFieldsIndex(opt.Filter, opt)
- if err != nil {
- return err
- }
- pageFieldsName := f.getPivotTableFieldsName(opt.Filter)
- for idx, pageField := range pageFieldsIndex {
- if pt.PageFields == nil {
- pt.PageFields = &xlsxPageFields{}
- }
- pt.PageFields.PageField = append(pt.PageFields.PageField, &xlsxPageField{
- Name: pageFieldsName[idx],
- Fld: pageField,
- })
- }
- // count page fields
- if pt.PageFields != nil {
- pt.PageFields.Count = len(pt.PageFields.PageField)
- }
- return err
- }
- // addPivotDataFields provides a method to add data fields for pivot table by
- // given pivot table options.
- func (f *File) addPivotDataFields(pt *xlsxPivotTableDefinition, opt *PivotTableOption) error {
- // data fields
- dataFieldsIndex, err := f.getPivotFieldsIndex(opt.Data, opt)
- if err != nil {
- return err
- }
- dataFieldsSubtotals := f.getPivotTableFieldsSubtotal(opt.Data)
- dataFieldsName := f.getPivotTableFieldsName(opt.Data)
- for idx, dataField := range dataFieldsIndex {
- if pt.DataFields == nil {
- pt.DataFields = &xlsxDataFields{}
- }
- pt.DataFields.DataField = append(pt.DataFields.DataField, &xlsxDataField{
- Name: dataFieldsName[idx],
- Fld: dataField,
- Subtotal: dataFieldsSubtotals[idx],
- })
- }
- // count data fields
- if pt.DataFields != nil {
- pt.DataFields.Count = len(pt.DataFields.DataField)
- }
- return err
- }
- // inStrSlice provides a method to check if an element is present in an array,
- // and return the index of its location, otherwise return -1.
- func inStrSlice(a []string, x string) int {
- for idx, n := range a {
- if x == n {
- return idx
- }
- }
- return -1
- }
- // inPivotTableField provides a method to check if an element is present in
- // pivot table fields list, and return the index of its location, otherwise
- // return -1.
- func inPivotTableField(a []PivotTableField, x string) int {
- for idx, n := range a {
- if x == n.Data {
- return idx
- }
- }
- return -1
- }
- // addPivotColFields create pivot column fields by given pivot table
- // definition and option.
- func (f *File) addPivotColFields(pt *xlsxPivotTableDefinition, opt *PivotTableOption) error {
- if len(opt.Columns) == 0 {
- if len(opt.Data) <= 1 {
- return nil
- }
- pt.ColFields = &xlsxColFields{}
- // in order to create pivot table in case there is no input from Columns
- pt.ColFields.Count = 1
- pt.ColFields.Field = append(pt.ColFields.Field, &xlsxField{
- X: -2,
- })
- return nil
- }
- pt.ColFields = &xlsxColFields{}
- // col fields
- colFieldsIndex, err := f.getPivotFieldsIndex(opt.Columns, opt)
- if err != nil {
- return err
- }
- for _, fieldIdx := range colFieldsIndex {
- pt.ColFields.Field = append(pt.ColFields.Field, &xlsxField{
- X: fieldIdx,
- })
- }
- //in order to create pivot in case there is many Columns and Many Datas
- if len(opt.Data) > 1 {
- pt.ColFields.Field = append(pt.ColFields.Field, &xlsxField{
- X: -2,
- })
- }
- // count col fields
- pt.ColFields.Count = len(pt.ColFields.Field)
- return err
- }
- // addPivotFields create pivot fields based on the column order of the first
- // row in the data region by given pivot table definition and option.
- func (f *File) addPivotFields(pt *xlsxPivotTableDefinition, opt *PivotTableOption) error {
- order, err := f.getPivotFieldsOrder(opt)
- if err != nil {
- return err
- }
- x := 0
- for _, name := range order {
- if inPivotTableField(opt.Rows, name) != -1 {
- defaultSubtotal, ok := f.getPivotTableFieldNameDefaultSubtotal(name, opt.Rows)
- var items []*xlsxItem
- if !ok || !defaultSubtotal {
- items = append(items, &xlsxItem{X: &x})
- } else {
- items = append(items, &xlsxItem{T: "default"})
- }
- pt.PivotFields.PivotField = append(pt.PivotFields.PivotField, &xlsxPivotField{
- Axis: "axisRow",
- Name: f.getPivotTableFieldName(name, opt.Rows),
- Items: &xlsxItems{
- Count: len(items),
- Item: items,
- },
- DefaultSubtotal: &defaultSubtotal,
- })
- continue
- }
- if inPivotTableField(opt.Filter, name) != -1 {
- pt.PivotFields.PivotField = append(pt.PivotFields.PivotField, &xlsxPivotField{
- Axis: "axisPage",
- Name: f.getPivotTableFieldName(name, opt.Columns),
- Items: &xlsxItems{
- Count: 1,
- Item: []*xlsxItem{
- {T: "default"},
- },
- },
- })
- continue
- }
- if inPivotTableField(opt.Columns, name) != -1 {
- defaultSubtotal, ok := f.getPivotTableFieldNameDefaultSubtotal(name, opt.Columns)
- var items []*xlsxItem
- if !ok || !defaultSubtotal {
- items = append(items, &xlsxItem{X: &x})
- } else {
- items = append(items, &xlsxItem{T: "default"})
- }
- pt.PivotFields.PivotField = append(pt.PivotFields.PivotField, &xlsxPivotField{
- Axis: "axisCol",
- Name: f.getPivotTableFieldName(name, opt.Columns),
- Items: &xlsxItems{
- Count: len(items),
- Item: items,
- },
- DefaultSubtotal: &defaultSubtotal,
- })
- continue
- }
- if inPivotTableField(opt.Data, name) != -1 {
- pt.PivotFields.PivotField = append(pt.PivotFields.PivotField, &xlsxPivotField{
- DataField: true,
- })
- continue
- }
- pt.PivotFields.PivotField = append(pt.PivotFields.PivotField, &xlsxPivotField{})
- }
- return err
- }
- // countPivotTables provides a function to get drawing files count storage in
- // the folder xl/pivotTables.
- func (f *File) countPivotTables() int {
- count := 0
- f.Pkg.Range(func(k, v interface{}) bool {
- if strings.Contains(k.(string), "xl/pivotTables/pivotTable") {
- count++
- }
- return true
- })
- return count
- }
- // countPivotCache provides a function to get drawing files count storage in
- // the folder xl/pivotCache.
- func (f *File) countPivotCache() int {
- count := 0
- f.Pkg.Range(func(k, v interface{}) bool {
- if strings.Contains(k.(string), "xl/pivotCache/pivotCacheDefinition") {
- count++
- }
- return true
- })
- return count
- }
- // getPivotFieldsIndex convert the column of the first row in the data region
- // to a sequential index by given fields and pivot option.
- func (f *File) getPivotFieldsIndex(fields []PivotTableField, opt *PivotTableOption) ([]int, error) {
- pivotFieldsIndex := []int{}
- orders, err := f.getPivotFieldsOrder(opt)
- if err != nil {
- return pivotFieldsIndex, err
- }
- for _, field := range fields {
- if pos := inStrSlice(orders, field.Data); pos != -1 {
- pivotFieldsIndex = append(pivotFieldsIndex, pos)
- }
- }
- return pivotFieldsIndex, nil
- }
- // getPivotTableFieldsSubtotal prepare fields subtotal by given pivot table fields.
- func (f *File) getPivotTableFieldsSubtotal(fields []PivotTableField) []string {
- field := make([]string, len(fields))
- enums := []string{"average", "count", "countNums", "max", "min", "product", "stdDev", "stdDevp", "sum", "var", "varp"}
- inEnums := func(enums []string, val string) string {
- for _, enum := range enums {
- if strings.EqualFold(enum, val) {
- return enum
- }
- }
- return "sum"
- }
- for idx, fld := range fields {
- field[idx] = inEnums(enums, fld.Subtotal)
- }
- return field
- }
- // getPivotTableFieldsName prepare fields name list by given pivot table
- // fields.
- func (f *File) getPivotTableFieldsName(fields []PivotTableField) []string {
- field := make([]string, len(fields))
- for idx, fld := range fields {
- if len(fld.Name) > 255 {
- field[idx] = fld.Name[0:255]
- continue
- }
- field[idx] = fld.Name
- }
- return field
- }
- // getPivotTableFieldName prepare field name by given pivot table fields.
- func (f *File) getPivotTableFieldName(name string, fields []PivotTableField) string {
- fieldsName := f.getPivotTableFieldsName(fields)
- for idx, field := range fields {
- if field.Data == name {
- return fieldsName[idx]
- }
- }
- return ""
- }
- func (f *File) getPivotTableFieldNameDefaultSubtotal(name string, fields []PivotTableField) (bool, bool) {
- for _, field := range fields {
- if field.Data == name {
- return field.DefaultSubtotal, true
- }
- }
- return false, false
- }
- // addWorkbookPivotCache add the association ID of the pivot cache in workbook.xml.
- func (f *File) addWorkbookPivotCache(RID int) int {
- wb := f.workbookReader()
- if wb.PivotCaches == nil {
- wb.PivotCaches = &xlsxPivotCaches{}
- }
- cacheID := 1
- for _, pivotCache := range wb.PivotCaches.PivotCache {
- if pivotCache.CacheID > cacheID {
- cacheID = pivotCache.CacheID
- }
- }
- cacheID++
- wb.PivotCaches.PivotCache = append(wb.PivotCaches.PivotCache, xlsxPivotCache{
- CacheID: cacheID,
- RID: fmt.Sprintf("rId%d", RID),
- })
- return cacheID
- }
|