source: XIOS/dev/common/src/data_treatment.hpp @ 1512

#ifndef __XMLIO_CDataTreatment__
#define __XMLIO_CDataTreatment__

/// xmlioserver headers ///
#include "xmlioserver_spl.hpp"
#include "node_type.hpp"
#include "mpi_manager.hpp"
#include "data_output.hpp"
#include "duration.hpp"
#include "client.hpp"
#include "xios_manager.hpp"
#include <mpi.h>

namespace xmlioserver
{
   namespace data
   {
      /// ////////////////////// Déclarations ////////////////////// ///
      class CDataTreatment
      {
         public :

            /// Construteurs ///
            CDataTreatment
               (boost::shared_ptr<CContext> _ctxt =
                  CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId()));

            CDataTreatment(const CDataTreatment & data_treatment);       // Not implemented yet.
            CDataTreatment(const CDataTreatment * const data_treatment); // Not implemented yet.

            /// Accesseurs ///
            const boost::shared_ptr<CContext> getCurrentContext(void) const;
            const std::vector<boost::shared_ptr<CFile> > & getEnabledFiles  (void) const;

            /// Ecriture et mise à jour du calendrier ///
            void write_data(const StdString & fieldId,
                            const StdString & fileId,
                            const std::deque<ARRAY(double, 1)> & data);

            template <StdSize N>
               void write_data(const StdString & fieldId, const ARRAY(double, N) & data);

            template <StdSize N>
               void write_data(const StdString & fieldId, const ARRAY(float, N) & data);

            void update_calendar(int step);
            void set_timestep(const date::CDuration & duration);
            void finalize(void) ;
            
            /// Création des sorties ///
            template <class T> void createDataOutput(MPI_Comm comm_server = CXIOSManager::Comm_Server);

           
            /// Destructeur ///
            ~CDataTreatment(void);

         private :

            /// Traitements ///
            void doTreatment(void);
            void closeAllFile(void ) ;

            void findAllEnabledFields(void);
            void solveAllGridRef(void);
            void solveAllOperation(void);
            void solveAllInheritance(void) const;
            void findEnabledFiles(void);

            /// Propriétés privées ///
            boost::shared_ptr<CContext> currentContext;
            std::vector<boost::shared_ptr<CFile> > enabledFiles;
     
      }; // CDataTreatment

      //----------------------------------------------------------------

      template <class T>
         void CDataTreatment::createDataOutput(MPI_Comm comm_server )
      {
         std::vector<boost::shared_ptr<CFile> >::const_iterator
            it = this->enabledFiles.begin(), end = this->enabledFiles.end();
         
         for (; it != end; it++)
         {
            CXIOSManager::Comm_Server = comm_server;
            boost::shared_ptr<CFile> file = *it;
            StdString filename = (!file->name.isEmpty())
                               ?   file->name.getValue() : file->getId();
            StdOStringStream oss;
            if (!CObjectFactory::GetObject<CContext>
                (CObjectFactory::GetCurrentContextId())->output_dir.isEmpty())
                oss << CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId())->output_dir.getValue();
            oss << filename;
            if (!file->name_suffix.isEmpty())
                oss << file->name_suffix.getValue();

            bool multifile=true ;
            if (!file->type.isEmpty())
            {
              StdString str=file->type.getValue() ; 
              if (str.compare("one_file")==0) multifile=false ;
              else if (str.compare("multi_file")==0) multifile=true ;
              else ERROR("void CDataTreatment::createDataOutput(MPI_Comm comm_server)",
                         "incorrect file <type> attribut : must be <multi_file> or <one_file>, "
                        <<"having : <"<<str<<">") ;
            } 
            if (multifile) 
            {
              if (comm::CMPIManager::GetCommSize(comm_server) > 1)
                       oss << "_" << comm::CMPIManager::GetCommRank(comm_server);
            }
            oss << ".nc";

            boost::shared_ptr<io::CDataOutput> dout(new T(oss.str(), false,comm_server,multifile));
//            file->initializeDataOutput(dout);
         }
      }

      template <StdSize N>
         void CDataTreatment::write_data
            (const StdString & fieldId, const ARRAY(float, N) & data)
      {
         typedef typename boost::multi_array<double, N>::size_type sizetp;
         std::vector<sizetp> shape;
         const sizetp * shapearr = data->shape();

         shape.assign(shapearr, shapearr + N);
         ARRAY(double, N) datad(new CArray<double, N>(shape));
         for (StdSize i = 0; i < datad->num_elements(); i++)
          datad->data()[i] = data->data()[i];

         this->write_data(fieldId, datad);
      }

      template <StdSize N>
         void CDataTreatment::write_data
            (const StdString & fieldId, const ARRAY(double, N) & data)
      {
         const date::CDate & currDate =
                this->currentContext->getCalendar()->getCurrentDate();
         const date::CDuration & timestep = 
               this->currentContext->getCalendar()->getTimeStep();
         const std::vector<boost::shared_ptr<CField> > & refField=
               CObjectFactory::GetObject<CField>(fieldId)->getAllReference();
         std::vector<boost::shared_ptr<CField> >::const_iterator
               it = refField.begin(), end = refField.end();
         for (; it != end; it++)
         {
            boost::shared_ptr<CField> field = *it;
            boost::shared_ptr<CFile>  file  = field->getRelFile();
            if (field->updateData(data))
            {
               if (CXIOSManager::GetStatus() == CXIOSManager::LOC_CLIENT)
               { 
                   boost::shared_ptr<comm::CClient> client = comm::CClient::GetClient();
                   client->sendData(field->getId(), file->getId(), field->getData());
               }
               else
               {
                  file->getDataOutput()->writeFieldData(field);
               }
            }
         }

      }

   } // namespace data

} // namespace xmlioserver

#endif // __XMLIO_CDataTreatment__