mirror of
https://github.com/xiaoqidun/ofdgo.git
synced 2026-10-29 14:53:44 +08:00
1 parent
d3904521b9
commit
3e87048a30
8 files changed
+336
-32
No files matched your search
@@ -0,0 +1,79 @@
|
||||
// Copyright 2025-2026 肖其顿 (XIAO QI DUN)
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package ofdgo
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"slices"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// ParsePageRange 解析一基页码及闭区间,返回升序去重的零基页面索引
|
||||
// 入参: value 页码表达式,如1-6,7,8-10, count 文档总页数
|
||||
// 返回: []int 页面索引, error 错误信息
|
||||
func ParsePageRange(value string, count int) ([]int, error) {
|
||||
selected := make(map[int]bool)
|
||||
for _, part := range strings.Split(strings.ReplaceAll(value, ",", ","), ",") {
|
||||
first, last, interval := strings.Cut(strings.TrimSpace(part), "-")
|
||||
start, err := strconv.Atoi(strings.TrimSpace(first))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("invalid page range %q", part)
|
||||
}
|
||||
end := start
|
||||
if interval {
|
||||
end, err = strconv.Atoi(strings.TrimSpace(last))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("invalid page range %q", part)
|
||||
}
|
||||
}
|
||||
if start < 1 || end < start || end > count {
|
||||
return nil, fmt.Errorf("page range %q outside 1-%d", part, count)
|
||||
}
|
||||
for index := start - 1; index < end; index++ {
|
||||
selected[index] = true
|
||||
}
|
||||
}
|
||||
indices := make([]int, 0, len(selected))
|
||||
for index := range selected {
|
||||
indices = append(indices, index)
|
||||
}
|
||||
slices.Sort(indices)
|
||||
return indices, nil
|
||||
}
|
||||
|
||||
// exportPageIndices 校验导出索引并按原页序去重,省略索引时选取全部页面
|
||||
// 入参: count 文档总页数, indices 零基页面索引
|
||||
// 返回: []int 页面索引, error 错误信息
|
||||
func exportPageIndices(count int, indices []int) ([]int, error) {
|
||||
if count == 0 {
|
||||
return nil, fmt.Errorf("no pages found")
|
||||
}
|
||||
if len(indices) == 0 {
|
||||
indices = make([]int, count)
|
||||
for index := range indices {
|
||||
indices[index] = index
|
||||
}
|
||||
return indices, nil
|
||||
}
|
||||
for _, index := range indices {
|
||||
if index < 0 || index >= count {
|
||||
return nil, fmt.Errorf("page index %d out of range", index)
|
||||
}
|
||||
}
|
||||
indices = slices.Clone(indices)
|
||||
slices.Sort(indices)
|
||||
return slices.Compact(indices), nil
|
||||
}
|
||||
Reference in new issue
Block a user