# ~*~ encoding: utf-8 ~*~ # import sys ################################################# # classe qui parse le header (binaire) en objet # ################################################# class BMPHeader: # CONSTRUCTEUR: initialise les variables def __init__(self): self.bin = 0; # header brut (format initial: bin) self.signature = 0; # signature (4D42) self.fileSize = 0; # taille du fichier bmp (bytes) self.offset = 0; # début de l'image (bytes) self.infoSize = 0; # taille du INFO_HEADER self.width = 0; # longueur de l'image (pixel) self.height = 0; # hauteur de l'image (pixel) self.plans = 0; # nombre de plans (default: 1) self.bpp = 0; # nombre de bits par pixel (1,4,8, 24) self.compType = 0; # type de compression (0=none, 1=RLE-8, 2=RLE-4) self.size = 0; # taille de l'image avec padding (bytes) self.horiRes = 0; # résolution horizontale (pixels) self.vertRes = 0; # résolution verticale (pixels) self.colorNb = 0; # nombre de couleurs de l'image (ou 0) self.colorINb = 0; # nombre d'images importantes (ou 0) # convertit les octets en entier def toInt(self, bytes): intReturn = 0; for i, byte in enumerate(bytes): intReturn += ord(byte) * (256 ** i) return intReturn # parse le header au format bin en objet def parse(self, binHeader): self.bin = binHeader # header brut (format initial: bin) self.signature = self.toInt(binHeader[ 0: 2]) # signature (4D42) self.fileSize = self.toInt(binHeader[ 2: 6]) # taille du fichier bmp (bytes) self.offset = self.toInt(binHeader[10:14]) # début de l'image (bytes) self.infoSize = self.toInt(binHeader[14:18]) # taille du INFO_HEADER self.width = self.toInt(binHeader[18:22]) # longueur de l'image (pixel) self.height = self.toInt(binHeader[22:26]) # hauteur de l'image (pixel) self.plans = self.toInt(binHeader[26:28]) # nombre de plans (default: 1) self.bpp = self.toInt(binHeader[28:30]) # nombre de bits par pixel (1,4,8, 24) self.compType = self.toInt(binHeader[30:34]) # type de compression (0=none, 1=RLE-8, 2=RLE-4) self.size = self.toInt(binHeader[34:38]) # taille de l'image avec padding (bytes) self.horiRes = self.toInt(binHeader[38:42]) # résolution horizontale (pixels) self.vertRes = self.toInt(binHeader[42:46]) # résolution verticale (pixels) self.colorNb = self.toInt(binHeader[46:50]) # nombre de couleurs de l'image (ou 0) self.colorINb = self.toInt(binHeader[50:54]) # nombre d'images importantes (ou 0) self.readableData = "" for byte in binHeader: self.readableData += str(ord(byte)) + " " # Affiche au format humain, toutes les infos du header def info(self): print print "INFORMATION DU HEADER" print "=====================" print "signature: %s" % ( hex(self.signature) ) print "taille du fichier: %s" % ( hex(self.fileSize ) ) print "offset du contenu: %s" % ( hex(self.offset ) ) print "taille infoHeader: %s" % ( hex(self.infoSize ) ) print "largeur: %s" % ( hex(self.width ) ) print "hauteur: %s" % ( hex(self.height ) ) print "nombre de plans: %s" % ( hex(self.plans ) ) print "bits par pixel: %s" % ( hex(self.bpp ) ) print "type compression: %s" % ( hex(self.compType ) ) print "taille(+padding): %s" % ( hex(self.size ) ) print "horizontal resol: %s" % ( hex(self.horiRes ) ) print "vertical resol: %s" % ( hex(self.vertRes ) ) print "nombre de couleur: %s" % ( hex(self.colorNb ) ) print "nb couleurs impor: %s" % ( hex(self.colorINb ) ) print "=====================" print "INFORMATIONS COMP." print "=====================" print "padding: %s" % ( (4- self.width*3 %4)%4 ) print "=====================" #################################################### # classe qui parse le content (binaire) en matrice # #################################################### class BMPContent: # CONSTRUCTEUR: instancie les attribus def __init__(self): self.map = [] self.bin = "" self.readableData = "" # parse le content (bin) avec les informations: #
BMPHeader de l'image en question def parse(self, binContent, header): # gestion du bpp if( header.bpp != 24 ): print "Ne prends pas en charge les versions autre que bmp24"; exit() # taille avec un padding de 1 padding = ( 4 - header.width*3 % 4 ) % 4 correctSize = header.width * ( padding + header.bpp/8 ) * header.height; # si le fichier a une mauvaise taille donc mauvais format if not len(binContent) == correctSize: print "Mauvais format (erreur de taille)" exit() # attribution de la map self.map = [] i = 0 for line in range(0, header.height): self.map.append( [] ) # on créé la colonne for pix in range(0, header.width): self.map[line].append( # on ajoute le pixel à la ligne RGBPixel( ord( binContent[i+2] ), # rouge ord( binContent[i+1] ), # vert ord( binContent[i+0] ) # bleu ) ); i += 3 # on passe à la suite i += padding # on saute le padding de saut de ligne self.map = self.map[::-1] # on inverse les lignes self.bin = binContent for byte in binContent: self.readableData += str(ord(byte)) + " " # unparse une map de pixels en binaire def unparse(self, map): self.map = map height = len( map ) # nb de lignes = taille de la map width = len( map[0] ) # nb de colonnes = taille des lignes de la map padding = ( 4 - width*3 % 4 ) % 4 # padding de bourrage de lignes self.bin = "" for line in self.map[::-1]: for pixel in line: self.bin += unichr(pixel.b) + unichr(pixel.g) + unichr(pixel.r) for zero in range(0, padding): self.bin += unichr(0) ################################# # classe contenant un pixel RGB # ################################# class RGBPixel: def __init__(self, r, g, b): self.r = r self.g = g self.b = b #################################################### # classe qui parse un fichier BMP complet en objet # #################################################### class BMPFile: # parse à partir du nom du fichier en > objets et def parse(self, filename): # si on a défini le fichier if filename == "": print "Missing argument 1: filename.bmp" exit() # gestion du format if not ".bmp" in filename[-4:]: print "must be a .bmp file" exit() self.bin = "" self.readableData = "" # lecture du fichier with open( filename ) as file: for byte in file.read(): self.bin += byte; self.readableData += str(hex(ord(byte))) + " " headerSize = 54 # parsing header self.header = BMPHeader() self.header.parse( self.bin[:headerSize] ) # parsing content self.content = BMPContent() self.content.parse( self.bin[self.header.offset:], self.header ) # unparse à partir d'un et d'un def unparse(self): print "To Do !" print "implement unparse function"