bscpkgs/garlic/tools.nix

76 lines
2.5 KiB
Nix
Raw Normal View History

2020-10-09 21:55:37 +08:00
{
stdenv
}:
with stdenv.lib;
2020-07-27 17:14:33 +08:00
2020-10-09 21:55:37 +08:00
let
2020-07-27 17:14:33 +08: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 21:55:37 +08:00
mergeConfig = (arr: new: flatten ( map (x: addAttrSets new.name new.value x) arr));
2020-07-27 17:14:33 +08: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 21:55:37 +08:00
genConfigs = (config: foldl mergeConfig [{}] (attrToList config));
2020-07-27 17:14:33 +08:00
# Generate multiple app versions by override with each config
2020-07-30 00:38:39 +08:00
genApp = (app: configs: map (conf: app.override conf // {conf=conf;}) configs);
2020-07-27 17:14:33 +08:00
# Generate app version from an array of apps
genApps = (apps: configs:
2020-10-09 21:55:37 +08: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 17:14:33 +08:00
2020-10-16 00:48:50 +08:00
/* Given a trebuchet, returns the experiment */
getExperimentStage = drv:
if (drv ? isExperiment) && drv.isExperiment then drv
else getExperimentStage drv.nextStage;
2020-11-14 02:06:31 +08: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));
# Generates a list of integers by halving number N until it reaches 1. Is
# sorted from the smallest to largest.
halfList = N:
2020-11-14 02:06:31 +08:00
let
_divList = n: if (n == 0) then [] else (_divList (n / 2)) ++ [ n ];
in
_divList N;
2020-11-16 18:59:11 +08: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-14 02:06:31 +08: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 17:14:33 +08:00
};
in
gen