Changeset 214 for XMLIO_V2/dev/dev_rv/src/xmlio/data_treatment.hpp

Timestamp:

06/14/11 16:20:27 (13 years ago)

Author:

hozdoba

Message:

 

File:

• XMLIO_V2/dev/dev_rv/src/xmlio/data_treatment.hpp (modified) (3 diffs)

XMLIO_V2/dev/dev_rv/src/xmlio/data_treatment.hpp

@@ -7 +7 @@
 #include "mpi_manager.hpp"
 #include "data_output.hpp"
+#include "duration.hpp"
+#include "client.hpp"
+#include "xios_manager.hpp"
 
 namespace xmlioserver
@@ -29 +32 @@
             const std::vector<boost::shared_ptr<CFile> > & getEnabledFiles  (void) const;
 
-            /// Creation des sorties ///
+            /// 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);
+
+            /// Création des sorties ///
             template <class T> void createDataOutput(void);
 
@@ -72 +89 @@
       }
 
+      template <StdSize N>
+         void CDataTreatment::write_data
+            (const StdString & fieldId, const ARRAY(float, N) & data)
+      {
+         std::vector<boost::multi_array<double, N>::size_type shape() > shape;
+         const size_type *      shapearr = data->shape();
+
+         shape.assign(shapearr, shapearr + N);
+         ARRAY_CREATE(datad, double, N, shape);
+         datad->assign(data->begin(), data->end());
+
+         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 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(currDate, data))
+            {
+               if (CXIOSManager::GetStatus() == CXIOSManager::LOC_CLIENT)
+               { 
+                   boost::shared_ptr<comm::CClient> client = comm::CClient::GetClient();
+                   client.sendData(fieldId, file->getId(), field->getData());
+               }
+               else
+               {
+                  file->getDataOutput()->writeFieldData(field);
+               }
+            }
+            return;
+         }
+
+      }
+
    } // namespace data