-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathset_float_data.cpp
More file actions
44 lines (36 loc) · 1.06 KB
/
set_float_data.cpp
File metadata and controls
44 lines (36 loc) · 1.06 KB
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
#include <iostream>
#include <fstream>
#include <cstdlib>
#include <string>
#include <stdexcept>
#include <cassert>
template <typename T>
static void dumpBufferToFile(const char* fileName, T* data, size_t dataNumItems)
{
std::ofstream outFile(fileName, std::ios::binary);
if(outFile)
{
outFile.write(reinterpret_cast<char*>(data), dataNumItems * sizeof(T));
outFile.close();
printf("Wrote output to file %s\n", fileName);
}
else
{
printf("Could not open file %s for writing\n", fileName);
}
}
int main(int argc, char* argv[])
{
if(argc != 4)
{
std::cerr << "Usage: " << argv[0] << " <file> <number of elements> <value>" << std::endl;
throw std::runtime_error("Invalid command argument!");
};
int numElements = atoi(argv[2]);
float value = static_cast<float>(atof(argv[3]));
assert(numElements > 0);
float* dataBuffer = new float[numElements];
for(int i = 0; i < numElements; i++)
dataBuffer[i] = value;
dumpBufferToFile(argv[1], dataBuffer, numElements);
};