denoising.py/code/classes.py

125 lines
4.1 KiB
Python
Raw Normal View History

2015-09-06 14:01:17 +00:00
# ~*~ encoding: utf-8 ~*~ #
2015-09-06 17:06:44 +00:00
import sys
2015-09-06 14:28:56 +00:00
#################################################
# classe qui parse le header (binaire) en objet #
#################################################
2015-09-06 14:01:17 +00:00
class BMPHeader:
2015-09-06 14:28:56 +00:00
# convertit les octets <bytes> en entier
2015-09-06 14:01:17 +00:00
def toInt(self, bytes):
intReturn = 0;
for i, byte in enumerate(bytes):
intReturn += ord(byte) * (256 ** i)
return intReturn
# CONSTRUCTEUR: parse le header au format bin en objet
def __init__(self, binHeader):
2015-09-06 16:35:58 +00:00
self.header = binHeader # header brut (format initial: bin)
2015-09-06 14:01:17 +00:00
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)
2015-09-06 16:35:58 +00:00
2015-09-06 15:05:52 +00:00
2015-09-06 14:28:56 +00:00
####################################################
# classe qui parse le content (binaire) en matrice #
####################################################
class BMPContent:
# CONSTRUCTEUR: parse le content (bin) <binContent> avec les informations:
2015-09-06 15:05:52 +00:00
# <header> BMPHeader de l'image en question
def __init__(self, binContent, header):
2015-09-06 14:28:56 +00:00
# gestion du bpp
2015-09-06 15:05:52 +00:00
if( header.bpp != 24 ):
2015-09-06 14:28:56 +00:00
print "ne prends pas en charge les versions autre que bmp24";
exit
2015-09-06 16:35:58 +00:00
# taille avec un padding de 1
correctSizes = [
( 1 + header.width * header.bpp/8 ) * header.height,
( 2 + header.width * header.bpp/8 ) * header.height
]
# si le fichier a une mauvaise taille donc mauvais format
2015-09-06 17:06:44 +00:00
padding = 0
2015-09-06 16:35:58 +00:00
if not len(binContent) in correctSizes:
print "Mauvais format"
exit
2015-09-06 17:06:44 +00:00
elif len(binContent) == correctSizes[0]:
padding = 1
else:
padding = 2
2015-09-06 16:35:58 +00:00
# attribution de la map
self.map = []
i = 0
2015-09-06 14:28:56 +00:00
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
2015-09-06 17:06:44 +00:00
i += padding # on saute le padding de saut de ligne
self.map = self.map[::-1] # on inverse les lignes
#################################
# classe contenant un pixel RGB #
#################################
class RGBPixel:
def __init__(self, r, g, b):
self.r = r
self.g = g
2015-09-06 17:06:44 +00:00
self.b = b
####################################################
# classe qui parse un fichier BMP complet en objet #
####################################################
class BMPFile:
# CONSTRUCTEUR #
# à partir du fichier <filename>
# parse le header dans une classe <BMPHeader>
# parse le contenu dans une classe <BMPContent>
def __init__(self, filename):
# gestion du format
if not ".bmp" in filename[-4:]:
print "must be a .bmp file"
exit
self.fileData = ""
self.readableData = ""
with open(sys.argv[1]) as file:
for byte in file.read():
self.fileData += byte;
self.readableData += str(ord(byte)) + " "
headerSize = 54
self.header = BMPHeader ( self.fileData[:headerSize] )
self.content = BMPContent( self.fileData[self.header.offset:], self.header )
self.map = self.content.map # lien