当前位置：文档之家 › 水印代码

水印代码

  • 格式：doc
  • 大小：64.50 KB
  • 文档页数：14

clear all;%logistic序列生成算法figure(1);an=linspace(3.1,3.99,400);hold on;box on;axis([min(an),max(an),-1,2]);N=64*64;xn=zeros(1,N);for a=an;x=rand;for k=1:20;x=a*x*(1-x);endfor k=1:N;x=a*x*(1-x);xn(k)=x;b(k,1)=x;endplot(a*ones(1,N),xn,'k.','markersize',1);endc=reshape(b,64,64);%二维矩阵调制for a1=1:64;for a2=1:64;if c(a1,a2)>=0.5;d(a1,a2)=1;else d(a1,a2)=0;end;end;end%水印加密e=imread('c:\new\123.BMP');e=double(e);f=bitxor(d,e);%mm=imread('c:\new\0.BMP');nn=double(mm);%h=bitxor(d,mm); g=bitxor(d,f);figure(2);subplot(2,2,1);imshow(d);title('混沌序列矩阵');subplot(2,2,2);imshow(e);title('水印图像');subplot(2,2,3);imshow(f);title('混沌加密水印图像');imwrite(f,'c:\new\0.BMP');subplot(2,2,4);imshow(g);title('水印图像恢复');M=256;N=64;[filename1,pathname]=uigetfile('*.*','select the image');image1=imread(num2str(filename1));figure(3);subplot(2,2,1);imshow(image1);title('original image');image1=double(image1);subplot(2,2,2);imshow(f);title('original watermark');%嵌入水印[ca,ch,cv,cd] = dwt2(image1,'db1');[cas,chs,cvs,cds] = dwt2(ca,'db1');for i=1:Nfor j=1:Nif f(i,j)==0a=-1;else a=1;endCa(i,j)=cas(i,j)*(1+a*0.03);endendIM= idwt2(Ca,chs,cvs,cds,'db1')markedimage=double(idwt2(IM,ch,cv,cd,'db1'));subplot(2,2,3);colormap(gray(256));image(markedimage);title('marked image');imwrite(markedimage,gray(256),'c:\new\watermarked.bmp','bmp');%提取水印image1=imread(num2str(filename1));image1=double(image1); imaged=imread('c:\new\watermarked.bmp');[ca,ch,cv,cd] = dwt2(image1,'db1');[cas,chs,cvs,cds]=dwt2(ca,'db1');[caa,chh,cvv,cdd]=dwt2(imaged,'db1');[caas,chhs,cvvs,cdds]=dwt2(caa,'db1');for p=1:Nfor q=1:Na=caas(p,q)/cas(p,q)-1;if a<0W(p,q)=0;elseW(p,q)=255;end%频谱pic1= imread('c:\new\3.jpg');%pic=rgb2gray(pic);subplot(2,2,1);imshow(pic1);title('原图');pic_fft1=fft2(double(pic1));pic_fft1=fftshift(pic_fft1);subplot(2,2,2);imshow(log(abs(pic_fft1)),[]);title('原图频谱图');endendsubplot(2,2,4);colormap(gray(256));image(W);title('从含水印图像中提取的水印'); imwrite(W,gray(256),'watermark.bmp','bmp');%figure(4);h=bitxor(d,W);figure(7);subplot(2,2,1);imshow(h);pic2= imread('c:\new\watermarked.BMP');%pic=rgb2gray(pic);subplot(2,2,3);imshow(pic2);title('新图');pic_fft2=fft2(double(pic2));pic_fft2=fftshift(pic_fft2);subplot(2,2,4);imshow(log(abs(pic_fft2)),[]);title('新图频谱图'%DWT水印源码%嵌入源码clcclear all;% 保存开始时间start_time=cputime;iTimes=20; %置乱次数k=20;blocksize=8; % 设置块的大小filter_m=[ 1,1,1,1,1,1,1,1; % 滤波矩阵1,1,1,1,1,1,1,1;1,1,0,0,0,0,1,1;1,1,0,0,0,0,1,1;1,1,0,0,0,0,1,1;1,1,0,0,0,0,1,1;1,1,1,1,1,1,1,1;1,1,1,1,1,1,1,1 ];% 读入原图像file_name='_lena_std_bw.bmp';cover_object=double(imread(file_name));% 原图像的行数与列数Mc=size(cover_object,1); %原图像行数Nc=size(cover_object,2); %原图像列数% 最大嵌入信息量max_message=Mc*Nc/(blocksize^2);% 读入水印图像file_name='c.bmp';message=double(imread(file_name));%%水印图像的行数与列数Mm=size(message,1); %水印行数Nm=size(message,2); %水印列数% 检查水印信息是否过大if (Mm*Nm> max_message)error('水印太大')end%对水印图像进行arnold置乱if Mm~=Nmerror('水印矩阵必须为方阵');endif Mm~=40error('必须为40*40大小,或者修改置乱次数'); endtempImg=message;for n=1:iTimes % 次数for u=1:Mmfor v=1:Nmtemp=tempImg(u,v);ax=mod((u-1)+(v-1),Mm)+1;ay=mod((u-1)+2*(v-1),Nm)+1;outImg(ax,ay)=temp;endendtempImg=outImg;endmessage_vector=reshape(tempImg,1,Mm*Nm);% 产生watermarked_image,并写入原图信息watermarked_image=cover_object;%置随机数发生器的状态为1100key=1100;rand('state',key);% 产生伪随机序列pn_sequence_zero=round(2*(rand(1,sum(sum(filter_m)))-0.5)); pn_sequence_one=round(2*(rand(1,sum(sum(filter_m)))-0.5)); % 将图像分块嵌入% 当(2,2) > (2,3) 且message_pad(kk)=0% 当(2,2) < (2,3) 且message_pad(kk)=1%%经过分析可以得出结论：在提取水印时，如果cD1(2,2)>cD1(2,3)便是嵌入了水印的黑色像素，%%反之则是嵌入了白色像素x=1;y=1;h=waitbar(0,'嵌入水印，请等待');for (kk = 1:length(message_vector))% 对块进行DWT变换[cA1,cH1,cV1,cD1] =dwt2(cover_object(x:x+blocksize-1,y:y+blocksize-1),'haar');a=cD1;% 如果message_pad(kk)== 0if (message_vector(kk) == 0)% 且(2,2) < (2,3) ，交换它们if (cD1(2,2) < cD1(2,3))temp=cD1(2,3);cD1(2,3)=cD1(2,2);cD1(2,2)=temp;end% 如果message_pad(kk) == 1, elseif (message_vector(kk) == 1)% 且(2,2) > (2,3) ，交换它们if (cD1(2,2) >= cD1(2,3))temp=cD1(2,3);cD1(2,3)=cD1(2,2);cD1(2,2)=temp;endend% 检查(2,2) ，(2,3)的差是否>= k if cD1(2,2) > cD1(2,3)if cD1(2,2) - cD1(2,3) < kcD1(2,2)=cD1(2,2)+(k/2);cD1(2,3)=cD1(2,3)-(k/2);endelseif cD1(2,3) - cD1(2,2) < kcD1(2,3)=cD1(2,3)+(k/2);cD1(2,2)=cD1(2,2)-(k/2);endend%IDWTwatermarked_image(x:x+blocksize-1,y:y+blocksize-1)=idwt2(cA1,cH1,cV1,cD1,'haar',[Mc,Nc]);% 移动到下一块if (x+blocksize) >= Ncx=1;y=y+blocksize;elsex=x+blocksize;endwaitbar(kk/length(message_vector),h);endclose(h);% 转换为uint8并写入dwt_watermarked.bmpwatermarked_image_uint=uint8(watermarked_image);imwrite(watermarked_image_uint,'dwt_watermarked.bmp','bmp'); % 显示运行时间elapsed_time=cputime-start_time,% 显示图像figure(1)subplot(2,2,1);imshow(message,[]);title('水印');subplot(2,2,4)imshow(outImg,[]);title('置乱水印');subplot(2,2,2);imshow(cover_object,[]);title('原图');subplot(2,2,3);imshow(watermarked_image,[]) title('嵌入水印图像')%DWT源码%提取源码clear all;% 保存开始时间start_time=cputime;iTimes=20; %置乱次数blocksize=8; % 设置块的大小filter_m=[ 1,1,1,1,1,1,1,1; % 滤波矩阵1,1,1,1,1,1,1,1;1,1,0,0,0,0,1,1;1,1,0,0,0,0,1,1;1,1,0,0,0,0,1,1;1,1,0,0,0,0,1,1;1,1,1,1,1,1,1,1;1,1,1,1,1,1,1,1 ];% 读入嵌入水印图像[filename,pathname]=uigetfile('*.bmp;*tiff;*tif;*jpg','输入嵌有水印的图像'); watermarked_image=double(imread(filename));% 嵌入水印图像的行数与列数Mw=size(watermarked_image,1); %嵌入水印图像的行数Nw=size(watermarked_image,2); %嵌入水印图像的列数% 最大嵌入信息量max_message=Mw*Nw/(blocksize^2);% 读入原始水印file_name='c.bmp';orig_watermark=double(imread(file_name));% 原始水印的行数与列数Mo=size(orig_watermark,1); %原始水印的行数No=size(orig_watermark,2); %原始水印的列数%置随机数发生器的状态为1100rand('state',1100);% 产生伪随机序列pn_sequence_zero=round(2*(rand(1,sum(sum(filter_m)))-0.5));pn_sequence_one=round(2*(rand(1,sum(sum(filter_m)))-0.5));% 将图像分块提取水印x=1;y=1;h=waitbar(0,'水印提取中，请等待');for (kk = 1:max_message)% 对块进行dwt变换[cA1,cH1,cV1,cD1] =dwt2(watermarked_image(x:x+blocksize-1,y:y+blocksize-1),'haar');% 如果cD1(2,2) > cD1(3,3) 那么message_vector(kk)=0% 否则message_vector(kk)=1if cD1(2,2)>cD1(3,3)message_vector(kk)=0;elsemessage_vector(kk)=1;end% 移动到下一块if (x+blocksize) >= Mwx=1;y=y+blocksize;elsex=x+blocksize;endwaitbar(kk/max_message,h);endclose(h);% arnold置乱tempImg=reshape(message_vector(1:Mo*No),Mo,No); message_arnold=tempImg;for n=1:iTimes % 次数for u=1:Mofor v=1:Notemp=tempImg(u,v);ax=mod((u-1)+(v-1),Mo)+1;ay=mod((u-1)+2*(v-1),No)+1;outImg(ax,ay)=temp;endendtempImg=outImg;endmessage=tempImg;% 计算运行时间elapsed_time=cputime-start_time, % 显示提取水印与原始水印figure(3)subplot(1,3,1);imshow(message_arnold,[]);title('提取的置乱水印');subplot(1,3,2);imshow(message,[]);title('再次置乱后水印');subplot(1,3,3);imshow(orig_watermark,[])title('原始水印');。

合集下载

java实现图片加水印效果

java实现图片加水印效果

java实现图⽚加⽔印效果图⽚加⽔印代码，这些代码不常⽤，但是⽤到的时候需要注意的地⽅也挺多的，每次都重写⽐较⿇烦，记下来备忘。

代码是图⽚加⽔印的⼀般流程，可根据实际项⽬需要⾃⾏修改。

注：代码在JPG和PNG格式图⽚下测试通过，其他图⽚格式请⾃⾏测试和修改import java.awt.AlphaComposite;import java.awt.Color;import java.awt.Font;import java.awt.FontMetrics;import java.awt.Graphics2D;import java.awt.RenderingHints;import java.awt.geom.AffineTransform;import java.awt.image.BufferedImage;import java.io.File;import java.util.Iterator;import javax.imageio.IIOImage;import javax.imageio.ImageIO;import javax.imageio.ImageReader;import javax.imageio.ImageWriteParam;import javax.imageio.ImageWriter;import javax.imageio.stream.FileImageInputStream;import javax.imageio.stream.FileImageOutputStream;import org.junit.Test;public class Wjt {@Testpublic void t3() throws Exception {File rootDir = new File("C:/Users/Administrator/Desktop/tem1");String[] fa = { "1.png", "2.jpg" };for (String s : fa) {String f = s;{FileImageInputStream fileImageInputStream = new FileImageInputStream(new File(rootDir, f));Iterator<ImageReader> imageReaders = ImageIO.getImageReaders(fileImageInputStream); //根据图⽚⽂件获取读取器ImageReader imageReader = imageReaders.hasNext() ? imageReaders.next() : null; //检测读取器是否存在imageReader.setInput(fileImageInputStream); //设置读取器输⼊流BufferedImage bufferedImage = imageReader.read(0); //从读取器取BufferedImage，这⾥只有⼀个，所以可以直接写0{ //开始加⽔印Graphics2D graphics2d = bufferedImage.createGraphics(); //取画笔graphics2d.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON); //设置画笔⽂字抗锯齿graphics2d.transform(AffineTransform.getTranslateInstance(bufferedImage.getWidth() / 2, bufferedImage.getHeight() / 2)); //将画笔移动到图⽚中央graphics2d.transform(AffineTransform.getRotateInstance(Math.toRadians(45))); //将画笔旋转⼀定弧度float radius = Double.valueOf(Math.sqrt(bufferedImage.getWidth() * bufferedImage.getWidth() + bufferedImage.getHeight() * bufferedImage.getHeight()) / 2).floatValue(); //计算图⽚外切圆半径 Font waterMarkerFont = new Font("微软雅⿊", Font.PLAIN, bufferedImage.getHeight() / 25); //设置画笔要⽤的字体，这⾥可以设置字体样式和⼤⼩graphics2d.setFont(waterMarkerFont); //设置画笔字体graphics2d.setColor(Color.WHITE); //设置画笔颜⾊graphics2d.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, 0.3F)); //设置画笔操作透明度FontMetrics fontMetrics = graphics2d.getFontMetrics(); //取画笔字体信息String waterMarker = "THIS IS WATERMARK."; //⽔印⽂字int stringWidth = fontMetrics.stringWidth(waterMarker); //计算⽔印⽂字宽度，很多⼈卡在了这⼀步for (float i = -radius; i <= radius; i += (stringWidth + bufferedImage.getHeight() / 25)) { //循环添加⽔印，这⾥可以设置⽔印⽔平距离，单位：pxfor (float j = -radius; j <= radius; j += (fontMetrics.getHeight() + bufferedImage.getHeight() / 25)) { //这⾥可以设置⽔印垂直距离，单位：pxgraphics2d.drawString(waterMarker, i, j); //画⽔印}}graphics2d.dispose(); //释放画笔资源}ImageWriter imageWriter = ImageIO.getImageWriter(imageReader); //根据图⽚⽂件读取器获取图⽚⽂件写⼊器String[] split = f.split("\\.");FileImageOutputStream fileOutputStream = new FileImageOutputStream(new File(rootDir, split[0] + "r." + split[1])); //⽣成⽬标图⽚⽂件路径imageWriter.setOutput(fileOutputStream); //设置写⼊器输出流IIOImage iioImage = new IIOImage(bufferedImage, null, null);ImageWriteParam imageWriteParam = imageWriter.getDefaultWriteParam();if (imageReader.getFormatName().equals("JPEG")) { //jpg格式图⽚可设置输出图⽚⽂件的质量imageWriteParam.setCompressionMode(ImageWriteParam.MODE_EXPLICIT); //这⼀⾏固定这样写，跟下⼀⾏搭配imageWriteParam.setCompressionQuality(0.947F); //跟上⼀⾏搭配，设置输出图⽚⽂件的质量}imageWriter.write(null, iioImage, imageWriteParam); //写⽂件imageWriter.dispose(); //关闭写⼊器imageReader.dispose(); //关闭读取器}}}}代码流程在注释中写的很详细了，不多做解释。

如何利用python给图片添加半透明水印

如何利用python给图片添加半透明水印

如何利⽤python给图⽚添加半透明⽔印前⾔本⽂主要给⼤家介绍了关于python图⽚添加半透明⽔印的相关资料，分享出来供⼤家参考学习，下⾯话不多说了，来⼀起看看详细的介绍吧⽰例代码：# coding:utf-8from PIL import Image, ImageDraw, ImageFontdef add_text_to_image(image, text):font = ImageFont.truetype('C:\Windows\Fonts\STXINGKA.TTF', 36)# 添加背景new_img = Image.new('RGBA', (image.size[0] * 3, image.size[1] * 3), (0, 0, 0, 0))new_img.paste(image, image.size)# 添加⽔印font_len = len(text)rgba_image = new_img.convert('RGBA')text_overlay = Image.new('RGBA', rgba_image.size, (255, 255, 255, 0))image_draw = ImageDraw.Draw(text_overlay)for i in range(0, rgba_image.size[0], font_len*40+100):for j in range(0, rgba_image.size[1], 200):image_draw.text((i, j), text, font=font, fill=(0, 0, 0, 50))text_overlay = text_overlay.rotate(-45)image_with_text = Image.alpha_composite(rgba_image, text_overlay)# 裁切图⽚image_with_text = image_with_text.crop((image.size[0], image.size[1], image.size[0] * 2, image.size[1] * 2))return image_with_textif __name__ == '__main__':img = Image.open("test.jpg")im_after = add_text_to_image(img, u'测试使⽤')im_after.save(u'测试使⽤.png')效果图：总结以上就是这篇⽂章的全部内容了，希望本⽂的内容对⼤家的学习或者⼯作具有⼀定的参考学习价值，谢谢⼤家对的⽀持。

Java在Excel中添加水印的实现（单一水印、平铺水印）

Java在Excel中添加水印的实现（单一水印、平铺水印）

Java在Excel中添加⽔印的实现（单⼀⽔印、平铺⽔印）在Excel中没有直接添加⽔印的功能，但依旧可以通过⼀定⽅式来实现类似⽔印效果。

本⽂通过Java程序代码介绍具体实现⽅法。

可添加单⼀⽔印效果，即⽔印是以单个⽂本字样来呈现；也可添加多个平铺⽔印效果，即⽔印是以多个⽂本字样来页⾯中平铺。

详细内容见下⽂。

程序环境：测试⽂档：Office Excel 2013编译环境：IntelliJ IDEA 2018JDK版本：1.8.0Excel库：Java系列 spire.xls.jar 3.9.11.单⼀⽔印效果import com.spire.xls.*;import java.awt.*;import java.awt.image.BufferedImage;import static java.awt.image.BufferedImage.TYPE_INT_ARGB;public class SingleWatermark {public static void main(String[] args) {//加载Excel测试⽂档Workbook wb = new Workbook();wb.loadFromFile("test.xlsx");//设置⽂本和字体⼤⼩Font font = new Font("仿宋", Font.PLAIN, 40);for (int i =0;i<wb.getWorksheets().getCount();i++){Worksheet sheet = wb.getWorksheets().get(i);//调⽤DrawText() ⽅法插⼊图⽚BufferedImage imgWtrmrk = drawText("内部专⽤", font, Color.pink, Color.white, sheet.getPageSetup().getPageHeight(), sheet.getPageSetup().getPageWidth());//将图⽚设置为页眉sheet.getPageSetup().setCenterHeaderImage(imgWtrmrk);sheet.getPageSetup().setCenterHeader("&G");//将显⽰模式设置为Layoutsheet.setViewMode(yout);}//保存⽂档wb.saveToFile("SingleWatermark.xlsx", ExcelVersion.Version2013);}private static BufferedImage drawText (String text, Font font, Color textColor, Color backColor,double height, double width){//定义图⽚宽度和⾼度BufferedImage img = new BufferedImage((int) width, (int) height, TYPE_INT_ARGB);Graphics2D loGraphic = img.createGraphics();//获取⽂本sizeFontMetrics loFontMetrics = loGraphic.getFontMetrics(font);int liStrWidth = loFontMetrics.stringWidth(text);int liStrHeight = loFontMetrics.getHeight();//⽂本显⽰样式及位置loGraphic.setColor(backColor);loGraphic.fillRect(0, 0, (int) width, (int) height);loGraphic.translate(((int) width - liStrWidth) / 2, ((int) height - liStrHeight) / 2);loGraphic.rotate(Math.toRadians(-45));loGraphic.translate(-((int) width - liStrWidth) / 2, -((int) height - liStrHeight) / 2);loGraphic.setFont(font);loGraphic.setColor(textColor);loGraphic.drawString(text, ((int) width - liStrWidth) / 2, ((int) height - liStrHeight) / 2);loGraphic.dispose();return img;}}单⼀⽔印效果：2.平铺⽔印效果import com.spire.xls.*;import java.awt.*;import java.awt.image.BufferedImage;import static java.awt.image.BufferedImage.TYPE_INT_ARGB;public class TiledWatermark {public static void main(String[] args) {//加载Excel测试⽂档Workbook wb = new Workbook();wb.loadFromFile("test.xlsx");//设置⽂本和字体⼤⼩Font font = new Font("仿宋", Font.PLAIN, 25);for (int i =0;i<wb.getWorksheets().getCount();i++){Worksheet sheet = wb.getWorksheets().get(i);//调⽤DrawText() ⽅法插⼊图⽚BufferedImage imgWtrmrk = drawText("内部专⽤内部专⽤内部专⽤内部专⽤", font, Color.pink, Color.white, sheet.getPageSetup().getPageHeight(), sheet.getPageSetup().getPageWidth()); //将图⽚设置为页眉sheet.getPageSetup().setCenterHeaderImage(imgWtrmrk);sheet.getPageSetup().setCenterHeader("&G");//将显⽰模式设置为Layoutsheet.setViewMode(yout);}//保存⽂档wb.saveToFile("TiledWatermark.xlsx", ExcelVersion.Version2013);}private static BufferedImage drawText (String text, Font font, Color textColor, Color backColor,double height, double width){//定义图⽚宽度和⾼度BufferedImage img = new BufferedImage((int) width, (int) height, TYPE_INT_ARGB);Graphics2D loGraphic = img.createGraphics();//获取⽂本sizeFontMetrics loFontMetrics = loGraphic.getFontMetrics(font);int liStrWidth = loFontMetrics.stringWidth(text);int liStrHeight = loFontMetrics.getHeight();//⽂本显⽰样式及位置loGraphic.setColor(backColor);loGraphic.fillRect(0, 0, (int) width, (int) height);loGraphic.translate(((int) width - liStrWidth) / 2, ((int) height - liStrHeight) / 2);//loGraphic.rotate(Math.toRadians(-45));loGraphic.translate(-((int) width - liStrWidth) / 2, -((int) height - liStrHeight) / 2);loGraphic.setFont(font);loGraphic.setColor(textColor);loGraphic.drawString(text, ((int) width - liStrWidth) /6 , ((int) height - liStrHeight) /6);loGraphic.drawString(text,((int) width - liStrWidth) /3, ((int) height - liStrHeight) /3);loGraphic.drawString(text,((int) width - liStrWidth) /2, ((int) height - liStrHeight) /2);loGraphic.dispose();return img;}}平铺⽔印效果：★需要注意的是：在添加完⽔印效果后，查看⽂档时，在“普通视图”⽔印不可见，需在“页⾯布局”模式或“打印预览”模式下查看。

ffmpeg代码实现视频添加水印功能

ffmpeg代码实现视频添加水印功能

ffmpeg代码实现视频添加⽔印功能备忘:⽤ffmpeg给视频添加⽔印的命令为:ffmpeg -i in.mp4 -i xxx.png -filter_complex "overlay=5:5" out.mp4代码实现如下, 主要是⽤了参考了ffmpeg.c⾥⾯的filter_complex的代码:[cpp]view plaincopy1. #include <unistd.h>2. #include <stdio.h>3. #include <assert.h>4.5. #include <libavcodec/avcodec.h>6. #include <libavformat/avformat.h>7. #include <libavfilter/avfiltergraph.h>8. #include <libavfilter/avcodec.h>9. #include <libavfilter/buffersink.h>10. #include <libavfilter/buffersrc.h>11. #include <libswscale/swscale.h>12.13. const char *filter_descr = "overlay=5:5";14.15. #define ENABLE_YUV_FILE 116.17. AVFormatContext *input_fmt_ctx;18. AVCodecContext *input_dec_ctx;19.20. AVFormatContext *overlay_fmt_ctx;21. AVCodecContext *overlay_dec_ctx;22.23. int input_video_stream_idx, overlay_video_stream_idx;24.25. AVFilterGraph *filter_graph;26.27. AVFilterInOut *inputs;28. AVFilterInOut *outputs;29.30. AVFilterContext *buffersrc_ctx;31. AVFilterContext *bufferoverlay_ctx;32. AVFilterContext *buffersink_ctx;33.34. int ret;35. int got_frame;36.37. int video_eof_reached = 0;38. int overlay_eof_reached = 0;39.40. int active_stream_index = -1;41.42. FILE* fp_yuv;43.44. void yuv420p_save(AVFrame *pFrame);45. int video_transcode_step(AVFrame* mVideoFrame);46. int overlay_transcode_step(AVFrame* mOverlayFrame);46. int overlay_transcode_step(AVFrame* mOverlayFrame);47. int video_output_eof_packet(const char* tag,48. AVStream* ist, AVFilterContext* ifilter);49.50. static int open_input_file(const char *filename)51. {52. int ret;53. AVCodec *dec;54.55. if ((ret = avformat_open_input(&input_fmt_ctx, filename, NULL, NULL)) < 0) {56. av_log(NULL, AV_LOG_ERROR, "Cannot open input file\n");57. return ret;58. }59.60. if ((ret = avformat_find_stream_info(input_fmt_ctx, NULL)) < 0) {61. av_log(NULL, AV_LOG_ERROR, "Cannot find stream information\n");62. return ret;63. }64.65. /* select the video stream */66. ret = av_find_best_stream(input_fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, &dec, 0);67. if (ret < 0) {68. av_log(NULL, AV_LOG_ERROR, "Cannot find a video stream in the input file\n");69. return ret;70. }71. input_video_stream_idx = ret;72. input_dec_ctx = input_fmt_ctx->streams[input_video_stream_idx]->codec;73.74. /* init the video decoder */75. if ((ret = avcodec_open2(input_dec_ctx, dec, NULL)) < 0) {76. av_log(NULL, AV_LOG_ERROR, "Cannot open video decoder\n");77. return ret;78. }79.80. return 0;81. }82.83.84. static int open_overlay_file(const char *filename)85. {86. int ret;87. AVCodec *dec;88.89. if ((ret = avformat_open_input(&overlay_fmt_ctx, filename, NULL, NULL)) < 0) {90. av_log(NULL, AV_LOG_ERROR, "Cannot open input file\n");91. return ret;92. }93.94. if ((ret = avformat_find_stream_info(overlay_fmt_ctx, NULL)) < 0) {95. av_log(NULL, AV_LOG_ERROR, "Cannot find stream information\n");96. return ret;97. }98.99. /* select the video stream */100. ret = av_find_best_stream(overlay_fmt_ctx, AVMEDIA_TYPE_VIDEO, -1, -1, &dec, 0); 101. if (ret < 0) {102. av_log(NULL, AV_LOG_ERROR, "Cannot find a video stream in the input file\n"); 103. return ret;104. }105. overlay_video_stream_idx = ret;106. overlay_dec_ctx = overlay_fmt_ctx->streams[overlay_video_stream_idx]->codec;107.108. /* init the video decoder */109. if ((ret = avcodec_open2(overlay_dec_ctx, dec, NULL)) < 0) {110. av_log(NULL, AV_LOG_ERROR, "Cannot open video decoder\n");111. return ret;112. }113.114. printf("overlay format = %s\n", overlay_fmt_ctx->iformat->name);115.116. return 0;117. }118.119.120. static int video_config_input_filter(AVFilterInOut* inputs, AVFilterContext** input_filter_ctx)121. {122. char args[512];123. memset(args, 0, sizeof(args));124.125. AVFilterContext *first_filter = inputs->filter_ctx;126. int pad_idx = inputs->pad_idx;127.128. AVFilter *filter = avfilter_get_by_name("buffer");129. // AVRational time_base = input_dec_ctx->time_base;130. AVStream* video_st = input_fmt_ctx->streams[input_video_stream_idx];131. AVRational time_base = video_st->time_base;132.133. snprintf(args, sizeof(args),134. "video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d:sws_param=flags=%d:frame_rate=%d/%d", 135. input_dec_ctx->width, input_dec_ctx->height, input_dec_ctx->pix_fmt,136. input_dec_ctx->time_base.num, input_dec_ctx->time_base.den,137. input_dec_ctx->sample_aspect_ratio.num, input_dec_ctx->sample_aspect_ratio.den,138. SWS_BILINEAR + ((video_st->codec->flags&CODEC_FLAG_BITEXACT) ? SWS_BITEXACT:0),139. video_st->r_frame_rate.num, video_st->r_frame_rate.den);140.141. printf("input args = %s\n", args);142.143. ret = avfilter_graph_create_filter(input_filter_ctx, filter, "src_in", args, NULL, filter_graph);144.145. if (ret < 0 ) {146. printf("video config input filter fail.\n");147. return -1;148. }149. ret = avfilter_link(*input_filter_ctx, 0, first_filter, pad_idx);150. assert(ret >= 0);151.152. printf("video_config_input_filter avfilter_link ret = %d\n", ret);153.154. return ret;155.156. }157.158. static int video_config_overlay_filter(AVFilterInOut* inputs, AVFilterContext** overlay_filter_ctx )159. {160. char args[512];161. memset(args, 0, sizeof(args));162.163. AVFilterContext *first_filter = inputs->filter_ctx;163. AVFilterContext *first_filter = inputs->filter_ctx;164.165. int pad_idx = inputs->pad_idx;166.167. AVFilter *filter = avfilter_get_by_name("buffer");168.169. //AVRational time_base = overlay_dec_ctx->time_base;170. AVStream* overlay_st = overlay_fmt_ctx->streams[overlay_video_stream_idx];171. AVRational time_base = overlay_st->time_base;172.173. snprintf(args, sizeof(args),174. "video_size=%dx%d:pix_fmt=%d:time_base=%d/%d:pixel_aspect=%d/%d:sws_param=flags=%d:frame_rate=%d/%d", 175. overlay_dec_ctx->width, overlay_dec_ctx->height, overlay_dec_ctx->pix_fmt,176. time_base.num, time_base.den,177. overlay_dec_ctx->sample_aspect_ratio.num, overlay_dec_ctx->sample_aspect_ratio.den,178. SWS_BILINEAR + ((overlay_st->codec->flags&CODEC_FLAG_BITEXACT) ? SWS_BITEXACT:0),179. overlay_st->r_frame_rate.num, overlay_st->r_frame_rate.den);180.181. printf("overlay args = %s\n", args);182.183. ret = avfilter_graph_create_filter(overlay_filter_ctx, filter, "overlay_in", args, NULL, filter_graph);184.185. if (ret < 0 ) {186. printf("video config overlay filter fail.\n");187. return -1;188. }189.190. ret = avfilter_link(*overlay_filter_ctx, 0, first_filter, pad_idx);191. assert(ret >= 0);192.193. printf("video_config_overlay_filter ret = %d\n", ret);194.195. avfilter_inout_free(&inputs);196.197. return ret;198.199. }200.201. static int video_config_output_filter(AVFilterInOut* outputs, AVFilterContext** out_filter_ctx)202. {203. char args[512];204.205. AVFilterContext *last_filter = outputs->filter_ctx;206. int pad_idx = outputs->pad_idx;207. AVFilter *buffersink = avfilter_get_by_name("ffbuffersink");208.209. int ret = avfilter_graph_create_filter(out_filter_ctx, buffersink, "video_out", NULL, NULL, filter_graph);210. assert(ret >= 0);211.212. if (ret < 0)213. return ret;214.215. ret = avfilter_link(last_filter, pad_idx, *out_filter_ctx, 0);216. assert(ret >= 0);217. if (ret < 0)218. return ret;219.220. avfilter_inout_free(&outputs);221.221.222. return 0;223.224. }225.226. static int init_input_filters()227. {228. filter_graph->scale_sws_opts = av_strdup("flags=0x4");229. av_opt_set(filter_graph, "aresample_swr_opts", "", 0);230.231. ret = avfilter_graph_parse2(filter_graph, filter_descr, &inputs, &outputs);232.233. assert(inputs && inputs->next && !inputs->next->next);234.235. ret = video_config_input_filter(inputs, &buffersrc_ctx);236.237. ret = video_config_overlay_filter(inputs->next, &bufferoverlay_ctx);238.239. return ret;240. }241.242. static int init_output_filters()243. {244. return video_config_output_filter(outputs, &buffersink_ctx);245. }246.247. int reap_filters() {248. AVFilterBufferRef *picref;249.250. while (1) {251. ret = av_buffersink_get_buffer_ref(buffersink_ctx, &picref, AV_BUFFERSINK_FLAG_NO_REQUEST); 252. if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {253. //printf("reap_filters fail ret = %d\n", ret);254. return 0; // no frame filtered.255. }256.257. printf("samplesref -------------------\n");258.259. AVFrame* filtered_frame = avcodec_alloc_frame();260. avcodec_get_frame_defaults(filtered_frame);261.262. avfilter_copy_buf_props(filtered_frame, picref);263.264. yuv420p_save(filtered_frame);265.266. avfilter_unref_bufferp(&picref);267. }268.269. }270.271. int transcode_from_filter(AVFilterContext** ifilters, int* eof_reached_arr, int* active_stream_indext) { 272. int ret = 0;273.274. ret = avfilter_graph_request_oldest(filter_graph);275.276. if (ret >= 0) {277. return ret;278. }279. if (ret == AVERROR_EOF) {280. return ret;280. return ret;281. }282. if (ret != AVERROR(EAGAIN)) {283. return ret;284. }285.286. int nb_requests_max = 0;287. int i;288. for (i = 0; i < 2; i++) {289. int eof_reached = eof_reached_arr[i];290. if (eof_reached) {291. continue;292. }293. AVFilterContext* ifilter = ifilters[i];294. int nb_requests = av_buffersrc_get_nb_failed_requests(ifilter); 295. if (nb_requests > nb_requests_max) {296. nb_requests_max = nb_requests;297. *active_stream_indext = i;298. }299. }300.301. return ret;302.303. }304.305.306. int main()307. {308. avcodec_register_all();309. av_register_all();310. avfilter_register_all();311. avformat_network_init();312.313.314. char* video_file = "outFileSrc.mp4";315. char* overlay_video_file = "my_logo.png"; // light1.mp4316.317.318. #if ENABLE_YUV_FILE319. const char* yuvFile = "outWater.yuv";320.321. fp_yuv = fopen(yuvFile, "wb");322.323. #endif324.325. open_input_file(video_file);326. open_overlay_file(overlay_video_file);327.328. filter_graph = avfilter_graph_alloc();329. if (!filter_graph) {330. printf("filter graph alloc fail.\n");331. return -1;332. }333.334. init_input_filters();335. init_output_filters();336.337. if ((ret = avfilter_graph_config(filter_graph, NULL)) < 0)338. return ret;338. return ret;339.340. AVFrame* mVideoFrame = avcodec_alloc_frame();341. AVFrame* mOverlayFrame = avcodec_alloc_frame();342.343. while(1) {344. if (video_eof_reached && overlay_eof_reached) {345. printf("stream EOF.\n");346. break;347. }348.349. AVFilterContext* ifilters[] = {buffersrc_ctx, bufferoverlay_ctx};350. int eof_reacheds[] = {video_eof_reached, overlay_eof_reached};351.352. ret = transcode_from_filter(ifilters, eof_reacheds, &active_stream_index); 353.354. if (ret >= 0) {355. ret = reap_filters();356. assert(ret >= 0);357. continue;358. }359. if (ret == AVERROR_EOF) {360. ret = reap_filters();361. assert(ret >= 0);362. continue;363. }364.365. if (ret == AVERROR(EAGAIN) && active_stream_index < 0) {366. continue;367. }368. assert(active_stream_index >= 0);369.370. printf("active_stream_index = %d\n", active_stream_index);371.372. if (active_stream_index == 0) {373. video_transcode_step(mVideoFrame);374. continue;375. }376.377. overlay_transcode_step(mOverlayFrame);378.379. }380.381. if (input_dec_ctx)382. avcodec_close(input_dec_ctx);383. avformat_close_input(&input_fmt_ctx);384.385. if (overlay_dec_ctx)386. avcodec_close(overlay_dec_ctx);387. avformat_close_input(&overlay_fmt_ctx);388.389.390.391. printf("my_filtering_video3 end -------\n");392. return 0;393. }394.395. int video_transcode_step(AVFrame* mVideoFrame) {396. int ret = 0;397. AVPacket pkt;397. AVPacket pkt;398.399. ret = av_read_frame(input_fmt_ctx, &pkt);400. if (ret == AVERROR(EAGAIN)) {401. return 0;402. }403. if (ret < 0) {404. video_eof_reached = 1;405. assert(ret == AVERROR_EOF);406. ret = video_output_eof_packet("video_eof", input_fmt_ctx->streams[input_video_stream_idx], buffersrc_ctx);407. assert (ret >= 0);408. return ret;409.410. }411.412. if (pkt.stream_index != input_video_stream_idx) {413. // av_free(&pkt);414. return ret;415. }416.417. ret = avcodec_decode_video2(input_dec_ctx, mVideoFrame, &got_frame, &pkt);418. if (ret < 0) {419. printf("Error decoding input video\n");420. }421.422. if (got_frame) {423. int64_t best_effort_timestamp = av_frame_get_best_effort_timestamp(mVideoFrame);424. mVideoFrame->pts = best_effort_timestamp;425.426. if (av_buffersrc_add_frame(buffersrc_ctx, mVideoFrame, AV_BUFFERSRC_FLAG_PUSH) < 0) {427. av_log(NULL, AV_LOG_ERROR, "Error while feeding the video filtergraph\n");428. return -1;429. }430. reap_filters();431. }432.433. return 0;434.435. }436.437. int overlay_transcode_step(AVFrame* mOverlayFrame) {438. int ret = 0;439. AVPacket pkt;440.441. ret = av_read_frame(overlay_fmt_ctx, &pkt);442. if (ret == AVERROR(EAGAIN)) {443. return 0;444. }445.446. if (ret < 0) {447. overlay_eof_reached = 1;448. ret = video_output_eof_packet("overlay_eof", overlay_fmt_ctx->streams[input_video_stream_idx], bufferoverlay_ctx); 449. assert(ret >=0 );450. return ret;451. }452.453. if (pkt.stream_index != overlay_video_stream_idx) {454. av_free_packet(&pkt);455. return ret;455. return ret;456. }457.458. ret = avcodec_decode_video2(overlay_dec_ctx, mOverlayFrame, &got_frame, &pkt);459. if (ret < 0) {460. printf("Error decoding overlay video\n");461. }462.463. if (got_frame) {464. int64_t best_effort_timestamp = av_frame_get_best_effort_timestamp(mOverlayFrame);465.466. mOverlayFrame->pts = best_effort_timestamp;467.468. if (av_buffersrc_add_frame(bufferoverlay_ctx, mOverlayFrame, AV_BUFFERSRC_FLAG_PUSH) < 0) { 469. av_log(NULL, AV_LOG_ERROR, "Error while feeding the audio filtergraph\n");470. return -1;471. }472. }473. return 0;474.475. }476.477. /**478. * output EOF packet to filter to flush479. */480. int video_output_eof_packet(const char* tag,481. AVStream* ist, AVFilterContext* ifilter)482. {483. int ret = 0;484.485. // alloc frame if NULL486. AVFrame* decoded_frame = avcodec_alloc_frame();487.488. AVPacket pkt;489. av_init_packet(&pkt);490. pkt.data = NULL;491. pkt.size = 0;492.493. int got_frame = 0;494. ret = avcodec_decode_video2(ist->codec, decoded_frame, &got_frame, &pkt);495. // EOF, assert got nothing and ret is 0.496. // TODO: here we still got frame, different to ffmpeg.497. assert(ret >= 0);498.499. // flush filter500. av_buffersrc_add_ref(ifilter, NULL, 0);501.502. printf("[%s] filter -> eof packet.\n", tag);503.504. return ret;505. }506.507.508. /**509. * save yuv420p frame [YUV]510. */511. void yuv420p_save(AVFrame *pFrame)512. {513. int i = 0;514.514.515. int width = pFrame->width, height = pFrame->height;516. int height_half = height / 2, width_half = width / 2;517. int y_wrap = pFrame->linesize[0];518. int u_wrap = pFrame->linesize[1];519. int v_wrap = pFrame->linesize[2];520.521. unsigned char *y_buf = pFrame->data[0];522. unsigned char *u_buf = pFrame->data[1];523. unsigned char *v_buf = pFrame->data[2];524.525. //save y526. for (i = 0; i < height; i++)527. fwrite(y_buf + i * y_wrap, 1, width, fp_yuv);528.529. //save u530. for (i = 0; i < height_half; i++)531. fwrite(u_buf + i * u_wrap, 1, width_half, fp_yuv);532.533. //save v534. for (i = 0; i < height_half; i++)535. fwrite(v_buf + i * v_wrap, 1, width_half, fp_yuv);536.537. fflush(fp_yuv);538. }PS: ⽤到avfilter的，如果有两个输⼊，不管⾳频还是视频，代码基本都是这样 (如视频的overlay ⾳频的amix 等等)[cpp]view plaincopy1.。