bscpkgs/garlic/tools.nix

83 lines
2.7 KiB
Nix
Raw Normal View History

2020-10-09 15:55:37 +02:00
{
stdenv
}:
with stdenv.lib;
2020-07-27 11:14:33 +02:00
2020-10-09 15:55:37 +02:00
let
2020-07-27 11:14:33 +02:00
gen = rec {
# genAttrSets "a" ["hello" "world"]
# [ { a = "hello"; } { a = "world"; } ]
genAttrSets = (name: arr: (map (x: {${name}=x; })) arr);
# addAttrSets "a" [1 2] {e=4;}
# [ { a = 1; e = 4; } { a = 2; e = 4; } ]
addAttrSets = (name: arr: set: (map (x: set // {${name}=x; })) arr);
# attrToList {a=1;}
# [ { name = "a"; value = 1; } ]
attrToList = (set: map (name: {name=name; value=set.${name};} ) (builtins.attrNames set));
# mergeConfig [{e=1;}] {name="a"; value=[1 2]
# [ { a = 1; e = 1; } { a = 2; e = 1; } ]
2020-10-09 15:55:37 +02:00
mergeConfig = (arr: new: flatten ( map (x: addAttrSets new.name new.value x) arr));
2020-07-27 11:14:33 +02:00
# genConfigs {a=[1 2]; b=[3 4];}
# [ { a = 1; b = 3; } { a = 1; b = 4; } { a = 2; b = 3; } { a = 2; b = 4; } ]
2020-10-09 15:55:37 +02:00
genConfigs = (config: foldl mergeConfig [{}] (attrToList config));
2020-07-27 11:14:33 +02:00
# Generate multiple app versions by override with each config
2020-07-29 18:38:39 +02:00
genApp = (app: configs: map (conf: app.override conf // {conf=conf;}) configs);
2020-07-27 11:14:33 +02:00
# Generate app version from an array of apps
genApps = (apps: configs:
2020-10-09 15:55:37 +02:00
flatten (map (app: genApp app configs) apps));
/* Returns the path of the executable of a stage */
stageProgram = stage:
if stage ? programPath
then "${stage}${stage.programPath}"
else "${stage}";
2020-07-27 11:14:33 +02:00
2020-10-15 18:48:50 +02:00
/* Given a trebuchet, returns the experiment */
getExperimentStage = drv:
if (drv ? isExperiment) && drv.isExperiment then drv
else getExperimentStage drv.nextStage;
2020-11-13 19:06:31 +01:00
# Computes the exponentiation operation
pow = x: n: fold (a: b: a*b) 1 (map (a: x) (range 1 n));
# Generates a list of exponents from a to b inclusive, and raises base to
# each element of the list.
expRange = base: a: b: (map (ex: pow base ex) (range a b));
2021-01-28 15:00:43 +01:00
# Generates a range from start to end (inclusive) by multiplying start by 2.
range2 = start: end:
let
_range2 = s: e: if (s > e) then [] else [ s ] ++ (_range2 (s * 2) e);
in
_range2 start end;
2020-11-13 19:06:31 +01:00
# Generates a list of integers by halving number N until it reaches 1. Is
# sorted from the smallest to largest.
halfList = N:
2020-11-13 19:06:31 +01:00
let
_divList = n: if (n == 0) then [] else (_divList (n / 2)) ++ [ n ];
in
_divList N;
2020-11-16 11:59:11 +01:00
# A list of all divisors of n, sorted in increased order:
divisors = n: filter (x: (mod n x == 0)) (range 1 n);
2020-11-13 19:06:31 +01:00
# Generates a set given a list of keys, where all values are null.
genNullAttr = l: genAttrs l (name: null);
# From the keys in the lis l, generates a set with the values in the set a,
# if they don't exist, they are not taken. Values set to null are removed.
optionalInherit = l: a: filterAttrs (n: v: v!=null)
(overrideExisting (genNullAttr l) a);
2020-07-27 11:14:33 +02:00
};
in
gen