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

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

#include "xmlioserver.hpp"

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

#include "icutil.hpp"

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

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

   // -------------------- Traitement des données ------------------------------
   
   void xios_dtreatment_start(XContextPtr context_hdl, XFileType filetype, int comm_client_server, int comm_server)
   {
      using namespace xmlioserver::tree;
      using namespace xmlioserver;
      try
      {
         boost::shared_ptr<CContext> context =
                CObjectFactory::GetObject<CContext>(context_hdl);
         StdOStringStream oss;
         oss << StdString("def_client_next.")
             << CMPIManager::GetCommRank(CMPIManager::GetCommWorld());
         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::GetCommWorld());
         CTreeManager::PrintTreeToFile(oss.str());

         if ((comm_client_server != -1))
         {
            MPI_Request request = 0;
            StdOStringStream ostrs;
            if (CMPIManager::GetCommRank(comm_client_server) == 1)
            {
               CTreeManager::ToBinary(ostrs);
               CLinearBuffer lbuffer(ostrs.str().size()+13);
               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()+13);
               lbuffer.appendString(ostrs.str());
               CMPIManager::SendLinearBuffer(comm_client_server, 0, lbuffer, request);
               CMPIManager::Wait(request);  // Pas encore en mode RPC
            }
            CXIOSManager::RunClient(false, CMPIManager::GetCommWorld());
            CClient::CreateClient(comm_client_server);
         }
         else
         {
            dt->createDataOutput<CNc4DataOutput>(CMPIManager::GetCommWorld());
         }
      }
      catch (CException & exc)
      {
         std::cerr << exc.getMessage() << std::endl;
         exit (EXIT_FAILURE);
      }
   }
   
   void xios_dtreatment_end(void)
   {
      try
      {
         CMPIManager::Finalize();
      }
      catch (CException & exc)
      {
         std::cerr << exc.getMessage() << std::endl;
         exit (EXIT_FAILURE);
      }
   }

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

   void xios_init_ioserver(MPIComm * comm_client)
   {
      try
      {
         MPIComm comm_client_server, comm_server;
         CMPIManager::Initialise(NULL, NULL);
         CMPIManager::DispatchClient(false, *comm_client, comm_client_server, comm_server);
      }
      catch (CException & exc)
      {
         std::cerr << exc.getMessage() << std::endl;
         exit (EXIT_FAILURE);
      }
   }
   
   // ---------------------- Ecriture des données ------------------------------
   
   void xios_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 xios_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 xios_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 xios_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 xios_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 xios_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"