source: XMLIO_V2/dev/common/src/xmlio/fortran/icdata.cpp @ 286

/* ************************************************************************** *
 *      Copyright © IPSL/LSCE, XMLIOServer, Avril 2010 - Octobre 2011         *
 * ************************************************************************** */

#include <boost/multi_array.hpp>
#include <boost/shared_ptr.hpp>
#include <string>
#include <iostream>


#include "xmlioserver.hpp"
#include "oasis_cinterface.hpp"

#include "attribute_template_impl.hpp"
#include "object_template_impl.hpp"
#include "group_template_impl.hpp"

#include "icutil.hpp"
#include "mpi_manager.hpp"
#include "buffer.hpp"

extern "C"
{
// /////////////////////////////// Définitions ////////////////////////////// //

   // ----------------------- Redéfinition de types ----------------------------
   
   typedef enum { NETCDF4 = 0 } XFileType;
   
   typedef xmlioserver::tree::CContext * XContextPtr;

   // -------------------- Traitement des données ------------------------------
   
   void cxios_dtreatment_start()
   {
      using namespace xmlioserver::tree;
      using namespace xmlioserver;
      try
      {
        MPI_Comm comm_client_server=comm::CMPIManager::GetCommClientServer() ;
        MPI_Comm comm_server=comm::CMPIManager::GetCommServer() ;
        
         boost::shared_ptr<CContext> context =
            CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId());
                
         StdOStringStream oss;
         oss << StdString("def_client_next.")
             << CMPIManager::GetCommRank(CMPIManager::GetCommClient());
         CTreeManager::PrintTreeToFile(oss.str());
         oss.str("");
         boost::shared_ptr<CDataTreatment> dt(new CDataTreatment (context));
         context->setDataTreatment(dt);

         oss << StdString("def_client_end.")
             << CMPIManager::GetCommRank(CMPIManager::GetCommClient());
         CTreeManager::PrintTreeToFile(oss.str());

         if ( !comm::CMPIManager::IsConnected() )
         {
            MPI_Request request = 0;
            StdOStringStream ostrs;
/*
            if (CMPIManager::GetCommRank(comm_client_server) == 1)
            {
               CTreeManager::ToBinary(ostrs);
               CLinearBuffer lbuffer(ostrs.str().size()+CBuffer::getDataHeaderSize());
               std::cout<<"lbuffer size "<<ostrs.str().size()<<std::endl ;
               lbuffer.appendString(ostrs.str());
               CMPIManager::SendLinearBuffer(comm_client_server, 0, lbuffer, request);
               CMPIManager::Wait(request);  // Pas encore en mode RPC
            }
            else
            {
               CTreeManager::DomainsToBinary(ostrs);
               CLinearBuffer lbuffer(ostrs.str().size()+CBuffer::getDataHeaderSize()); 
               std::cout<<"lbuffer size "<<ostrs.str().size()<<std::endl ;
               lbuffer.appendString(ostrs.str());
               CMPIManager::SendLinearBuffer(comm_client_server, 0, lbuffer, request);
               CMPIManager::Wait(request);  // Pas encore en mode RPC
            }
*/
            CTreeManager::ToBinary(ostrs);
            CLinearBuffer lbuffer(ostrs.str().size()+CBuffer::getDataHeaderSize());
            lbuffer.appendString(ostrs.str());
            CMPIManager::SendLinearBuffer(comm_client_server, 0, lbuffer, request);
            CMPIManager::Wait(request);  // Pas encore en mode RPC
               
               
            CXIOSManager::RunClient(false, CMPIManager::GetCommClient());
            CClient::CreateClient(CMPIManager::GetCommClientServer());
         }
         else
         {
            dt->createDataOutput<CNc4DataOutput>(CMPIManager::GetCommClient());
         }
      }
      catch (CException & exc)
      {
         std::cerr << exc.getMessage() << std::endl;
         exit (EXIT_FAILURE);
      }
   }
   
   void cxios_dtreatment_end(void)
   {
      try
      {
         boost::shared_ptr<xmlioserver::tree::CContext> context =
         xmlioserver::CObjectFactory::GetObject<xmlioserver::tree::CContext>
            (CObjectFactory::GetCurrentContextId());
         boost::shared_ptr<xmlioserver::data::CDataTreatment> dtreat = context->getDataTreatment();
         dtreat->finalize();
         
//         CMPIManager::Finalize();
      }
      catch (CException & exc)
      {
         std::cerr << exc.getMessage() << std::endl;
         exit (EXIT_FAILURE);
      }
   }

   // ---------------------- Finalize du serveur -------------------------

   void cxios_finalize_ioserver(void)
   {
      try
      {
         CMPIManager::Finalize();
      }
      catch (CException & exc)
      {
         std::cerr << exc.getMessage() << std::endl;
         exit (EXIT_FAILURE);
      }
    }

   // ---------------------- Initialisation du serveur -------------------------

   void cxios_init_ioserver(MPIComm * f_comm_client, MPIComm * f_comm_parent)
   {
      try
      {
         MPI_Comm comm_client_server, comm_server,comm_client,comm_parent;
         xmlioserver::CTreeManager::ParseFile("iodef.xml");
         CTreeManager::SetCurrentContextId(StdString("xios"));
         CMPIManager::InitialiseClient(NULL, NULL);
         comm_parent=MPI_Comm_f2c(*f_comm_parent) ;
         CMPIManager::DispatchClient(false, comm_client, comm_client_server, comm_server,comm_parent);
         *f_comm_client=MPI_Comm_c2f(comm_client) ;
      }
      catch (CException & exc)
      {
         std::cerr << exc.getMessage() << std::endl;
         exit (EXIT_FAILURE);
      }
   }
   
   // ---------------------- Ecriture des données ------------------------------
   
   void cxios_write_data_k81(const char * fieldid, int fieldid_size, double * data_k8, int data_Xsize)
   {
      std::string fieldid_str;
      boost::shared_ptr<CContext> context =
      CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId());
      boost::shared_ptr<data::CDataTreatment> dtreat = context->getDataTreatment();
      if (!cstr2string(fieldid, fieldid_size, fieldid_str)) return; 
      
//      boost::const_multi_array_ref<double, 1> array_(data_k8,
//          boost::extents [data_Xsize],
//          boost::fortran_storage_order());
      ARRAY(double, 1) data(new CArray<double, 1>(boost::extents [data_Xsize]));
      std::copy(data_k8, &(data_k8[data->num_elements()]), data->data());
      dtreat->write_data(fieldid_str, data);
   }
   
   void cxios_write_data_k82(const char * fieldid, int fieldid_size, double * data_k8, int data_Xsize, int data_Ysize)
   {
      std::string fieldid_str;
      boost::shared_ptr<CContext> context =
      CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId());
      boost::shared_ptr<data::CDataTreatment> dtreat = context->getDataTreatment();
      if (!cstr2string(fieldid, fieldid_size, fieldid_str)) return;
      
//      boost::const_multi_array_ref<double, 2> array_(data_k8,
//          boost::extents [data_Xsize][data_Ysize],
//          boost::fortran_storage_order());
      ARRAY(double, 2) data(new CArray<double, 2>(boost::extents [data_Xsize][data_Ysize]));
      std::copy(data_k8, &(data_k8[data->num_elements()]), data->data());
      dtreat->write_data(fieldid_str, data);
   }
   
   void cxios_write_data_k83(const char * fieldid, int fieldid_size, double * data_k8, int data_Xsize, int data_Ysize, int data_Zsize)
   {
      std::string fieldid_str;
      boost::shared_ptr<CContext> context =
      CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId());
      boost::shared_ptr<data::CDataTreatment> dtreat = context->getDataTreatment();
      if (!cstr2string(fieldid, fieldid_size, fieldid_str)) return; 
      
//      boost::const_multi_array_ref<double, 3> array_(data_k8,
//          boost::extents [data_Xsize][data_Ysize][data_Zsize],
//          boost::fortran_storage_order());
      ARRAY(double, 3) data(new CArray<double, 3>(boost::extents [data_Xsize][data_Ysize][data_Zsize]));
      std::copy(data_k8, &(data_k8[data->num_elements()]), data->data());
      dtreat->write_data(fieldid_str, data);
   }
   
   void cxios_write_data_k41(const char * fieldid, int fieldid_size, float * data_k4, int data_Xsize)
   {
      std::string fieldid_str;
      boost::shared_ptr<CContext> context =
      CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId());
      boost::shared_ptr<data::CDataTreatment> dtreat = context->getDataTreatment();
      if (!cstr2string(fieldid, fieldid_size, fieldid_str)) return; 
      
//      boost::const_multi_array_ref<float, 1> array_(data_k4,
//          boost::extents [data_Xsize],
//          boost::fortran_storage_order());
      ARRAY(float, 1) data(new CArray<float, 1>(boost::extents [data_Xsize]));
      std::copy(data_k4, &(data_k4[data->num_elements()]), data->data());
      dtreat->write_data(fieldid_str, data);
   }
   
   void cxios_write_data_k42(const char * fieldid, int fieldid_size, float * data_k4, int data_Xsize, int data_Ysize)
   {
      std::string fieldid_str;
      boost::shared_ptr<CContext> context =
      CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId());
      boost::shared_ptr<data::CDataTreatment> dtreat = context->getDataTreatment();
      if (!cstr2string(fieldid, fieldid_size, fieldid_str)) return; 
      
//      boost::const_multi_array_ref<float, 2> array_(data_k4,
//          boost::extents [data_Xsize][data_Ysize],
//          boost::fortran_storage_order());
      ARRAY(float, 2) data(new CArray<float, 2>(boost::extents [data_Xsize][data_Ysize]));
      std::copy(data_k4, &(data_k4[data->num_elements()]), data->data());
      dtreat->write_data(fieldid_str, data);
   }
   
   void cxios_write_data_k43(const char * fieldid, int fieldid_size, float * data_k4, int data_Xsize, int data_Ysize, int data_Zsize)
   {
      std::string fieldid_str;
      boost::shared_ptr<CContext> context =
      CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId());
      boost::shared_ptr<data::CDataTreatment> dtreat = context->getDataTreatment();
      if (!cstr2string(fieldid, fieldid_size, fieldid_str)) return; 
      
//      boost::const_multi_array_ref<float, 3> array_(data_k4,
//          boost::extents [data_Xsize][data_Ysize][data_Zsize],
//          boost::fortran_storage_order());
      ARRAY(float, 3) data(new CArray<float, 3>(boost::extents [data_Xsize][data_Ysize][data_Zsize]));
      std::copy(data_k4, &(data_k4[data->num_elements()]), data->data());
      dtreat->write_data(fieldid_str, data);
   } 

} // extern "C"