morton_order.h

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//
// Copyright (C) 2021 CERN & University of Surrey for the benefit of the
// BioDynaMo collaboration. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
//
// See the LICENSE file distributed with this work for details.
// See the NOTICE file distributed with this work for additional information
// regarding copyright ownership.
//
// -----------------------------------------------------------------------------
 
#ifndef CORE_ENVIRONMENT_MORTON_ORDER_H_
#define CORE_ENVIRONMENT_MORTON_ORDER_H_
 
#include <array>
#include <vector>
#include "core/container/fixed_size_vector.h"
#include "core/container/math_array.h"
#include "core/functor.h"
#include "core/util/iterator.h"
 
namespace bdm {
 
class MortonOrder {
 public:
  void Update(const std::array<uint64_t, 3>& num_boxes_axis);
  uint64_t GetMortonCode(uint64_t box_index) const;
  void CallMortonIteratorConsumer(uint64_t start_index, uint64_t end_index,
                                  Functor<void, Iterator<uint64_t>*>& f) const;
 
 private:
  uint64_t num_boxes_;
  std::vector<std::pair<uint64_t, uint64_t>> offset_index_;
};
 
}  // namespace bdm
 
#endif  // CORE_ENVIRONMENT_MORTON_ORDER_H_