Get pixel dimensions

Add args to input width and height of the image
This commit is contained in:
Marc-Antoine Lafreniere
2019-02-16 20:16:47 -05:00
parent ef266883ec
commit ff7a860c2d
4 changed files with 62 additions and 8 deletions

View File

@@ -2,10 +2,10 @@ import math
class FileUtils:
@staticmethod
def pbmToCsv(pbmFile, dimensionLineIndex = 2):
def pbmToMatrix(pbmFilePath, dimensionLineIndex = 2):
completeFile = []
file = open(pbmFile, 'r')
file = open(pbmFilePath, 'r')
lines = file.readlines()
width, height = (int(val) for val in lines[dimensionLineIndex].split())
file.close()
@@ -18,7 +18,7 @@ class FileUtils:
for index, value in enumerate(completeFile):
matrix[int(math.floor(index / width))][index % width] = value
return matrix
return matrix, height, width
@staticmethod
def saveMatrixToFile(matrix, filePath):
@@ -27,4 +27,17 @@ class FileUtils:
for y in x:
f.write('%s ' % y )
f.write('\n')
f.close()
f.close()
@staticmethod
def getPixelSize(matHeight, matWidth, pcbHeight, pcbWidth, unit = 'mm'):
if unit == 'mm':
return pcbHeight / matHeight, pcbWidth / matWidth
elif unit == 'cm':
return pcbHeight / matHeight * 10, pcbWidth / matWidth * 10
elif unit == 'm':
return pcbHeight / matHeight * 100, pcbWidth / matWidth * 100
elif unit == 'in':
return pcbHeight / matHeight * 25.4, pcbWidth / matWidth * 25.4
else:
raise RuntimeError('Unit not handle')