photoprism/pkg/clusters/csv_importer.go

63 lines
895 B
Go
Raw Normal View History

package clusters
import (
"bufio"
"encoding/csv"
"io"
"os"
"strconv"
)
type csvImporter struct {
}
func CsvImporter() Importer {
return &csvImporter{}
}
func (i *csvImporter) Import(file string, start, end int) ([][]float64, error) {
if start < 0 || end < 0 || start > end {
return [][]float64{}, errInvalidRange
}
f, err := os.Open(file)
if err != nil {
return [][]float64{}, err
}
defer f.Close()
var (
d = make([][]float64, 0)
r = csv.NewReader(bufio.NewReader(f))
s = end - start + 1
g []float64
)
Main:
for {
record, err := r.Read()
if err == io.EOF {
break
} else if err != nil {
return [][]float64{}, err
}
g = make([]float64, 0, s)
for j := start; j <= end; j++ {
f, err := strconv.ParseFloat(record[j], 64)
if err == nil {
g = append(g, f)
} else {
continue Main
}
}
d = append(d, g)
}
return d, nil
}