mirror of
https://github.com/simon987/imhashdb.git
synced 2025-04-04 02:22:58 +00:00
123 lines
2.4 KiB
Go
123 lines
2.4 KiB
Go
package imhashdb
|
|
|
|
import "github.com/simon987/fastimagehash-go"
|
|
|
|
type HashType string
|
|
|
|
const (
|
|
DHash8 HashType = "dhash8"
|
|
DHash16 HashType = "dhash16"
|
|
DHash32 HashType = "dhash32"
|
|
|
|
MHash8 HashType = "mhash8"
|
|
MHash16 HashType = "mhash16"
|
|
MHash32 HashType = "mhash32"
|
|
|
|
PHash8 HashType = "phash8"
|
|
PHash16 HashType = "phash16"
|
|
PHash32 HashType = "phash32"
|
|
|
|
WHash8Haar HashType = "whash8haar"
|
|
WHash16Haar HashType = "whash16haar"
|
|
WHash32Haar HashType = "whash32haar"
|
|
)
|
|
|
|
var HashTypes = []HashType{
|
|
DHash8, DHash16, DHash32,
|
|
MHash8, MHash16, MHash32,
|
|
PHash8, PHash16, PHash32,
|
|
WHash8Haar, WHash16Haar, WHash32Haar,
|
|
}
|
|
|
|
func (h HashType) HashLength() int {
|
|
switch h {
|
|
case DHash8:
|
|
fallthrough
|
|
case MHash8:
|
|
fallthrough
|
|
case PHash8:
|
|
fallthrough
|
|
case WHash8Haar:
|
|
return 8
|
|
|
|
case DHash16:
|
|
fallthrough
|
|
case MHash16:
|
|
fallthrough
|
|
case PHash16:
|
|
fallthrough
|
|
case WHash16Haar:
|
|
return 32
|
|
|
|
case DHash32:
|
|
fallthrough
|
|
case MHash32:
|
|
fallthrough
|
|
case PHash32:
|
|
fallthrough
|
|
case WHash32Haar:
|
|
return 128
|
|
default:
|
|
panic("Invalid invalid hash")
|
|
}
|
|
}
|
|
|
|
type HashReq struct {
|
|
Data []byte `json:"data"`
|
|
}
|
|
type Hashes struct {
|
|
DHash8 *fastimagehash.Hash `json:"dhash8"`
|
|
DHash16 *fastimagehash.Hash `json:"dhash16"`
|
|
DHash32 *fastimagehash.Hash `json:"dhash32"`
|
|
|
|
MHash8 *fastimagehash.Hash `json:"mhash8"`
|
|
MHash16 *fastimagehash.Hash `json:"mhash16"`
|
|
MHash32 *fastimagehash.Hash `json:"mhash32"`
|
|
|
|
PHash8 *fastimagehash.Hash `json:"phash8"`
|
|
PHash16 *fastimagehash.Hash `json:"phash16"`
|
|
PHash32 *fastimagehash.Hash `json:"phash32"`
|
|
|
|
WHash8 *fastimagehash.Hash `json:"whash8haar"`
|
|
WHash16 *fastimagehash.Hash `json:"whash16haar"`
|
|
WHash32 *fastimagehash.Hash `json:"whash32haar"`
|
|
}
|
|
|
|
type QueryReq struct {
|
|
HashType HashType `json:"type"`
|
|
Hash []byte `json:"hash"`
|
|
Distance uint `json:"distance"`
|
|
Limit uint `json:"limit"`
|
|
Offset uint `json:"offset"`
|
|
}
|
|
|
|
type ImageList struct {
|
|
Images []*Image `json:"images"`
|
|
}
|
|
|
|
type QueryResp struct {
|
|
Err string `json:"err,omitempty"`
|
|
}
|
|
|
|
type Meta struct {
|
|
RetrievedAt int64 `json:"retrieved_at"`
|
|
Id string `json:"id"`
|
|
Meta interface{} `json:"meta"`
|
|
}
|
|
|
|
type ImageHasMeta struct {
|
|
Url string `json:"url"`
|
|
Meta Meta `json:"meta"`
|
|
}
|
|
|
|
type Image struct {
|
|
id int64
|
|
Size int `json:"size"`
|
|
Sha1 []byte `json:"sha1"`
|
|
Md5 []byte `json:"md5"`
|
|
Sha256 []byte `json:"sha256"`
|
|
Crc32 uint32 `json:"crc32"`
|
|
|
|
Meta []ImageHasMeta `json:"meta"`
|
|
}
|