def decode(img, args): if (args.format == "nifti"): data = img.get_data() # data = affine.nifti2dicom(data) elif (args.format == "dicom"): data = img.pixel_array data = img.get_data() data = data.ravel() if (args.mode == "lsb"): msgDataList = lsb.decode(data) return lsb.listToText(msgDataList) elif (args.mode == "haar"): return "Haar not implemented yet"
def testEncodeAndDecodeTile(self): data = numpy.zeros(1000, dtype=numpy.uint32) encData = lsbstego.encode(data, self.msg, "tile") decMsgList = lsbstego.decode(encData) decMsg = lsbstego.listToText(decMsgList) self.assertEqual(decMsg, "[{0}]".format(self.msg))
def testEncodeAndDecodeOnRandomNotile(self): data = numpy.random.randint(0, high=256, size=1000).astype(numpy.uint32) encData = lsbstego.encode(data, self.msg, "notile") decMsgList = lsbstego.decode(encData) decMsg = lsbstego.listToText(decMsgList) self.assertEqual(decMsg, "[{0}]".format(self.msg))