tensor_op_transform.hpp

 
#ifndef EXATN_NUMERICS_TENSOR_OP_TRANSFORM_HPP_
#define EXATN_NUMERICS_TENSOR_OP_TRANSFORM_HPP_
 
#include "Identifiable.hpp"
 
#include "tensor_basic.hpp"
#include "tensor_operation.hpp"
 
#include "tensor_method.hpp"
 
namespace exatn{
 
namespace numerics{
 
class TensorOpTransform: public TensorOperation{
public:
 
 TensorOpTransform();
 
 TensorOpTransform(const TensorOpTransform &) = default;
 TensorOpTransform & operator=(const TensorOpTransform &) = default;
 TensorOpTransform(TensorOpTransform &&) noexcept = default;
 TensorOpTransform & operator=(TensorOpTransform &&) noexcept = default;
 virtual ~TensorOpTransform() = default;
 
 virtual bool isSet() const override;
 
 virtual int accept(runtime::TensorNodeExecutor & node_executor,
                    runtime::TensorOpExecHandle * exec_handle) override;
 
 static std::unique_ptr<TensorOperation> createNew();
 
 void resetFunctor(std::shared_ptr<talsh::TensorFunctor<Identifiable>> functor){
  functor_ = functor;
  return;
 }
 
 int apply(talsh::Tensor & local_tensor){
  if(functor_) return functor_->apply(local_tensor);
  return 0;
 }
 
private:
 
 std::shared_ptr<talsh::TensorFunctor<Identifiable>> functor_; //tensor functor (method)
 
};
 
} //namespace numerics
 
} //namespace exatn
 
#endif //EXATN_NUMERICS_TENSOR_OP_TRANSFORM_HPP_