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
|
#include "sdsl/int_vector.hpp"
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <string>
using namespace std;
using namespace sdsl;
ptrdiff_t myrandom(ptrdiff_t i)
{
return rand()%i;
}
ptrdiff_t (*p_myrandom)(ptrdiff_t) = myrandom;
int main(int argc, char* argv[])
{
if (argc < 5) {
cout << "Usage: " << argv[0] << " FILE SIZE WIDTH DEFAULT_VALUE [PERM_SEED]" << endl;
cout << " (1) Generates an int_vector<>(SIZE, DEFAULT_VALUE, WIDTH)" << endl;
cout << " Vector will be initialized with random bits, if " << endl;
cout << " DEFAULT_VALUE=r. If DEFAULT_VALUE=i, v will be set to" << endl;
cout << " the identity." << endl;
cout << " (2) If PERM_SEED is specified, a random_shuffle seeded with" << endl;
cout << " PERM_SEED will be performed." << endl;
cout << " (3) Stores the vector to FILE." << endl;
cout << "Only "<<argc<<" elements provided at current program call." << endl;
return 1;
}
uint64_t size = stoull(argv[2]);
uint64_t width = stoull(argv[3]);
int_vector<> v(size, 0, width);
if ('r' == argv[4][0]) {
util::set_random_bits(v);
} else if ('i' == argv[4][0]) {
util::set_to_id(v);
} else {
uint64_t default_value = stoull(argv[4]);
util::set_to_value(v, default_value);
}
if (argc > 5) {
unsigned long seed = stoul(argv[5]);
srand(seed);
random_shuffle(v.begin(), v.end(), p_myrandom);
}
store_to_file(v, argv[1]);
}
|
