wartank/vendor/github.com/visualfc/atk/tk/image.go

// Copyright 2018 visualfc. All rights reserved.

package tk

import (
	"errors"
	"fmt"
	"image"
	"image/color"
	_ "image/png"
	"os"
	"path/filepath"
	"strings"

	"github.com/visualfc/atk/tk/interp"
)

type Image struct {
	id        string
	photo     *interp.Photo
	tk85alpha color.Color
}

func (i *Image) Id() string {
	return i.id
}

type ImageAttr struct {
	key   string
	value interface{}
}

func ImageAttrGamma(gamma float64) *ImageAttr {
	return &ImageAttr{"gamma", gamma}
}

func ImageAttrTk85AlphaColor(color color.Color) *ImageAttr {
	return &ImageAttr{"tk85alphacolor", color}
}

func LoadImage(file string, attributes ...*ImageAttr) (*Image, error) {
	if file == "" {
		return nil, ErrInvalid
	}
	var fileImage image.Image
	if filepath.Ext(file) == ".gif" {
		attributes = append(attributes, &ImageAttr{"file", file})
	} else {
		file, err := os.Open(file)
		if err != nil {
			return nil, err
		}
		im, _, err := image.Decode(file)
		file.Close()
		if err != nil {
			return nil, err
		}
		fileImage = im
	}
	im := NewImage(attributes...)
	if im == nil {
		return nil, errors.New("NewImage failed")
	}
	if fileImage != nil {
		im.SetImage(fileImage)
	}
	return im, nil
}

func NewImage(attributes ...*ImageAttr) *Image {
	var attrList []string
	var tk85alphacolor color.Color
	for _, attr := range attributes {
		if attr == nil {
			continue
		}
		if attr.key == "tk85alphacolor" {
			if clr, ok := attr.value.(color.Color); ok {
				tk85alphacolor = clr
			}
			continue
		}
		if s, ok := attr.value.(string); ok {
			pname := "atk_tmp_" + attr.key
			setObjText(pname, s)
			attrList = append(attrList, fmt.Sprintf("-%v $%v", attr.key, pname))
			continue
		}
		attrList = append(attrList, fmt.Sprintf("-%v {%v}", attr.key, attr.value))
	}
	iid := makeNamedId("atk_image")
	script := fmt.Sprintf("image create photo %v", iid)
	if len(attrList) > 0 {
		script += " " + strings.Join(attrList, " ")
	}
	err := eval(script)
	if err != nil {
		return nil
	}
	photo := interp.FindPhoto(mainInterp, iid)
	if photo == nil {
		return nil
	}
	return &Image{iid, photo, tk85alphacolor}
}

func (i *Image) IsValid() bool {
	return i.id != "" && i.photo != nil
}

func (i *Image) SetImage(img image.Image) *Image {
	err := i.photo.PutImage(img, i.tk85alpha)
	if err != nil {
		dumpError(err)
	}
	return i
}

func (i *Image) SetZoomedImage(img image.Image, zoomX, zoomY, subsampleX, subsampleY int) *Image {
	err := i.photo.PutZoomedImage(img, zoomX, zoomY, subsampleX, subsampleY, i.tk85alpha)
	if err != nil {
		dumpError(err)
	}
	return i
}

func (i *Image) ToImage() image.Image {
	return i.photo.ToImage()
}

func (i *Image) Blank() *Image {
	i.photo.Blank()
	return i
}

func (i *Image) SizeN() (width int, height int) {
	return i.photo.Size()
}

func (i *Image) Size() Size {
	w, h := i.SizeN()
	return Size{w, h}
}

func (i *Image) SetSizeN(width int, height int) *Image {
	err := i.photo.SetSize(width, height)
	if err != nil {
		dumpError(err)
	}
	return i
}

func (i *Image) SetSize(sz Size) *Image {
	return i.SetSizeN(sz.Width, sz.Height)
}

func (i *Image) Gamma() float64 {
	v, _ := evalAsFloat64(fmt.Sprintf("%v cget -gamma", i.id))
	return v
}

func (i *Image) SetGamma(v float64) *Image {
	eval(fmt.Sprintf("%v configure -gamma {%v}", i.id, v))
	return i
}

func parserImageResult(id string, err error) *Image {
	if err != nil {
		return nil
	}
	photo := interp.FindPhoto(mainInterp, id)
	if photo == nil {
		return nil
	}
	return &Image{id, photo, nil}
}