Tsqr_generateStack.hpp

// @HEADER
// ***********************************************************************
//
//                 Anasazi: Block Eigensolvers Package
//                 Copyright (2010) Sandia Corporation
//
// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
// license for use of this work by or on behalf of the U.S. Government.
//
// This library is free software; you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation; either version 2.1 of the
// License, or (at your option) any later version.
//
// This library 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
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
// USA
// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
//
// ***********************************************************************
// @HEADER

#ifndef __TSQR_Test_generateStack_hpp
#define __TSQR_Test_generateStack_hpp

#include <Tsqr_Blas.hpp>
#include <Tsqr_Lapack.hpp>
#include <Tsqr_Matrix.hpp>
#include <Tsqr_Util.hpp>
#include <Tsqr_Random_MatrixGenerator.hpp>
#include <Tsqr_RMessenger.hpp>

#include <algorithm>
#include <functional>
#include <sstream>
#include <stdexcept>


namespace TSQR {
  namespace Test {

    template< class Ordinal, class Scalar, class Generator >
    static void
    generateStack (Generator& generator,
       Matrix< Ordinal, Scalar >& A_global,
       const typename ScalarTraits< Scalar >::magnitude_type singular_values[],
       const int nprocs,
       const Ordinal ncols)
    {
      TSQR::Random::MatrixGenerator< Ordinal, Scalar, Generator > matGen (generator);
      const Ordinal nrows = nprocs * ncols;
      A_global.reshape (nrows, ncols);
      A_global.fill (Scalar(0));

      for (int p = 0; p < nprocs; ++p)
  {
    Scalar* const curptr = A_global.get() + p*ncols;
    MatView< Ordinal, Scalar > R_cur (ncols, ncols, curptr, nrows);
    matGen.fill_random_R (ncols, R_cur.get(), nrows, singular_values);
  }
    }

    template< class Ordinal, class Scalar, class Generator >
    void
    par_tsqr_test_problem (Generator& generator,
         Matrix< Ordinal, Scalar >& A_local,
         Matrix< Ordinal, Scalar >& A_global, 
         const Ordinal ncols,
         const Teuchos::RCP< MessengerBase< Scalar > >& messenger)
    {
      const int nprocs = messenger->size();
      const int my_rank = messenger->rank();
      A_local.reshape (ncols, ncols);

      if (my_rank == 0)
  {
    typedef typename ScalarTraits< Scalar >::magnitude_type magnitude_type;

    std::vector< magnitude_type > singular_values (ncols);
    singular_values[0] = magnitude_type(1);
    for (Ordinal k = 1; k < ncols; ++k)
      singular_values[k] = singular_values[k-1] / magnitude_type(2);

    generateStack (generator, A_global, &singular_values[0], nprocs, ncols);
    scatterStack (A_global, A_local, messenger);
  }
      else
  scatterStack (A_global, A_local, messenger);
    }


  } // namespace Test
} // namespace TSQR

#endif // __TSQR_Test_generateStack_hpp