STK_ExportToArray2D.h

Go to the documentation of this file.

/*--------------------------------------------------------------------*/
/*     Copyright (C) 2004-2011  Serge Iovleff

 This program 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 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 Lesser General Public License for more details.

 You should have received a copy of the GNU Lesser General Public
 License along with this program; if not, write to the
 Free Software Foundation, Inc.,
 59 Temple Place,
 Suite 330,
 Boston, MA 02111-1307
 USA

 Contact : Serge.Iovleff@stkpp.org
 */

/*
 * Project:  stkpp::DManager
 * created on: 9 juin 2011
 * Purpose:  Create an utility class in order to transfer the Data from
 * a DataFrame in an Array2D.
 * Author:   iovleff, serge.iovleff@stkpp.org
 *
 **/

#ifndef STK_EXPORTTOARRAY2D_H
#define STK_EXPORTTOARRAY2D_H

#include "../../STKernel/include/STK_Array2D.h"

namespace STK
{

template<class TYPE>
class ExportToArray2D
{
  public:
    ExportToArray2D( DataFrame const& df)
                   : p_data_(0)
    {
      // create an empty Array2D
      p_data_ = new Array2D<TYPE>();

      // for each field Try a type conversion
      for(Integer iVar = df.firstCol(); iVar<=df.lastCol(); iVar++)
      {
        IVariable *const p_var = df.elt(iVar);
        // if there is a variable
        if (p_var)
        {
          // check the type of the variable
          if (p_var->getType() == IdTypeImpl<TYPE>::returnType())
          {
            Variable<TYPE>* p_variable = static_cast<Variable<TYPE>* >(p_var);
            p_data_->pushBackCol(*p_variable);
          }
        }
      }
    }

    virtual ~ExportToArray2D()
    { if (p_data_) delete p_data_;}

    inline Array2D<TYPE>* const& array2D() const
    { return p_data_;}

    inline void release() { p_data_ =0;}

  private:
    Array2D<TYPE>* p_data_;
};

}

#endif /* STK_EXPORTTOARRAY2D_H */