/**
* Set the detector positions given the r,theta and phi.
* @param detID :: A vector of detector IDs
* @param l2 :: A vector of l2 distances
* @param theta :: A vector of theta distances
* @param phi :: A vector of phi values
*/
void UpdateInstrumentFromFile::setDetectorPositions(
const std::vector<int32_t> &detID, const std::vector<float> &l2,
const std::vector<float> &theta, const std::vector<float> &phi) {
const int numDetector = static_cast<int>(detID.size());
g_log.information() << "Setting new positions for " << numDetector
<< " detectors\n";
auto &detectorInfo = m_workspace->mutableDetectorInfo();
for (int i = 0; i < numDetector; ++i) {
try {
auto index = detectorInfo.indexOf(detID[i]);
double p{phi[i]};
if (m_ignorePhi) {
double r, t;
detectorInfo.position(index).getSpherical(r, t, p);
}
setDetectorPosition(detectorInfo, index, l2[i], theta[i],
static_cast<float>(p));
} catch (std::out_of_range &) {
// Invalid detID[i]
continue;
}
progress(static_cast<double>(i) / numDetector,
"Updating Detector Positions from File");
}
}
/**
* Set the detector positions given the r,theta and phi.
* @param detID :: A vector of detector IDs
* @param l2 :: A vector of l2 distances
* @param theta :: A vector of theta distances
* @param phi :: A vector of phi values
*/
void UpdateInstrumentFromFile::setDetectorPositions(const std::vector<int32_t> & detID, const std::vector<float> & l2,
const std::vector<float> & theta, const std::vector<float> & phi)
{
Geometry::Instrument_const_sptr inst = m_workspace->getInstrument();
const int numDetector = static_cast<int>(detID.size());
g_log.information() << "Setting new positions for " << numDetector << " detectors\n";
for (int i = 0; i < numDetector; ++i)
{
try
{
Geometry::IDetector_const_sptr det = inst->getDetector(detID[i]);
setDetectorPosition(det, l2[i], theta[i], phi[i]);
}
catch (Kernel::Exception::NotFoundError&)
{
continue;
}
progress(static_cast<double>(i)/numDetector,"Updating Detector Positions from File");
}
}
/** Executes the algorithm. Moving detectors of input workspace to positions indicated in table workspace
*
* @throw FileError Thrown if unable to get instrument from workspace,
* table workspace is incompatible with instrument
*/
void ApplyCalibration::exec()
{
// Get pointers to the workspace, parameter map and table
API::MatrixWorkspace_sptr inputWS = getProperty("Workspace");
m_pmap = &(inputWS->instrumentParameters()); // Avoids a copy if you get the reference before the instrument
API::ITableWorkspace_sptr PosTable = getProperty("PositionTable");
Geometry::Instrument_const_sptr instrument = inputWS->getInstrument();
if(!instrument)
{
throw std::runtime_error("Workspace to apply calibration to has no defined instrument");
}
size_t numDetector = PosTable->rowCount();
ColumnVector<int> detID = PosTable->getVector("Detector ID");
ColumnVector<V3D> detPos = PosTable->getVector("Detector Position");
// numDetector needs to be got as the number of rows in the table and the detID got from the (i)th row of table.
for (size_t i = 0; i < numDetector; ++i)
{
setDetectorPosition(instrument, detID[i], detPos[i], false );
}
// Ensure pointer is only valid for execution
m_pmap = NULL;
}
/**
* Updates from a more generic ascii file
* @param filename :: The input filename
*/
void UpdateInstrumentFromFile::updateFromAscii(const std::string & filename)
{
AsciiFileHeader header;
const bool isSpectrum = parseAsciiHeader(header);
Geometry::Instrument_const_sptr inst = m_workspace->getInstrument();
// Throws for multiple detectors
const spec2index_map specToIndex(m_workspace->getSpectrumToWorkspaceIndexMap());
std::ifstream datfile(filename.c_str(), std::ios_base::in);
const int skipNLines = getProperty("SkipFirstNLines");
std::string line;
int lineCount(0);
while(lineCount < skipNLines)
{
std::getline(datfile,line);
++lineCount;
}
std::vector<double> colValues(header.colCount - 1, 0.0);
while(std::getline(datfile,line))
{
boost::trim(line);
std::istringstream is(line);
// Column 0 should be ID/spectrum number
int32_t detOrSpec(-1000);
is >> detOrSpec;
// If first thing read is not a number then skip the line
if(is.fail())
{
g_log.debug() << "Skipping \"" << line << "\". Cannot interpret as list of numbers.\n";
continue;
}
Geometry::IDetector_const_sptr det;
try
{
if(isSpectrum)
{
auto it = specToIndex.find(detOrSpec);
if(it != specToIndex.end())
{
const size_t wsIndex = it->second;
det = m_workspace->getDetector(wsIndex);
}
else
{
g_log.debug() << "Skipping \"" << line << "\". Spectrum is not in workspace.\n";
continue;
}
}
else
{
det = inst->getDetector(detOrSpec);
}
}
catch(Kernel::Exception::NotFoundError&)
{
g_log.debug() << "Skipping \"" << line << "\". Spectrum in workspace but cannot find associated detector.\n";
continue;
}
// Special cases for detector r,t,p. Everything else is
// attached as an detector parameter
double R(0.0),theta(0.0), phi(0.0);
for(size_t i = 1; i < header.colCount; ++i)
{
double value(0.0);
is >> value;
if(i < header.colCount - 1 && is.eof())
{
//If stringstream is at EOF & we are not at the last column then
// there aren't enought columns in the file
throw std::runtime_error("UpdateInstrumentFromFile::updateFromAscii - "
"File contains fewer than expected number of columns, check AsciiHeader property.");
}
if(i == header.rColIdx) R = value;
else if(i == header.thetaColIdx) theta = value;
else if(i == header.phiColIdx) phi = value;
else if(header.detParCols.count(i) == 1)
{
Geometry::ParameterMap & pmap = m_workspace->instrumentParameters();
pmap.addDouble(det->getComponentID(), header.colToName[i],value);
}
}
// Check stream state. stringstream::EOF should have been reached, if not then there is still more to
// read and the file has more columns than the header indicated
if(!is.eof())
{
throw std::runtime_error("UpdateInstrumentFromFile::updateFromAscii - "
"File contains more than expected number of columns, check AsciiHeader property.");
}
// If not supplied use current values
double r,t,p;
det->getPos().getSpherical(r,t,p);
if(header.rColIdx == 0) R = r;
if(header.thetaColIdx == 0) theta = t;
if(header.phiColIdx == 0) phi = p;
setDetectorPosition(det, static_cast<float>(R), static_cast<float>(theta), static_cast<float>(phi));
}
}
/**
* Updates from a more generic ascii file
* @param filename :: The input filename
*/
void UpdateInstrumentFromFile::updateFromAscii(const std::string &filename) {
AsciiFileHeader header;
const bool isSpectrum = parseAsciiHeader(header);
// Throws for multiple detectors
const spec2index_map specToIndex(
m_workspace->getSpectrumToWorkspaceIndexMap());
std::ifstream datfile(filename.c_str(), std::ios_base::in);
const int skipNLines = getProperty("SkipFirstNLines");
std::string line;
int lineCount(0);
while (lineCount < skipNLines) {
std::getline(datfile, line);
++lineCount;
}
Geometry::ParameterMap &pmap = m_workspace->instrumentParameters();
auto &detectorInfo = m_workspace->mutableDetectorInfo();
const auto &spectrumInfo = m_workspace->spectrumInfo();
std::vector<double> colValues(header.colCount - 1, 0.0);
while (std::getline(datfile, line)) {
boost::trim(line);
std::istringstream is(line);
// Column 0 should be ID/spectrum number
int32_t detOrSpec(-1000);
is >> detOrSpec;
// If first thing read is not a number then skip the line
if (is.fail()) {
g_log.debug() << "Skipping \"" << line
<< "\". Cannot interpret as list of numbers.\n";
continue;
}
bool skip{false};
size_t index = static_cast<size_t>(-1);
const Geometry::IDetector *det{nullptr};
if (isSpectrum) {
auto it = specToIndex.find(detOrSpec);
if (it != specToIndex.end()) {
index = it->second;
if (spectrumInfo.hasDetectors(index)) {
det = &spectrumInfo.detector(index);
} else {
skip = true;
}
} else {
g_log.debug() << "Skipping \"" << line
<< "\". Spectrum is not in workspace.\n";
continue;
}
} else {
try {
index = detectorInfo.indexOf(detOrSpec);
det = &detectorInfo.detector(index);
} catch (std::out_of_range &) {
skip = true;
}
}
if (skip || index == static_cast<size_t>(-1)) {
g_log.debug()
<< "Skipping \"" << line
<< "\". Spectrum in workspace but cannot find associated detector.\n";
continue;
}
std::vector<size_t> indices;
if (isSpectrum) {
if (auto group = dynamic_cast<const Geometry::DetectorGroup *>(det)) {
for (const auto detID : group->getDetectorIDs())
indices.push_back(detectorInfo.indexOf(detID));
} else {
indices.push_back(detectorInfo.indexOf(det->getID()));
}
} else {
indices.push_back(index);
}
// Special cases for detector r,t,p. Everything else is
// attached as an detector parameter
double R(0.0), theta(0.0), phi(0.0);
for (size_t i = 1; i < header.colCount; ++i) {
double value(0.0);
is >> value;
if (i < header.colCount - 1 && is.eof()) {
// If stringstream is at EOF & we are not at the last column then
// there aren't enought columns in the file
throw std::runtime_error("UpdateInstrumentFromFile::updateFromAscii - "
"File contains fewer than expected number of "
"columns, check AsciiHeader property.");
}
if (i == header.rColIdx)
R = value;
else if (i == header.thetaColIdx)
theta = value;
else if (i == header.phiColIdx)
phi = value;
else if (header.detParCols.count(i) == 1) {
for (const auto index : indices) {
auto id = detectorInfo.detector(index).getComponentID();
pmap.addDouble(id, header.colToName[i], value);
}
}
}
// Check stream state. stringstream::EOF should have been reached, if not
// then there is still more to
// read and the file has more columns than the header indicated
if (!is.eof()) {
throw std::runtime_error("UpdateInstrumentFromFile::updateFromAscii - "
"File contains more than expected number of "
"columns, check AsciiHeader property.");
}
// If not supplied use current values
double r, t, p;
if (isSpectrum)
spectrumInfo.position(index).getSpherical(r, t, p);
else
detectorInfo.position(index).getSpherical(r, t, p);
if (header.rColIdx == 0)
R = r;
if (header.thetaColIdx == 0)
theta = t;
if (header.phiColIdx == 0 || m_ignorePhi)
phi = p;
for (const auto index : indices)
setDetectorPosition(detectorInfo, index, static_cast<float>(R),
static_cast<float>(theta), static_cast<float>(phi));
}
}
