watermark.go 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. package processing
  2. import (
  3. "context"
  4. "github.com/imgproxy/imgproxy/v3/config"
  5. "github.com/imgproxy/imgproxy/v3/imagedata"
  6. "github.com/imgproxy/imgproxy/v3/imath"
  7. "github.com/imgproxy/imgproxy/v3/options"
  8. "github.com/imgproxy/imgproxy/v3/vips"
  9. )
  10. var watermarkPipeline = pipeline{
  11. prepare,
  12. scaleOnLoad,
  13. importColorProfile,
  14. scale,
  15. rotateAndFlip,
  16. padding,
  17. }
  18. func prepareWatermark(wm *vips.Image, wmData *imagedata.ImageData, opts *options.WatermarkOptions, imgWidth, imgHeight int) error {
  19. if err := wm.Load(wmData, 1, 1.0, 1); err != nil {
  20. return err
  21. }
  22. po := options.NewProcessingOptions()
  23. po.ResizingType = options.ResizeFit
  24. po.Dpr = 1
  25. po.Enlarge = true
  26. po.Format = wmData.Type
  27. if opts.Scale > 0 {
  28. po.Width = imath.Max(imath.Scale(imgWidth, opts.Scale), 1)
  29. po.Height = imath.Max(imath.Scale(imgHeight, opts.Scale), 1)
  30. }
  31. if opts.Replicate {
  32. po.Padding.Enabled = true
  33. po.Padding.Left = int(opts.Gravity.X / 2)
  34. po.Padding.Right = int(opts.Gravity.X) - po.Padding.Left
  35. po.Padding.Top = int(opts.Gravity.Y / 2)
  36. po.Padding.Bottom = int(opts.Gravity.Y) - po.Padding.Top
  37. }
  38. if err := watermarkPipeline.Run(context.Background(), wm, po, wmData); err != nil {
  39. return err
  40. }
  41. if opts.Replicate {
  42. return wm.Replicate(imgWidth, imgHeight)
  43. }
  44. left, top := calcPosition(imgWidth, imgHeight, wm.Width(), wm.Height(), &opts.Gravity, true)
  45. return wm.Embed(imgWidth, imgHeight, left, top)
  46. }
  47. func applyWatermark(img *vips.Image, wmData *imagedata.ImageData, opts *options.WatermarkOptions, framesCount int) error {
  48. if err := img.RgbColourspace(); err != nil {
  49. return err
  50. }
  51. wm := new(vips.Image)
  52. defer wm.Clear()
  53. width := img.Width()
  54. height := img.Height()
  55. if err := prepareWatermark(wm, wmData, opts, width, height/framesCount); err != nil {
  56. return err
  57. }
  58. if framesCount > 1 {
  59. if err := wm.Replicate(width, height); err != nil {
  60. return err
  61. }
  62. }
  63. opacity := opts.Opacity * config.WatermarkOpacity
  64. return img.ApplyWatermark(wm, opacity)
  65. }
  66. func watermark(pctx *pipelineContext, img *vips.Image, po *options.ProcessingOptions, imgdata *imagedata.ImageData) error {
  67. if !po.Watermark.Enabled || imagedata.Watermark == nil {
  68. return nil
  69. }
  70. return applyWatermark(img, imagedata.Watermark, &po.Watermark, 1)
  71. }