blob: ac1b68dbd9b42b9b00c9df432e0d55f54ab79b9c (
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
// Copyright 2018 The Amber Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "src/datum_type.h"
#include <string>
#include "src/format_parser.h"
namespace amber {
DatumType::DatumType() = default;
DatumType::DatumType(const DatumType&) = default;
DatumType::~DatumType() = default;
DatumType& DatumType::operator=(const DatumType&) = default;
uint32_t DatumType::ElementSizeInBytes() const {
uint32_t s;
if (type_ == DataType::kInt8 || type_ == DataType::kUint8)
s = 1;
else if (type_ == DataType::kInt16 || type_ == DataType::kUint16)
s = 2;
else if (type_ == DataType::kInt32 || type_ == DataType::kUint32)
s = 4;
else if (type_ == DataType::kInt64 || type_ == DataType::kUint64)
s = 8;
else if (type_ == DataType::kFloat)
s = sizeof(float);
else
s = sizeof(double);
return s;
}
uint32_t DatumType::SizeInBytes() const {
uint32_t s = ElementSizeInBytes();
uint32_t bytes = s * column_count_ * row_count_;
// For a vector of size 3 in std140 we need to have alignment of 4N. For a
// matrix, we update each row to 4N.
if (is_std140_ && row_count_ == 3)
bytes += (s * column_count_);
return bytes;
}
Format DatumType::AsFormat() const {
uint32_t bits_per_element = ElementSizeInBytes() * 8;
static const char* prefixes = "RGBA";
std::string name = "";
for (size_t i = 0; i < row_count_; ++i)
name += prefixes[i] + std::to_string(bits_per_element);
name += "_";
if (IsFloat() || IsDouble())
name += "SFLOAT";
else if (IsInt8() || IsInt16() || IsInt32() || IsInt64())
name += "SINT";
else
name += "UINT";
FormatParser fp;
auto fmt = fp.Parse(name);
return *(fmt.get());
}
} // namespace amber
|
