unary_engine.h

Go to the documentation of this file.

/*
 *  This file is a part of TiledArray.
 *  Copyright (C) 2014  Virginia Tech
 *
 *  This program 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.
 *
 *  This program 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/>.
 *
 *  Justus Calvin
 *  Department of Chemistry, Virginia Tech
 *
 *  unary_engine.h
 *  Apr 1, 2014
 *
 */

#ifndef TILEDARRAY_EXPRESSIONS_UNARY_ENGINE_H__INCLUDED
#define TILEDARRAY_EXPRESSIONS_UNARY_ENGINE_H__INCLUDED

#include <TiledArray/expressions/expr_engine.h>
#include <TiledArray/dist_eval/unary_eval.h>

namespace TiledArray {
  namespace expressions {


    // Forward declarations
    template <typename> class UnaryExpr;


    template <typename Derived>
    class UnaryEngine : ExprEngine<Derived> {
    public:
      // Class hierarchy typedefs
      typedef UnaryEngine<Derived> UnaryEngine_; 
      typedef ExprEngine<Derived> ExprEngine_; 

      // Argument typedefs
      typedef typename EngineTrait<Derived>::argument_type argument_type; 

      // Operational typedefs
      typedef typename EngineTrait<Derived>::value_type value_type; 
      typedef typename EngineTrait<Derived>::op_type op_type; 
      typedef typename EngineTrait<Derived>::policy policy; 
      typedef typename EngineTrait<Derived>::dist_eval_type dist_eval_type; 

      // Meta data typedefs
      typedef typename EngineTrait<Derived>::size_type size_type; 
      typedef typename EngineTrait<Derived>::trange_type trange_type; 
      typedef typename EngineTrait<Derived>::shape_type shape_type; 
      typedef typename EngineTrait<Derived>::pmap_interface pmap_interface; 


      static constexpr bool consumable = true;
      static constexpr unsigned int leaves = argument_type::leaves;

    protected:

      // Import base class variables to this scope
      using ExprEngine_::world_;
      using ExprEngine_::vars_;
      using ExprEngine_::perm_;
      using ExprEngine_::trange_;
      using ExprEngine_::shape_;
      using ExprEngine_::pmap_;
      using ExprEngine_::permute_tiles_;

      argument_type arg_; 

    private:

      // Not allowed
      UnaryEngine(const UnaryEngine_&);
      UnaryEngine_& operator=(const UnaryEngine_&);

    public:

      template <typename D>
      UnaryEngine(const UnaryExpr<D>& expr) :
        ExprEngine_(expr), arg_(expr.arg())
      { }

      // Pull base class functions into this class.
      using ExprEngine_::derived;
      using ExprEngine_::vars;


      void perm_vars(const VariableList& target_vars) {
        TA_ASSERT(permute_tiles_);

        vars_ = target_vars;
        if(arg_.vars() != target_vars)
          arg_.perm_vars(target_vars);
      }


      void init_vars(const VariableList& target_vars) {
        arg_.init_vars(target_vars);
        perm_vars(target_vars);
      }

      void init_vars() {
        arg_.init_vars();
        vars_ = arg_.vars();
      }



      void init_struct(const VariableList& target_vars) {
        arg_.init_struct(ExprEngine_::vars());
        ExprEngine_::init_struct(target_vars);
      }


      void init_distribution(World* world,
          const std::shared_ptr<pmap_interface>& pmap)
      {
        arg_.init_distribution(world, pmap);
        ExprEngine_::init_distribution(world, arg_.pmap());
      }


      trange_type make_trange() const { return arg_.trange(); }


      trange_type make_trange(const Permutation& perm) const {
        return perm ^ arg_.trange();
      }


      dist_eval_type make_dist_eval() const {
        typedef TiledArray::detail::UnaryEvalImpl<typename argument_type::dist_eval_type,
            typename Derived::op_type, typename dist_eval_type::policy> impl_type;

        // Construct left and right distributed evaluators
        const typename argument_type::dist_eval_type arg = arg_.make_dist_eval();

        // Construct the distributed evaluator type
        std::shared_ptr<impl_type> pimpl =
            std::make_shared<impl_type>(arg, *world_, trange_, shape_, pmap_,
                                        perm_, ExprEngine_::make_op());

        return dist_eval_type(pimpl);
      }


      void print(ExprOStream os, const VariableList& target_vars) const {
        ExprEngine_::print(os, target_vars);
        os.inc();
        arg_.print(os, vars_);
        os.dec();
      }

    }; // class UnaryEngine

  }  // namespace expressions
} // namespace TiledArray

#endif // TILEDARRAY_EXPRESSIONS_UNARY_ENGINE_H__INCLUDED