add AddInlineDrawingFrom

apidrawing.go

@@ -1,7 +1,95 @@
 package docxlib
 
-// AddDrawing adds drawing to paragraph
-func (p *Paragraph) AddDrawing(pic []byte) *Run {
-	//TODO: finish add drawing
-	return nil
+import (
+	"bytes"
+	"fmt"
+	"math/rand"
+	"os"
+	"strconv"
+	"sync/atomic"
+
+	"github.com/fumiama/imgsz"
+)
+
+// AddInlineDrawing adds inline drawing to paragraph
+func (p *Paragraph) AddInlineDrawing(pic []byte) (*Run, error) {
+	sz, format, err := imgsz.DecodeSize(bytes.NewReader(pic))
+	if err != nil {
+		return nil, err
+	}
+	id := strconv.Itoa(int(atomic.AddUintptr(&p.file.imageId, 1)))
+	rId := p.file.addImage(Media{Name: "image" + id + "." + format, Data: pic})
+	w, h := sz.Width, sz.Height
+	if float64(w)/float64(h) > 1.2 {
+		h = A4_EMU_MAX_WIDTH * h / w
+		w = A4_EMU_MAX_WIDTH
+	} else {
+		h = A4_EMU_MAX_WIDTH * h / w / 2
+		w = A4_EMU_MAX_WIDTH / 2
+	}
+	d := &Drawing{
+		Inline: &WPInline{
+			AnchorID: fmt.Sprintf("%08X", rand.Uint32()),
+			EditID:   fmt.Sprintf("%08X", rand.Uint32()),
+
+			Extent: &WPExtent{
+				CX: w,
+				CY: h,
+			},
+			EffectExtent: &WPEffectExtent{},
+			DocPr: &WPDocPr{
+				ID:   id,
+				Name: "图片 " + id,
+			},
+			CNvGraphicFramePr: &WPCNvGraphicFramePr{
+				Locks: &AGraphicFrameLocks{
+					NoChangeAspect: 1,
+				},
+			},
+			Graphic: &AGraphic{
+				GraphicData: &AGraphicData{
+					URI: XMLNS_PICTURE,
+					Pic: &PICPic{
+						NonVisualPicProperties: &PICNonVisualPicProperties{
+							NonVisualDrawingProperties: PICNonVisualDrawingProperties{
+								ID: id,
+							},
+						},
+						BlipFill: &PICBlipFill{
+							Blip: ABlip{
+								Embed:  rId,
+								Cstate: "print",
+							},
+						},
+						SpPr: &PICSpPr{
+							Xfrm: AXfrm{
+								Ext: AExt{
+									CX: w,
+									CY: h,
+								},
+							},
+							PrstGeom: APrstGeom{
+								Prst: "rect",
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+	run := &Run{
+		Drawing:       d,
+		RunProperties: &RunProperties{},
+	}
+	p.Children = append(p.Children, ParagraphChild{Run: run})
+	return run, nil
+}
+
+// AddInlineDrawingFrom adds drawing from file to paragraph
+func (p *Paragraph) AddInlineDrawingFrom(file string) (*Run, error) {
+	data, err := os.ReadFile(file)
+	if err != nil {
+		return nil, err
+	}
+	return p.AddInlineDrawing(data)
 }