default.nix

{
  hostPkgs ? import <nixpkgs> {},
  system ? builtins.currentSystem,
  pkgs         ? import ../pin.nix { inherit hostPkgs; jsonpath=../nixpkgs-18.03.json; depset = {inherit system;}; },
  kapack       ? import ../pin.nix { hostPkgs = pkgs ; jsonpath=../kapack.json; depset = {hostPkgs=pkgs;};},
 supportedSystems ? [ "x86_64-linux" "i686-linux" ]
}:
let
  releaseTools = import "${pkgs}/top-level/release-lib.nix" {
    inherit supportedSystems;
  };

  gitpin = f: pkgs.fetchgit {inherit (pkgs.lib.importJSON f) url sha256 rev fetchSubmodules;};
  compiler_path = "/opt/intel/compilers_and_libraries_2017.1.132/linux";
  callPackage = pkgs.lib.callPackageWith pkgs;
  callPackagePinWith =
  { folderPath
  , nixName ? "default"
  , jsonName ? nixName
  , depset ? {}
  } :
  let
    nixPath = folderPath + "/${nixName}.nix";
    jsonPath = folderPath + "/${jsonName}.json";
    src = gitpin jsonPath;
  in
  callPackage nixPath (depset // {inherit src;});

  cpDefaultPin = folderPath: depset: callPackagePinWith { inherit folderPath depset;};
  cpAlternatePin = folderPath: depset: jsonName: callPackagePinWith {inherit folderPath depset jsonName;};
  cpBothAlternatePin = folderPath: depset: jsonName: callPackagePinWith {inherit folderPath depset jsonName; nixName=jsonName;};
  pythonPackages = pkgs.python27Packages;
  
  argopkgs = rec {
    #NRM components
    tools = rec {
      inherit (kapack) zymake;
    };

    nodelevel = rec {
      libnrm = cpDefaultPin ./libnrm {};
      libnrm-pmpi = libnrm.overrideAttrs (oldAttrs: rec {
        configureFlags = "--enable-pmpi";
        buildInputs = oldAttrs.buildInputs ++ [ pkgs.mpich2 ];
      });
      containers = cpDefaultPin ./containers { inherit pythonPackages; };
      containers-bandit = cpAlternatePin ./containers { inherit pythonPackages; } "bandit";
      nrm = cpDefaultPin ./nrm { inherit pythonPackages; nrm-containers = containers; };
      nrm-bandit = cpAlternatePin ./nrm { inherit pythonPackages; nrm-containers=containers-bandit; } "bandit";
    };

    #Instrumented applications
    applications =
    let
      paths = {
       amg = applications/amg;
       #qmcpack = applications/qmcpack;
       #openmc = applications/openmc;
       stream = applications/stream;
       lammps = applications/lammps;
      };
      rawApplication = name : path: cpDefaultPin path {};
      instrumentedApplication = name: path:
        (cpAlternatePin path {} "nrm").overrideAttrs (oldAttrs:
        {
          buildInputs = oldAttrs.buildInputs ++ [nodelevel.libnrm];
          cmakeFlags = if (name=="qmcpack") then
          (oldAttrs.cmakeFlags ++ [ ''-DQMC_INCLUDE=${nodelevel.libnrm}/include/'' ''-DQMC_EXTRA_LIBS=${nodelevel.libnrm}/lib/libnrm.so'' ])
          else
          [];
        });
    in
    rec {
      raw = pkgs.lib.mapAttrs rawApplication paths;
      nrm = (pkgs.lib.mapAttrs instrumentedApplication paths) // {
        dgemm_randomwalk = cpBothAlternatePin applications/simple {inherit intel; inherit (nodelevel) libnrm;} "dgemm_randomwalk";
        graph500 = cpBothAlternatePin applications/simple { inherit (nodelevel) libnrm;} "graph500";
      };
    };

    #Intel compilers
    intel = with pkgs; rec {
      icc = callPackage intel/icc { inherit compiler_path; };
      mkl = callPackage intel/mkl { inherit compiler_path; };
      WrappedICC = (import intel/cc-wrapper  {
          inherit stdenv binutils coreutils ;
          libc = glibc;
          nativeTools = false;
          nativeLibc = false;
          cc = icc;
      });
    };

    argopkgs = pkgs.releaseTools.channel {
      name = "argopkgs";
      src = ./.;
      constituents = [ nodelevel ];
    };

  };

in argopkgs