Rodrigo Arias Mallo
8bc5656461
It allows getExperimentStage to be called from any stage above the experiment.
76 lines
2.5 KiB
Nix
76 lines
2.5 KiB
Nix
{
|
|
stdenv
|
|
}:
|
|
|
|
with stdenv.lib;
|
|
|
|
let
|
|
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; } ]
|
|
mergeConfig = (arr: new: flatten ( map (x: addAttrSets new.name new.value x) arr));
|
|
|
|
# genConfigs {a=[1 2]; b=[3 4];}
|
|
# [ { a = 1; b = 3; } { a = 1; b = 4; } { a = 2; b = 3; } { a = 2; b = 4; } ]
|
|
genConfigs = (config: foldl mergeConfig [{}] (attrToList config));
|
|
|
|
# Generate multiple app versions by override with each config
|
|
genApp = (app: configs: map (conf: app.override conf // {conf=conf;}) configs);
|
|
|
|
# Generate app version from an array of apps
|
|
genApps = (apps: configs:
|
|
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}";
|
|
|
|
/* Given a trebuchet, returns the experiment */
|
|
getExperimentStage = drv:
|
|
if (drv ? isExperiment) && drv.isExperiment then drv
|
|
else getExperimentStage drv.nextStage;
|
|
|
|
# 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:
|
|
let
|
|
_divList = n: if (n == 0) then [] else (_divList (n / 2)) ++ [ n ];
|
|
in
|
|
_divList N;
|
|
|
|
# A list of all divisors of n, sorted in increased order:
|
|
divisors = n: filter (x: (mod n x == 0)) (range 1 n);
|
|
|
|
# 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);
|
|
|
|
};
|
|
in
|
|
gen
|