1 //===- EndianStream.h - Stream ops with endian specific data ----*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file defines utilities for operating on streams that have endian
10 // specific data.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #ifndef LLVM_SUPPORT_ENDIANSTREAM_H
15 #define LLVM_SUPPORT_ENDIANSTREAM_H
16 
17 #include "llvm/ADT/ArrayRef.h"
18 #include "llvm/Support/Endian.h"
19 #include "llvm/Support/MathExtras.h"
20 #include "llvm/Support/raw_ostream.h"
21 
22 namespace llvm {
23 namespace support {
24 
25 namespace endian {
26 
27 template <typename value_type>
28 inline void write(raw_ostream &os, value_type value, endianness endian) {
29   value = byte_swap<value_type>(value, endian);
30   os.write((const char *)&value, sizeof(value_type));
31 }
32 
33 template <>
34 inline void write<float>(raw_ostream &os, float value, endianness endian) {
35   write(os, FloatToBits(value), endian);
36 }
37 
38 template <>
39 inline void write<double>(raw_ostream &os, double value,
40                           endianness endian) {
41   write(os, DoubleToBits(value), endian);
42 }
43 
44 template <typename value_type>
45 inline void write(raw_ostream &os, ArrayRef<value_type> vals,
46                   endianness endian) {
47   for (value_type v : vals)
48     write(os, v, endian);
49 }
50 
51 /// Adapter to write values to a stream in a particular byte order.
52 struct Writer {
53   raw_ostream &OS;
54   endianness Endian;
55   Writer(raw_ostream &OS, endianness Endian) : OS(OS), Endian(Endian) {}
56   template <typename value_type> void write(ArrayRef<value_type> Val) {
57     endian::write(OS, Val, Endian);
58   }
59   template <typename value_type> void write(value_type Val) {
60     endian::write(OS, Val, Endian);
61   }
62 };
63 
64 } // end namespace endian
65 
66 } // end namespace support
67 } // end namespace llvm
68 
69 #endif
70