USReconstructInputData USSavingRecorder::getDataForStream(unsigned videoRecorderIndex)
{
if (!mSession)
return USReconstructInputData();
if (videoRecorderIndex>=mVideoRecorder.size())
return USReconstructInputData();
SavingVideoRecorderPtr videoRecorder = mVideoRecorder[videoRecorderIndex];
videoRecorder->completeSave(); // just in case - should have been done earlier.
TimedTransformMap trackerRecordedData = RecordSession::getToolHistory_prMt(mRecordingTool, mSession, true);
std::map<double, cx::ToolPositionMetadata> trackerMetadata = RecordSession::getToolHistory_metadata(mRecordingTool, mSession, true);
std::map<double, cx::ToolPositionMetadata> referenceTrackerMetadata = RecordSession::getToolHistory_metadata(mReference, mSession, true);
std::cout << "----------- "
"trackerMetadata : " << trackerMetadata.size() << std::endl;
CachedImageDataContainerPtr imageData = videoRecorder->getImageData();
std::vector<TimeInfo> imageTimestamps = videoRecorder->getTimestamps();
QString streamSessionName = mSession->getDescription()+"_"+videoRecorder->getSource()->getUid();
UsReconstructionFileMakerPtr fileMaker;
fileMaker.reset(new UsReconstructionFileMaker(streamSessionName));
USReconstructInputData reconstructData = fileMaker->getReconstructData(imageData,
imageTimestamps,
trackerRecordedData,
trackerMetadata,
referenceTrackerMetadata,
mRecordingTool,
videoRecorder->getSource()->getUid(),
mDoWriteColor,
m_rMpr);
return reconstructData;
}
USReconstructInputData USSavingRecorder::getDataForStream(QString streamUid)
{
for (unsigned i=0; i<mVideoRecorder.size(); ++i)
{
if (mVideoRecorder[i]->getSource()->getUid() == streamUid)
return this->getDataForStream(i);
}
return USReconstructInputData();
}
void USAcquisition::recordStarted()
{
if (!mBase->getCurrentContext().testFlag(AcquisitionService::tUS))
return;
this->getReconstructer()->selectData(USReconstructInputData()); // clear old data in reconstructeer
ToolPtr tool = this->getServices()->tracking()->getFirstProbe();
mCore->setWriteColor(this->getWriteColor());
mCore->startRecord(mBase->getLatestSession(),
tool,
this->getServices()->tracking()->getReferenceTool(),
this->getRecordingVideoSources(tool));
}
void TemporalCalibration::selectData(QString filename)
{
mFilename = filename;
mFileData = USReconstructInputData();
if (!QFileInfo(filename).exists())
return;
UsReconstructionFileReader fileReader;
mFileData = fileReader.readAllFiles(filename);
if (!mFileData.mUsRaw)
{
reportWarning("Failed to load data from " + filename);
return;
}
report("Temporal Calibration: Successfully loaded data from " + filename);
}
void UsReconstructionImplService::clearAll()
{
mOriginalFileData = USReconstructInputData();
mOutputVolumeParams = OutputVolumeParams();
}
