Squashed commit of the following:

commit b54a13480c9da962c17481fa4b6b09a08af1ae95
Author: antares <antares@alyr.dev>
Date:   Fri Jul 25 22:21:51 2025 +0800

    Update

commit e6ff757
Author: antares <antares@alyr.dev>
Date:   Fri Jul 18 21:51:45 2025 +0800

    Add nix files

Author: Antares0982

.gitignore

@@ -1,5 +1,7 @@
 benchmark_result*.json
 /*.pdf
+/.nix-pyenv
+/.direnv
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

.vscode/settings.json

@@ -0,0 +1,15 @@
+{
+    "python.analysis.stubPath": ".nix-pyenv/lib",
+    "terminal.integrated.profiles.linux": {
+        "nix-shell": {
+            "path": "/run/current-system/sw/bin/nix-shell",
+            "icon": "terminal-bash"
+        },
+        "nix-dev": {
+            "path": "/run/current-system/sw/bin/nix",
+            "args": ["develop", "-c", "nix-zshell"],
+            "icon": "terminal-bash"
+        }
+    },
+    "terminal.integrated.defaultProfile.linux": "nix-dev"
+}

flake.lock

@@ -0,0 +1,34 @@
+{
+  description = "A simple flake for a simple python environment";
+
+  inputs = {
+    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
+  };
+
+  outputs =
+    { self, nixpkgs }:
+    let
+      forAllSystems =
+        function:
+        nixpkgs.lib.genAttrs
+          [
+            "x86_64-linux"
+            "aarch64-linux"
+            "x86_64-darwin"
+            "aarch64-darwin"
+          ]
+          (
+            system:
+            function (
+              import nixpkgs {
+                inherit system;
+              }
+            )
+          );
+    in
+    {
+      devShells = forAllSystems (pkgs: rec {
+        default = pkgs.callPackage ./shell.nix { persist = true; };
+      });
+    };
+}

flake.nix

@@ -0,0 +1,10 @@
+{ pkgs, ... }:
+pypkgs: with pypkgs; [
+  matplotlib
+  orjson
+  psutil
+  reportlab
+  svglib
+  build
+  pip
+]

py_requirements.nix

@@ -0,0 +1 @@
+python benchmark.py

run_benchmark.sh

@@ -0,0 +1,49 @@
+{
+  pkgs ? import <nixpkgs> { },
+  lib ? pkgs.lib,
+  persist ? false,
+  mkShell ? pkgs.mkShell,
+  callPackage ? pkgs.callPackage,
+}:
+let
+  optionalAttrs = lib.attrsets.optionalAttrs;
+  # define the nix-pyenv directory
+  nix-pyenv-directory = ".nix-pyenv";
+  # define version
+  usingPython = pkgs.python313;
+  # import required python packages
+  requiredPythonPackages = callPackage ./py_requirements.nix { };
+  # create python environment
+  pyenv = usingPython.withPackages requiredPythonPackages;
+  #
+  callShellHookParam = {
+    inherit
+      nix-pyenv-directory
+      pyenv
+      usingPython
+      persist
+      pkgs
+      ;
+  };
+  internalShell = mkShell (
+    {
+      packages = [
+        pyenv
+        pkgs.cmake
+      ];
+    }
+    // (optionalAttrs (!persist) {
+      shellHook = callPackage ./shellhook.nix callShellHookParam;
+    })
+  );
+in
+internalShell.overrideAttrs (
+  optionalAttrs persist {
+    shellHook = callPackage ./shellhook.nix (
+      callShellHookParam
+      // {
+        inherit (internalShell) inputDerivation;
+      }
+    );
+  }
+)

shell.nix

@@ -0,0 +1,62 @@
+{
+  nix,
+  nix-pyenv-directory,
+  pyenv,
+  usingPython,
+  persist,
+  inputDerivation ? "",
+  ...
+}:
+let
+  optionalCommentOut = if persist then "" else "#";
+  sitePackages = usingPython.sitePackages;
+in
+''
+  if [[ $name == nix-shell ]]; then
+      cd ${builtins.toString ./.}
+  fi
+
+  # ensure the nix-pyenv directory exists
+  if [[ ! -d ${nix-pyenv-directory} ]]; then mkdir ${nix-pyenv-directory}; fi
+  if [[ ! -d ${nix-pyenv-directory}/lib ]]; then mkdir ${nix-pyenv-directory}/lib; fi
+  if [[ ! -d ${nix-pyenv-directory}/bin ]]; then mkdir ${nix-pyenv-directory}/bin; fi
+
+  ensure_symlink() {
+      local link_path="$1"
+      local target_path="$2"
+      if [[ -L "$link_path" ]] && [[ "$(readlink "$link_path")" = "$target_path" ]]; then
+          return 0
+      fi
+      rm -f "$link_path" > /dev/null 2>&1
+      ln -s "$target_path" "$link_path"
+  }
+
+  # creating python library symlinks
+  for file in ${pyenv}/${sitePackages}/*; do
+      basefile=$(basename $file)
+      if [ -d "$file" ]; then
+          if [[ "$basefile" != *dist-info && "$basefile" != __pycache__ ]]; then
+              ensure_symlink "${nix-pyenv-directory}/lib/$basefile" $file
+          fi
+      else
+          # the typing_extensions.py will make the vscode type checker not working!
+          if [[ $basefile == *.so ]] || ([[ $basefile == *.py ]] && [[ $basefile != typing_extensions.py ]]); then
+              ensure_symlink "${nix-pyenv-directory}/lib/$basefile" $file
+          fi
+      fi
+  done
+  for file in ${nix-pyenv-directory}/lib/*; do
+      if [[ -L "$file" ]] && [[ "$(dirname $(readlink "$file"))" != "${pyenv}/${sitePackages}" ]]; then
+          rm -f "$file"
+      fi
+  done
+
+  # ensure the typing_extensions.py is not in the lib directory
+  rm ${nix-pyenv-directory}/lib/typing_extensions.py > /dev/null 2>&1
+
+  # add python executable to the bin directory
+  ensure_symlink "${nix-pyenv-directory}/bin/python" ${pyenv}/bin/python
+  export PATH=${nix-pyenv-directory}/bin:$PATH
+
+  ${optionalCommentOut} ${nix}/bin/nix-store --add-root ${nix-pyenv-directory}/.nix-shell-inputs --realise ${inputDerivation}
+''