opencv給圖片添加文字水印<轉>

其中有一些改動為了文字大小等還有一些圖片的尺寸，真正使用的時候可以把尺寸的屏蔽掉

頭文件：

//====================================================================
//====================================================================
//
// 文件: CvxText.h
//
// 說明: OpenCV漢字輸出
//
// 時間: 
// 作者: xx
//
//====================================================================
//====================================================================

#ifndef OPENCV_CVX_TEXT_2007_08_31_H
#define OPENCV_CVX_TEXT_2007_08_31_H

/**
* \file CvxText.h
* \brief OpenCV漢字輸出接口
*
* 實現了漢字輸出功能。
*/

#include <ft2build.h>
#include FT_FREETYPE_H

#include <cv.h>
#include <highgui.h>

/**
* \class CvxText
* \brief OpenCV中輸出漢字
*
* OpenCV中輸出漢字。字庫提取采用了開源的FreeFype庫。由於FreeFype是
* GPL版權發布的庫，和OpenCV版權並不一致，因此目前還沒有合並到OpenCV
* 擴展庫中。
*
* 顯示漢字的時候需要一個漢字字庫文件，字庫文件系統一般都自帶了。
* 這里采用的是一個開源的字庫：“文泉驛正黑體”。
*
* 關於"OpenCV擴展庫"的細節請訪問
* http://code.google.com/p/opencv-extension-library/
*
* 關於FreeType的細節請訪問
* http://www.freetype.org/
*
* 例子：
*
* \code
int main(int argc, char *argv)
{
    // 定義CvxApplication對象
    CvxApplication app(argc, argv);
    // 打開一個影象
    IplImage *img = cvLoadImage("d:\\test.jpg", 1);
    if(!img)
        return -1;
    // 輸出漢字
    {
        // "wqy-zenhei.ttf"為文泉驛正黑體
        CvText text("wqy-zenhei.ttf");
        const char *msg = "在OpenCV中輸出漢字！";
        float p = 0.5;
        text.setFont(NULL, NULL, NULL, &p);        // 透明處理
        text.putText(img, msg, cvPoint(100, 150), CV_RGB(255,0,0));
    }
    // 定義窗口，並顯示影象
    CvxWindow myWin("myWin");
    myWin.showImage(img);
    try
    {

        cvSaveImage("d;\\test1.jpg", img, 0);
    }
    catch (runtime_error& ex)
    {
        cvReleaseImage(&img);
        fprintf(stderr, "Exception converting image to PNG format: %s\n", ex.what());
        return -2;
    }

    //cvWaitKey(0);
    cvReleaseImage(&img);
    // 進入消息循環
    return app.exec();
}
* \endcode
*/

class CvxText
{
    // 禁止copy
    CvxText& operator=(const CvxText&);

    //================================================================
    //================================================================

public:

    /**
    * 裝載字庫文件
    *
    *
    * \param freetype  字體庫目錄
    * \param size_base 原始尺寸640*480 為true
    */
    CvxText(const char *freeType, bool size_base);//size_base 原始尺寸640x480
    virtual ~CvxText();

    //================================================================
    //================================================================

    /**
    * 獲取字體。目前有些參數尚不支持。
    *
    * \param font      字體類型, 目前不支持
    * \param size      字體大小/空白比例/間隔比例/旋轉角度
    * \param underline   下畫線
    * \param diaphaneity 透明度
    *
    * \sa setFont, restoreFont
    */
    void getFont(int *type,
        CvScalar *size = NULL, bool *underline = NULL, float *diaphaneity = NULL);

    /**
    * 設置字體。目前有些參數尚不支持。
    *
    * \param font      字體類型, 目前不支持
    * \param size      字體大小/空白比例/間隔比例/旋轉角度
    * \param underline   下畫線
    * \param diaphaneity 透明度
    *
    * \sa getFont, restoreFont
    */
    void setFont(int *type,
        CvScalar *size = NULL, bool *underline = NULL, float *diaphaneity = NULL);

    /**
    * 恢復原始的字體設置。
    *
    * \sa getFont, setFont
    * \ fontsize  字體大小
    */
    void restoreFont(int fontsize);

    //================================================================
    //================================================================

    /**
    * 輸出漢字(顏色默認為黑色)。遇到不能輸出的字符將停止。
    *
    * \param img輸出的影象
    * \param text 文本內容
    * \param pos文本位置
    *
    * \return 返回成功輸出的字符長度，失敗返回-1。
    */
    int putText(IplImage *img, const char    *text, CvPoint pos);

    /**
    * 輸出漢字(顏色默認為黑色)。遇到不能輸出的字符將停止。
    *
    * \param img輸出的影象
    * \param text 文本內容
    * \param pos文本位置
    *
    * \return 返回成功輸出的字符長度，失敗返回-1。
    */
    int putText(IplImage *img, const wchar_t *text, CvPoint pos);

    /**
    * 輸出漢字。遇到不能輸出的字符將停止。
    *
    * \param img   輸出的影象
    * \param text文本內容
    * \param pos   文本位置
    * \param color 文本顏色
    *
    * \return 返回成功輸出的字符長度，失敗返回-1。
    */
    int putText(IplImage *img, const char    *text, CvPoint pos, CvScalar color);

    /**
    * 輸出漢字。遇到不能輸出的字符將停止。
    *
    * \param img   輸出的影象
    * \param text文本內容
    * \param pos   文本位置
    * \param color 文本顏色
    *
    * \return 返回成功輸出的字符長度，失敗返回-1。
    */
    int putText(IplImage *img, const wchar_t *text, CvPoint pos, CvScalar color);

    //================================================================
    //================================================================

private:

    // 輸出當前字符, 更新m_pos位置
    void putWChar(IplImage *img, wchar_t wc, CvPoint &pos, CvScalar color);

    //================================================================
    //================================================================

private:

    FT_Library        m_library;        // 字庫
    FT_Face                m_face;                // 字體

    //================================================================
    //================================================================

    // 默認的字體輸出參數

    int                        m_fontType;
    CvScalar        m_fontSize;
    bool                m_fontUnderline;
    float                m_fontDiaphaneity;

    //================================================================
    //================================================================
};

#endif // OPENCV_CVX_TEXT_2007_08_31_H

 

源文件:

#include "stdafx.h"
#include <wchar.h>
#include <assert.h>
#include <locale.h>
#include <ctype.h>

#include "CvxText.h"

//====================================================================
//====================================================================

// 打開字庫

CvxText::CvxText(const char *freeType, bool size_base)
{
    assert(freeType != NULL);

    // 打開字庫文件, 創建一個字體

    if (FT_Init_FreeType(&m_library)) throw;
    if (FT_New_Face(m_library, freeType, 0, &m_face)) throw;

    // 設置字體輸出參數

    if (size_base)
    {
        restoreFont(18);
    }
    else
    {
        restoreFont(16);
    }
    

    // 設置C語言的字符集環境

    setlocale(LC_ALL, "");
}

// 釋放FreeType資源

CvxText::~CvxText()
{
    FT_Done_Face(m_face);
    FT_Done_FreeType(m_library);
}

// 設置字體參數:
//
// font                        - 字體類型, 目前不支持
// size                        - 字體大小/空白比例/間隔比例/旋轉角度
// underline        - 下畫線
// diaphaneity        - 透明度

void CvxText::getFont(int *type, CvScalar *size, bool *underline, float *diaphaneity)
{
    if (type) *type = m_fontType;
    if (size) *size = m_fontSize;
    if (underline) *underline = m_fontUnderline;
    if (diaphaneity) *diaphaneity = m_fontDiaphaneity;
}

void CvxText::setFont(int *type, CvScalar *size, bool *underline, float *diaphaneity)
{
    // 參數合法性檢查

    if (type)
    {
        if (type >= 0) m_fontType = *type;
    }
    if (size)
    {
        m_fontSize.val[0] = fabs(size->val[0]);
        m_fontSize.val[1] = fabs(size->val[1]);
        m_fontSize.val[2] = fabs(size->val[2]);
        m_fontSize.val[3] = fabs(size->val[3]);
    }
    if (underline)
    {
        m_fontUnderline = *underline;
    }
    if (diaphaneity)
    {
        m_fontDiaphaneity = *diaphaneity;
    }
}

// 恢復原始的字體設置

void CvxText::restoreFont(int fontsize)
{
    m_fontType = 0;                                // 字體類型(不支持)

    m_fontSize.val[0] = fontsize;                // 字體大小
    m_fontSize.val[1] = 0.5;        // 空白字符大小比例
    m_fontSize.val[2] = 0.1;        // 間隔大小比例
    m_fontSize.val[3] = 0;                // 旋轉角度(不支持)

    m_fontUnderline = false;        // 下畫線(不支持)

    m_fontDiaphaneity = 1.0;        // 色彩比例(可產生透明效果)

    // 設置字符大小

    FT_Set_Pixel_Sizes(m_face, (int)m_fontSize.val[0], 0);
}

// 輸出函數(顏色默認為黑色)

int CvxText::putText(IplImage *img, const char    *text, CvPoint pos)
{
    return putText(img, text, pos, CV_RGB(255, 255, 255));
}
int CvxText::putText(IplImage *img, const wchar_t *text, CvPoint pos)
{
    return putText(img, text, pos, CV_RGB(255, 255, 255));
}

//

int CvxText::putText(IplImage *img, const char    *text, CvPoint pos, CvScalar color)
{
    if (img == NULL) return -1;
    if (text == NULL) return -1;

    //

    int i;
    for (i = 0; text[i] != '\0'; ++i)
    {
        wchar_t wc = text[i];

        // 解析雙字節符號

        if (!isascii(wc)) mbtowc(&wc, &text[i++], 2);

        // 輸出當前的字符

        putWChar(img, wc, pos, color);
    }
    return i;
}
int CvxText::putText(IplImage *img, const wchar_t *text, CvPoint pos, CvScalar color)
{
    if (img == NULL) return -1;
    if (text == NULL) return -1;

    //

    int i;
    for (i = 0; text[i] != '\0'; ++i)
    {
        // 輸出當前的字符

        putWChar(img, text[i], pos, color);
    }
    return i;
}

// 輸出當前字符, 更新m_pos位置

void CvxText::putWChar(IplImage *img, wchar_t wc, CvPoint &pos, CvScalar color)
{
    // 根據unicode生成字體的二值位圖

    FT_UInt glyph_index = FT_Get_Char_Index(m_face, wc);
    FT_Load_Glyph(m_face, glyph_index, FT_LOAD_DEFAULT);
    FT_Render_Glyph(m_face->glyph, FT_RENDER_MODE_MONO);

    //

    FT_GlyphSlot slot = m_face->glyph;

    // 行列數

    int rows = slot->bitmap.rows;
    int cols = slot->bitmap.width;

    //

    for (int i = 0; i < rows; ++i)
    {
        for (int j = 0; j < cols; ++j)
        {
            int off = ((img->origin == 0) ? i : (rows - 1 - i))
                * slot->bitmap.pitch + j / 8;

            if (slot->bitmap.buffer[off] & (0xC0 >> (j % 8)))
            {
                int r = (img->origin == 0) ? pos.y - (rows - 1 - i) : pos.y + i;;
                int c = pos.x + j;

                if (r >= 0 && r < img->height
                    && c >= 0 && c < img->width)
                {
                    CvScalar scalar = cvGet2D(img, r, c);

                    // 進行色彩融合

                    float p = m_fontDiaphaneity;
                    for (int k = 0; k < 4; ++k)
                    {
                        scalar.val[k] = scalar.val[k] * (1 - p) + color.val[k] * p;
                    }

                    cvSet2D(img, r, c, scalar);
                }
            }
        } // end for
    } // end for

    // 修改下一個字的輸出位置

    double space = m_fontSize.val[0] * m_fontSize.val[1];
    double sep = m_fontSize.val[0] * m_fontSize.val[2];

    pos.x += (int)((cols ? cols : space) + sep);
}

=========================================

備注：需要依賴第三方庫 freetype 和opencv環境。

源：https://github.com/msly/codemanage