# -*- coding: UTF-8 -*-
 
from PIL import Image
 
 
 
def plus(str):
 
      #Python zfill() 方法返回指定长度的字符串，原字符串右对齐，前面填充0。
 
    return str.zfill(8)
 
 
 
def get_key(strr):
 
      #获取要隐藏的文件内容
 
    tmp = strr
 
    f = file(tmp,"rb")
 
    str = ""
 
    s = f.read()
 
    for i in range(len(s)):
 
         #逐个字节将要隐藏的文件内容转换为二进制，并拼接起来
 
         #1.先用ord()函数将s的内容逐个转换为ascii码
 
         #2.使用bin()函数将十进制的ascii码转换为二进制
 
         #3.由于bin()函数转换二进制后，二进制字符串的前面会有"0b"来表示这个字符串是二进制形式，所以用replace()替换为空
 
         #4.又由于ascii码转换二进制后是七位，而正常情况下每个字符由8位二进制组成，所以使用自定义函数plus将其填充为8位
 
        str = str+plus(bin(ord(s[i])).replace('0b',''))
 
        #print str
 
    f.closed
 
    return str
 
 
 
def mod(x,y):
 
    return x%y;
 
#str1为载体图片路径，str2为隐写文件，str3为加密图片保存的路径
 
def func(str1,str2,str3):  
 
    im = Image.open(str1)
 
    #获取图片的宽和高
 
    width = im.size[0]
 
    print "width:"+str(width)+"\n"
 
    height = im.size[1]
 
    print "height:"+str(height)+"\n"
 
    count = 0
 
    #获取需要隐藏的信息
 
    key = get_key(str2)
 
    keylen = len(key)
 
    for h in range(0,height):
 
        for w in range(0,width):
 
            pixel = im.getpixel((w,h))
 
            a=pixel[0]
 
            b=pixel[1]
 
            c=pixel[2]
 
            if count == keylen:
 
                break
 
            #下面的操作是将信息隐藏进去
 
            #分别将每个像素点的RGB值余2，这样可以去掉最低位的值
 
            #再从需要隐藏的信息中取出一位，转换为整型
 
            #两值相加，就把信息隐藏起来了
 
            a= a-mod(a,2)+int(key[count])
 
            count+=1
 
            if count == keylen:
 
                im.putpixel((w,h),(a,b,c))
 
                break
 
            b =b-mod(b,2)+int(key[count])
 
            count+=1
 
            if count == keylen:
 
                im.putpixel((w,h),(a,b,c))
 
                break
 
            c= c-mod(c,2)+int(key[count])
 
            count+=1
 
            if count == keylen:
 
                im.putpixel((w,h),(a,b,c))
 
                break
 
            if count % 3 == 0:
 
                im.putpixel((w,h),(a,b,c))
 
    im.save(str3)
 
#原图
 
old = "1.png"
 
#处理后输出的图片路径
 
new = "LSB.png"
 
#需要隐藏的信息
 
enc = "flag.txt"
 
func(old,enc,new)