blob: eb1c5d4b780ec9ab8ddda2d6aca029430e43e238 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
#pragma once
#include "geometry/rect2d.hpp"
#include "coding/file_writer.hpp"
#include <cstdint>
#include <limits>
#include <string>
#include <vector>
namespace feature
{
class FeatureBuilder;
// Writes features to file.
class FeaturesCollector
{
public:
static size_t constexpr kBufferSize = 48000;
FeaturesCollector(std::string const & fName, FileWriter::Op op = FileWriter::Op::OP_WRITE_TRUNCATE);
virtual ~FeaturesCollector();
static uint64_t GetCurrentPosition();
std::string const & GetFilePath() const { return m_dataFile.GetName(); }
/// \brief Serializes |f|.
/// \returns Feature id of serialized feature.
virtual uint32_t Collect(FeatureBuilder const & f);
virtual uint32_t Collect(FeatureBuilder & f)
{
return Collect(const_cast<FeatureBuilder const &>(f));
}
virtual void Finish() {}
protected:
/// \return Feature offset in the file, which is used as an ID later
uint32_t WriteFeatureBase(std::vector<char> const & bytes, FeatureBuilder const & fb);
void Flush();
FileWriter m_dataFile;
m2::RectD m_bounds;
private:
void Write(char const * src, size_t size);
void FlushBuffer();
std::vector<char> m_writeBuffer;
size_t m_writePosition = 0;
uint32_t m_featureID = 0;
};
class FeaturesAndRawGeometryCollector : public FeaturesCollector
{
FileWriter m_rawGeometryFileStream;
size_t m_rawGeometryCounter = 0;
public:
FeaturesAndRawGeometryCollector(std::string const & featuresFileName,
std::string const & rawGeometryFileName);
~FeaturesAndRawGeometryCollector() override;
uint32_t Collect(FeatureBuilder const & f) override;
};
uint32_t CheckedFilePosCast(FileWriter const & f);
} // namespace feature
|