tt

[canny.git] / stc / exp / raphus / sobel555_for_ensemble.py

diff --git a/stc/exp/raphus/sobel555_for_ensemble.py b/stc/exp/raphus/sobel555_for_ensemble.py
index 89f8d988275b90170b8f12b339621f2c7a81bbac..89e0cbdd39d07aa7bc39d413c1ce4004607af201 100644 (file)
--- a/stc/exp/raphus/sobel555_for_ensemble.py
+++ b/stc/exp/raphus/sobel555_for_ensemble.py
@@ -8,8 +8,14 @@ import cv
 import os
 from random import *
 from math import *
 import os
 from random import *
 from math import *

+from bbs import *
+

 
 infinity = 1000000000
 
 infinity = 1000000000

+M=18532395500947174450709383384936679868383424444311405679463280782405796233163977*39688644836832882526173831577536117815818454437810437210221644553381995813014959
+X=18532395500947174450709383384936679868383424444311
+
+

 
 
 
 
 
 


@@ -372,7 +378,12 @@ def compute_list_bit_to_change(threshold,processed):
     return List
 
 
     return List
 
 

-def compute_filter_canny(level,image):
+def compute_filter_canny(level,image,MsgLen):
+
+    bbs = BlumBlumShub();
+    bbs.setN(M)
+    bbs.setSeed(X)
+

     level2=level.copy()
     level2= array(level2.getdata()).flatten()
     l=0
     level2=level.copy()
     level2= array(level2.getdata()).flatten()
     l=0


@@ -418,32 +429,44 @@ def compute_filter_canny(level,image):
     processed= array(processed.getdata()).flatten()
     List7=set(compute_list_bit_to_change(100,processed))
 
     processed= array(processed.getdata()).flatten()
     List7=set(compute_list_bit_to_change(100,processed))
 

-    nb_bit_embedded=(512*512/10)+40
+    #nb_bit_embedded=(512*512/10)+40
+    nb_bit_embedded=max(2*MsgLen,int(len(List3)/MsgLen)*MsgLen)
+    print "nb_bit_embedded",nb_bit_embedded

     AvailablePixel3=List3
     AvailablePixel5=AvailablePixel3.union(List5)
     AvailablePixel7=AvailablePixel5.union(List7)
     AvailablePixel3=List3
     AvailablePixel5=AvailablePixel3.union(List5)
     AvailablePixel7=AvailablePixel5.union(List7)

+    sub=set()
+

     if len(AvailablePixel3)>nb_bit_embedded:
         step=1
         WorkingPixel=AvailablePixel3
     if len(AvailablePixel3)>nb_bit_embedded:
         step=1
         WorkingPixel=AvailablePixel3

+        sub = bbs.sample(AvailablePixel3,nb_bit_embedded)

     elif len(AvailablePixel5)>nb_bit_embedded:
         step=2
         WorkingPixel=AvailablePixel5
     elif len(AvailablePixel5)>nb_bit_embedded:
         step=2
         WorkingPixel=AvailablePixel5

+        sub = AvailablePixel3.union(
+            bbs.sample(AvailablePixel5-AvailablePixel3,nb_bit_embedded-len(AvailablePixel3)))
+

     elif len(AvailablePixel7)>nb_bit_embedded:
         step=3
         WorkingPixel=AvailablePixel7
     elif len(AvailablePixel7)>nb_bit_embedded:
         step=3
         WorkingPixel=AvailablePixel7

+        sub = AvailablePixel5.union(
+            bbs.sample(AvailablePixel7-AvailablePixel5,nb_bit_embedded-len(AvailablePixel5)))

     else:
         step=4
         WorkingPixel=range(len(level2))
     else:
         step=4
         WorkingPixel=range(len(level2))

+        sub = range(len(level2))
+

 
 

-    print "avail P3",len(AvailablePixel3)
-    print "avail P5",len(AvailablePixel5)
-    print "avail P7",len(AvailablePixel7)
+    #print "avail P3",len(AvailablePixel3)
+    #print "avail P5",len(AvailablePixel5)
+    #print "avail P7",len(AvailablePixel7)

 
 

-    print "size WorkingPixel",len(WorkingPixel)
-    Weight=[0 for _ in WorkingPixel]
+
+    Weight=[0 for _ in sub]

 
     l=0
 
     l=0

-    for i in WorkingPixel:
+    for i in sub:

         if step>=1 and i in List3:
             Weight[l]=1
         if step>=2 and i in List5 and Weight[l]==0:
         if step>=1 and i in List3:
             Weight[l]=1
         if step>=2 and i in List5 and Weight[l]==0:


@@ -454,16 +477,10 @@ def compute_filter_canny(level,image):
             Weight[l]=1000
         l+=1
 
             Weight[l]=1000
         l+=1
 

-            
-        
-
-    List_bit_to_change=WorkingPixel
+    List_bit_to_change=sub

         
         
         
         

-
-
-
-    return [List_bit_to_change,Weight]
+    return [sub,Weight]

 
 
 
 
 
 


@@ -476,54 +493,39 @@ def compute_filter_canny(level,image):
 
 
 def mystego(filein, fileout):
 
 
 def mystego(filein, fileout):

+    bit_to_read = 1
+    global M,X
+    bbs = BlumBlumShub();
+    bbs.setN(M)
+    bbs.setSeed(X)
+

     dd = im.open(filein)
     dd = dd.convert('RGB') 
     red, green, blue = dd.split()
     level=red.copy()
 
     dd = im.open(filein)
     dd = dd.convert('RGB') 
     red, green, blue = dd.split()
     level=red.copy()
 

-    [List_bit_to_change,Weight]=compute_filter_canny(level,dd)
-    level= array(level.getdata()).flatten()
-
-
-    bit_to_read=1  
-
-
-
-
-    M=18532395500947174450709383384936679868383424444311405679463280782405796233163977*39688644836832882526173831577536117815818454437810437210221644553381995813014959
-    # M clef
-    X=18532395500947174450709383384936679868383424444311
-    # X clef

 
 

+    """

 
 
 
 

+    message="Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete  Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete  Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete  Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete  Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete voila c'est la fin blablabla:-)"

 
 

-    l=0
-
-
+    message="Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela."

 
 

-    message="Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete  Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete  Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete  Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete  Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete Ce que j'écris est très original... Bref, je suis un poete   Salut christophe, arrives tu à lire ce message? normalement tu dois lire cela. Bon voici un test avec un message un peu plus long. Bon j'en rajoute pour voir. Ce que j'écris est très original... Bref, je suis un poete voila c'est la fin blablabla:-)"
-    message=message[0:len(message)/1]
+    
+   
+    

     leng_msg=len(message)
     leng_msg=len(message)

+    print "taille du message en caracteres",leng_msg

     message=message+((leng_msg+7)/8*8-leng_msg)*" "
     leng_msg=len(message)
     message=message+((leng_msg+7)/8*8-leng_msg)*" "
     leng_msg=len(message)

-

     leng='%08d'%len(message)
     leng='%08d'%len(message)

-
-
-
-
+    

     len_leng=len(leng)
     leng_error=int(len_leng)
     leng_cor=leng
     len_leng=len(leng)
     leng_error=int(len_leng)
     leng_cor=leng

+    

     List_pack=a2b_list(leng_cor)
     List_pack=a2b_list(leng_cor)

-    print len(str(M))
-   
-    List_random=[]
-    while len(List_random)<len(List_bit_to_change):
-        X=(X*X)%M
-        List_random.extend(Denary2Binary(X))
-

     size=0
     for i in range(leng_msg/8):
         m=message[i*8:(i+1)*8]
     size=0
     for i in range(leng_msg/8):
         m=message[i*8:(i+1)*8]


@@ -532,33 +534,52 @@ def mystego(filein, fileout):
         size=size+len(m_bin)
         List_pack.extend(m_bin) 
 
         size=size+len(m_bin)
         List_pack.extend(m_bin) 
 

+    """    

 
 
 
 
 
 

+    message = []
+    for c in [x if x==0 else 1 for x in im.open("invader.png").getdata()]:
+        message +=[c]
+    
+    leng_msg=len(message)
+    leng='%08d'%len(message)
+    len_leng=len(leng)
+    leng_error=int(len_leng)
+    leng_cor=leng    
+    List_pack=a2b_list(leng_cor)
+    List_pack += message
+    #print List_pack

 
 

+    leng_msg=len(List_pack)

 
 

+    [List_bit_to_change,Weight]=compute_filter_canny(level,dd,leng_msg)
+    level= array(level.getdata()).flatten()

    
    

+    List_random=[]
+    while len(List_random)<len(List_bit_to_change):
+        List_random.extend(Denary2Binary(bbs.next()))

 
 

-    #List_bit_to_change = support
-    #print len(List_bit_to_change)
-
+        

 
     #print List_bit_to_change
 
     #print List_bit_to_change

-    Support=[getBit(level[l],bit_to_read) for l in List_bit_to_change]
-    #print len(List_pack)
+    Support=[]
+    for l in List_bit_to_change : 
+        Support += [getBit(level[l],bit_to_read) ]
+

     #List_pack = message
 
 
     Message=[(int(List_pack[l])^int(List_random[l])) for l in xrange(len(List_pack))]
 
     #List_pack = message
 
 
     Message=[(int(List_pack[l])^int(List_random[l])) for l in xrange(len(List_pack))]
 

-    print "support",len(List_bit_to_change)
-    print "message",len(Message)
-    print "weight",len(Weight)
+    #print "support",len(List_bit_to_change)
+    #print "message",len(Message)
+    #print "weight",len(Weight)

 
     (x_b,Stc_message,H_hat) = stc(Support,Weight,Message)
 
 
     (x_b,Stc_message,H_hat) = stc(Support,Weight,Message)
 

-    print "pourcentage modif",nbdif(x_b,Stc_message)
-    print "taille Stc_message",len(Stc_message)
+    #print "pourcentage modif",nbdif(x_b,Stc_message)
+    #print "taille Stc_message",len(Stc_message)

 #    Stc_message=Message
 
 
 #    Stc_message=Message
 
 


@@ -576,9 +597,9 @@ def mystego(filein, fileout):
             level[l]=float64(setBit(level[l],bit_to_read,Stc_message[size_mesg]))
             size_mesg+=1
 
             level[l]=float64(setBit(level[l],bit_to_read,Stc_message[size_mesg]))
             size_mesg+=1
 

-    print 'size mesg',size_mesg
-    print 'val mod',val_mod
-    print 'len message',len(Message),len(List_pack)
+    #print 'size mesg',size_mesg
+    #print 'val mod',val_mod
+    #print 'len message',len(Message),len(List_pack)

 
     
 
 
     
 


@@ -589,24 +610,23 @@ def mystego(filein, fileout):
     #zz4.show()
 
 
     #zz4.show()
 
 

-
+    """

     
     dd2 = im.open(fileout)
     dd2 = dd2.convert('RGB') 
     red2, green, blue = dd2.split()
     level2=red2.copy()
 
     
     dd2 = im.open(fileout)
     dd2 = dd2.convert('RGB') 
     red2, green, blue = dd2.split()
     level2=red2.copy()
 

-    [List_bit_to_change2,Weight2]=compute_filter_canny(level2,dd2)
+    [List_bit_to_change2,Weight2]=compute_filter_canny(level2,dd2,leng_msg)

 
 
     level2= array(level2.getdata()).flatten()
 
     print "support2",len(List_bit_to_change2)
     print "message2",len(Message)
 
 
     level2= array(level2.getdata()).flatten()
 
     print "support2",len(List_bit_to_change2)
     print "message2",len(Message)

-    print "weight2",len(Weight)
+    print "weight2",len(Weight2)

 
     alpha = float(len(List_bit_to_change2))/len(Message)
 
     alpha = float(len(List_bit_to_change2))/len(Message)

-    print alpha

     assert alpha >= 2 
     index = min(int(alpha),9)
     H_hat2 = {
     assert alpha >= 2 
     index = min(int(alpha),9)
     H_hat2 = {


@@ -624,11 +644,11 @@ def mystego(filein, fileout):
     print H_hat,H_hat2
 
 
     print H_hat,H_hat2
 
 

-    Stc_message2=[getBit(level2[l],bit_to_read) for l in List_bit_to_change]
+    Stc_message2=[getBit(level2[l],bit_to_read) for l in List_bit_to_change2]

     LL1=list(List_bit_to_change);
     LL2=list(List_bit_to_change2)
 
     LL1=list(List_bit_to_change);
     LL2=list(List_bit_to_change2)
 

-    print "Level",max([level2[l]-level[l] for l in xrange(len(Stc_message))])
+    print "Level",max([level2[l]-level[l] for l in xrange(len(Stc_message2))])

     print "List bit to change",max([LL2[l]-LL1[l] for l in xrange(len(Stc_message))])
     print "Stc message", max([Stc_message[l]-Stc_message2[l] for l in xrange(len(Stc_message))])
 
     print "List bit to change",max([LL2[l]-LL1[l] for l in xrange(len(Stc_message))])
     print "Stc message", max([Stc_message[l]-Stc_message2[l] for l in xrange(len(Stc_message))])
 


@@ -651,6 +671,7 @@ def mystego(filein, fileout):
 #    print conv_list_bit(''.join([`MessageDecoded2[i]` for i in xrange(leng_error*8)]))
 #    print int(conv_list_bit(''.join([`MessageDecoded2[i]` for i in xrange(leng_error*8)])))
 
 #    print conv_list_bit(''.join([`MessageDecoded2[i]` for i in xrange(leng_error*8)]))
 #    print int(conv_list_bit(''.join([`MessageDecoded2[i]` for i in xrange(leng_error*8)])))
 

+    print "MessageDecoded2",MessageDecoded2

     print conv_list_bit(''.join([`MessageDecoded2[i]` for i in xrange(len(Message2))]))
  
 
     print conv_list_bit(''.join([`MessageDecoded2[i]` for i in xrange(len(Message2))]))
  
 


@@ -680,35 +701,37 @@ def mystego(filein, fileout):
     print decoded_msg#[0:20]
     print len(List_bit_to_change)
     list_nb_bit.append(filein)
     print decoded_msg#[0:20]
     print len(List_bit_to_change)
     list_nb_bit.append(filein)

-    list_nb_bit.append(len(List_bit_to_change))
-
+    list_nb_bit.append(len(List_bit_to_change2))

 
 

+"""

 
 
 #path_cover = '/localhome/couturie/ensemble/cover_bad_laplace/'
 
 
 #path_cover = '/localhome/couturie/ensemble/cover_bad_laplace/'

-path_stego = '/localhome/couturie/ensemble/BossBase-1.0-canny_new/'
-path_cover = '/localhome/couturie/ensemble/BossBase-1.0-cover/'
+path_stego = '/home/couchot/rech/CCG12/canny/stc/exp/raphus/stego/'
+path_cover = '/home/couchot/rech/BCG10/Oxford11/experiments/images/'

 #path_stego = '/tmp/'
 #path_stego = '/home/couturie/BossBase-1.0-canny/'
 #listing = os.listdir(path_cover)
 
 #path_stego = '/tmp/'
 #path_stego = '/home/couturie/BossBase-1.0-canny/'
 #listing = os.listdir(path_cover)
 

+

 list_nb_bit=[]
 l=0
 
 
 list_nb_bit=[]
 l=0
 
 

-mystego(argv[1],argv[2])
+

 # entree sortie 
 
 # entree sortie 
 

-raise SystemExit
+
+listing = [957, 108, 106, 29, 431, 924, 262, 477, 346, 809]

 
 for infile in listing:
 #    if l<10:
 
 for infile in listing:
 #    if l<10:

-    if infile[0]==argv[1]:
-        print "current file is: " + infile, path_stego+infile
-        mystego(path_cover+infile,path_stego+infile)
+    fi = path_cover+str(infile)+'.png'
+    fo = path_stego+str(infile)+'.png'
+    mystego(fi,fo)

     l+=1
 
 #f = open('histogram_boss_canny_100255', 'w')
 #f = open('histogram', 'w')
 #for item in list_nb_bit:
     l+=1
 
 #f = open('histogram_boss_canny_100255', 'w')
 #f = open('histogram', 'w')
 #for item in list_nb_bit:

-#  f.write("%s\n" % item)
+#f.write("%s\n" % item)