Golang 处理相机的照片保存后旋转的问题

1、指定路径处理

package main
 
import (
    "bytes"
    "fmt"
    "image"
    "image/jpeg"
    "io/ioutil"
    "math"
    "os"
 
    "code.google.com/p/graphics-go/graphics"
    "github.com/rwcarlsen/goexif/exif"
)
 
// png图片没有Orientation信息
func ReadOrientation(filename string) int {
    file, err := os.Open(filename)
    if err != nil {
        fmt.Println("failed to open file, err: ", err)
        return 0
    }
    defer file.Close()
 
    x, err := exif.Decode(file)
    if err != nil {
        fmt.Println("failed to decode file, err: ", err)
        return 0
    }
 
    orientation, err := x.Get(exif.Orientation)
    if err != nil {
        fmt.Println("failed to get orientation, err: ", err)
        return 0
    }
    orientVal, err := orientation.Int(0)
    if err != nil {
        fmt.Println("failed to convert type of orientation, err: ", err)
        return 0
    }
 
    fmt.Println("the value of photo orientation is :", orientVal)
    return orientVal
}
 
func RotateImage(src []byte, angle int) ([]byte, error) {
    var img, _, err = image.Decode(bytes.NewReader(src))
    //var img, err = GetImage(src)
    if err != nil {
        return src, err
    }
    angle = angle % 360
    //弧度转换
    radian := float64(angle) * math.Pi / 180.0
    cos := math.Cos(float64(radian))
    sin := math.Sin(radian)
    //原图的宽高
    w := float64(img.Bounds().Dx())
    h := float64(img.Bounds().Dy())
 
    //新图高宽
    W := int((math.Max(math.Abs(float64(w*cos-h*sin)), math.Abs(w*cos+h*sin))))
    H := int((math.Max(math.Abs(w*sin-h*cos), math.Abs(w*sin+h*cos))))
 
    dst := image.NewNRGBA(image.Rect(0, 0, W, H))
    err = graphics.Rotate(dst, img, &graphics.RotateOptions{radian})
    if err != nil {
        return src, err
    }
 
    file, err := os.Create("newImage.jpg")
    if err != nil {
        return src, err
    }
    defer file.Close()
    err = jpeg.Encode(file, dst, &jpeg.Options{50})
    if err != nil {
        return src, err
    }
 
    return nil, nil
}
 
func main() {
    ori := ReadOrientation("bebRotated.jpg")//先读取该图片的原始方向，方向值参考上图，
    data, err := ioutil.ReadFile("bebRotated.jpg")根据方向值，旋转图片
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(ori)
    //根据图片原始信息的Orientation判断需要转多少度
    if ori == 6 {
        RotateImage(data, 90)
    }
    if ori == 3 {
        RotateImage(data, 180)
    }
    if ori == 8 {
        RotateImage(data, 270)
    }
 
}

2、上传文件方式处理

package main

import (
    "bytes"
    "fmt"
    "image"
    "image/jpeg"
    "image/png"
    "io/ioutil"
    "mime/multipart"
    "net/http"
    "os"
    "path"
    "strings"
    "testing"

    "github.com/labstack/echo"
)

func uploadImage(c echo.Context) error {
    return createFiles(c, "out/", "/index.html")
}

// createFiles 批量创建文件createFiles(file,"public/crops/","/result_crop")
func createFiles(c echo.Context, savePath, resultUrl string) (err error) {
    savePath = GetPath(savePath)
    var form *multipart.Form
    form, err = c.MultipartForm()
    if err != nil {
        err = c.String(http.StatusOK, fmt.Sprintf("获取文件失败，error：%s", err.Error()))
        return
    }

    files := form.File["files"]
    if len(files) <= 0 {
        err = c.String(http.StatusOK, "未上传任何文件")
        return
    }

    // 清除历史目录中的图片
    var dir []os.FileInfo
    dir, err = ioutil.ReadDir(savePath)
    if err != nil {
        err = c.String(http.StatusOK, fmt.Sprintf("文件名目录不存在，savePath：%s", savePath))
    }
    for _, d := range dir {
        os.RemoveAll(path.Join([]string{savePath, d.Name()}...))
    }

    // 批量创建文件
    for _, file := range files {
        err = createFile(file, savePath)
        if err != nil {
            return c.String(http.StatusOK, err.Error())
        }
    }
    return c.HTML(http.StatusOK, fmt.Sprintf("<p> successfully </p><script>location.href='%s'</script>", resultUrl))
}

// createFile 创建文件 createFile(file,"public/crops/")
func createFile(file *multipart.FileHeader, savePath string) (err error) {
    var img image.Image
    img, err = getFileImage(file)
    if err != nil {
        return
    }

    // create image file
    out, err := os.Create(savePath + file.Filename)
    if err != nil {
        return err
    }
    defer out.Close()

    // write new image to file
    err = jpeg.Encode(out, img, nil)
    if err != nil {
        return err
    }
    return
}

func getFileImage(file *multipart.FileHeader) (img image.Image, err error) {
    var fileName, ext string
    src, err := file.Open()
    if err != nil {
        err = fmt.Errorf("文件【%s】打开失败，error：%s", fileName, err.Error())
        return
    }
    defer src.Close()

    fileName = strings.ToLower(file.Filename)
    ext = path.Ext(file.Filename)
    if ext == ".jpg" || ext == ".jpeg" {
        var data []byte
        data, err = ioutil.ReadAll(src)
        if err != nil {
            err = fmt.Errorf("文件【%s】读取失败，error：%s", fileName, err.Error())
            return
        }

        // ======= 根据相机图片原始信息的Orientation判断需要转多少度 =======
        // 为什么不直接使用imgFile 而使用 bytes.NewReader(data)，原因：无论image.Decode和image.DecodeConfig消耗从传入的字节io.Reade，imgFile已在ioutil.ReadAll中消费过，说明：https://stackoverflow.com/questions/62846156/image-decode-unknown-format
        ori := ReadOrientation(bytes.NewReader(data))
        if ori == 6 {
            img, err = RotateImage(bytes.NewReader(data), 90)
        } else if ori == 3 {
            img, err = RotateImage(bytes.NewReader(data), 180)
        } else if ori == 8 {
            img, err = RotateImage(bytes.NewReader(data), 270)
        } else {
            // 非相机照片处理逻辑
            img, err = jpeg.Decode(bytes.NewReader(data))
        }
        if err != nil {
            err = fmt.Errorf("处理文件【%s】失败，ori：%d，error：%s", fileName, ori, err.Error())
            return
        }
    } else if ext == ".png" {
        img, err = png.Decode(src)
    } else {
        err = fmt.Errorf("文件【%s】格式错误，仅支持.jpg、.jpeg、.png格式", fileName)
        return
    }
    if err != nil {
        err = fmt.Errorf(fmt.Sprintf("文件【%s】读取发送错误，error：%s", fileName, err.Error()))
        return
    }
    return
}

func TestImage(t *testing.T) {
    e := echo.New()
    e.POST("/uploadImage", uploadImage)
    e.Logger.Fatal(e.Start(":8081"))
}
最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 217,826评论 6赞 506
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 92,968评论 3赞 395
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 164,234评论 0赞 354
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 58,562评论 1赞 293
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 67,611评论 6赞 392
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 51,482评论 1赞 302
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 40,271评论 3赞 418
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 39,166评论 0赞 276
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 45,608评论 1赞 314
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,814评论 3赞 336
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,926评论 1赞 348
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 35,644评论 5赞 346
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 41,249评论 3赞 329
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,866评论 0赞 22
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,991评论 1赞 269
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 48,063评论 3赞 370
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 44,871评论 2赞 354