jacob1123/dupe-finder
1
0
Fork 0
mirror of https://github.com/JaCoB1123/dupe-finder.git synced 2025-05-18 06:01:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

Calculate dhash for jpgs

Browse Source
This commit is contained in:
Jan Bader 2023-12-04 22:59:55 +01:00
parent b58151efb7
commit bbdc296cbd
4 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
file.go
View File

@ -3,10 +3,14 @@ package main
import (
"crypto/sha1"
"encoding/base64"
"image/jpeg"
"io"
"os"
"path/filepath"
"strconv"
"strings"
"github.com/corona10/goimagehash"
)
func remove(path string) {
@ -48,6 +52,19 @@ func calculateHash(path string) (string, error) {
}
defer f.Close()
if strings.HasSuffix(path, ".jpg") {
img, err := jpeg.Decode(f)
if err != nil {
return "", err
}
hash, err := goimagehash.DifferenceHash(img)
if err != nil {
return "", err
}
return hash.ToString(), nil
}
h := sha1.New()
if _, err := io.Copy(h, f); err != nil {
return "", err

5
filesmap.go
View File

@ -6,6 +6,7 @@ import (
"log"
"os"
"path/filepath"
"strings"
"sync"
"github.com/vbauerster/mpb/v7"
@ -88,6 +89,10 @@ func (fm *FilesMap) WalkDirectories() int {
return nil
}
if !strings.HasSuffix(path, ".jpg") {
size = 123456789123456
}
fm.incomingBar.Increment()
countFiles++
fm.incomingBar.SetTotal(int64(countFiles), false)

2
go.mod
View File

@ -7,7 +7,9 @@ require github.com/vbauerster/mpb/v7 v7.0.5
require (
github.com/VividCortex/ewma v1.2.0 // indirect
github.com/acarl005/stripansi v0.0.0-20180116102854-5a71ef0e047d // indirect
github.com/corona10/goimagehash v1.1.0 // indirect
github.com/mattn/go-runewidth v0.0.13 // indirect
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 // indirect
github.com/rivo/uniseg v0.2.0 // indirect
golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c // indirect
)

4
go.sum
View File

@ -2,8 +2,12 @@ github.com/VividCortex/ewma v1.2.0 h1:f58SaIzcDXrSy3kWaHNvuJgJ3Nmz59Zji6XoJR/q1o
github.com/VividCortex/ewma v1.2.0/go.mod h1:nz4BbCtbLyFDeC9SUHbtcT5644juEuWfUAUnGx7j5l4=
github.com/acarl005/stripansi v0.0.0-20180116102854-5a71ef0e047d h1:licZJFw2RwpHMqeKTCYkitsPqHNxTmd4SNR5r94FGM8=
github.com/acarl005/stripansi v0.0.0-20180116102854-5a71ef0e047d/go.mod h1:asat636LX7Bqt5lYEZ27JNDcqxfjdBQuJ/MM4CN/Lzo=
github.com/corona10/goimagehash v1.1.0 h1:teNMX/1e+Wn/AYSbLHX8mj+mF9r60R1kBeqE9MkoYwI=
github.com/corona10/goimagehash v1.1.0/go.mod h1:VkvE0mLn84L4aF8vCb6mafVajEb6QYMHl2ZJLn0mOGI=
github.com/mattn/go-runewidth v0.0.13 h1:lTGmDsbAYt5DmK6OnoV7EuIF1wEIFAcxld6ypU4OSgU=
github.com/mattn/go-runewidth v0.0.13/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 h1:zYyBkD/k9seD2A7fsi6Oo2LfFZAehjjQMERAvZLEDnQ=
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646/go.mod h1:jpp1/29i3P1S/RLdc7JQKbRpFeM1dOBd8T9ki5s+AY8=
github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY=
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
github.com/vbauerster/mpb/v7 v7.0.5 h1:/CQxyoPjdlON6kqqq3Uq3UUw5tFjuBCjOmLQYaYvBmM=