parent
07cd0c252d
commit
d5dd86ed24
GPG Key ID:
284488987C31F630
@ -1,50 +1,73 @@
|
|||||||
SRC = $(wildcard src/*.c)
|
SRC_C = $(wildcard src/*.c)
|
||||||
INCLUDE = include/
|
SRC_CXX = $(wildcard src/*.cpp)
|
||||||
|
SRC_CXX+= $(wildcard src/rng/*.cpp)
|
||||||
|
|
||||||
|
INCLUDE = include
|
||||||
|
|
||||||
PROJECT=shuffle3
|
PROJECT=shuffle3
|
||||||
|
|
||||||
BUILD=build
|
COMMON_FLAGS = -Wall -pedantic $(addprefix -I,$(INCLUDE)) -fno-strict-aliasing
|
||||||
|
|
||||||
OPT_FLAGS?= -march=native -fgraphite -fopenmp -floop-parallelize-all -ftree-parallelize-loops=4 \
|
OPT_FLAGS?= -march=native -fgraphite -fopenmp -floop-parallelize-all -ftree-parallelize-loops=4 \
|
||||||
-floop-interchange -ftree-loop-distribution -floop-strip-mine -floop-block
|
-floop-interchange -ftree-loop-distribution -floop-strip-mine -floop-block \
|
||||||
|
-fno-stack-check
|
||||||
|
|
||||||
|
CXX_OPT_FLAGS?= $(OPT_FLAGS) -felide-constructors
|
||||||
|
|
||||||
CFLAGS+= $(addprefix -I,$(INCLUDE)) -Wall -pedantic --std=gnu11
|
CFLAGS += $(COMMON_FLAGS) --std=gnu11
|
||||||
LDFLAGS+= -lm
|
CXXFLAGS += $(COMMON_FLAGS) --std=gnu++20 -fno-exceptions
|
||||||
|
LDFLAGS += -lfmt
|
||||||
|
|
||||||
|
STRIP=strip
|
||||||
|
|
||||||
RELEASE_CFLAGS?= -O3 -flto $(OPT_FLAGS)
|
RELEASE_CFLAGS?= -O3 -flto $(OPT_FLAGS)
|
||||||
|
RELEASE_CXXFLAGS?= -O3 -flto $(CXX_OPT_FLAGS)
|
||||||
RELEASE_LDFLAGS?= -O3 -flto
|
RELEASE_LDFLAGS?= -O3 -flto
|
||||||
|
|
||||||
DEBUG_CFLAGS?= -g -O0
|
DEBUG_CFLAGS?= -O0 -g -DDEBUG
|
||||||
DEBUG_LDFLAGS?= -O0
|
DEBUG_CXXFLAGS?= $(DEBUG_CFLAGS)
|
||||||
|
DEBUG_LDFLAGS?=
|
||||||
|
|
||||||
STRIP=strip
|
# Objects
|
||||||
|
|
||||||
OBJ = $(addprefix obj/,$(SRC:.c=.o))
|
OBJ_C = $(addprefix obj/c/,$(SRC_C:.c=.o))
|
||||||
|
OBJ_CXX = $(addprefix obj/cxx/,$(SRC_CXX:.cpp=.o))
|
||||||
|
OBJ = $(OBJ_C) $(OBJ_CXX)
|
||||||
|
|
||||||
|
# Phonies
|
||||||
|
|
||||||
.PHONY: release
|
.PHONY: release
|
||||||
release: | dirs $(BUILD)/$(PROJECT)-release
|
release: | dirs $(PROJECT)-release
|
||||||
|
|
||||||
.PHONY: debug
|
.PHONY: debug
|
||||||
debug: | dirs $(BUILD)/$(PROJECT)-debug
|
debug: | dirs $(PROJECT)-debug
|
||||||
|
|
||||||
|
# Targets
|
||||||
|
|
||||||
dirs:
|
dirs:
|
||||||
@mkdir -p obj/src
|
@mkdir -p obj/c{,xx}/src
|
||||||
@mkdir -p $(BUILD)
|
@mkdir -p obj/cxx/src/rng
|
||||||
|
|
||||||
obj/%.o: %.c
|
obj/c/%.o: %.c
|
||||||
$(CC) -c $< $(CFLAGS) -o $@ $(LDFLAGS)
|
$(CC) -c $< $(CFLAGS) -o $@ $(LDFLAGS)
|
||||||
|
|
||||||
$(BUILD)/$(PROJECT)-release: CFLAGS+= $(RELEASE_CFLAGS)
|
obj/cxx/%.o: %.cpp
|
||||||
$(BUILD)/$(PROJECT)-release: LDFLAGS+= $(RELEASE_LDFLAGS)
|
$(CXX) -c $< $(CXXFLAGS) -o $@ $(LDFLAGS)
|
||||||
$(BUILD)/$(PROJECT)-release: $(OBJ)
|
|
||||||
$(CC) $^ $(CFLAGS) -o $@ $(LDFLAGS)
|
|
||||||
|
$(PROJECT)-release: CFLAGS+= $(RELEASE_CFLAGS)
|
||||||
|
$(PROJECT)-release: CXXFLAGS += $(RELEASE_CXXFLAGS)
|
||||||
|
$(PROJECT)-release: LDFLAGS += $(RELEASE_LDFLAGS)
|
||||||
|
$(PROJECT)-release: $(OBJ)
|
||||||
|
$(CXX) $^ $(CXXFLAGS) -o $@ $(LDFLAGS)
|
||||||
$(STRIP) $@
|
$(STRIP) $@
|
||||||
|
|
||||||
$(BUILD)/$(PROJECT)-debug: CFLAGS+= $(DEBUG_CFLAGS)
|
$(PROJECT)-debug: CFLAGS+= $(DEBUG_CFLAGS)
|
||||||
$(BUILD)/$(PROJECT)-debug: LDFLAGS+= $(DEBUG_LDFLAGS)
|
$(PROJECT)-debug: CXXFLAGS += $(DEBUG_CXXFLAGS)
|
||||||
$(BUILD)/$(PROJECT)-debug: $(OBJ)
|
$(PROJECT)-debug: LDFLAGS += $(DEBUG_LDFLAGS)
|
||||||
$(CC) $^ $(CFLAGS) -o $@ $(LDFLAGS)
|
$(PROJECT)-debug: $(OBJ)
|
||||||
|
$(CXX) $^ $(CXXFLAGS) -o $@ $(LDFLAGS)
|
||||||
|
|
||||||
clean:
|
clean:
|
||||||
rm -rf obj
|
rm -rf obj
|
||||||
rm -f $(BUILD)/*
|
rm -f $(PROJECT)-{release,debug}
|
||||||
|
|||||||
@ -1 +0,0 @@
|
|||||||
Complete rewrite using `mmap` et al. Maybe in Rust of C++ idk
|
|
||||||
@ -1,25 +1,59 @@
|
|||||||
#ifndef _RNG_H
|
#ifndef _RNG_H
|
||||||
#define _RNG_H
|
#define _RNG_H
|
||||||
|
|
||||||
typedef struct rng_algo *RNG; //RNG algorithm reference
|
#include "shuffle3.h"
|
||||||
|
|
||||||
#define rng_reinterpret(rng, value, type) (*((type*)rng_next(rng, &value, sizeof(type))))
|
#ifdef __cplusplus
|
||||||
#define rng_reinterpret_new(rng, value, type) (*((type*)rng_next(rng, malloc(sizeof(type)), sizeof(type))))
|
extern "C" {
|
||||||
#define rng_reinterpret_stackalloc(rng, value, type) rng_reinterpret(rng, alloca(sizeof(type)), type)
|
#endif
|
||||||
|
|
||||||
#define RNG_LEXBIND(rng, type, name) type name = rng_reinterpret(rng, name, type)
|
enum rng_kind {
|
||||||
|
RNG_KIND_FRNG,
|
||||||
double rng_next_double(RNG algo);
|
RNG_KIND_DRNG,
|
||||||
int rng_next_int_bounded(RNG algo, int min, int max);
|
RNG_KIND_XORNG,
|
||||||
int rng_next_int(RNG algo, int max);
|
};
|
||||||
void* rng_next(RNG algo, void* data, int len);
|
|
||||||
int rng_chance(RNG algo, double d);
|
typedef struct rng_init
|
||||||
void rng_free(RNG algo);
|
{
|
||||||
void rng_seed(RNG algo, void* seed);
|
enum rng_kind kind;
|
||||||
RNG rng_new(RNG (*instantiate)(void));
|
union {
|
||||||
|
struct {
|
||||||
#define RNG_IMPL_DEFINITION(name) RNG __rng_impl_ ## name(void)
|
double state[2];
|
||||||
#define RNG_ALGO(name) &__rng_impl_ ## name
|
} frng;
|
||||||
#define RNG_NEW(name) rng_new(RNG_ALGO(name))
|
struct {
|
||||||
|
int32_t state;
|
||||||
|
} drng;
|
||||||
|
struct {
|
||||||
|
uint64_t state[2];
|
||||||
|
} xorng;
|
||||||
|
} init;
|
||||||
|
|
||||||
|
} rng_init_opt;
|
||||||
|
|
||||||
|
typedef struct rng_impl* _UNIQUE rng_t;
|
||||||
|
|
||||||
|
rng_t rng_new(rng_init_opt kind);
|
||||||
|
#define RNG_INIT(_kind,...) ((rng_init_opt){.kind=(_kind), .init.__VA_ARGS__ })
|
||||||
|
void rng_free(rng_t ptr);
|
||||||
|
|
||||||
|
// Tests
|
||||||
|
extern void rng_test();
|
||||||
|
extern void rng_test_spec(rng_t rng);
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
// RNG interfaces
|
||||||
|
#include <rng/frng.hpp>
|
||||||
|
#include <rng/drng.hpp>
|
||||||
|
#include <rng/xoroshiro128plus.hpp>
|
||||||
|
|
||||||
|
namespace rng {
|
||||||
|
void test_algo(RNG&& rng);
|
||||||
|
|
||||||
|
template<class R,
|
||||||
|
typename std::enable_if<std::is_base_of<RNG, R>::value>::value __fuck>
|
||||||
|
inline void test_algo(R&& rng) { test_algo(static_cast<RNG&&>(rng)); }
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
#endif /* _RNG_H */
|
#endif /* _RNG_H */
|
||||||
|
|||||||
@ -1,73 +0,0 @@
|
|||||||
SRC_C = $(wildcard src/*.c)
|
|
||||||
SRC_CXX = $(wildcard src/*.cpp)
|
|
||||||
SRC_CXX+= $(wildcard src/rng/*.cpp)
|
|
||||||
|
|
||||||
INCLUDE = include
|
|
||||||
|
|
||||||
PROJECT=shuffle3
|
|
||||||
|
|
||||||
COMMON_FLAGS = -Wall -pedantic $(addprefix -I,$(INCLUDE)) -fno-strict-aliasing
|
|
||||||
|
|
||||||
OPT_FLAGS?= -march=native -fgraphite -fopenmp -floop-parallelize-all -ftree-parallelize-loops=4 \
|
|
||||||
-floop-interchange -ftree-loop-distribution -floop-strip-mine -floop-block \
|
|
||||||
-fno-stack-check
|
|
||||||
|
|
||||||
CXX_OPT_FLAGS?= $(OPT_FLAGS) -felide-constructors
|
|
||||||
|
|
||||||
CFLAGS += $(COMMON_FLAGS) --std=gnu11
|
|
||||||
CXXFLAGS += $(COMMON_FLAGS) --std=gnu++20 -fno-exceptions
|
|
||||||
LDFLAGS += -lfmt
|
|
||||||
|
|
||||||
STRIP=strip
|
|
||||||
|
|
||||||
RELEASE_CFLAGS?= -O3 -flto $(OPT_FLAGS)
|
|
||||||
RELEASE_CXXFLAGS?= -O3 -flto $(CXX_OPT_FLAGS)
|
|
||||||
RELEASE_LDFLAGS?= -O3 -flto
|
|
||||||
|
|
||||||
DEBUG_CFLAGS?= -O0 -g -DDEBUG
|
|
||||||
DEBUG_CXXFLAGS?= $(DEBUG_CFLAGS)
|
|
||||||
DEBUG_LDFLAGS?=
|
|
||||||
|
|
||||||
# Objects
|
|
||||||
|
|
||||||
OBJ_C = $(addprefix obj/c/,$(SRC_C:.c=.o))
|
|
||||||
OBJ_CXX = $(addprefix obj/cxx/,$(SRC_CXX:.cpp=.o))
|
|
||||||
OBJ = $(OBJ_C) $(OBJ_CXX)
|
|
||||||
|
|
||||||
# Phonies
|
|
||||||
|
|
||||||
.PHONY: release
|
|
||||||
release: | dirs $(PROJECT)-release
|
|
||||||
|
|
||||||
.PHONY: debug
|
|
||||||
debug: | dirs $(PROJECT)-debug
|
|
||||||
|
|
||||||
# Targets
|
|
||||||
|
|
||||||
dirs:
|
|
||||||
@mkdir -p obj/c{,xx}/src
|
|
||||||
@mkdir -p obj/cxx/src/rng
|
|
||||||
|
|
||||||
obj/c/%.o: %.c
|
|
||||||
$(CC) -c $< $(CFLAGS) -o $@ $(LDFLAGS)
|
|
||||||
|
|
||||||
obj/cxx/%.o: %.cpp
|
|
||||||
$(CXX) -c $< $(CXXFLAGS) -o $@ $(LDFLAGS)
|
|
||||||
|
|
||||||
|
|
||||||
$(PROJECT)-release: CFLAGS+= $(RELEASE_CFLAGS)
|
|
||||||
$(PROJECT)-release: CXXFLAGS += $(RELEASE_CXXFLAGS)
|
|
||||||
$(PROJECT)-release: LDFLAGS += $(RELEASE_LDFLAGS)
|
|
||||||
$(PROJECT)-release: $(OBJ)
|
|
||||||
$(CXX) $^ $(CXXFLAGS) -o $@ $(LDFLAGS)
|
|
||||||
$(STRIP) $@
|
|
||||||
|
|
||||||
$(PROJECT)-debug: CFLAGS+= $(DEBUG_CFLAGS)
|
|
||||||
$(PROJECT)-debug: CXXFLAGS += $(DEBUG_CXXFLAGS)
|
|
||||||
$(PROJECT)-debug: LDFLAGS += $(DEBUG_LDFLAGS)
|
|
||||||
$(PROJECT)-debug: $(OBJ)
|
|
||||||
$(CXX) $^ $(CXXFLAGS) -o $@ $(LDFLAGS)
|
|
||||||
|
|
||||||
clean:
|
|
||||||
rm -rf obj
|
|
||||||
rm -f $(PROJECT)-{release,debug}
|
|
||||||
@ -1,59 +0,0 @@
|
|||||||
#ifndef _RNG_H
|
|
||||||
#define _RNG_H
|
|
||||||
|
|
||||||
#include "shuffle3.h"
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
extern "C" {
|
|
||||||
#endif
|
|
||||||
|
|
||||||
enum rng_kind {
|
|
||||||
RNG_KIND_FRNG,
|
|
||||||
RNG_KIND_DRNG,
|
|
||||||
RNG_KIND_XORNG,
|
|
||||||
};
|
|
||||||
|
|
||||||
typedef struct rng_init
|
|
||||||
{
|
|
||||||
enum rng_kind kind;
|
|
||||||
union {
|
|
||||||
struct {
|
|
||||||
double state[2];
|
|
||||||
} frng;
|
|
||||||
struct {
|
|
||||||
int32_t state;
|
|
||||||
} drng;
|
|
||||||
struct {
|
|
||||||
uint64_t state[2];
|
|
||||||
} xorng;
|
|
||||||
} init;
|
|
||||||
|
|
||||||
} rng_init_opt;
|
|
||||||
|
|
||||||
typedef struct rng_impl* _UNIQUE rng_t;
|
|
||||||
|
|
||||||
rng_t rng_new(rng_init_opt kind);
|
|
||||||
#define RNG_INIT(_kind,...) ((rng_init_opt){.kind=(_kind), .init.__VA_ARGS__ })
|
|
||||||
void rng_free(rng_t ptr);
|
|
||||||
|
|
||||||
// Tests
|
|
||||||
extern void rng_test();
|
|
||||||
extern void rng_test_spec(rng_t rng);
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
}
|
|
||||||
// RNG interfaces
|
|
||||||
#include <rng/frng.hpp>
|
|
||||||
#include <rng/drng.hpp>
|
|
||||||
#include <rng/xoroshiro128plus.hpp>
|
|
||||||
|
|
||||||
namespace rng {
|
|
||||||
void test_algo(RNG&& rng);
|
|
||||||
|
|
||||||
template<class R,
|
|
||||||
typename std::enable_if<std::is_base_of<RNG, R>::value>::value __fuck>
|
|
||||||
inline void test_algo(R&& rng) { test_algo(static_cast<RNG&&>(rng)); }
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif /* _RNG_H */
|
|
||||||
@ -1,78 +0,0 @@
|
|||||||
#include <stdlib.h>
|
|
||||||
#include <stdio.h>
|
|
||||||
|
|
||||||
#include <string.h>
|
|
||||||
|
|
||||||
|
|
||||||
#include <shuffle3.h>
|
|
||||||
#include <panic.h>
|
|
||||||
#include <reinterpret.h>
|
|
||||||
#include <map.h>
|
|
||||||
#include <rng.h>
|
|
||||||
#include <work.h>
|
|
||||||
|
|
||||||
#include <debug.h>
|
|
||||||
|
|
||||||
#define noreturn __attribute__((noreturn)) void
|
|
||||||
|
|
||||||
_Static_assert(sizeof(float)==sizeof(uint32_t), "float is not 32 bits");
|
|
||||||
|
|
||||||
const char* _prog_name;
|
|
||||||
|
|
||||||
noreturn help_then_exit()
|
|
||||||
{
|
|
||||||
fprintf(stderr, "Try passing `-h`\n");
|
|
||||||
exit(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
void usage()
|
|
||||||
{
|
|
||||||
printf( "shuffle3 - 3 pass binary shuffler\n"
|
|
||||||
"Usage: %s -s <file>\n"
|
|
||||||
"Usage: %s -u <file>\n", _prog_name, _prog_name);
|
|
||||||
printf("\nOPTIONS:\n"
|
|
||||||
" -s\tShuffle file in place\n"
|
|
||||||
" -u\tUnshuffle file in place\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
int main(int argc, char** argv)
|
|
||||||
{
|
|
||||||
_prog_name = argv[0];
|
|
||||||
|
|
||||||
work_args_t parsed;
|
|
||||||
|
|
||||||
if( !argv[1] || *(argv[1]) != '-') help_then_exit();
|
|
||||||
|
|
||||||
D_dprintf("Parsing `%c'", argv[1][1]);
|
|
||||||
switch(argv[1][1])
|
|
||||||
{
|
|
||||||
case 's':
|
|
||||||
parsed.op = OP_SHUFFLE_IP;
|
|
||||||
if(!(parsed.data.op_shuffle_ip.file = argv[2]))
|
|
||||||
{
|
|
||||||
fprintf(stderr, "Error: -s expected file argument.\n");
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
D_dprintf("parsed.op = %d", OP_SHUFFLE_IP);
|
|
||||||
break;
|
|
||||||
case 'u':
|
|
||||||
parsed.op = OP_UNSHUFFLE_IP;
|
|
||||||
if(!(parsed.data.op_unshuffle_ip.file = argv[2]))
|
|
||||||
{
|
|
||||||
fprintf(stderr, "Error: -u expected file argument.\n");
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
D_dprintf("parsed.op = %d", OP_UNSHUFFLE_IP);
|
|
||||||
break;
|
|
||||||
case 'h':
|
|
||||||
usage();
|
|
||||||
return 0;
|
|
||||||
default:
|
|
||||||
fprintf(stderr, "Error: unknown argument `%s'\n\n", argv[1]);
|
|
||||||
help_then_exit();
|
|
||||||
panic("Unreachable");
|
|
||||||
}
|
|
||||||
|
|
||||||
return do_work(parsed);
|
|
||||||
}
|
|
||||||
|
|
||||||
@ -0,0 +1,50 @@
|
|||||||
|
SRC = $(wildcard src/*.c)
|
||||||
|
INCLUDE = include/
|
||||||
|
|
||||||
|
PROJECT=shuffle3
|
||||||
|
|
||||||
|
BUILD=build
|
||||||
|
|
||||||
|
OPT_FLAGS?= -march=native -fgraphite -fopenmp -floop-parallelize-all -ftree-parallelize-loops=4 \
|
||||||
|
-floop-interchange -ftree-loop-distribution -floop-strip-mine -floop-block
|
||||||
|
|
||||||
|
CFLAGS+= $(addprefix -I,$(INCLUDE)) -Wall -pedantic --std=gnu11
|
||||||
|
LDFLAGS+= -lm
|
||||||
|
|
||||||
|
RELEASE_CFLAGS?= -O3 -flto $(OPT_FLAGS)
|
||||||
|
RELEASE_LDFLAGS?= -O3 -flto
|
||||||
|
|
||||||
|
DEBUG_CFLAGS?= -g -O0
|
||||||
|
DEBUG_LDFLAGS?= -O0
|
||||||
|
|
||||||
|
STRIP=strip
|
||||||
|
|
||||||
|
OBJ = $(addprefix obj/,$(SRC:.c=.o))
|
||||||
|
|
||||||
|
.PHONY: release
|
||||||
|
release: | dirs $(BUILD)/$(PROJECT)-release
|
||||||
|
|
||||||
|
.PHONY: debug
|
||||||
|
debug: | dirs $(BUILD)/$(PROJECT)-debug
|
||||||
|
|
||||||
|
dirs:
|
||||||
|
@mkdir -p obj/src
|
||||||
|
@mkdir -p $(BUILD)
|
||||||
|
|
||||||
|
obj/%.o: %.c
|
||||||
|
$(CC) -c $< $(CFLAGS) -o $@ $(LDFLAGS)
|
||||||
|
|
||||||
|
$(BUILD)/$(PROJECT)-release: CFLAGS+= $(RELEASE_CFLAGS)
|
||||||
|
$(BUILD)/$(PROJECT)-release: LDFLAGS+= $(RELEASE_LDFLAGS)
|
||||||
|
$(BUILD)/$(PROJECT)-release: $(OBJ)
|
||||||
|
$(CC) $^ $(CFLAGS) -o $@ $(LDFLAGS)
|
||||||
|
$(STRIP) $@
|
||||||
|
|
||||||
|
$(BUILD)/$(PROJECT)-debug: CFLAGS+= $(DEBUG_CFLAGS)
|
||||||
|
$(BUILD)/$(PROJECT)-debug: LDFLAGS+= $(DEBUG_LDFLAGS)
|
||||||
|
$(BUILD)/$(PROJECT)-debug: $(OBJ)
|
||||||
|
$(CC) $^ $(CFLAGS) -o $@ $(LDFLAGS)
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -rf obj
|
||||||
|
rm -f $(BUILD)/*
|
||||||
@ -0,0 +1,25 @@
|
|||||||
|
#ifndef _RNG_H
|
||||||
|
#define _RNG_H
|
||||||
|
|
||||||
|
typedef struct rng_algo *RNG; //RNG algorithm reference
|
||||||
|
|
||||||
|
#define rng_reinterpret(rng, value, type) (*((type*)rng_next(rng, &value, sizeof(type))))
|
||||||
|
#define rng_reinterpret_new(rng, value, type) (*((type*)rng_next(rng, malloc(sizeof(type)), sizeof(type))))
|
||||||
|
#define rng_reinterpret_stackalloc(rng, value, type) rng_reinterpret(rng, alloca(sizeof(type)), type)
|
||||||
|
|
||||||
|
#define RNG_LEXBIND(rng, type, name) type name = rng_reinterpret(rng, name, type)
|
||||||
|
|
||||||
|
double rng_next_double(RNG algo);
|
||||||
|
int rng_next_int_bounded(RNG algo, int min, int max);
|
||||||
|
int rng_next_int(RNG algo, int max);
|
||||||
|
void* rng_next(RNG algo, void* data, int len);
|
||||||
|
int rng_chance(RNG algo, double d);
|
||||||
|
void rng_free(RNG algo);
|
||||||
|
void rng_seed(RNG algo, void* seed);
|
||||||
|
RNG rng_new(RNG (*instantiate)(void));
|
||||||
|
|
||||||
|
#define RNG_IMPL_DEFINITION(name) RNG __rng_impl_ ## name(void)
|
||||||
|
#define RNG_ALGO(name) &__rng_impl_ ## name
|
||||||
|
#define RNG_NEW(name) rng_new(RNG_ALGO(name))
|
||||||
|
|
||||||
|
#endif /* _RNG_H */
|
||||||
@ -0,0 +1,373 @@
|
|||||||
|
#include <stdio.h>
|
||||||
|
#include <float.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <assert.h>
|
||||||
|
|
||||||
|
#include <rng.h>
|
||||||
|
#include <rng_algos.h>
|
||||||
|
#include <ptr_store.h>
|
||||||
|
#include <array.h>
|
||||||
|
|
||||||
|
_Static_assert(sizeof(float)==4, "Float must be 32 bits.");
|
||||||
|
|
||||||
|
int surpress_out=0;
|
||||||
|
|
||||||
|
/*void test_destructor(object* ptr)
|
||||||
|
{
|
||||||
|
printf("die\n");
|
||||||
|
}*/
|
||||||
|
void _rng_object_destructor(object* obj)
|
||||||
|
{
|
||||||
|
rng_free(obj->state);
|
||||||
|
}
|
||||||
|
RNG rng_object(S_LEXENV, RNG rng)
|
||||||
|
{
|
||||||
|
object proto = OBJ_PROTO;
|
||||||
|
proto.state = rng;
|
||||||
|
proto.destructor = &_rng_object_destructor;
|
||||||
|
|
||||||
|
return snew_obj(proto)->state;
|
||||||
|
}
|
||||||
|
|
||||||
|
void shuffle(RNG with, array_t data)
|
||||||
|
{
|
||||||
|
if(!surpress_out) printf(" -> shuffling %d objects...", (int)ar_size(data));
|
||||||
|
for(int i=ar_size(data)-1;i>0;i--)
|
||||||
|
{
|
||||||
|
int j = rng_next_int(with, i);
|
||||||
|
ar_swap(data, i, j);
|
||||||
|
}
|
||||||
|
if(!surpress_out) printf(" Okay\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
void unshuffle(RNG with, array_t data)
|
||||||
|
{
|
||||||
|
int rng_values[ar_size(data)-1];
|
||||||
|
int k=0;
|
||||||
|
if(!surpress_out) printf(" -> unshuffling %d objects...", (int)ar_size(data));
|
||||||
|
for(int i=ar_size(data)-1;i>0;i--)
|
||||||
|
rng_values[k++] = rng_next_int(with, i);
|
||||||
|
|
||||||
|
for(int i=1;i<ar_size(data);i++)
|
||||||
|
{
|
||||||
|
ar_swap(data, i, rng_values[--k]);
|
||||||
|
}
|
||||||
|
if(!surpress_out) printf(" Okay\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
void minmax_int64_ts(int64_t* min, int64_t* max, const array_t data)
|
||||||
|
{
|
||||||
|
for(register int i=0;i<ar_size(data);i++)
|
||||||
|
{
|
||||||
|
if(ar_get_v(data, int64_t, i) > *max) *max = ar_get_v(data, int64_t, i);
|
||||||
|
if(ar_get_v(data, int64_t, i) < *min) *min = ar_get_v(data, int64_t, i);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
int valid_float(float f)
|
||||||
|
{
|
||||||
|
return !( (f!=f) || (f< -FLT_MAX || f> FLT_MAX));
|
||||||
|
|
||||||
|
}
|
||||||
|
void minmax_floats(float* min, float* max, const array_t data)
|
||||||
|
{
|
||||||
|
for(register int i=0;i<ar_size(data);i++)
|
||||||
|
{
|
||||||
|
if(!valid_float(ar_get_v(data,float,i))) continue;
|
||||||
|
if(ar_get_v(data, float, i) > *max) *max = ar_get_v(data, float, i);
|
||||||
|
if(ar_get_v(data, float, i) < *min) *min = ar_get_v(data, float, i);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void minmax_sbytes(int8_t* min, int8_t* max, const array_t data)
|
||||||
|
{
|
||||||
|
for(register int i=0;i<ar_size(data);i++)
|
||||||
|
{
|
||||||
|
if(ar_get_v(data, int8_t, i) > *max) *max = ar_get_v(data, int8_t, i);
|
||||||
|
if(ar_get_v(data, int8_t, i) < *min) *min = ar_get_v(data, int8_t, i);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void print_array(const array_t data)
|
||||||
|
{
|
||||||
|
printf("---%d elements---\n", (int)ar_size(data));
|
||||||
|
for(register int i=0;i<ar_size(data);i++)
|
||||||
|
printf("%d ", (int)ar_get_v(data, unsigned char, i));
|
||||||
|
printf("\n---\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
void unshuffle3(S_LEXENV, array_t data)
|
||||||
|
{
|
||||||
|
if(ar_type(data)!=sizeof(int8_t))
|
||||||
|
ar_reinterpret(data, sizeof(int8_t));
|
||||||
|
RNG xoro = S_LEXCALL(rng_object, rng_new(RNG_ALGO(xoroshiro128plus)));
|
||||||
|
RNG frng = S_LEXCALL(rng_object, rng_new(RNG_ALGO(frng)));
|
||||||
|
RNG drng = S_LEXCALL(rng_object, rng_new(RNG_ALGO(drng)));
|
||||||
|
|
||||||
|
int8_t bmax = INT8_MIN;
|
||||||
|
int8_t bmin = INT8_MAX;
|
||||||
|
minmax_sbytes(&bmin, &bmax, data);
|
||||||
|
uint32_t seed = (0xfffa << 16) | (bmin<<8) | bmax;
|
||||||
|
rng_seed(drng, &seed);
|
||||||
|
unshuffle(drng, data);
|
||||||
|
|
||||||
|
float fmin = (float)DBL_MAX;
|
||||||
|
float fmax = (float)-DBL_MAX;
|
||||||
|
ar_reinterpret(data, sizeof(float));
|
||||||
|
minmax_floats(&fmin, &fmax, data);
|
||||||
|
double fseed[2];
|
||||||
|
fseed[0] = fmin;
|
||||||
|
fseed[1] = fmax;
|
||||||
|
rng_seed(frng, fseed);
|
||||||
|
unshuffle(frng, data);
|
||||||
|
|
||||||
|
int64_t min = INT64_MAX;
|
||||||
|
int64_t max = INT64_MIN;
|
||||||
|
ar_reinterpret(data, sizeof(int64_t));
|
||||||
|
minmax_int64_ts(&min, &max, data);
|
||||||
|
uint64_t xseed[2];
|
||||||
|
xseed[0] = min;
|
||||||
|
xseed[1] = max;
|
||||||
|
rng_seed(xoro, xseed);
|
||||||
|
unshuffle(xoro, data);
|
||||||
|
}
|
||||||
|
void shuffle3(S_LEXENV, array_t data)
|
||||||
|
{
|
||||||
|
if(ar_type(data)!=sizeof(int64_t))
|
||||||
|
ar_reinterpret(data, sizeof(int64_t));
|
||||||
|
RNG xoro = S_LEXCALL(rng_object, rng_new(RNG_ALGO(xoroshiro128plus)));
|
||||||
|
RNG frng = S_LEXCALL(rng_object, rng_new(RNG_ALGO(frng)));
|
||||||
|
RNG drng = S_LEXCALL(rng_object, rng_new(RNG_ALGO(drng)));
|
||||||
|
|
||||||
|
int64_t min = INT64_MAX;
|
||||||
|
int64_t max = INT64_MIN;
|
||||||
|
|
||||||
|
float fmin = (float)DBL_MAX;
|
||||||
|
float fmax = (float)-DBL_MAX;
|
||||||
|
|
||||||
|
int8_t bmax = INT8_MIN;
|
||||||
|
int8_t bmin = INT8_MAX;
|
||||||
|
|
||||||
|
minmax_int64_ts(&min, &max, data);
|
||||||
|
uint64_t xseed[2];
|
||||||
|
xseed[0] = min;
|
||||||
|
xseed[1] = max;
|
||||||
|
|
||||||
|
rng_seed(xoro, xseed);
|
||||||
|
shuffle(xoro, data);
|
||||||
|
|
||||||
|
ar_reinterpret(data, sizeof(float));
|
||||||
|
minmax_floats(&fmin, &fmax, data);
|
||||||
|
|
||||||
|
double fseed[2];
|
||||||
|
fseed[0] = fmin;
|
||||||
|
fseed[1] = fmax;
|
||||||
|
rng_seed(frng, fseed);
|
||||||
|
shuffle(frng, data);
|
||||||
|
|
||||||
|
ar_reinterpret(data,sizeof(int8_t));
|
||||||
|
minmax_sbytes(&bmin, &bmax, data);
|
||||||
|
uint32_t seed = (0xfffa << 16) | (bmin<<8) | bmax;
|
||||||
|
rng_seed(drng, &seed);
|
||||||
|
shuffle(drng, data);
|
||||||
|
}
|
||||||
|
|
||||||
|
void print_usage(char** argv)
|
||||||
|
{
|
||||||
|
printf("Usage: %s -[b]s <file> [<outfile>]\nUsage: %s -[b]u <file> [<outfile>]\n", argv[0], argv[0]);
|
||||||
|
printf("Usage: %s -S <string>|-\nUsage: %s -U <string>|-\n", argv[0], argv[0]);
|
||||||
|
printf(" <file>\t\tFile to shuffle\n");
|
||||||
|
printf(" <outfile>\tOutput file (only valid for buffered mode)\n");
|
||||||
|
printf(" <string>\tThe string to shuffle (for -(S|U)). If `-`, read from stdin\n");
|
||||||
|
printf("\ts\tShuffle\n");
|
||||||
|
printf("\tu\tReverse shuffle\n");
|
||||||
|
printf("\tS\tShuffle string\n");
|
||||||
|
printf("\tU\tReverse shuffle string\n");
|
||||||
|
printf("\n\tb\tUse buffered mode instead of shuffling in place (must specify output file)\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
array_t read_whole_file(S_NAMED_LEXENV(base), FILE* fp)
|
||||||
|
{
|
||||||
|
fseek(fp,0,SEEK_END);
|
||||||
|
int64_t sz = ftell(fp);
|
||||||
|
fseek(fp,0,SEEK_SET);
|
||||||
|
|
||||||
|
array_t ar;
|
||||||
|
MANAGED({
|
||||||
|
void* buf = smalloc(sz);
|
||||||
|
|
||||||
|
register size_t read;
|
||||||
|
register size_t full=0;
|
||||||
|
while ((read=fread(buf, 1, sz-full, fp))>0)
|
||||||
|
full+=read;
|
||||||
|
|
||||||
|
assert(full == sz);
|
||||||
|
|
||||||
|
ar = ar_create_memory_from(base, buf, 1, sz);
|
||||||
|
});
|
||||||
|
return ar;
|
||||||
|
}
|
||||||
|
|
||||||
|
#define BUFSIZE 512
|
||||||
|
#define BUFMIN 128
|
||||||
|
|
||||||
|
char* read_stdin()
|
||||||
|
{
|
||||||
|
char* input, *p;
|
||||||
|
int len,remain,n,size;
|
||||||
|
|
||||||
|
size = BUFSIZE;
|
||||||
|
input = malloc(size);
|
||||||
|
memset(input,0,size);
|
||||||
|
len=0;
|
||||||
|
remain = size;
|
||||||
|
|
||||||
|
while(!feof(stdin))
|
||||||
|
{
|
||||||
|
if(remain<= BUFMIN)
|
||||||
|
{
|
||||||
|
remain+=size;
|
||||||
|
size *=2;
|
||||||
|
p = realloc(input, size);
|
||||||
|
if(p==NULL) {
|
||||||
|
free(input);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
input = p;
|
||||||
|
}
|
||||||
|
fgets(input+len, remain, stdin);
|
||||||
|
n+=strlen(input+len);
|
||||||
|
len+=n;
|
||||||
|
remain-=n;
|
||||||
|
}
|
||||||
|
return input;
|
||||||
|
}
|
||||||
|
|
||||||
|
int string_shuffle(char* string, int un)
|
||||||
|
{
|
||||||
|
if(!string)
|
||||||
|
{
|
||||||
|
char* ptr = read_stdin();
|
||||||
|
if(!ptr) {
|
||||||
|
printf("! read from stdin failed\n");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
int ret = string_shuffle(ptr, un);
|
||||||
|
free(ptr);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
else if(strcmp(string, "-") == 0)
|
||||||
|
{
|
||||||
|
return string_shuffle(NULL, un);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
surpress_out=1;
|
||||||
|
MANAGED({
|
||||||
|
array_t ar = ar_create_memory_from(LEXENV, string, 1, strlen(string));
|
||||||
|
|
||||||
|
if(un) unshuffle3(LEXENV, ar);
|
||||||
|
else shuffle3(LEXENV, ar);
|
||||||
|
|
||||||
|
ar_reinterpret(ar, sizeof(char));
|
||||||
|
|
||||||
|
for(int i=0;i<ar_size(ar);i++)
|
||||||
|
{
|
||||||
|
printf("%c", ar_get_v(ar, char, i));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char** argv)
|
||||||
|
{
|
||||||
|
int is_buffered=0;
|
||||||
|
int i=1;
|
||||||
|
int is_unshuffling =0;
|
||||||
|
if(!argv[1] || !argv[2] || argv[1][0] != '-')
|
||||||
|
{
|
||||||
|
print_usage(argv);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
do_switch:
|
||||||
|
switch(argv[1][i])
|
||||||
|
{
|
||||||
|
case 'b':
|
||||||
|
if(is_buffered)
|
||||||
|
{
|
||||||
|
print_usage(argv);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
is_buffered = 1;
|
||||||
|
i+=1;
|
||||||
|
goto do_switch;
|
||||||
|
case 'u':
|
||||||
|
is_unshuffling=1;
|
||||||
|
break;
|
||||||
|
case 's':
|
||||||
|
break;
|
||||||
|
case 'S':
|
||||||
|
return string_shuffle(argv[2], 0);
|
||||||
|
case 'U':
|
||||||
|
return string_shuffle(argv[2], 1);
|
||||||
|
default:
|
||||||
|
print_usage(argv);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if(is_buffered && !argv[3])
|
||||||
|
{
|
||||||
|
printf("Buffered option requires an output file\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
MANAGED_RETURNABLE(int ,{
|
||||||
|
array_t array;
|
||||||
|
|
||||||
|
if(is_buffered)
|
||||||
|
{
|
||||||
|
FILE* infile = fopen(argv[2], "rb");
|
||||||
|
if(!infile)
|
||||||
|
{
|
||||||
|
printf("! could not open file for reading\n");
|
||||||
|
MANAGED_RETURN(1);
|
||||||
|
}
|
||||||
|
array = read_whole_file(LEXENV, infile);
|
||||||
|
fclose(infile);
|
||||||
|
printf(" buffered file (%ld bytes)\n", ar_size(array));
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
FILE* infile = fopen(argv[2], "r+b");
|
||||||
|
if(!infile)
|
||||||
|
{
|
||||||
|
printf("! could not open file for reading+writing\n");
|
||||||
|
MANAGED_RETURN(1);
|
||||||
|
}
|
||||||
|
array = ar_create_file(LEXENV, infile, 1, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
//print_array(array);
|
||||||
|
|
||||||
|
if(is_unshuffling) unshuffle3(LEXENV, array);
|
||||||
|
else shuffle3(LEXENV, array);
|
||||||
|
|
||||||
|
if(is_buffered)
|
||||||
|
{
|
||||||
|
FILE* outfile = fopen(argv[3], "wb");
|
||||||
|
if(!outfile)
|
||||||
|
{
|
||||||
|
printf("! could not open outfile for writing\n");
|
||||||
|
MANAGED_RETURN(1);
|
||||||
|
}
|
||||||
|
void* wbuf = smalloc(ar_full_size(array));
|
||||||
|
if(!ar_ndump(array, wbuf, ar_full_size(array), 0, ar_size(array)))
|
||||||
|
printf("W memory dump failed, continuing anyway\n");
|
||||||
|
fwrite(wbuf, 1, ar_full_size(array), outfile);
|
||||||
|
fclose(outfile);
|
||||||
|
printf(" write completed (%d bytes)\n", (int)ar_size(array));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Loading…
Reference in new issue