Function mergeExampleInfo

src/data/common.cc:50–73

}

ExampleInfo mergeExampleInfo(const ExampleInfo& A, const ExampleInfo& B) {
  std::map<int, SlotInfo> slots;
  for (int i = 0; i < A.slot_size(); ++i) {
    slots[A.slot(i).id()] = A.slot(i);
  }

  for (int i = 0; i < B.slot_size(); ++i) {
    int id = B.slot(i).id();
    if (slots.count(id) == 0) { slots[id] = B.slot(i); continue; }
    auto a = slots[id];
    auto b = B.slot(i);
    CHECK_EQ(a.format(), b.format());
    a.set_min_key(std::min(a.min_key(), b.min_key()));
    a.set_max_key(std::max(a.max_key(), b.max_key()));
    a.set_nnz_ele(a.nnz_ele() + b.nnz_ele());
    a.set_nnz_ex(a.nnz_ex() + b.nnz_ex());
    slots[id] = a;
  }

  ExampleInfo C;
  C.set_num_ex(A.num_ex() + B.num_ex());
  for (const auto& it : slots) *C.add_slot() = it.second;
  return C;
}

// InstanceInfo mergeInstanceInfo(const InstanceInfo& A, const InstanceInfo& B) {
//   auto as = A.fea_grp_size();