Changeset 321 for XIOS/trunk/src/node/file.cpp

Timestamp:

02/23/12 14:57:50 (12 years ago)

Author:

ymipsl

Message:

Add splitting file functionality
new file attribut "split_freq" to put the frequency of splitting

YM

File:

• XIOS/trunk/src/node/file.cpp (modified) (4 diffs)

XIOS/trunk/src/node/file.cpp

@@ -141 +141 @@
       return false ;
     }
-      
+    
+   void CFile::initFile(void)
+   {
+      shared_ptr<CContext> context=CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId()) ;
+      date::CDate& currentDate=context->calendar->getCurrentDate() ;
+      
+      if (! sync_freq.isEmpty()) syncFreq = date::CDuration::FromString(sync_freq.getValue());
+      if (! split_freq.isEmpty()) splitFreq = date::CDuration::FromString(split_freq.getValue());
+      if (! output_freq.isEmpty()) outputFreq = date::CDuration::FromString(output_freq.getValue());
+      lastSync=new date::CDate(currentDate) ;
+      lastSplit=new date::CDate(currentDate) ;
+      isOpen=false ;
+    }
+    
+    void CFile::checkFile(void)
+    {
+      if (!isOpen) createHeader() ;
+      checkSync() ;
+      checkSplit() ;
+    }
+      
+     
+   bool CFile::checkSync(void)
+   {
+     shared_ptr<CContext> context=CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId()) ;
+     date::CDate& currentDate=context->calendar->getCurrentDate() ;
+     if (! sync_freq.isEmpty())
+     {
+       if (*lastSync+syncFreq < currentDate)
+       {
+         *lastSync=currentDate ;
+         data_out->syncFile() ;
+         return true ;
+        }
+      }
+      return false ;
+    }
+    
+    
+    bool CFile::checkSplit(void)
+    {
+      shared_ptr<CContext> context=CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId()) ;
+      date::CDate& currentDate=context->calendar->getCurrentDate() ;
+      if (! split_freq.isEmpty())
+      {
+        if (*lastSplit+splitFreq < currentDate)
+        {
+          *lastSplit=currentDate-outputFreq ;
+        
+          std::vector<boost::shared_ptr<CField> >::iterator it, end = this->enabledFields.end();
+          for (it = this->enabledFields.begin() ;it != end; it++)  (*it)->resetNStep() ;
+          createHeader() ;
+          return true ;
+        }
+      }
+      return false ;
+    }
+    
    void CFile::createHeader(void)
    {
       shared_ptr<CContext> context=CObjectFactory::GetObject<CContext>(CObjectFactory::GetCurrentContextId()) ;
-
-      if (! sync_freq.isEmpty()) syncFreq = date::CDuration::FromString(sync_freq.getValue());
-      lastSync=new date::CDate(context->calendar->getCurrentDate()) ;
+      date::CDate& currentDate=context->calendar->getCurrentDate() ;
       
       std::vector<boost::shared_ptr<CField> >::iterator it, end = this->enabledFields.end();
@@ -168 +223 @@
          oss << filename;
          if (!name_suffix.isEmpty()) oss << name_suffix.getValue();
-
+//         if (!split_freq.isEmpty()) oss<<"-["<<currentDate.toString()<<"]" ;
+         if (!split_freq.isEmpty()) oss<<"_"<<lastSplit->getStryyyymmdd()<<"-"<< (*lastSplit+(splitFreq-1*date::Second)).getStryyyymmdd();
          bool multifile=true ;
          if (!type.isEmpty())
@@ -187 +243 @@
          oss << ".nc";
 
+         if (isOpen) data_out->closeFile() ;
+         
          data_out=shared_ptr<io::CDataOutput>(new io::CNc4DataOutput(oss.str(), false,server->intraComm,multifile));
+         isOpen=true ;
 
          data_out->writeFile(CObjectFactory::GetObject<CFile>(this));
@@ -208 +267 @@
    void CFile::close(void)
    {
+     delete lastSync ;
+     delete lastSplit ;
      if (!AllDomainEmpty ||  type.getValue()=="one_file")
-       this->data_out->closeFile();
-     delete lastSync ;
+       if (isOpen) 
+       {
+         this->data_out->closeFile();
+       }
    }
    //----------------------------------------------------------------