item_struct.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#include "item_struct.h"
#include <cstdlib>
 
#define FAILURE 1
#define SUCCESS 0
 
template <typename T> int item_struct<T>::getItem(const std::string &item_name, T &value) {
  int n;
  long spec_no;
  // handle case of item_spectrum which means we average the array
  // item[ndet] for that particular spectrum
  n = item_name.find('_');
  if (n != std::string::npos) {
    spec_no = atol(item_name.c_str() + n + 1);
    return getItem(item_name.substr(0, n), &spec_no, 1, &value);
  } else {
    spec_no = 0;
    return getItem(item_name, &spec_no, 0, &value);
  }
}
 
template <typename T>
int item_struct<T>::getItem(const std::string &item_name, const long *spec_array, int nspec, T *lVal) {
  int i, j, n;
  const T *pVal = NULL;
  const item_t *item;
  item = findItem(item_name, false);
  if (item != NULL) {
    for (j = 0; j < (nspec == 0 ? 1 : nspec); j++) {
      lVal[j] = *(item->value);
    }
    return SUCCESS;
  }
  if (nspec == 0) {
    return FAILURE;
  }
  item = findItem(item_name, true);
  if (item != NULL) {
    if (item->dim1 == 0) {
      n = *(item->dim0);
    } else {
      n = *(item->dim0) * *(item->dim1);
    }
    if (n == m_ndet) {
      pVal = item->value;
    }
  }
  if (pVal == NULL) {
    return FAILURE;
  }
  for (j = 0; j < nspec; j++) {
    lVal[j] = 0;
    n = 0;
    for (i = 0; i < m_ndet; i++) {
      if (m_spec_array[i] == spec_array[j]) {
        lVal[j] += pVal[i];
        n++;
      }
    }
    if (n > 0) {
      lVal[j] = lVal[j] / n;
    }
  }
  return SUCCESS;
}
 
template <typename T> int item_struct<T>::getArrayItemSize(const std::string &item_name, int *dims_array, int &ndims) {
  const item_t *item;
  item = findItem(item_name, false);
  if (item == NULL) {
    item = findItem(item_name, true);
  }
  if (item != NULL) {
    if (item->dim1 == 0) {
      dims_array[0] = *(item->dim0);
      ndims = 1;
    } else {
      dims_array[0] = *(item->dim0);
      dims_array[1] = *(item->dim1);
      ndims = 2;
    }
    return SUCCESS;
  }
  return FAILURE;
}
 
template <typename T> int item_struct<T>::getArrayItem(const std::string &item_name, int nspec, T *larray) {
  const item_t *item;
  item = findItem(item_name, false);
  if (item == NULL) {
    item = findItem(item_name, true);
  }
  if (item != NULL) {
    int n;
    if (item->dim1 == 0) {
      n = *(item->dim0);
    } else {
      n = *(item->dim0) * *(item->dim1);
    }
    for (int k = 0; k < nspec; k++) {
      for (int j = 0; j < n; j++) {
        larray[j + k * n] = item->value[j];
      }
    }
    return SUCCESS;
  }
  return FAILURE;
}
 
template <typename T> int item_struct<T>::getArrayItem(const std::string &item_name, T *larray) {
  int n;
  long spec_no;
  n = item_name.find('_');
  if (n != std::string::npos) {
    spec_no = atol(item_name.c_str() + n + 1);
    return getIntArrayItem(item_name.substr(0, n), &spec_no, 1, larray);
  } else {
    spec_no = 0;
    return getIntArrayItem(item_name, &spec_no, 1, larray);
  }
}