Maison  >  Article  >  développement back-end  >  Convertir l'image en pdf golang

Convertir l'image en pdf golang

WBOY
WBOYoriginal
2023-05-15 09:04:07878parcourir

Récemment, lorsque je développais un outil de conversion de fichiers, j'avais besoin de convertir plusieurs images en un fichier PDF. Depuis que j'utilise le langage Golang, j'ai choisi d'utiliser le langage Go pour écrire un programme permettant de convertir des images en PDF.

Dans cet article, je partagerai l'expérience et quelques détails clés que j'ai acquis au cours du processus de développement. Voici les principales fonctions et processus du programme :

  • Lire plusieurs fichiers image
  • Créer des fichiers PDF
  • Convertir toutes les images en pages PDF
  • Enregistrer toutes les pages dans des fichiers PDF

Tout d'abord, nous devons traiter lecture de fichiers. Il existe une bibliothèque standard "io/ioutil" dans Go pour lire des fichiers, et elle est très pratique et facile à utiliser. Nous pouvons obtenir tous les fichiers du répertoire spécifié en utilisant la méthode ReadDir() dans cette bibliothèque.

func getImagesFromDir(dir string) ([]string, error) {
    files, err := ioutil.ReadDir(dir)
    images := []string{}

    if err != nil {
        return images, err
    }

    for _, file := range files {
        if !file.IsDir() && strings.Contains(file.Name(), ".jpg") {
            images = append(images, filepath.Join(dir, file.Name()))
        }
    }

    return images, nil
}

Ensuite, nous devons créer un fichier PDF. Il existe plusieurs bibliothèques tierces dans Go qui peuvent créer des fichiers PDF, et nous pouvons choisir d'utiliser la bibliothèque GoFPDF. La bibliothèque offre plusieurs options et fonctionnalités de personnalisation telles que le redimensionnement de la page, la définition des polices et des couleurs, etc.

pdf := gofpdf.New("P", "mm", "A4", "")
pdf.AddPage()
pdf.SetFont("Arial", "B", 16)
pdf.Cell(40, 10, "Hello, world!")
pdf.OutputFileAndClose("hello.pdf")

Nous avons maintenant créé avec succès un fichier PDF, mais nous n'y avons pas encore ajouté d'images. L'étape suivante consiste à convertir toutes les images en pages PDF, ce qui peut être réalisé en ajoutant des images comme arrière-plans de page. Nous pouvons utiliser les bibliothèques standard d’images et d’images/dessins dans Go pour ouvrir et traiter des images.

func imageToPdf(imagePath string, pdf *gofpdf.Fpdf) {
    image.RegisterFormat("jpeg", "jpeg", jpeg.Decode, jpeg.DecodeConfig)

    f, err := os.Open(imagePath)
    defer f.Close()

    if err != nil {
        log.Fatal("Failed to open file:", err)
    }

    // decode the image
    img, _, err := image.Decode(f)

    if err != nil {
        log.Fatal("Failed to decode image:", err)
    }

    // get image dimensions
    w := float64(img.Bounds().Max.X)
    h := float64(img.Bounds().Max.Y)

    // add page to pdf
    pdf.AddPageFormat("P", gofpdf.SizeType{Wd: w, Ht: h})
    pdf.Image(imagePath, 0, 0, w, h, false, "", 0, "")
}

La dernière étape consiste à enregistrer toutes les pages dans un fichier PDF. Nous pouvons utiliser la méthode WriteFile() dans Golang pour écrire toutes les pages dans un fichier avec le suffixe pdf.

func savePdf(pdf *gofpdf.Fpdf, outputPath string) error {
    return pdf.OutputFileAndClose(outputPath)
}

Maintenant, nous pouvons intégrer tous les codes ci-dessus ensemble pour implémenter un programme complet d'image en PDF.

package main

import (
    "fmt"
    "github.com/jung-kurt/gofpdf"
    "image"
    "image/jpeg"
    "io/ioutil"
    "log"
    "os"
    "path/filepath"
    "strings"
)

func getImagesFromDir(dir string) ([]string, error) {
    files, err := ioutil.ReadDir(dir)
    images := []string{}

    if err != nil {
        return images, err
    }

    for _, file := range files {
        if !file.IsDir() && strings.Contains(file.Name(), ".jpg") {
            images = append(images, filepath.Join(dir, file.Name()))
        }
    }

    return images, nil
}

func imageToPdf(imagePath string, pdf *gofpdf.Fpdf) {
    image.RegisterFormat("jpeg", "jpeg", jpeg.Decode, jpeg.DecodeConfig)

    f, err := os.Open(imagePath)
    defer f.Close()

    if err != nil {
        log.Fatal("Failed to open file:", err)
    }

    // decode the image
    img, _, err := image.Decode(f)

    if err != nil {
        log.Fatal("Failed to decode image:", err)
    }

    // get image dimensions
    w := float64(img.Bounds().Max.X)
    h := float64(img.Bounds().Max.Y)

    // add page to pdf
    pdf.AddPageFormat("P", gofpdf.SizeType{Wd: w, Ht: h})
    pdf.Image(imagePath, 0, 0, w, h, false, "", 0, "")
}

func savePdf(pdf *gofpdf.Fpdf, outputPath string) error {
    return pdf.OutputFileAndClose(outputPath)
}

func main() {
    inputDir := "input"
    outputPdf := "output.pdf"

    fmt.Printf("Reading images from '%v'
", inputDir)
    images, err := getImagesFromDir(inputDir)
    if err != nil {
        log.Fatal("Failed to read images:", err)
    }

    if len(images) == 0 {
        log.Fatal("No images found in directory")
    }

    fmt.Printf("Found %v images
", len(images))

    pdf := gofpdf.New("P", "mm", "A4", "")

    for _, imagePath := range images {
        fmt.Printf("Converting '%v'
", imagePath)
        imageToPdf(imagePath, pdf)
    }

    if err = savePdf(pdf, outputPdf); err != nil {
        log.Fatal("Failed to save PDF:", err)
    }

    fmt.Printf("Saved PDF to '%v'
", outputPdf)
}

Quelques suggestions :

  • Améliorez l'extension. Si votre application doit utiliser plus d'extensions de fichiers, n'oubliez pas d'apporter les modifications appropriées dans la fonction getImagesFromDir.
  • Zoomez les images. Vous pouvez utiliser les méthodes de la bibliothèque image/draw pour redimensionner l'image afin d'éviter de déborder de la page PDF.
  • Ajoutez des numéros de page ou du texte. En plus des images affichées dans l'application, vous pouvez ajouter du texte, des titres, des numéros de page, etc.

Conclusion :

La conversion d'images en PDF est une tâche courante, mais cela ne veut pas dire qu'elle devrait être trop difficile ou compliquée. Vous pouvez créer votre propre programme de conversion en vous concentrant principalement sur le processus de lecture de fichiers, de création de fichiers PDF, de conversion d'images en pages PDF et d'enregistrement de toutes les pages dans un seul fichier. Si votre projet repose sur la conversion d'images en fichiers PDF, nous vous recommandons d'utiliser le langage Golang.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!

Déclaration:
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn