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閱讀 228,461評論 6 532
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,538評論 3 417
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,423評論 0 375
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,991評論 1 312
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,761評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,207評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,268評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,419評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,959評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,782評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,983評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,528評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,222評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,653評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,901評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,678評論 3 392
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,978評論 2 374