def __createImageDescription(self, scene, renderFile):
'''
Initializes the instance of @see: ImageDescription class
and saves it into file
'''
# creating the object
name = renderFile.replace(".py", self.generatorDescription.outputFormat)
translate = scene.camera.translate
rotate = scene.camera.rotate
img = ImageDescription(name=name,
x=translate[0],
y=translate[1],
z=translate[2],
p=rotate[0],
q=rotate[1],
r=rotate[2],
points=[])
self.logger.debug("Image description object created \n'%s'" % img)
# saving to file
fileName = renderFile.replace(".py", ".imd")
fileName = fileName.replace("scripts", self.generatorDescription.outputFolder)
writer = ImageDescriptionWriter()
with open(fileName, 'w') as fileStream :
writer.write(stream=fileStream, imageDescription=img)
self.logger.info("Image description written to '%s'" % fileName)
def testOutputWithAllElement(self):
'''
tests if the output stream is good when parameters passed
are good
'''
# get the image description and the writer
imageDesc = ImageDescription(self.name, self.x, self.y, self.z,
self.p, self.q, self.r, self.points)
imageWriter = ImageDescriptionWriter()
#get the string steam
with open(self.fileName, 'r+') as fileStream:
imageWriter.write(fileStream, imageDesc)
# read this file
with open(self.fileName, 'r+') as fileStream:
allLines = ''.join(fileStream.readlines())
fileStream.close()
self.assertEqual(self.expectedString, allLines , "Provided strings should be identical")
