-
Notifications
You must be signed in to change notification settings - Fork 1
/
contextImages.go
64 lines (56 loc) · 2.11 KB
/
contextImages.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
// Licensed to NASA JPL under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. NASA JPL licenses this file to you 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 jplbreadboard
import (
"path/filepath"
"strconv"
"strings"
"github.com/pixlise/core/v4/core/fileaccess"
"github.com/pixlise/core/v4/core/logger"
)
func processContextImages(path string, jobLog logger.ILogger, fs fileaccess.FileAccess) (map[int32]string, error) {
jobLog.Infof(" Reading context image files from directory: %v", path)
contextImgDirFiles, err := fs.ListObjects(path, "")
//contextImgDirFiles, err := importerutils.GetDirListing(path, "", jobLog)
if err != nil {
return nil, err
}
return getContextImagesPerPMCFromListing(contextImgDirFiles, jobLog), nil
}
func getContextImagesPerPMCFromListing(paths []string, jobLog logger.ILogger) map[int32]string {
result := make(map[int32]string)
for _, pathitem := range paths {
_, file := filepath.Split(pathitem)
extension := filepath.Ext(file)
if extension == ".jpg" {
fileNameBits := strings.Split(file, "_")
if len(fileNameBits) != 3 {
jobLog.Infof("Ignored unexpected image file name \"%v\" when searching for context images.", pathitem)
} else {
pmcStr := fileNameBits[len(fileNameBits)-1]
pmcStr = pmcStr[0 : len(pmcStr)-len(extension)]
pmcI, err := strconv.Atoi(pmcStr)
if err != nil {
jobLog.Infof("Ignored unexpected image file name \"%v\", couldn't parse PMC.", pathitem)
} else {
result[int32(pmcI)] = file
}
}
}
}
return result
}