bool mitk::NavigationToolStorage::AddTool(mitk::NavigationTool::Pointer tool)
{
if (m_storageLocked)
{
MITK_WARN << "Storage is locked, cannot modify it!";
return false;
}
else if (GetTool(tool->GetIdentifier()).IsNotNull())
{
MITK_WARN << "Tool ID already exists in storage, can't add!";
return false;
}
else
{
m_ToolCollection.push_back(tool);
if (m_DataStorage.IsNotNull())
{
if (!m_DataStorage->Exists(tool->GetDataNode()))
m_DataStorage->Add(tool->GetDataNode());
}
//This line is important so that other widgets can get a notice that the toolStorage has changed!
this->UpdateMicroservice();
return true;
}
}
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;
}
bool mitk::NavigationToolStorage::AddTool(mitk::NavigationTool::Pointer tool)
{
if (GetTool(tool->GetIdentifier()).IsNotNull()) return false;
else
{
m_ToolCollection.push_back(tool);
if(m_DataStorage.IsNotNull())
{
if (!m_DataStorage->Exists(tool->GetDataNode()))
m_DataStorage->Add(tool->GetDataNode());
}
return true;
}
}
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;
}
void QmitkNavigationToolCreationWidget::SetDefaultData(mitk::NavigationTool::Pointer DefaultTool)
{
//Set Members. This can either be the new initialized tool from call of Initialize() or a tool which already exists in the toolStorage
m_ToolToBeEdited = DefaultTool->Clone();
//Set all gui variables
SetGuiElements();
}
void QmitkNavigationToolCreationWidget::SetDefaultData(mitk::NavigationTool::Pointer DefaultTool)
{
m_Controls->m_ToolNameEdit->setText(QString(DefaultTool->GetDataNode()->GetName().c_str()));
m_Controls->m_IdentifierEdit->setText(QString(DefaultTool->GetIdentifier().c_str()));
m_Controls->m_SerialNumberEdit->setText(QString(DefaultTool->GetSerialNumber().c_str()));
switch(DefaultTool->GetTrackingDeviceType())
{
case mitk::NDIAurora:
m_Controls->m_TrackingDeviceTypeChooser->setCurrentIndex(0);break;
case mitk::NDIPolaris:
m_Controls->m_TrackingDeviceTypeChooser->setCurrentIndex(1);break;
case mitk::ClaronMicron:
m_Controls->m_TrackingDeviceTypeChooser->setCurrentIndex(2);break;
default:
m_Controls->m_TrackingDeviceTypeChooser->setCurrentIndex(0);
}
m_Controls->m_CalibrationFileName->setText(QString(DefaultTool->GetCalibrationFile().c_str()));
m_Controls->m_Surface_Use_Other->setChecked(true);
switch(DefaultTool->GetType())
{
case mitk::NavigationTool::Instrument:
m_Controls->m_ToolTypeChooser->setCurrentIndex(0); break;
case mitk::NavigationTool::Fiducial:
m_Controls->m_ToolTypeChooser->setCurrentIndex(1); break;
case mitk::NavigationTool::Skinmarker:
m_Controls->m_ToolTypeChooser->setCurrentIndex(2); break;
case mitk::NavigationTool::Unknown:
m_Controls->m_ToolTypeChooser->setCurrentIndex(3); break;
}
m_Controls->m_SurfaceChooser->SetSelectedNode(DefaultTool->GetDataNode());
}