
from DataController import DataController
from DataReader import DataReader
from vtk import vtkMetaImageWriter
from vtk import vtkXMLImageDataWriter


class DataWriter(DataController):
	"""
	DataWriter writes an image data object to
	disk using the provided format.
	"""
	def __init__(self):
		super(DataWriter, self).__init__()

		self.supportedExtensions = [DataReader.TypeMHD,
									DataReader.TypeVTI]

	def WriteToFile(self, imageData, exportFileName, fileType):
		if fileType == DataReader.TypeMHD:
			if not exportFileName.endswith(".mhd"):
				exportFileName = exportFileName + ".mhd"
			writer = vtkMetaImageWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		elif fileType == DataReader.TypeVTI:
			writer = vtkXMLImageDataWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		else:
			raise NotImplementedError("No writing support for type " + str(fileType))


from DataController import DataController
from DataReader import DataReader
from vtk import vtkMetaImageWriter
from vtk import vtkXMLImageDataWriter


class DataWriter(DataController):
	"""
	DataWriter writes an image data object to
	disk using the provided format.
	"""
	def __init__(self):
		super(DataWriter, self).__init__()

		self.supportedExtensions = [DataReader.TypeMHD,
									DataReader.TypeVTI,
									DataReader.TypeMHA]

	def WriteToFile(self, imageData, exportFileName, fileType):
		if fileType == DataReader.TypeMHD:
			if not exportFileName.endswith(".mhd"):
				exportFileName = exportFileName + ".mhd"
			writer = vtkMetaImageWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		elif fileType == DataReader.TypeVTI:
			writer = vtkXMLImageDataWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		elif fileType == DataReader.TypeMHA:
			writer = vtkMetaImageWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		else:
			raise NotImplementedError("No writing support for type " + str(fileType))

Build in support for writing mha files.
Build in support for writing mha files.

  
  from DataController import DataController
  from DataReader import DataReader
  from vtk import vtkMetaImageWriter
  from vtk import vtkXMLImageDataWriter
  
  
  class DataWriter(DataController):
  	"""
  	DataWriter writes an image data object to
  	disk using the provided format.
  	"""
  	def __init__(self):
  		super(DataWriter, self).__init__()
  
  		self.supportedExtensions = [DataReader.TypeMHD,
- 									DataReader.TypeVTI]
+ 									DataReader.TypeVTI,
+ 									DataReader.TypeMHA]
  
  	def WriteToFile(self, imageData, exportFileName, fileType):
  		if fileType == DataReader.TypeMHD:
  			if not exportFileName.endswith(".mhd"):
  				exportFileName = exportFileName + ".mhd"
  			writer = vtkMetaImageWriter()
  			writer.SetFileName(exportFileName)
  			writer.SetInputData(imageData)
  			writer.Write()
  		elif fileType == DataReader.TypeVTI:
  			writer = vtkXMLImageDataWriter()
  			writer.SetFileName(exportFileName)
  			writer.SetInputData(imageData)
  			writer.Write()
+ 		elif fileType == DataReader.TypeMHA:
+ 			writer = vtkMetaImageWriter()
+ 			writer.SetFileName(exportFileName)
+ 			writer.SetInputData(imageData)
+ 			writer.Write()
  		else:
  			raise NotImplementedError("No writing support for type " + str(fileType))
  
Build in support for writing mha files.
## Code Before:

from DataController import DataController
from DataReader import DataReader
from vtk import vtkMetaImageWriter
from vtk import vtkXMLImageDataWriter


class DataWriter(DataController):
	"""
	DataWriter writes an image data object to
	disk using the provided format.
	"""
	def __init__(self):
		super(DataWriter, self).__init__()

		self.supportedExtensions = [DataReader.TypeMHD,
									DataReader.TypeVTI]

	def WriteToFile(self, imageData, exportFileName, fileType):
		if fileType == DataReader.TypeMHD:
			if not exportFileName.endswith(".mhd"):
				exportFileName = exportFileName + ".mhd"
			writer = vtkMetaImageWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		elif fileType == DataReader.TypeVTI:
			writer = vtkXMLImageDataWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		else:
			raise NotImplementedError("No writing support for type " + str(fileType))

## Instruction:
Build in support for writing mha files.
## Code After:

from DataController import DataController
from DataReader import DataReader
from vtk import vtkMetaImageWriter
from vtk import vtkXMLImageDataWriter


class DataWriter(DataController):
	"""
	DataWriter writes an image data object to
	disk using the provided format.
	"""
	def __init__(self):
		super(DataWriter, self).__init__()

		self.supportedExtensions = [DataReader.TypeMHD,
									DataReader.TypeVTI,
									DataReader.TypeMHA]

	def WriteToFile(self, imageData, exportFileName, fileType):
		if fileType == DataReader.TypeMHD:
			if not exportFileName.endswith(".mhd"):
				exportFileName = exportFileName + ".mhd"
			writer = vtkMetaImageWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		elif fileType == DataReader.TypeVTI:
			writer = vtkXMLImageDataWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		elif fileType == DataReader.TypeMHA:
			writer = vtkMetaImageWriter()
			writer.SetFileName(exportFileName)
			writer.SetInputData(imageData)
			writer.Write()
		else:
			raise NotImplementedError("No writing support for type " + str(fileType))
