lsst.sphgeom g0261d94427+c9b10be330
Chunker.h
Go to the documentation of this file.
1/*
2 * LSST Data Management System
3 * Copyright 2014-2015 AURA/LSST.
4 *
5 * This product includes software developed by the
6 * LSST Project (http://www.lsst.org/).
7 *
8 * This program is free software: you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation, either version 3 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the LSST License Statement and
19 * the GNU General Public License along with this program. If not,
20 * see <https://www.lsstcorp.org/LegalNotices/>.
21 */
22
23#ifndef LSST_SPHGEOM_CHUNKER_H_
24#define LSST_SPHGEOM_CHUNKER_H_
25
29
30#include <stdint.h>
31#include <vector>
32
33#include "Angle.h"
34#include "Box.h"
35
36
37namespace lsst {
38namespace sphgeom {
39
43struct SubChunks {
44 int32_t chunkId;
45 std::vector<int32_t> subChunkIds;
46
47 SubChunks() : chunkId(-1) {}
48
49 void swap(SubChunks & sc) {
50 std::swap(chunkId, sc.chunkId);
51 subChunkIds.swap(sc.subChunkIds);
52 }
53};
54
55
66class Chunker {
67public:
68 Chunker(int32_t numStripes,
69 int32_t numSubStripesPerStripe);
70
71 bool operator==(Chunker const & c) const {
72 return _numStripes == c._numStripes &&
73 _numSubStripesPerStripe == c._numSubStripesPerStripe;
74 }
75
76 bool operator!=(Chunker const & c) const {
77 return _numStripes != c._numStripes ||
78 _numSubStripesPerStripe != c._numSubStripesPerStripe;
79 }
80
83 int32_t getNumStripes() const {
84 return _numStripes;
85 }
86
89 int32_t getNumSubStripesPerStripe() const {
90 return _numSubStripesPerStripe;
91 }
92
95 std::vector<int32_t> getChunksIntersecting(Region const & r) const;
96
99 std::vector<SubChunks> getSubChunksIntersecting(Region const & r) const;
100
103 std::vector<int32_t> getAllChunks() const;
104
107 std::vector<int32_t> getAllSubChunks(int32_t chunkId) const;
108
110 bool valid(int32_t chunkId) const;
111
112 Box getChunkBoundingBox(int32_t stripe, int32_t chunk) const;
113 Box getSubChunkBoundingBox(int32_t subStripe, int32_t subChunk) const;
114
116 int32_t getStripe(int32_t chunkId) const {
117 return chunkId / (2 * _numStripes);
118 }
119
121 int32_t getChunk(int32_t chunkId, int32_t stripe) const {
122 return chunkId - stripe*2*_numStripes;
123 }
124
125private:
126 struct Stripe {
127 Angle chunkWidth;
128 int32_t numChunksPerStripe;
129 int32_t numSubChunksPerChunk;
130
131 Stripe() :
132 chunkWidth(0),
133 numChunksPerStripe(0),
134 numSubChunksPerChunk(0)
135 {}
136 };
137
138 struct SubStripe {
139 Angle subChunkWidth;
140 int32_t numSubChunksPerChunk;
141
142 SubStripe() : subChunkWidth(), numSubChunksPerChunk(0) {}
143 };
144
145 int32_t _getChunkId(int32_t stripe, int32_t chunk) const {
146 return stripe * 2 * _numStripes + chunk;
147 }
148
149 int32_t _getSubChunkId(int32_t stripe, int32_t subStripe,
150 int32_t chunk, int32_t subChunk) const {
151 int32_t y = subStripe - stripe * _numSubStripesPerStripe;
152 int32_t x = subChunk -
153 chunk * _subStripes[subStripe].numSubChunksPerChunk;
154 return y * _maxSubChunksPerSubStripeChunk + x;
155 }
156
157 void _getSubChunks(std::vector<SubChunks> & subChunks,
158 Region const & r,
159 NormalizedAngleInterval const & lon,
160 int32_t stripe,
161 int32_t chunk,
162 int32_t minSS,
163 int32_t maxSS) const;
164
165 int32_t _numStripes;
166 int32_t _numSubStripesPerStripe;
167 int32_t _numSubStripes;
168 int32_t _maxSubChunksPerSubStripeChunk;
169 Angle _subStripeHeight;
170 std::vector<Stripe> _stripes;
171 std::vector<SubStripe> _subStripes;
172};
173
174}} // namespace lsst::sphgeom
175
176#endif // LSST_SPHGEOM_CHUNKER_H_
This file declares a class for representing angles.
This file declares a class for representing longitude/latitude angle boxes on the unit sphere.
Definition: Angle.h:43
Definition: Box.h:54
Definition: Chunker.h:66
std::vector< int32_t > getAllChunks() const
Definition: Chunker.cc:258
int32_t getStripe(int32_t chunkId) const
Return the stripe for the specified chunkId.
Definition: Chunker.h:116
int32_t getNumSubStripesPerStripe() const
Definition: Chunker.h:89
int32_t getChunk(int32_t chunkId, int32_t stripe) const
Return the chunk for the given chunkId and stripe.
Definition: Chunker.h:121
std::vector< int32_t > getAllSubChunks(int32_t chunkId) const
Definition: Chunker.cc:269
std::vector< SubChunks > getSubChunksIntersecting(Region const &r) const
Definition: Chunker.cc:148
int32_t getNumStripes() const
Definition: Chunker.h:83
bool valid(int32_t chunkId) const
Return 'true' if the specified chunk number is valid.
Definition: Chunker.cc:285
std::vector< int32_t > getChunksIntersecting(Region const &r) const
Definition: Chunker.cc:103
Definition: Region.h:79
Definition: Chunker.h:43