mitk::NavigationTool::Pointer mitk::NavigationToolReader::DoRead(std::string filename)
{
//decompress all files into a temporary directory
std::ifstream file(filename.c_str(), std::ios::binary);
if (!file.good())
{
m_ErrorMessage = "Cannot open '" + filename + "' for reading";
return nullptr;
}
std::string tempDirectory = m_ToolfilePath + GetFileWithoutPath(filename);
Poco::Zip::Decompress unzipper(file, Poco::Path(tempDirectory));
unzipper.decompressAllFiles();
//use SceneSerialization to load the DataStorage
mitk::SceneIO::Pointer mySceneIO = mitk::SceneIO::New();
mitk::DataStorage::Pointer loadedStorage = mySceneIO->LoadScene(tempDirectory + Poco::Path::separator() + GetFileWithoutPath(filename) + ".storage");
if (loadedStorage->GetAll()->size() == 0 || loadedStorage.IsNull())
{
m_ErrorMessage = "Invalid file: cannot parse tool data.";
return nullptr;
}
//convert the DataStorage back to a NavigationTool-Object
mitk::DataNode::Pointer myNode = loadedStorage->GetAll()->ElementAt(0);
mitk::NavigationTool::Pointer returnValue = ConvertDataNodeToNavigationTool(myNode, tempDirectory);
//delete the data-storage file which is not needed any more. The toolfile must be left in the temporary directory becauses it is linked in the datatreenode of the tool
std::remove((std::string(tempDirectory + Poco::Path::separator() + GetFileWithoutPath(filename) + ".storage")).c_str());
return returnValue;
}
mitk::DataNode::Pointer mitk::NavigationToolWriter::ConvertToDataNode(mitk::NavigationTool::Pointer Tool)
{
mitk::DataNode::Pointer thisTool = mitk::DataNode::New();
//Name
if (Tool->GetDataNode().IsNull()) thisTool->SetName("none");
else thisTool->SetName(Tool->GetDataNode()->GetName().c_str());
//Identifier
thisTool->AddProperty("identifier",mitk::StringProperty::New(Tool->GetIdentifier().c_str()));
//Serial Number
thisTool->AddProperty("serial number",mitk::StringProperty::New(Tool->GetSerialNumber().c_str()));
//Tracking Device
thisTool->AddProperty("tracking device type",mitk::IntProperty::New(Tool->GetTrackingDeviceType()));
//Tool Type
thisTool->AddProperty("tracking tool type",mitk::IntProperty::New(Tool->GetType()));
//Calibration File Name
thisTool->AddProperty("toolfileName",mitk::StringProperty::New(GetFileWithoutPath(Tool->GetCalibrationFile())));
//Surface
if (Tool->GetDataNode().IsNotNull()) if (Tool->GetDataNode()->GetData()!=NULL) thisTool->SetData(Tool->GetDataNode()->GetData());
//Tool Landmarks
thisTool->AddProperty("ToolRegistrationLandmarks",mitk::StringProperty::New(ConvertPointSetToString(Tool->GetToolRegistrationLandmarks())));
thisTool->AddProperty("ToolCalibrationLandmarks",mitk::StringProperty::New(ConvertPointSetToString(Tool->GetToolCalibrationLandmarks())));
//Tool Tip
thisTool->AddProperty("ToolTipPosition",mitk::StringProperty::New(ConvertPointToString(Tool->GetToolTipPosition())));
thisTool->AddProperty("ToolTipOrientation",mitk::StringProperty::New(ConvertQuaternionToString(Tool->GetToolTipOrientation())));
//Material is not needed, to avoid errors in scene serialization we have to do this:
thisTool->ReplaceProperty("material",NULL);
return thisTool;
}
bool mitk::NavigationToolWriter::DoWrite(std::string FileName,mitk::NavigationTool::Pointer Tool)
{
//convert whole data to a mitk::DataStorage
mitk::StandaloneDataStorage::Pointer saveStorage = mitk::StandaloneDataStorage::New();
mitk::DataNode::Pointer thisTool = ConvertToDataNode(Tool);
saveStorage->Add(thisTool);
//use SceneSerialization to save the DataStorage
std::string DataStorageFileName = mitk::StandardFileLocations::GetInstance()->GetOptionDirectory() + Poco::Path::separator() + GetFileWithoutPath(FileName) + ".storage";
mitk::SceneIO::Pointer mySceneIO = mitk::SceneIO::New();
mySceneIO->SaveScene(saveStorage->GetAll(),saveStorage,DataStorageFileName);
//now put the DataStorage and the Toolfile in a ZIP-file
std::ofstream file( FileName.c_str(), std::ios::binary | std::ios::out);
if (!file.good())
{
m_ErrorMessage = "Could not open a zip file for writing: '" + FileName + "'";
return false;
}
else
{
Poco::Zip::Compress zipper( file, true );
zipper.addFile(DataStorageFileName,GetFileWithoutPath(DataStorageFileName));
if (Tool->GetCalibrationFile()!="none") zipper.addFile(Tool->GetCalibrationFile(),GetFileWithoutPath(Tool->GetCalibrationFile()));
zipper.close();
}
//delete the data storage
std::remove(DataStorageFileName.c_str());
return true;
}