DataProcessorAlgorithm.h 4.9 KB
Newer Older
1
2
3
4
5
#ifndef MANTID_API_DATAPROCESSORALGORITHM_H_
#define MANTID_API_DATAPROCESSORALGORITHM_H_

#include "MantidKernel/System.h"
#include "MantidAPI/Algorithm.h"
6
#include "MantidAPI/AlgorithmManager.h"
7
#include "MantidAPI/ITableWorkspace.h"
8
#include "MantidAPI/IEventWorkspace.h"
9
#include "MantidKernel/PropertyManager.h"
10
11
12
13
14
15
#include <vector>

namespace Mantid
{
namespace API
{
16
17
/**

18
19
   Data processor algorithm to be used as a parent to workflow algorithms.
   This algorithm provides utility methods to load and process data.
20
21
22

   @date 2012-04-04

23
   Copyright &copy; 2012 ISIS Rutherford Appleton Laboratory, NScD Oak Ridge National Laboratory & European Spallation Source
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

   This file is part of Mantid.

   Mantid is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 3 of the License, or
   (at your option) any later version.

   Mantid is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.

40
   File change history is stored at: <https://github.com/mantidproject/mantid>
41
42
43
44
45
46
47
48
49
   Code Documentation is available at: <http://doxygen.mantidproject.org>
 */
class DLLExport DataProcessorAlgorithm : public Algorithm
{
public:
  DataProcessorAlgorithm();
  virtual ~DataProcessorAlgorithm();

protected:
50
  virtual boost::shared_ptr<Algorithm> createChildAlgorithm(const std::string& name, const double startProgress = -1.,
51
      const double endProgress = -1., const bool enableLogging=true, const int& version = -1);
52
  void setLoadAlg(const std::string & alg);
53
  void setLoadAlgFileProp(const std::string & filePropName);
54
55
56
  void setAccumAlg(const std::string & alg);
  ITableWorkspace_sptr determineChunk();
  void loadChunk();
57
  Workspace_sptr load(const std::string &inputData, const bool loadQuiet = false);
58
59
  std::vector<std::string> splitInput(const std::string & input);
  void forwardProperties();
60
  boost::shared_ptr<Kernel::PropertyManager> getProcessProperties(const std::string &propertyManager);
61
62
63
  /// MPI option. If false, we will use one job event if MPI is available
  bool m_useMPI;
  Workspace_sptr assemble(const std::string &partialWSName, const std::string &outputWSName);
64
65
66
  void saveNexus(const std::string &outputWSName, const std::string &outputFile);
  bool isMainThread();
  int getNThreads();
67

68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
  /// Divide a matrix workspace by another matrix workspace 
  MatrixWorkspace_sptr divide(const MatrixWorkspace_sptr lhs, const MatrixWorkspace_sptr rhs);
  /// Divide a matrix workspace by a single value 
  MatrixWorkspace_sptr divide(const MatrixWorkspace_sptr lhs, const double& rhsValue);
  
  /// Multiply a matrix workspace by another matrix workspace 
  MatrixWorkspace_sptr multiply(const MatrixWorkspace_sptr lhs, const MatrixWorkspace_sptr rhs);
  /// Multiply a matrix workspace by a single value 
  MatrixWorkspace_sptr multiply(const MatrixWorkspace_sptr lhs, const double& rhsValue);
  
  /// Add a matrix workspace to another matrix workspace 
  MatrixWorkspace_sptr plus(const MatrixWorkspace_sptr lhs, const MatrixWorkspace_sptr rhs);
  /// Add a single value to a matrix workspace
  MatrixWorkspace_sptr plus(const MatrixWorkspace_sptr lhs, const double& rhsValue);
  
  /// Subract a matrix workspace by another matrix workspace 
  MatrixWorkspace_sptr minus(const MatrixWorkspace_sptr lhs, const MatrixWorkspace_sptr rhs);
  /// Subract a single value from a matrix workspace 
  MatrixWorkspace_sptr minus(const MatrixWorkspace_sptr lhs, const double& rhsValue);

88
private:
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
  template<typename LHSType, typename RHSType, typename ResultType>
  ResultType executeBinaryAlgorithm(const std::string & algorithmName, const LHSType lhs, const RHSType rhs)
  {
    auto alg = createChildAlgorithm(algorithmName);
    alg->initialize();

    alg->setProperty<LHSType>("LHSWorkspace",lhs);
    alg->setProperty<RHSType>("RHSWorkspace",rhs);
    alg->execute();

    if (alg->isExecuted())
    {
      // Get the output workspace property
      return alg->getProperty("OutputWorkspace");
    }
    else
    {
      std::string message = "Error while executing operation: " + algorithmName;
      throw std::runtime_error(message);
    }
  }

  /// Create a matrix workspace from a single number
  MatrixWorkspace_sptr createWorkspaceSingleValue(const double& rhsValue);

114
115
116
117
  /// The name of the algorithm to invoke when loading data
  std::string m_loadAlg;
  /// The name of the algorithm to invoke when accumulating data chunks
  std::string m_accumulateAlg;
118
119
  /// An alternate filename property for the load algorithm
  std::string m_loadAlgFileProp;
120
121
122
123
124
125
};

} // namespace API
} // namespace Mantid

#endif  /* MANTID_API_DATAPROCESSORALGORITHM_H_ */