source: XIOS/dev/XIOS_DEV_CMIP6/src/filter/store_filter.cpp @ 1251

Last change on this file since 1251 was 1251, checked in by oabramkina, 5 years ago

Fixing a bug in case of activated missing values. If nan values are replaced by default values or vice versa data package (or data) is copied.

File size: 4.1 KB
Line 
1#include "store_filter.hpp"
2#include "context.hpp"
3#include "grid.hpp"
4#include "timer.hpp"
5
6namespace xios
7{
8  CStoreFilter::CStoreFilter(CGarbageCollector& gc, CContext* context, CGrid* grid,
9                             bool detectMissingValues /*= false*/, double missingValue /*= 0.0*/)
10    : CInputPin(gc, 1)
11    , gc(gc)
12    , context(context)
13    , grid(grid)
14    , detectMissingValues(detectMissingValues)
15    , missingValue(missingValue)
16  {
17    if (!context)
18      ERROR("CStoreFilter::CStoreFilter(CContext* context, CGrid* grid)",
19            "Impossible to construct a store filter without providing a context.");
20    if (!grid)
21      ERROR("CStoreFilter::CStoreFilter(CContext* context, CGrid* grid)",
22            "Impossible to construct a store filter without providing a grid.");
23  }
24
25  CConstDataPacketPtr CStoreFilter::getPacket(Time timestamp)
26  {
27    CTimer timer("CStoreFilter::getPacket");
28    CConstDataPacketPtr packet;
29    const double timeout = CXios::recvFieldTimeout;
30
31    do
32    {
33      if (canBeTriggered())
34        trigger(timestamp);
35
36      timer.resume();
37
38      std::map<Time, CDataPacketPtr>::const_iterator it = packets.find(timestamp);
39      if (it != packets.end())
40        packet = it->second;
41      else // if the packet is not available yet, check if it can be received
42        context->checkBuffersAndListen();
43
44      timer.suspend();
45    } while (!packet && timer.getCumulatedTime() < timeout);
46
47    if (!packet)
48    {
49      std::map<Time, CDataPacketPtr>::const_iterator it ;
50      info(0)<<"Impossible to get the packet with timestamp = " << timestamp<<std::endl<<"Available timestamp are : "<<std::endl ;
51      for(it=packets.begin();it!=packets.end();++it) info(0)<<it->first<<"  ";
52      info(0)<<std::endl ;
53      ERROR("CConstDataPacketPtr CStoreFilter::getPacket(Time timestamp) const",
54            << "Impossible to get the packet with timestamp = " << timestamp);
55    }
56    return packet;
57  }
58
59  template <int N>
60  CDataPacket::StatusCode CStoreFilter::getData(Time timestamp, CArray<double, N>& data)
61  {
62    CConstDataPacketPtr packet = getPacket(timestamp);
63
64    if (packet->status == CDataPacket::NO_ERROR)
65      grid->outputField(packet->data, data);
66
67    return packet->status;
68  }
69
70  template CDataPacket::StatusCode CStoreFilter::getData<1>(Time timestamp, CArray<double, 1>& data);
71  template CDataPacket::StatusCode CStoreFilter::getData<2>(Time timestamp, CArray<double, 2>& data);
72  template CDataPacket::StatusCode CStoreFilter::getData<3>(Time timestamp, CArray<double, 3>& data);
73  template CDataPacket::StatusCode CStoreFilter::getData<4>(Time timestamp, CArray<double, 4>& data);
74  template CDataPacket::StatusCode CStoreFilter::getData<5>(Time timestamp, CArray<double, 5>& data);
75  template CDataPacket::StatusCode CStoreFilter::getData<6>(Time timestamp, CArray<double, 6>& data);
76  template CDataPacket::StatusCode CStoreFilter::getData<7>(Time timestamp, CArray<double, 7>& data);
77
78  void CStoreFilter::onInputReady(std::vector<CDataPacketPtr> data)
79  {
80
81    CDataPacketPtr packet;
82    if (detectMissingValues)
83    {
84      CArray<double, 1> dataArray = data[0]->data.copy();
85
86      const size_t nbData = dataArray.numElements();
87      for (size_t idx = 0; idx < nbData; ++idx)
88      {
89        if (NumTraits<double>::isnan(dataArray(idx)))
90          dataArray(idx) = missingValue;
91      }
92
93      packet = CDataPacketPtr(new CDataPacket);
94      packet->date = data[0]->date;
95      packet->timestamp = data[0]->timestamp;
96      packet->status = data[0]->status;
97      packet->data.resize(dataArray.numElements());
98      packet->data = dataArray;
99    }
100
101    else
102    {
103      packet = data[0];
104    }
105
106    packets.insert(std::make_pair(packet->timestamp, packet));
107    // The packet is always destroyed by the garbage collector
108    // so we register but never unregister
109    gc.registerObject(this, packet->timestamp);
110
111  }
112
113  bool CStoreFilter::isDataExpected(const CDate& date) const
114  {
115    return true;
116  }
117
118  void CStoreFilter::invalidate(Time timestamp)
119  {
120    CInputPin::invalidate(timestamp);
121    packets.erase(packets.begin(), packets.lower_bound(timestamp));
122  }
123} // namespace xios
Note: See TracBrowser for help on using the repository browser.